更新 tpl/server.go

tpl/server.go

@@ -13,6 +13,7 @@ import (
 	"git.apinb.com/bsm-sdk/core/vars"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/reflection"
+	"google.golang.org/protobuf/proto"
 	gwRuntime "github.com/grpc-ecosystem/grpc-gateway/v2/runtime"
 )
 
@@ -23,7 +24,9 @@ type Server struct {
 }
 
 func New(addr string) *Server {
-	srv := &Server{Ctx: context.Background(), Grpc: grpc.NewServer(), Mux: gwRuntime.NewServeMux()}
+	srv := &Server{Ctx: context.Background(), Grpc: grpc.NewServer(), Mux: gwRuntime.NewServeMux(
+		gwRuntime.WithForwardResponseRewriter(responseEnvelope),
+	)}
 
 	// register service to grpc.Server
 	{register}
@@ -58,6 +61,19 @@ func (s *Server) RegisterSwagger() {
 	})
 }
 
+// response envelope
+func responseEnvelope(_ context.Context, response proto.Message) (interface{}, error) {
+	name := string(response.ProtoReflect().Descriptor().Name())
+	if name == "Status" || name == "Error" || name == "StatusReply" {
+		return response, nil
+	}
+	return map[string]any{
+		"code": 0,
+		"message": string(response.ProtoReflect().Descriptor().Name()),
+		"result": response,
+	  }, nil
+}
+
 `
 
 var Server = `