feat 包命名规范

internal/net/httpgateway/router.go

@@ -0,0 +1,127 @@
+package httpgateway
+
+import (
+	"context"
+	"git.hlsq.asia/mmorpg/service-common/discover/common"
+	"git.hlsq.asia/mmorpg/service-common/log"
+	"git.hlsq.asia/mmorpg/service-common/net/grpc/grpcclient"
+	"git.hlsq.asia/mmorpg/service-common/net/http/httpresp"
+	"git.hlsq.asia/mmorpg/service-common/proto/rs/rspb"
+	"git.hlsq.asia/mmorpg/service-gateway/internal/handler/httphandler"
+	"git.hlsq.asia/mmorpg/service-gateway/internal/net/httpgateway/wrapper"
+	"github.com/gin-contrib/cors"
+	"github.com/gin-gonic/gin"
+	"github.com/grpc-ecosystem/grpc-gateway/v2/runtime"
+	"go.opentelemetry.io/contrib/instrumentation/github.com/gin-gonic/gin/otelgin"
+	"google.golang.org/protobuf/encoding/protojson"
+	"strings"
+)
+
+func InitServeMux() *runtime.ServeMux {
+	baseMarshaler := &runtime.JSONPb{
+		MarshalOptions: protojson.MarshalOptions{
+			UseProtoNames:   false,
+			EmitUnpopulated: true,
+		},
+		UnmarshalOptions: protojson.UnmarshalOptions{
+			DiscardUnknown: true,
+		},
+	}
+	unifiedMarshaler := wrapper.NewWrappedMarshaler(baseMarshaler)
+
+	mux := runtime.NewServeMux(
+		runtime.WithMarshalerOption(runtime.MIMEWildcard, unifiedMarshaler),
+		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() *gin.Engine {
+	gin.SetMode(gin.ReleaseMode)
+
+	r := gin.New()
+	r.Use(
+		gin.Recovery(),
+		ginLogger(log.GetLogger().Named("GIN")),
+		cors.New(corsConfig()),
+		otelgin.Middleware(
+			common.KeyDiscoverServiceNameGateway,
+			otelgin.WithSpanNameFormatter(func(c *gin.Context) string {
+				method := strings.ToUpper(c.Request.Method)
+				return method + " " + c.Request.URL.Path
+			}),
+		),
+	)
+
+	r.HandleMethodNotAllowed = true
+	r.NoMethod(func(c *gin.Context) {
+		httpresp.JsonMethodNotAllowed(c)
+		c.Abort()
+	})
+	r.NoRoute(func(c *gin.Context) {
+		httpresp.JsonNotFound(c)
+		c.Abort()
+	})
+
+	auth := r.Group("/")
+	auth.Use(authJwt())
+
+	// 网关
+	initGatewayPath(auth)
+	// 用户中心
+	initUserPath(auth)
+	// 奇怪的知识-服务端
+	initQgdzsPath(auth)
+
+	return r
+}
+
+func initGatewayPath(r *gin.RouterGroup) {
+	g := r.Group("/" + common.KeyDiscoverServiceNameGateway)
+
+	g.POST("/open/login", httphandler.Login)
+	g.POST("/open/refresh_token", httphandler.RefreshToken)
+	g.POST("/open/logout", httphandler.Logout)
+}
+
+func initUserPath(r *gin.RouterGroup) {
+	g := r.Group("/" + common.KeyDiscoverServiceNameUser)
+
+	client, err := grpcclient.UserNewClientLB()
+	if err != nil {
+		log.Errorf("get user conn failed: %v", err)
+		return
+	}
+
+	gwMux := InitServeMux()
+	if err = rspb.RegisterUserHandlerClient(context.Background(), gwMux, client); err != nil {
+		log.Errorf("RegisterUserHandlerClient err: %v", err)
+		return
+	}
+
+	g.Any("/*path", gin.WrapH(gwMux))
+}
+
+func initQgdzsPath(r *gin.RouterGroup) {
+	g := r.Group("/" + common.KeyDiscoverServiceNameQgdzs)
+
+	client, err := grpcclient.QgdzsNewClientLB()
+	if err != nil {
+		log.Errorf("get qgdzs conn failed: %v", err)
+		return
+	}
+
+	gwMux := InitServeMux()
+	if err = rspb.RegisterQgdzsHandlerClient(context.Background(), gwMux, client); err != nil {
+		log.Errorf("RegisterQgdzsHandlerClient err: %v", err)
+		return
+	}
+
+	g.Any("/*path", gin.WrapH(gwMux))
+}