This commit is contained in:
Sakurasan
2025-02-01 23:52:55 +08:00
parent 65d6d12972
commit bc223d6530
30 changed files with 2683 additions and 242 deletions

47
wire/wire.go Normal file
View File

@@ -0,0 +1,47 @@
//go:build wireinject
// +build wireinject
package wire
import (
"context"
"opencatd-open/team/dao"
handler "opencatd-open/team/handler/team"
"opencatd-open/team/service"
"github.com/google/wire"
"gorm.io/gorm"
)
// 定义 provider set
var userSet = wire.NewSet(
dao.NewUserDAO,
wire.Bind(new(dao.UserRepository), new(*dao.UserDAO)),
service.NewUserService,
)
var keySet = wire.NewSet(
dao.NewApiKeyDAO,
wire.Bind(new(dao.ApiKeyRepository), new(*dao.ApiKeyDAO)),
service.NewApiKeyService,
)
var tokenSet = wire.NewSet(
dao.NewTokenDAO,
wire.Bind(new(dao.TokenRepository), new(*dao.TokenDAO)),
service.NewTokenService,
)
var usageSet = wire.NewSet(
dao.NewUsageDAO,
wire.Bind(new(dao.UsageRepository), new(*dao.UsageDAO)),
dao.NewDailyUsageDAO,
wire.Bind(new(dao.DailyUsageRepository), new(*dao.DailyUsageDAO)),
service.NewUsageService,
)
// 初始化 TeamHandler
func InitTeamHandler(ctx context.Context, db *gorm.DB) (*handler.TeamHandler, error) {
wire.Build(userSet, keySet, tokenSet, usageSet, handler.NewTeamHandler)
return nil, nil
}

43
wire/wire_gen.go Normal file
View File

@@ -0,0 +1,43 @@
// Code generated by Wire. DO NOT EDIT.
//go:generate go run github.com/google/wire/cmd/wire
//+build !wireinject
package wire
import (
"context"
"github.com/google/wire"
"gorm.io/gorm"
"opencatd-open/team/dao"
"opencatd-open/team/handler/team"
"opencatd-open/team/service"
)
// Injectors from wire.go:
// 初始化 TeamHandler
func InitTeamHandler(ctx context.Context, db *gorm.DB) (*handler.TeamHandler, error) {
userDAO := dao.NewUserDAO(db)
userService := service.NewUserService(userDAO, db)
tokenDAO := dao.NewTokenDAO(db)
tokenService := service.NewTokenService(tokenDAO)
apiKeyDAO := dao.NewApiKeyDAO(db)
apiKeyService := service.NewApiKeyService(apiKeyDAO, db)
usageDAO := dao.NewUsageDAO(db)
dailyUsageDAO := dao.NewDailyUsageDAO(db)
usageService := service.NewUsageService(ctx, db, usageDAO, dailyUsageDAO)
teamHandler := handler.NewTeamHandler(userService, tokenService, apiKeyService, usageService)
return teamHandler, nil
}
// wire.go:
// 定义 provider set
var userSet = wire.NewSet(dao.NewUserDAO, wire.Bind(new(dao.UserRepository), new(*dao.UserDAO)), service.NewUserService)
var keySet = wire.NewSet(dao.NewApiKeyDAO, wire.Bind(new(dao.ApiKeyRepository), new(*dao.ApiKeyDAO)), service.NewApiKeyService)
var tokenSet = wire.NewSet(dao.NewTokenDAO, wire.Bind(new(dao.TokenRepository), new(*dao.TokenDAO)), service.NewTokenService)
var usageSet = wire.NewSet(dao.NewUsageDAO, wire.Bind(new(dao.UsageRepository), new(*dao.UsageDAO)), dao.NewDailyUsageDAO, wire.Bind(new(dao.DailyUsageRepository), new(*dao.DailyUsageDAO)), service.NewUsageService)