refactor: complete backend rewrite for multi-protocol proxy
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:
+11
-96
@@ -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
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user