2026-04-22 16:01:59 +08:00
|
|
|
|
package wire
|
2026-03-13 20:40:20 +08:00
|
|
|
|
|
|
|
|
|
|
import (
|
2026-04-22 16:01:59 +08:00
|
|
|
|
"with_you/internal/config"
|
|
|
|
|
|
"with_you/internal/grpc/runner"
|
2026-03-13 20:40:20 +08:00
|
|
|
|
|
|
|
|
|
|
"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()
|
|
|
|
|
|
}
|