feat: register reflection on grpc server (#1943)

Signed-off-by: Gaius <gaius.qi@gmail.com>
dragonflyoss · Dec 21, 2022 · 92b01f0 · 92b01f0
1 parent c5aae85
commit 92b01f0
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 0 deletions.
diff --git a/pkg/rpc/cdnsystem/server/server.go b/pkg/rpc/cdnsystem/server/server.go
@@ -30,6 +30,7 @@ import (
 	"google.golang.org/grpc/health"
 	healthpb "google.golang.org/grpc/health/grpc_health_v1"
 	"google.golang.org/grpc/keepalive"
+	"google.golang.org/grpc/reflection"
 
 	cdnsystemv1 "d7y.io/api/pkg/apis/cdnsystem/v1"
 
@@ -89,5 +90,8 @@ func New(svr cdnsystemv1.SeederServer, opts ...grpc.ServerOption) *grpc.Server {
 
 	// Register health on grpc server.
 	healthpb.RegisterHealthServer(grpcServer, health.NewServer())
+
+	// Register reflection on grpc server.
+	reflection.Register(grpcServer)
 	return grpcServer
 }
diff --git a/pkg/rpc/dfdaemon/server/server.go b/pkg/rpc/dfdaemon/server/server.go
@@ -30,6 +30,7 @@ import (
 	"google.golang.org/grpc/health"
 	healthpb "google.golang.org/grpc/health/grpc_health_v1"
 	"google.golang.org/grpc/keepalive"
+	"google.golang.org/grpc/reflection"
 
 	dfdaemonv1 "d7y.io/api/pkg/apis/dfdaemon/v1"
 
@@ -89,5 +90,8 @@ func New(svr dfdaemonv1.DaemonServer, opts ...grpc.ServerOption) *grpc.Server {
 
 	// Register health on grpc server.
 	healthpb.RegisterHealthServer(grpcServer, health.NewServer())
+
+	// Register reflection on grpc server.
+	reflection.Register(grpcServer)
 	return grpcServer
 }
diff --git a/pkg/rpc/manager/server/server.go b/pkg/rpc/manager/server/server.go
@@ -30,6 +30,7 @@ import (
 	"google.golang.org/grpc/health"
 	healthpb "google.golang.org/grpc/health/grpc_health_v1"
 	"google.golang.org/grpc/keepalive"
+	"google.golang.org/grpc/reflection"
 
 	managerv1 "d7y.io/api/pkg/apis/manager/v1"
 	securityv1 "d7y.io/api/pkg/apis/security/v1"
@@ -85,9 +86,14 @@ func New(managerServer managerv1.ManagerServer, securityServer securityv1.Certif
 
 	// Register servers on grpc server.
 	managerv1.RegisterManagerServer(grpcServer, managerServer)
+
+	// Register security on grpc server.
 	securityv1.RegisterCertificateServiceServer(grpcServer, securityServer)
 
 	// Register health on grpc server.
 	healthpb.RegisterHealthServer(grpcServer, health.NewServer())
+
+	// Register reflection on grpc server.
+	reflection.Register(grpcServer)
 	return grpcServer
 }
diff --git a/pkg/rpc/scheduler/server/server.go b/pkg/rpc/scheduler/server/server.go
@@ -30,6 +30,7 @@ import (
 	"google.golang.org/grpc/health"
 	healthpb "google.golang.org/grpc/health/grpc_health_v1"
 	"google.golang.org/grpc/keepalive"
+	"google.golang.org/grpc/reflection"
 
 	schedulerv1 "d7y.io/api/pkg/apis/scheduler/v1"
 
@@ -89,5 +90,8 @@ func New(svr schedulerv1.SchedulerServer, opts ...grpc.ServerOption) *grpc.Serve
 
 	// Register health on grpc server.
 	healthpb.RegisterHealthServer(grpcServer, health.NewServer())
+
+	// Register reflection on grpc server.
+	reflection.Register(grpcServer)
 	return grpcServer
 }