Merge pull request #504 from hello-wn/feat-rpc

feat(template): enable grpc reflection
shanbay · Aug 28, 2023 · f8e9963 · f8e9963
2 parents 1dbd7d6 + 5bb1956
commit f8e9963
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/cmd/gobay/pkged.go b/cmd/gobay/pkged.go
diff --git a/cmd/gobay/templates/app/grpc/server.go.tmpl b/cmd/gobay/templates/app/grpc/server.go.tmpl
@@ -24,6 +24,7 @@ import (
 	"google.golang.org/grpc/health/grpc_health_v1"
 
 	myapp "{{ $.Url }}/app"
+	"google.golang.org/grpc/reflection"
 )
 
 // Serve grpc server
@@ -64,6 +65,8 @@ func Serve(app *gobay.Application) error {
 
 	configureAPI(server, &{{ toLowerCamel $.Name}}Server{app: app})
 
+	// Register reflection service on gRPC server.
+	reflection.Register(server)
 	// 启动 prometheus handler
 	grpc_prometheus.Register(server)
 	promSrv := http.Server{Addr: ":9000"}