feat：完善项目

router/middleware.go

@@ -0,0 +1,98 @@
+package router
+
+import (
+	"SciencesServer/serve/logger"
+	"context"
+	"fmt"
+	"net/http"
+	"time"
+
+	"github.com/gin-gonic/gin"
+	"github.com/sirupsen/logrus"
+)
+
+func Cors() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		method := c.Request.Method
+		c.Header("Access-Control-Allow-Origin", "*")
+		c.Header("Access-Control-Allow-Headers", "access-control-allow-origin, access-control-allow-headers, application/octet-stream")
+		c.Header("Access-Control-Allow-Methods", "POST, GET, OPTIONS")
+		c.Header("Access-Control-Expose-Headers", "Content-Length, Access-Control-Allow-Origin, Access-Control-Allow-Headers, Content-Type")
+		c.Header("Access-Control-Allow-Credentials", "true")
+
+		if method == "OPTIONS" {
+			c.AbortWithStatus(http.StatusNoContent)
+			return
+		}
+		c.Next()
+	}
+}
+
+// NoMethodHandler 未找到请求方法的处理函数
+func NoMethodHandler() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		c.JSON(http.StatusMethodNotAllowed, gin.H{
+			"message": "未找到请求路由的处理函数",
+		})
+		c.Abort()
+		return
+	}
+}
+
+// NoRouteHandler 未找到请求路由的处理函数
+func NoRouteHandler() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		c.JSON(http.StatusNotFound, gin.H{
+			"message": "未找到请求路由的处理函数",
+		})
+		c.Abort()
+		return
+	}
+}
+
+func LoggerHandle(log string, leastDay uint) gin.HandlerFunc {
+	_logger := logger.NewLogger().Init(&logger.Option{File: log, LeastDay: leastDay, Level: "debug", IsStdout: false}).Logger
+
+	return func(c *gin.Context) {
+		_logger.Log(logrus.InfoLevel, map[string]interface{}{
+			"Status": c.Writer.Status(),
+			"IP":     c.ClientIP(),
+			"Method": c.Request.Method,
+			"Url":    c.Request.RequestURI,
+		})
+	}
+}
+
+// TimeoutHandle 超时处理
+func TimeoutHandle(timeout time.Duration) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		ctx, cancel := context.WithTimeout(c.Request.Context(), timeout)
+
+		defer func() {
+			if ctx.Err() == context.DeadlineExceeded {
+				c.Writer.WriteHeader(http.StatusGatewayTimeout)
+				c.Abort()
+			}
+			cancel()
+		}()
+		c.Request = c.Request.WithContext(ctx)
+		c.Next()
+	}
+}
+
+// RecoveryHandler 崩溃恢复中间件
+func RecoveryHandler() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		defer func() {
+			if err := recover(); err != nil {
+				fmt.Printf("Recover：request【%s】 error：【%v】\n", c.Request.URL, err)
+				c.JSON(http.StatusInternalServerError, gin.H{
+					"message": "Internal Server Error！",
+				})
+				c.Abort()
+				return
+			}
+		}()
+		c.Next()
+	}
+}