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

refactor: complete backend rewrite for multi-protocol proxy

Browse Source 
Major rewrite of the Go backend to support:
- Three API format imports: openai, anthropic, compatible
- Three protocol conversions: Chat Completions, Responses, Messages
- Hub-and-spoke architecture with Chat as intermediate format

Deleted:
- opencat.go (old entry)
- store/, team/, pkg/team/, pkg/store/ (old data layer)
- internal/model/, internal/consts/ (old types)
- internal/service/team/, internal/controller/team/ (old handlers)
- llm/ (removed LLM client library, pure proxy mode)
- dist/, assets/ (old build artifacts)

Added:
- internal/store/ — 9 GORM models + multi-DB support
- internal/pkg/ — crypto (AES-GCM), apikey, jwt, ratelimit, resp, tokenizer
- internal/channel/ — channel selection, weighted LB, health checks
- internal/proxy/convert/ — 6 protocol conversion functions + SSE streaming
- internal/proxy/ — gateway with request dispatch and upstream selection
- internal/usage/ — async usage recorder with batch writes
- internal/api/ — management API (auth, users, keys, channels, models)
- Makefile for build/test/deploy

Fixed API to match frontend expectations:
- Login response wraps token in { data: { token } }
- GET /api/profile route added
- Profile response wraps user in { code, data }
- Role returned as number (10=admin, 1=user)
This commit is contained in:
Sakurasan
2026-08-30 11:49:31 +08:00
parent aa0d87f132
commit ef3025dd80
127 changed files with 4623 additions and 10500 deletions
Download Patch File Download Diff File Expand all files Collapse all files
wire/wire.go
+11 -96
View File
@@ -5,117 +5,32 @@ package wire
import (
"context"
"opencatd-open/internal/controller"
proxy "opencatd-open/internal/controller/proxy"
team_controller "opencatd-open/internal/controller/team"
"opencatd-open/internal/channel"
"opencatd-open/internal/controller/proxy"
"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"
)
// 定义 provider set
var userSet = wire.NewSet(
dao.NewUserDAO,
wire.Bind(new(dao.UserRepository), new(*dao.UserDAO)),
teamService.NewUserService,
)
var keySet = wire.NewSet(
dao.NewApiKeyDAO,
wire.Bind(new(dao.ApiKeyRepository), new(*dao.ApiKeyDAO)),
teamService.NewApiKeyService,
)
var tokenSet = wire.NewSet(
dao.NewTokenDAO,
wire.Bind(new(dao.TokenRepository), new(*dao.TokenDAO)),
teamService.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)),
teamService.NewUsageService,
)
// 初始化 TeamHandler
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(
var daoSet = wire.NewSet(
dao.NewUserDAO,
dao.NewApiKeyDAO,
dao.NewTokenDAO,
dao.NewUsageDAO,
dao.NewDailyUsageDAO,
dao.NewChannelDAO,
dao.NewModelDAO,
)
var channelSet = wire.NewSet(
channel.NewService,
channel.NewHealthChecker,
)
func InitProxyHandler(ctx context.Context, cfg *config.Config, db *gorm.DB, wg *sync.WaitGroup) (*proxy.Proxy, error) {
wire.Build(proxySet, proxy.NewProxy)
wire.Build(daoSet, channelSet, proxy.NewProxy)
return nil, nil
}
wire/wire_gen.go
-80
View File
@@ -1,80 +0,0 @@
// 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"
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, cfg *config.Config, db *gorm.DB) (*controller.Team, error) {
userDAO := dao.NewUserDAO(db)
userService := service.NewUserService(db, userDAO)
tokenDAO := dao.NewTokenDAO(db)
tokenService := service.NewTokenService(tokenDAO)
apiKeyDAO := dao.NewApiKeyDAO(cfg, db)
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(cfg, db, userDAO)
tokenDAO := dao.NewTokenDAO(db)
tokenServiceImpl := service2.NewTokenService(db, tokenDAO)
apiKeyDAO := dao.NewApiKeyDAO(cfg, db)
apiKeyServiceImpl := service2.NewApiKeyService(db, apiKeyDAO)
webAuthnService, err := service2.NewWebAuthnService(cfg, db)
if err != nil {
return nil, err
}
usageService := service2.NewUsageService(ctx, cfg, db)
api := controller2.NewApi(cfg, 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(cfg, 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:
// 定义 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)
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)