Bläddra i källkod

refactor(wire): 更新 shardingManager 初始化方式

- 在 NewShardingManager 函数中增加 viperViper 参数
- 优化 shardingManager 的初始化,支持配置文件读取
fusu 10 timmar sedan
förälder
incheckning
3232109ff3
1 ändrade filer med 1 tillägg och 1 borttagningar
  1. 1 1
      cmd/task/wire/wire_gen.go

+ 1 - 1
cmd/task/wire/wire_gen.go

@@ -40,7 +40,7 @@ func NewWire(viperViper *viper.Viper, logger *log.Logger) (*app.App, func(), err
 	database := repository.NewMongoDB(qmgoClient, viperViper)
 	rabbitMQ, cleanup := repository.NewRabbitMQ(viperViper, logger)
 	syncedEnforcer := repository.NewCasbinEnforcer(viperViper, logger, db)
-	shardingManager := repository.NewShardingManager(logger)
+	shardingManager := repository.NewShardingManager(logger, viperViper)
 	repositoryRepository := repository.NewRepository(logger, db, client, qmgoClient, database, rabbitMQ, syncedEnforcer, shardingManager)
 	transaction := repository.NewTransaction(repositoryRepository)
 	sidSid := sid.NewSid()