소스 검색

chore: refine code (#1215)

Kevin Wan 3 년 전
부모
커밋
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()