26 lines
561 B
Go
26 lines
561 B
Go
|
package wire
|
||
|
|
|
||
|
|
import (
|
||
|
|
"carrot_bbs/internal/config"
|
||
|
|
"carrot_bbs/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()
|
||
|
|
}
|