Hericium-sys
87 lines
2.1 KiB
Go
87 lines
2.1 KiB
Go
package rest
|
|
|
|
import (
|
|
"github.com/gin-gonic/gin"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
// 子路由注册函数签名
|
|
type Registrar func(*gin.RouterGroup)
|
|
|
|
// 两类注册器：公开/私有（是否需要 JWT）
|
|
var (
|
|
publicRegistrars []Registrar
|
|
privateRegistrars []Registrar
|
|
)
|
|
|
|
// RegisterPublic 注册公开路由（无需认证）
|
|
func RegisterPublic(fn Registrar) {
|
|
publicRegistrars = append(publicRegistrars, fn)
|
|
zap.L().Debug("公开路由注册器已添加",
|
|
zap.Int("total_public", len(publicRegistrars)),
|
|
)
|
|
}
|
|
|
|
// RegisterPrivate 注册私有路由（需要 JWT 认证）
|
|
func RegisterPrivate(fn Registrar) {
|
|
privateRegistrars = append(privateRegistrars, fn)
|
|
zap.L().Debug("私有路由注册器已添加",
|
|
zap.Int("total_private", len(privateRegistrars)),
|
|
)
|
|
}
|
|
|
|
// ApplyPublic 应用所有公开路由
|
|
func ApplyPublic(group *gin.RouterGroup) {
|
|
zap.L().Info("开始应用公开路由",
|
|
zap.Int("count", len(publicRegistrars)),
|
|
)
|
|
|
|
for i, fn := range publicRegistrars {
|
|
fn(group)
|
|
zap.L().Debug("公开路由已应用",
|
|
zap.Int("index", i+1),
|
|
)
|
|
}
|
|
|
|
zap.L().Info("公开路由应用完成",
|
|
zap.Int("applied_count", len(publicRegistrars)),
|
|
)
|
|
}
|
|
|
|
// ApplyPrivate 应用所有私有路由（带 JWT 中间件）
|
|
func ApplyPrivate(group *gin.RouterGroup) {
|
|
zap.L().Info("开始应用私有路由",
|
|
zap.Int("count", len(privateRegistrars)),
|
|
)
|
|
|
|
for i, fn := range privateRegistrars {
|
|
fn(group)
|
|
zap.L().Debug("私有路由已应用",
|
|
zap.Int("index", i+1),
|
|
)
|
|
}
|
|
|
|
zap.L().Info("私有路由应用完成",
|
|
zap.Int("applied_count", len(privateRegistrars)),
|
|
)
|
|
}
|
|
|
|
// GetRegistrarStats 获取注册器统计信息（调试用）
|
|
func GetRegistrarStats() map[string]int {
|
|
return map[string]int{
|
|
"public": len(publicRegistrars),
|
|
"private": len(privateRegistrars),
|
|
"total": len(publicRegistrars) + len(privateRegistrars),
|
|
}
|
|
}
|
|
|
|
// PrintStats 打印注册器统计信息
|
|
func PrintStats() {
|
|
stats := GetRegistrarStats()
|
|
zap.L().Info("路由注册器统计",
|
|
zap.Int("public_count", stats["public"]),
|
|
zap.Int("private_count", stats["private"]),
|
|
zap.Int("total_count", stats["total"]),
|
|
)
|
|
}
|