ok

tpl/server.go

@@ -5,22 +5,59 @@ var NewFile = `
 package server
 
 import (
+	"context"
+	"net/http"
+	"os"
+	"strings"
 	{import}
+	"git.apinb.com/bsm-sdk/core/vars"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/reflection"
+	gwRuntime "github.com/grpc-ecosystem/grpc-gateway/v2/runtime"
 )
 
-func New() *grpc.Server {
-	srv := grpc.NewServer()
+type Server struct {
+	Grpc *grpc.Server
+	Ctx  context.Context
+	Mux  *gwRuntime.ServeMux
+}
+
+func New(addr string) *Server {
+	srv := &Server{Ctx: context.Background(), Grpc: grpc.NewServer(), Mux: gwRuntime.NewServeMux()}
 
 	// register service to grpc.Server
 	{register}
 
-	reflection.Register(srv)
+	reflection.Register(srv.Grpc)
+
+	// 将服务注册到Gateway
+	opts := []grpc.DialOption{grpc.WithInsecure()}
+	{gw}
+
+	// Register services swagger
+	srv.RegisterSwagger()
+
 
 	return srv
 }
 
+// RegisterSwagger 注册swagger
+func (s *Server) RegisterSwagger() {
+	srvKey := strings.ToLower(vars.ServiceKey)
+	s.Mux.HandlePath("GET", "/"+srvKey+".swagger.json", func(w http.ResponseWriter, r *http.Request, pathParams map[string]string) {
+		w.Header().Set("Content-Type", "application/json")
+		bytes, err := os.ReadFile("./swagger/" + srvKey + ".swagger.json")
+		if err != nil {
+			w.WriteHeader(http.StatusNotFound)
+			w.Write([]byte(err.Error()))
+			return
+		}
+		w.Write(bytes)
+		return
+
+	})
+}
+
 `
 
 var Server = `