feat: register reflection on grpc server (#1943)

Signed-off-by: Gaius <gaius.qi@gmail.com>
This commit is contained in:
Gaius 2022-12-21 15:13:49 +08:00
parent 7cb802aeb7
commit 1edd36da59
No known key found for this signature in database
GPG Key ID: 8B4E5D1290FA2FFB
4 changed files with 18 additions and 0 deletions

View File

@ -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
}

View File

@ -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
}

View File

@ -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
}

View File

@ -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
}