cas_tt_cloud_backend/router/router.go

package router

import (
	"SciencesServer/config"
	"SciencesServer/router/rate"
	"github.com/prometheus/client_golang/prometheus/promhttp"
	"io"
	"net/http"
	"os"
	"time"

	"github.com/chenjiandongx/ginprom"
	"github.com/gin-gonic/gin"
)

type Router struct {
	*Option
}

type (
	Option struct {
		Mode   string
		IsCors bool
		*RateLimitConfig
	}
	// RateLimitConfig 限流配置
	RateLimitConfig struct {
		IsRate   bool `json:"is_rate"`
		Limit    int  `json:"limit"`
		Capacity int  `json:"capacity"`
	}
)

func (this *Router) Init() *gin.Engine {
	gin.SetMode(this.Mode)

	app := gin.New()

	if this.IsCors {
		app.Use(Cors())
	}
	app.NoRoute(NoRouteHandler())
	app.NoMethod(NoMethodHandler())
	app.Use(ginprom.PromMiddleware(nil))
	app.Use(LoggerHandle("log/gin.log", 3))
	app.Use(TimeoutHandle(time.Second * 30))
	//app.Use(RecoveryHandler())

	app.GET("/metrics", ginprom.PromHandler(promhttp.Handler()))

	if this.RateLimitConfig != nil {
		if this.RateLimitConfig.IsRate {
			rate.NewIPRateLimiter()(this.RateLimitConfig.Limit, this.RateLimitConfig.Capacity)
			app.Use(rate.RequestIPRateLimiter()(this.RateLimitConfig.Limit, this.RateLimitConfig.Capacity))
		}
	}
	if config.IsDebug() {
		gin.DefaultWriter = io.MultiWriter(os.Stdout)
		app.StaticFS("/api-docs", http.Dir("./doc"))
	}
	app.StaticFS("/upload", http.Dir("./upload"))
	// 注册路由
	registerAPI(app)
	registerAdminAPI(app)
	registerEnterpriseAPI(app)

	app.MaxMultipartMemory = 4 << 20

	return app
}

func NewRouter(option *Option) *Router {
	return &Router{option}
}