diff --git a/pkg/rpc/cdnsystem/server/server.go b/pkg/rpc/cdnsystem/server/server.go index 0b513720d..d018664c5 100644 --- 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 index db0339524..aff72ab45 100644 --- 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 index 8257cbee1..a8d58967c 100644 --- 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 index e2a78156b..5e577b5fc 100644 --- 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 }