package wire import ( "with_you/internal/config" "with_you/internal/grpc/runner" "github.com/google/wire" "go.uber.org/zap" ) // GRPCSet gRPC 相关的 Provider Set var GRPCSet = wire.NewSet( ProvideGRPCServer, ProvideTaskManager, ) // ProvideGRPCServer 提供 gRPC 服务器 func ProvideGRPCServer(cfg *config.Config, logger *zap.Logger) *runner.Server { return runner.NewServer(&cfg.GRPC, logger) } // ProvideTaskManager 提供 TaskManager func ProvideTaskManager(server *runner.Server) *runner.TaskManager { return server.GetTaskManager() }