71 lines
1.7 KiB
Go
71 lines
1.7 KiB
Go
|
//go:build wireinject
|
||
|
|
// +build wireinject
|
||
|
|
|
||
|
|
// The build tag makes sure the stub is not built in the final build.
|
||
|
|
package wire
|
||
|
|
|
||
|
|
import (
|
||
|
|
"ego/internal/conf"
|
||
|
|
"ego/internal/handler"
|
||
|
|
"ego/internal/service"
|
||
|
|
|
||
|
|
"github.com/google/wire"
|
||
|
|
)
|
||
|
|
|
||
|
|
// DBSet 提供数据库连接
|
||
|
|
var DBSet = wire.NewSet(
|
||
|
|
conf.NewDb, // 依赖 conf 包的 NewDb
|
||
|
|
)
|
||
|
|
|
||
|
|
// HandlerSet 处理器集合
|
||
|
|
var HandlerSet = wire.NewSet(
|
||
|
|
handler.NewSysUserHandler,
|
||
|
|
handler.NewSysLoginLogHandler,
|
||
|
|
handler.NewSysUploadHandler,
|
||
|
|
handler.NewSysDeployFileHandler,
|
||
|
|
)
|
||
|
|
|
||
|
|
// UserServiceSet 定义 service 层依赖
|
||
|
|
var UserServiceSet = wire.NewSet(
|
||
|
|
service.NewSysUserService,
|
||
|
|
)
|
||
|
|
|
||
|
|
// LoginLogServiceSet 服务器信息服务层依赖
|
||
|
|
var LoginLogServiceSet = wire.NewSet(
|
||
|
|
service.NewSysLoginLogService,
|
||
|
|
)
|
||
|
|
|
||
|
|
// UploadServiceSet 文件上传服务层依赖
|
||
|
|
var UploadServiceSet = wire.NewSet(
|
||
|
|
service.NewSysUploadService,
|
||
|
|
)
|
||
|
|
|
||
|
|
// DeployFileServiceSet 部署文件服务层依赖
|
||
|
|
var DeployFileServiceSet = wire.NewSet(
|
||
|
|
service.NewSysDeployFileService,
|
||
|
|
)
|
||
|
|
|
||
|
|
// InjectSysUserHandler 注入 handler
|
||
|
|
func InjectSysUserHandler() *handler.SysUserHandler {
|
||
|
|
panic(wire.Build(
|
||
|
|
HandlerSet,
|
||
|
|
UserServiceSet,
|
||
|
|
DBSet,
|
||
|
|
))
|
||
|
|
}
|
||
|
|
|
||
|
|
// InjectSysLoginLogHandler 注入登录记录信息处理器
|
||
|
|
func InjectSysLoginLogHandler() *handler.SysLoginLogHandler {
|
||
|
|
panic(wire.Build(HandlerSet, LoginLogServiceSet, DBSet))
|
||
|
|
}
|
||
|
|
|
||
|
|
// InjectSysUploadHandler 注入文件上传处理器
|
||
|
|
func InjectSysUploadHandler() *handler.SysUploadHandler {
|
||
|
|
panic(wire.Build(HandlerSet, UploadServiceSet, DBSet))
|
||
|
|
}
|
||
|
|
|
||
|
|
// InjectSysDeployFileHandler 注入部署文件处理器
|
||
|
|
func InjectSysDeployFileHandler() *handler.SysDeployFileHandler {
|
||
|
|
panic(wire.Build(HandlerSet, DeployFileServiceSet, DBSet))
|
||
|
|
}
|