zeromicro/go-zero
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

chore: refine code (#1215)

Browse Source
This commit is contained in:
Kevin Wan
2021-11-08 23:12:13 +08:00
committed by GitHub
parent a72b0a689b
commit f8367856e8
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -40,12 +40,9 @@ 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 { if c.Mode == service.DevMode || c.Mode == service.TestMode {
case service.DevMode,service.TestMode:
reflection.Register(grpcServer) reflection.Register(grpcServer)
default:
} }
}) })
defer s.Stop() defer s.Stop()