测试

goutils/proto_build/tpls/main.tpl

@@ -30,9 +30,8 @@ func main() {
 		log.Fatalf("failed to listen: %v", err)
 	}
 	s := grpc.NewServer(sopt)                             //新建一个grpc服务
-	{{range .StructServiceNames}}
-	service.Register{{.StructServiceName}}Server(s, &{{.LogicPackageName}}.{{.StructServiceName}}LogicGrpc{}) // {{.LogicPackageName}} 服务注册
-	{{- end}}
+
+	MainRegisterServer(s) // 注册所有的service
 	
 	if err := s.Serve(lis); err != nil {                  
 		log.Fatalf("failed to serve: %v", err)

goutils/proto_build/tpls/main_gen.tpl

@@ -0,0 +1,15 @@
+package main
+
+import (
+	"{{.ProjectName}}/gen/go/service"
+	"{{.ProjectName}}/server/logics/info_logic"
+	"{{.ProjectName}}/server/logics/product_logic"
+
+	"google.golang.org/grpc"
+)
+
+func MainRegisterServer(s *grpc.Server) {
+    {{range .StructServiceNames}}
+	    service.Register{{.StructServiceName}}Server(s, &{{.LogicPackageName}}.{{.StructServiceName}}LogicGrpc{}) // {{.LogicPackageName}} 服务注册
+	{{- end}}
+}