Kevin Wan 3 лет назад
Родитель
Сommit
f8367856e8
1 измененных файлов с 1 добавлено и 4 удалено
  1. 1 4
      tools/goctl/rpc/generator/genmain.go

+ 1 - 4
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()