chore: refine code (#1215)

2021-11-08 23:12:13 +08:00
parent a72b0a689b
commit f8367856e8
1 changed files with 1 additions and 4 deletions
--- a/tools/goctl/rpc/generator/genmain.go
+++ b/tools/goctl/rpc/generator/genmain.go
@@ -40,12 +40,9 @@ 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:
+		if c.Mode == service.DevMode || c.Mode == service.TestMode {
 			reflection.Register(grpcServer)
-		default:
 		}
-
 	})
 	defer s.Stop()