fork-it/gpt-load
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 代理认证中间件

Browse Source
This commit is contained in:
tbphp
2025-07-24 17:17:18 +08:00
parent 746c9f3108
commit abb8fa1d19
3 changed files with 66 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
internal/router/router.go
View File

@@ -2,7 +2,6 @@ package router
import (
"embed"
"gpt-load/internal/channel"
"gpt-load/internal/handler"
"gpt-load/internal/middleware"
"gpt-load/internal/proxy"
@@ -43,7 +42,6 @@ func NewRouter(
proxyServer *proxy.ProxyServer,
configManager types.ConfigManager,
groupManager *services.GroupManager,
channelFactory *channel.Factory,
buildFS embed.FS,
indexPage []byte,
) *gin.Engine {
@@ -66,7 +64,7 @@ func NewRouter(
// 注册路由
registerSystemRoutes(router, serverHandler)
registerAPIRoutes(router, serverHandler, configManager)
registerProxyRoutes(router, proxyServer, configManager)
registerProxyRoutes(router, proxyServer, groupManager)
registerFrontendRoutes(router, buildFS, indexPage)
return router
@@ -159,12 +157,11 @@ func registerProtectedAPIRoutes(api *gin.RouterGroup, serverHandler *handler.Ser
func registerProxyRoutes(
router *gin.Engine,
proxyServer *proxy.ProxyServer,
configManager types.ConfigManager,
groupManager *services.GroupManager,
) {
proxyGroup := router.Group("/proxy")
authConfig := configManager.GetAuthConfig()
proxyGroup.Use(middleware.Auth(authConfig))
proxyGroup.Use(middleware.ProxyAuth(groupManager))
proxyGroup.Any("/:group_name/*path", proxyServer.HandleProxy)
}