瀏覽代碼

feat: reflection grpc service (#1107)

* feat: reflection grpc service

* feat: reflection grpc service
小小小下 3 年之前
父節點
當前提交
295ec27e1b
共有 1 個文件被更改,包括 9 次插入0 次删除
  1. 9 0
      tools/goctl/rpc/generator/genmain.go

+ 9 - 0
tools/goctl/rpc/generator/genmain.go

@@ -23,6 +23,8 @@ import (
 	"github.com/tal-tech/go-zero/core/conf"
 	"github.com/tal-tech/go-zero/zrpc"
 	"google.golang.org/grpc"
+	"github.com/tal-tech/go-zero/core/service"
+	"google.golang.org/grpc/reflection"
 )
 
 var configFile = flag.String("f", "etc/{{.serviceName}}.yaml", "the config file")
@@ -37,6 +39,13 @@ func main() {
 
 	s := zrpc.MustNewServer(c.RpcServerConf, func(grpcServer *grpc.Server) {
 		{{.pkg}}.Register{{.service}}Server(grpcServer, srv)
+
+		switch c.Mode {
+		case service.DevMode,service.TestMode:
+			reflection.Register(grpcServer)
+		default:
+		}
+
 	})
 	defer s.Stop()