feat 网关鉴权

Server/gateway/internal/net/http_gateway/router.go

@@ -2,13 +2,16 @@ package http_gateway
 
 import (
 	"common/log"
+	"common/net/grpc/service"
 	"common/net/http/http_resp"
-	wrapper2 "gateway/internal/net/http_gateway/wrapper"
+	"common/proto/ss/grpc_pb"
+	"context"
+	"gateway/internal/handler/http_handler"
+	"gateway/internal/net/http_gateway/wrapper"
 	"github.com/gin-contrib/cors"
 	"github.com/gin-gonic/gin"
 	"github.com/grpc-ecosystem/grpc-gateway/v2/runtime"
 	"google.golang.org/protobuf/encoding/protojson"
-	"net/http"
 )
 
 func InitServeMux() *runtime.ServeMux {
@@ -21,16 +24,22 @@ func InitServeMux() *runtime.ServeMux {
 			DiscardUnknown: true,
 		},
 	}
-	unifiedMarshaler := wrapper2.NewWrappedMarshaler(baseMarshaler)
+	unifiedMarshaler := wrapper.NewWrappedMarshaler(baseMarshaler)
 
 	mux := runtime.NewServeMux(
 		runtime.WithMarshalerOption(runtime.MIMEWildcard, unifiedMarshaler),
-		runtime.WithErrorHandler(wrapper2.ErrorHandler),
+		runtime.WithErrorHandler(wrapper.ErrorHandler),
+		runtime.WithIncomingHeaderMatcher(func(header string) (string, bool) {
+			if header == "X-Usn" {
+				return "X-Usn", true
+			}
+			return runtime.DefaultHeaderMatcher(header)
+		}),
 	)
 	return mux
 }
 
-func InitRouter(mux *runtime.ServeMux) *gin.Engine {
+func InitRouter() *gin.Engine {
 	gin.SetMode(gin.ReleaseMode)
 
 	r := gin.New()
@@ -42,13 +51,42 @@ func InitRouter(mux *runtime.ServeMux) *gin.Engine {
 
 	r.HandleMethodNotAllowed = true
 	r.NoMethod(func(c *gin.Context) {
-		c.JSON(http.StatusMethodNotAllowed, http_resp.Error(http_resp.Failed.Code(), "Method Not Allowed"))
+		http_resp.JsonMethodNotAllowed(c)
 	})
 	r.NoRoute(func(c *gin.Context) {
-		c.JSON(http.StatusNotFound, http_resp.Error(http_resp.Failed.Code(), "Endpoint Not Found"))
+		http_resp.JsonNotFound(c)
 	})
 
-	r.Any("/*any", gin.WrapH(mux))
+	initBaseRoute(r.Group("/"))
+
+	auth := r.Group("/")
+	auth.Use(authJwt())
+
+	// 用户中心
+	initUserPath(auth)
 
 	return r
 }
+
+func initBaseRoute(r *gin.RouterGroup) {
+	g := r.Group("/gw")
+	g.POST("/login", http_handler.Login)
+	g.POST("/refresh_token", http_handler.RefreshToken)
+}
+
+func initUserPath(r *gin.RouterGroup) {
+	g := r.Group("/user")
+	client, err := service.UserNewClientLB()
+	if err != nil {
+		log.Errorf("get user conn failed: %v", err)
+		return
+	}
+
+	gwMux := InitServeMux()
+	if err = grpc_pb.RegisterUserHandlerClient(context.Background(), gwMux, client); err != nil {
+		log.Errorf("RegisterUserHandlerClient err: %v", err)
+		return
+	}
+
+	g.Any("/*path", gin.WrapH(gwMux))
+}