MTP20_IF/routers/router.go

package routers

import (
	"mtp2_if/controllers/cptrade"
	"mtp2_if/controllers/wrtrade"
	"mtp2_if/logger"
	"mtp2_if/token"
	"net/http"
	"time"

	"github.com/gin-gonic/gin"

	// Swagger生成的文档
	_ "mtp2_if/docs"

	ginSwagger "github.com/swaggo/gin-swagger"
	"github.com/swaggo/gin-swagger/swaggerFiles"
)

// InitRouter 初始化路由器的方法
func InitRouter() *gin.Engine {
	r := gin.New()
	// 设置日志中间件
	r.Use(ginLoggerMiddleware())
	// 设置奔溃中间件
	r.Use(ginRecoveryMiddleware())
	// 跨域
	r.Use(CORSMiddleware())
	// Swagger
	r.GET("/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler))

	// 路由分组
	apiR := r.Group("/api")
	apiR.Use(token.Auth())
	// ************************ 仓单贸易 ************************
	wrTradeR := apiR.Group("WRTrade")
	wrTradeR.Use()
	{
		wrTradeR.GET("/GetAllDeliveryGoods", wrtrade.GetAllDeliveryGoods)
	}
	// ************************ 产能预售 ************************
	cpTradeR := apiR.Group("CPTrade")
	cpTradeR.Use()
	{
		// 查询产能预售申请表
		cpTradeR.GET("/QueryPreasleApply", cptrade.QueryPreasleApply)
		// 查询远期订单信息
		cpTradeR.GET("/QueryUserGoodsData", cptrade.QueryUserGoodsData)
		// 查询远期订单注销申请信息
		cpTradeR.GET("/QueryPositionCancel", cptrade.QueryPositionCancel)
		// 查询产能预售商品扩展信息
		cpTradeR.GET("/QueryPresaleGoodsEx", cptrade.QueryPresaleGoodsEx)
	}

	return r
}

func ginLoggerMiddleware() gin.HandlerFunc {
	return func(c *gin.Context) {
		start := time.Now()
		c.Next()
		end := time.Now()
		latency := end.Sub(start)
		path := c.Request.URL.RequestURI()
		clientip := c.ClientIP()
		method := c.Request.Method
		statuscode := c.Writer.Status()
		logger.GetLogger().Infof("|%3d|%13v|%15s|%s %s",
			statuscode,
			latency,
			clientip,
			method,
			path)
	}
}

func ginRecoveryMiddleware() gin.HandlerFunc {
	return gin.RecoveryWithWriter(logger.GetLogWriter())
}

// CORSMiddleware cors跨域.
func CORSMiddleware() gin.HandlerFunc {
	return func(c *gin.Context) {
		c.Writer.Header().Set("Access-Control-Allow-Origin", "*")
		c.Writer.Header().Set("Access-Control-Allow-Credentials", "true")
		c.Writer.Header().Set("Access-Control-Allow-Headers", "Content-Type, Content-Length, Accept-Encoding, X-CSRF-Token, Authorization, accept, origin, Cache-Control, X-Requested-With, X-session-Token")
		c.Writer.Header().Set("Access-Control-Allow-Methods", "POST, OPTIONS, GET, PUT")
		c.Writer.Header().Set("Access-Control-Expose-Headers", "Content-Length, Access-Control-Allow-Origin, Access-Control-Allow-Headers, Content-Type, X-session-Token")
		if c.Request.Method == "OPTIONS" {
			c.AbortWithStatus(http.StatusNoContent)
			return
		}

		c.Next()
	}
}