rootandClaude Opus 4.7
85c0071fd8
来源：hgdjnet-server 9bfbafe10d417deea959bd53117bc88a10e60a83 项目身份 HKDJ / JHA-216，独立官网仓库，Jenkins 守卫键 hkdjnet 测试目录 /server/hkdj_net，App/Web API 30331/30332 Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
119 lines
3.0 KiB
Go
119 lines
3.0 KiB
Go
package router
|
|
|
|
import (
|
|
"hkdj-web-server/app/api/home"
|
|
"hkdj-web-server/app/appg"
|
|
"hkdj-web-server/app/middleware/auth"
|
|
"hkdj-web-server/app/middleware/cors"
|
|
"hkdj-web-server/app/middleware/ipblacklist"
|
|
"hkdj-web-server/app/middleware/ipstat"
|
|
"hkdj-web-server/app/middleware/reptilerecord"
|
|
"hkdj-web-server/app/middleware/templatecache"
|
|
"hkdj-web-server/app/middleware/visit"
|
|
"hkdj-web-server/pub/common"
|
|
"hkdj-web-server/pub/common/enum/consts"
|
|
"hkdj-web-server/pub/common/log"
|
|
"hkdj-web-server/pub/common/useragent"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/DeanThompson/ginpprof"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func Init() (engine *gin.Engine) {
|
|
// 设置启动模式
|
|
switch appg.Conf.RunMod {
|
|
case "dev":
|
|
gin.SetMode(gin.DebugMode)
|
|
case "test":
|
|
gin.SetMode(gin.TestMode)
|
|
case "prod":
|
|
gin.SetMode(gin.ReleaseMode)
|
|
default:
|
|
gin.SetMode(gin.DebugMode)
|
|
}
|
|
// 初始化gin引擎
|
|
engine = gin.Default()
|
|
if appg.Conf.TrustedProxies != "" {
|
|
proxies := strings.Split(appg.Conf.TrustedProxies, ",")
|
|
for i, p := range proxies {
|
|
proxies[i] = strings.TrimSpace(p)
|
|
}
|
|
_ = engine.SetTrustedProxies(proxies)
|
|
log.Infof("SetTrustedProxies: %v", proxies)
|
|
} else {
|
|
_ = engine.SetTrustedProxies([]string{})
|
|
log.Infof("SetTrustedProxies: none (direct TCP IP mode)")
|
|
}
|
|
engine.HandleMethodNotAllowed = true
|
|
engine.MaxMultipartMemory = 60 << 20
|
|
|
|
common.InitGinSecret(appg.Conf.ResKey, consts.RunMod(appg.Conf.RunMod))
|
|
// 注册中间件
|
|
registeMiddleware(engine)
|
|
// 注册路由
|
|
registeRouter(engine)
|
|
// 是否开启监控服务
|
|
if appg.Conf.Pprof {
|
|
ginpprof.Wrap(engine)
|
|
}
|
|
return
|
|
}
|
|
|
|
func registeMiddleware(engine *gin.Engine) {
|
|
engine.Use(
|
|
// gin.Recovery(),
|
|
cors.Cors(&cors.Options{
|
|
Origin: nil, // "*"
|
|
Methods: []string{http.MethodGet, http.MethodPost, http.MethodPut, http.MethodHead, http.MethodDelete},
|
|
AllowedHeaders: []string{
|
|
"Content-Length", "Content-Type", "Authorization", "X-User-Agent",
|
|
"device", "api_version", "x-api-key",
|
|
"sid",
|
|
"device_brand", "device-brand",
|
|
"device_model", "device-model",
|
|
"system_name", "system-name",
|
|
"system_version", "system-version",
|
|
useragent.SDKCommonHeader,
|
|
},
|
|
ExposedHeaders: []string{"Content-Disposition", "Refresh-Authorization"},
|
|
OptionsSuccessStatus: http.StatusOK,
|
|
}),
|
|
// 必须是此顺序, happens before原则
|
|
//monitor.Check,
|
|
ipblacklist.IpBlackListMiddleware,
|
|
reptilerecord.ReptileRecord,
|
|
ipstat.IpStatMiddleware,
|
|
useragent.UserAgent,
|
|
auth.Auth,
|
|
// calllimit.Check,
|
|
visit.Visit,
|
|
)
|
|
}
|
|
|
|
// 注册路由
|
|
func registeRouter(r *gin.Engine) {
|
|
videoRouter(r)
|
|
playletRouter(r)
|
|
playletThemeRouter(r)
|
|
commentRouter(r)
|
|
coinRouter(r)
|
|
feedbackRouter(r)
|
|
contactUsRouter(r)
|
|
userCenterRouter(r)
|
|
authRouter(r)
|
|
rechargeRouter(r)
|
|
rechargePackageRouter(r)
|
|
uploadRouter(r)
|
|
healthRouter(r)
|
|
configRouter(r)
|
|
appInfoRouter(r)
|
|
translationRouter(r)
|
|
navigationRouter(r)
|
|
statcenterRouter(r)
|
|
r.GET("/notpage", templatecache.WithCache(time.Minute*1), home.PageNotFound)
|
|
|
|
}
|