mirrors2/opencatd-open
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

reface to openteam

Browse Source
This commit is contained in:
Sakurasan
2025-04-16 18:01:27 +08:00
parent bc223d6530
commit e7ffc9e8b9
92 changed files with 5345 additions and 1273 deletions
Download Patch File Download Diff File Expand all files Collapse all files

92
wire/wire.go
View File

@@ -5,9 +5,15 @@ package wire
import (
"context"
"opencatd-open/team/dao"
handler "opencatd-open/team/handler/team"
"opencatd-open/team/service"
"opencatd-open/internal/controller"
proxy "opencatd-open/internal/controller/proxy"
team_controller "opencatd-open/internal/controller/team"
"opencatd-open/internal/dao"
"opencatd-open/pkg/config"
"sync"
service "opencatd-open/internal/service"
teamService "opencatd-open/internal/service/team"
"github.com/google/wire"
"gorm.io/gorm"
@@ -17,19 +23,19 @@ import (
var userSet = wire.NewSet(
dao.NewUserDAO,
wire.Bind(new(dao.UserRepository), new(*dao.UserDAO)),
service.NewUserService,
teamService.NewUserService,
)
var keySet = wire.NewSet(
dao.NewApiKeyDAO,
wire.Bind(new(dao.ApiKeyRepository), new(*dao.ApiKeyDAO)),
service.NewApiKeyService,
teamService.NewApiKeyService,
)
var tokenSet = wire.NewSet(
dao.NewTokenDAO,
wire.Bind(new(dao.TokenRepository), new(*dao.TokenDAO)),
service.NewTokenService,
teamService.NewTokenService,
)
var usageSet = wire.NewSet(
@@ -37,11 +43,79 @@ var usageSet = wire.NewSet(
wire.Bind(new(dao.UsageRepository), new(*dao.UsageDAO)),
dao.NewDailyUsageDAO,
wire.Bind(new(dao.DailyUsageRepository), new(*dao.DailyUsageDAO)),
service.NewUsageService,
teamService.NewUsageService,
)
// 初始化 TeamHandler
func InitTeamHandler(ctx context.Context, db *gorm.DB) (*handler.TeamHandler, error) {
wire.Build(userSet, keySet, tokenSet, usageSet, handler.NewTeamHandler)
func InitTeamHandler(ctx context.Context, cfg *config.Config, db *gorm.DB) (*team_controller.Team, error) {
wire.Build(userSet, keySet, tokenSet, usageSet, team_controller.NewTeam)
return nil, nil
}
// var userApi = wire.NewSet(
// dao.NewUserDAO,
// wire.Bind(new(dao.UserRepository), new(*dao.UserDAO)),
// service.NewUserService,
// )
// var keyApi = wire.NewSet(
// dao.NewApiKeyDAO,
// wire.Bind(new(dao.ApiKeyRepository), new(*dao.ApiKeyDAO)),
// service.NewApiKeyService,
// )
// var tokenApi = wire.NewSet(
// dao.NewTokenDAO,
// wire.Bind(new(dao.TokenRepository), new(*dao.TokenDAO)),
// service.NewTokenService,
// )
// func InitAPIHandler(ctx context.Context, db *gorm.DB) (*controller.Api, error) {
// wire.Build(userApi, keyApi, tokenApi, controller.NewApi)
// return nil, nil
// }
var repositorySet = wire.NewSet(
dao.NewUserDAO,
wire.Bind(new(dao.UserRepository), new(*dao.UserDAO)),
dao.NewTokenDAO,
wire.Bind(new(dao.TokenRepository), new(*dao.TokenDAO)),
dao.NewApiKeyDAO,
wire.Bind(new(dao.ApiKeyRepository), new(*dao.ApiKeyDAO)),
dao.NewUsageDAO,
wire.Bind(new(dao.UsageRepository), new(*dao.UsageDAO)),
dao.NewDailyUsageDAO,
wire.Bind(new(dao.DailyUsageRepository), new(*dao.DailyUsageDAO)),
)
var serviceSet = wire.NewSet(
service.NewUserService,
service.NewTokenService,
service.NewApiKeyService,
service.NewWebAuthnService,
service.NewUsageService,
)
func InitAPIHandler(ctx context.Context, cfg *config.Config, db *gorm.DB) (*controller.Api, error) {
wire.Build(
repositorySet,
serviceSet,
controller.NewApi,
)
return nil, nil
}
var proxySet = wire.NewSet(
dao.NewUserDAO,
dao.NewApiKeyDAO,
dao.NewTokenDAO,
dao.NewUsageDAO,
dao.NewDailyUsageDAO,
)
func InitProxyHandler(ctx context.Context, cfg *config.Config, db *gorm.DB, wg *sync.WaitGroup) (*proxy.Proxy, error) {
wire.Build(proxySet, proxy.NewProxy)
return nil, nil
}

59
wire/wire_gen.go
View File

@@ -9,26 +9,57 @@ import (
"context"
"github.com/google/wire"
"gorm.io/gorm"
"opencatd-open/team/dao"
"opencatd-open/team/handler/team"
"opencatd-open/team/service"
controller2 "opencatd-open/internal/controller"
controller3 "opencatd-open/internal/controller/proxy"
"opencatd-open/internal/controller/team"
"opencatd-open/internal/dao"
service2 "opencatd-open/internal/service"
"opencatd-open/internal/service/team"
"opencatd-open/pkg/config"
"sync"
)
// Injectors from wire.go:
// 初始化 TeamHandler
func InitTeamHandler(ctx context.Context, db *gorm.DB) (*handler.TeamHandler, error) {
func InitTeamHandler(ctx context.Context, cfg *config.Config, db *gorm.DB) (*controller.Team, error) {
userDAO := dao.NewUserDAO(db)
userService := service.NewUserService(userDAO, db)
userService := service.NewUserService(db, userDAO)
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
apiKeyService := service.NewApiKeyService(db, apiKeyDAO)
usageDAO := dao.NewUsageDAO(cfg, db)
dailyUsageDAO := dao.NewDailyUsageDAO(cfg, db)
usageService := service.NewUsageService(ctx, cfg, db, usageDAO, dailyUsageDAO)
team := controller.NewTeam(userService, tokenService, apiKeyService, usageService)
return team, nil
}
func InitAPIHandler(ctx context.Context, cfg *config.Config, db *gorm.DB) (*controller2.Api, error) {
userDAO := dao.NewUserDAO(db)
userServiceImpl := service2.NewUserService(db, userDAO)
tokenDAO := dao.NewTokenDAO(db)
tokenServiceImpl := service2.NewTokenService(db, tokenDAO)
apiKeyDAO := dao.NewApiKeyDAO(db)
apiKeyServiceImpl := service2.NewApiKeyService(db, apiKeyDAO)
webAuthnService, err := service2.NewWebAuthnService(db, cfg)
if err != nil {
return nil, err
}
usageService := service2.NewUsageService(ctx, cfg, db)
api := controller2.NewApi(db, userServiceImpl, tokenServiceImpl, apiKeyServiceImpl, webAuthnService, usageService)
return api, nil
}
func InitProxyHandler(ctx context.Context, cfg *config.Config, db *gorm.DB, wg *sync.WaitGroup) (*controller3.Proxy, error) {
userDAO := dao.NewUserDAO(db)
apiKeyDAO := dao.NewApiKeyDAO(db)
tokenDAO := dao.NewTokenDAO(db)
usageDAO := dao.NewUsageDAO(cfg, db)
dailyUsageDAO := dao.NewDailyUsageDAO(cfg, db)
proxy := controller3.NewProxy(ctx, cfg, db, wg, userDAO, apiKeyDAO, tokenDAO, usageDAO, dailyUsageDAO)
return proxy, nil
}
// wire.go:
@@ -41,3 +72,9 @@ var keySet = wire.NewSet(dao.NewApiKeyDAO, wire.Bind(new(dao.ApiKeyRepository),
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)
var repositorySet = wire.NewSet(dao.NewUserDAO, wire.Bind(new(dao.UserRepository), new(*dao.UserDAO)), dao.NewTokenDAO, wire.Bind(new(dao.TokenRepository), new(*dao.TokenDAO)), dao.NewApiKeyDAO, wire.Bind(new(dao.ApiKeyRepository), new(*dao.ApiKeyDAO)), dao.NewUsageDAO, wire.Bind(new(dao.UsageRepository), new(*dao.UsageDAO)), dao.NewDailyUsageDAO, wire.Bind(new(dao.DailyUsageRepository), new(*dao.DailyUsageDAO)))
var serviceSet = wire.NewSet(service2.NewUserService, service2.NewTokenService, service2.NewApiKeyService, service2.NewWebAuthnService, service2.NewUsageService)
var proxySet = wire.NewSet(dao.NewUserDAO, dao.NewApiKeyDAO, dao.NewTokenDAO, dao.NewUsageDAO, dao.NewDailyUsageDAO)