浏览代码

feat(wire): 集成 MongoDB 存储

- 在 wire.go 中添加 NewMongoDB 函数
- 在 wire_gen.go 中创建 MongoDB 实例并传递给 NewRepository
- 更新 repositorySet 以包含 NewMongoDB
fusu 2 月之前
父节点
当前提交
60b996db42
共有 2 个文件被更改,包括 4 次插入2 次删除
  1. 1 0
      cmd/task/wire/wire.go
  2. 3 2
      cmd/task/wire/wire_gen.go

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

@@ -19,6 +19,7 @@ import (
 var repositorySet = wire.NewSet(
 	repository.NewDB,
 	//repository.NewRedis,
+	repository.NewMongoDB,
 	repository.NewRepository,
 	repository.NewTransaction,
 	repository.NewUserRepository,

+ 3 - 2
cmd/task/wire/wire_gen.go

@@ -23,7 +23,8 @@ import (
 
 func NewWire(viperViper *viper.Viper, logger *log.Logger) (*app.App, func(), error) {
 	db := repository.NewDB(viperViper, logger)
-	repositoryRepository := repository.NewRepository(logger, db)
+	mongoDB := repository.NewMongoDB(viperViper)
+	repositoryRepository := repository.NewRepository(logger, db, mongoDB)
 	transaction := repository.NewTransaction(repositoryRepository)
 	sidSid := sid.NewSid()
 	taskTask := task.NewTask(transaction, logger, sidSid)
@@ -56,7 +57,7 @@ func NewWire(viperViper *viper.Viper, logger *log.Logger) (*app.App, func(), err
 
 // wire.go:
 
-var repositorySet = wire.NewSet(repository.NewDB, repository.NewRepository, repository.NewTransaction, repository.NewUserRepository, repository.NewGameShieldRepository, repository.NewGameShieldBackendRepository, repository.NewGameShieldPublicIpRepository, repository.NewHostRepository, repository.NewGameShieldUserIpRepository, repository.NewGameShieldSdkIpRepository)
+var repositorySet = wire.NewSet(repository.NewDB, repository.NewMongoDB, repository.NewRepository, repository.NewTransaction, repository.NewUserRepository, repository.NewGameShieldRepository, repository.NewGameShieldBackendRepository, repository.NewGameShieldPublicIpRepository, repository.NewHostRepository, repository.NewGameShieldUserIpRepository, repository.NewGameShieldSdkIpRepository)
 
 var taskSet = wire.NewSet(task.NewTask, task.NewUserTask, task.NewGameShieldTask)