go_zh/router/init.go

package router

import (
	"github.com/gin-gonic/gin"
)

// 路由配置函数类型
type RouteSetupFunc func(*gin.Engine)

var routeRegistrations []RouteSetupFunc

// 注册路由配置（自动被各个路由文件调用）
func RegisterRoute(setup RouteSetupFunc) {
	routeRegistrations = append(routeRegistrations, setup)
}

// 统一设置路由入口
func SetupRoutes(engine *gin.Engine) {
	for _, setup := range routeRegistrations {
		setup(engine)
	}
}