github_mirrors/go-zero
1
0
Fork 0
mirror of https://github.com/zeromicro/go-zero.git synced 2025-02-03 00:38:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: reflection grpc service (#1107)

Browse Source 
* feat: reflection grpc service

* feat: reflection grpc service
This commit is contained in:
小小小下 2021-10-01 22:02:04 +08:00 committed by GitHub
parent d1e702e8a3
commit 295ec27e1b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tools/goctl/rpc/generator/genmain.go
View File

@ -23,6 +23,8 @@ import (
"github.com/tal-tech/go-zero/core/conf" "github.com/tal-tech/go-zero/core/conf"
"github.com/tal-tech/go-zero/zrpc" "github.com/tal-tech/go-zero/zrpc"
"google.golang.org/grpc" "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") 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) { s := zrpc.MustNewServer(c.RpcServerConf, func(grpcServer *grpc.Server) {
{{.pkg}}.Register{{.service}}Server(grpcServer, srv) {{.pkg}}.Register{{.service}}Server(grpcServer, srv)
switch c.Mode {
case service.DevMode,service.TestMode:
reflection.Register(grpcServer)
default:
}
}) })
defer s.Stop() defer s.Stop()