feat app 模块化启动

Server/Gateway/net/http_gateway/middleward.go

@@ -0,0 +1,37 @@
+package http_gateway
+
+import (
+	"fmt"
+	"github.com/gin-contrib/cors"
+	"github.com/gin-gonic/gin"
+	"go.uber.org/zap"
+	"time"
+)
+
+func corsConfig() cors.Config {
+	return cors.Config{
+		AllowMethods:     []string{"GET", "POST", "OPTIONS"},
+		AllowHeaders:     []string{"Content-Type", "Authorization"},
+		AllowCredentials: false,
+		AllowAllOrigins:  true,
+		MaxAge:           12 * time.Hour,
+	}
+}
+
+func ginLogger(logger *zap.SugaredLogger) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		start := time.Now()
+		path := c.Request.URL.Path
+		c.Next()
+		cost := time.Since(start)
+
+		logger.Infof(fmt.Sprintf(
+			"HTTP Method:%v Code:%v Time:%v IP:%v Path:%v",
+			c.Request.Method,
+			c.Writer.Status(),
+			cost,
+			c.ClientIP(),
+			path),
+		)
+	}
+}

Server/Gateway/net/http_gateway/router.go

@@ -1,44 +1,43 @@
 package http_gateway
 
 import (
+	"common/log"
 	"gateway/config"
+	"gateway/handler/http_handler"
 	"github.com/gin-contrib/cors"
 	"github.com/gin-gonic/gin"
 	"net/http"
-	"time"
 )
 
 func InitRouter(cfg *config.Config) *gin.Engine {
-	r := gin.Default()
-	r.Use(gin.Recovery())
-	r.Use(cors.New(getCorsConfig()))
+	gin.SetMode(gin.ReleaseMode)
+
+	r := gin.New()
+	r.Use(
+		gin.Recovery(),
+		ginLogger(log.GetLogger().Named("GIN")),
+		cors.New(corsConfig()),
+	)
 
-	r.MaxMultipartMemory = 8 << 20
 	r.HandleMethodNotAllowed = true
 	r.NoMethod(func(c *gin.Context) {
 		c.JSON(http.StatusMethodNotAllowed, gin.H{
 			"result": false,
 			"error":  "Method Not Allowed",
 		})
-		return
 	})
 	r.NoRoute(func(c *gin.Context) {
 		c.JSON(http.StatusNotFound, gin.H{
 			"result": false,
 			"error":  "Endpoint Not Found",
 		})
-		return
 	})
+	initBaseRouter(r)
 
 	return r
 }
 
-func getCorsConfig() cors.Config {
-	return cors.Config{
-		AllowMethods:     []string{"GET", "POST", "PUT", "PATCH", "DELETE", "HEAD"},
-		AllowHeaders:     []string{"Origin", "Content-Length", "Content-Type", "Authorization", "X-Forwarded-For", "User-Agent", "Referer", "X-Token", "token", "Token", "company-id", "lang", "source"},
-		AllowCredentials: false,
-		AllowAllOrigins:  true,
-		MaxAge:           12 * time.Hour,
-	}
+func initBaseRouter(router *gin.Engine) {
+	g := router.Group("/b")
+	g.POST("/snowflake", http_handler.GenSnowflake) // 生成雪花
 }