Compare commits

...

5 Commits
v0.0.2 ... main

Author SHA1 Message Date
yanweidong 87734c972f 更新 tpl/logic.go 2025-04-18 12:14:38 +08:00
yanweidong 3b283780ab 更新 tpl/server.go 2025-04-17 23:24:09 +08:00
yanweidong cf7e34c61d 更新 tpl/server.go 2025-04-17 23:16:13 +08:00
yanweidong 9c3ac59467 更新 tpl/logic.go 2025-04-17 22:02:56 +08:00
yanweidong 689afaff97 更新 tpl/logic.go 2025-04-17 16:35:07 +08:00
2 changed files with 20 additions and 4 deletions

View File

@ -45,8 +45,8 @@ var FetchValidCode = `
var StatusReplyCode = ` var StatusReplyCode = `
return &pb.StatusReply{ return &pb.StatusReply{
Status: 200, Code: 0,
Message: "ok", Message: "OK",
Timeseq: time.Now().UnixNano(), Timeseq: time.Now().UnixMilli(),
}, nil }, nil
` `

View File

@ -13,6 +13,7 @@ import (
"git.apinb.com/bsm-sdk/core/vars" "git.apinb.com/bsm-sdk/core/vars"
"google.golang.org/grpc" "google.golang.org/grpc"
"google.golang.org/grpc/reflection" "google.golang.org/grpc/reflection"
"google.golang.org/protobuf/proto"
gwRuntime "github.com/grpc-ecosystem/grpc-gateway/v2/runtime" gwRuntime "github.com/grpc-ecosystem/grpc-gateway/v2/runtime"
) )
@ -23,7 +24,9 @@ type Server struct {
} }
func New(addr string) *Server { 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 service to grpc.Server
{register} {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": "OK",
"result": response,
}, nil
}
` `
var Server = ` var Server = `