Explorar o código

refactor(wire): 重构 Wire 依赖注入

- 移除了对 di 包的依赖
- 新增了 repositorySet、serviceSet 等新的 ProviderSet
- 更新了 NewWire 函数的实现,使用新的 ProviderSet
-增加了 sid 和 jwt包的依赖
fusu hai 2 meses
pai
achega
be4c08df2d
Modificáronse 1 ficheiros con 45 adicións e 8 borrados
  1. 45 8
      cmd/task/wire/wire.go

+ 45 - 8
cmd/task/wire/wire.go

@@ -4,18 +4,55 @@
 package wire
 
 import (
-	"github.com/go-nunu/nunu-layout-advanced/internal/di"
+	"github.com/go-nunu/nunu-layout-advanced/internal/repository"
 	"github.com/go-nunu/nunu-layout-advanced/internal/server"
+	"github.com/go-nunu/nunu-layout-advanced/internal/service"
 	"github.com/go-nunu/nunu-layout-advanced/internal/task"
 	"github.com/go-nunu/nunu-layout-advanced/pkg/app"
+	"github.com/go-nunu/nunu-layout-advanced/pkg/jwt"
 	"github.com/go-nunu/nunu-layout-advanced/pkg/log"
+	"github.com/go-nunu/nunu-layout-advanced/pkg/sid"
 	"github.com/google/wire"
 	"github.com/spf13/viper"
 )
 
-// 使用共享的Provider Set
-var taskSet = task.TaskSet
-var serverSet = server.TaskServerSet
+var repositorySet = wire.NewSet(
+	repository.NewDB,
+	//repository.NewRedis,
+	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,
+)
+var serverSet = wire.NewSet(
+	server.NewTaskServer,
+)
+
+var serviceSet = wire.NewSet(
+	service.NewService,
+	service.NewGameShieldService,
+	service.NewCrawlerService,
+	service.NewGameShieldPublicIpService,
+	service.NewDuedateService,
+	service.NewFormatterService,
+	service.NewParserService,
+	service.NewRequiredService,
+	service.NewHostService,
+	service.NewGameShieldBackendService,
+	service.NewGameShieldSdkIpService,
+	service.NewGameShieldUserIpService,
+)
 
 // build App
 func newApp(
@@ -29,12 +66,12 @@ func newApp(
 
 func NewWire(*viper.Viper, *log.Logger) (*app.App, func(), error) {
 	panic(wire.Build(
-		// 使用模块化依赖注入
-		di.RepositorySet,
-		di.ServiceSet,
-		di.CoreSet,
+		repositorySet,
 		taskSet,
 		serverSet,
+		serviceSet,
 		newApp,
+		sid.NewSid,
+		jwt.NewJwt,
 	))
 }