mirror of
https://github.com/rocky-linux/peridot.git
synced 2024-11-23 21:51:27 +00:00
38 lines
1.1 KiB
Go
38 lines
1.1 KiB
Go
package grpcutil
|
|
|
|
import (
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/reflection"
|
|
rpb "google.golang.org/grpc/reflection/grpc_reflection_v1alpha"
|
|
)
|
|
|
|
// NewAuthlessReflectionInterceptor creates a proxy GRPCServer which automatically converts
|
|
// ServerReflectionServer instances to onces that skip grpc auth middleware.
|
|
//
|
|
// change:
|
|
// reflection.Register(srv)
|
|
// to:
|
|
// reflection.Register(grpcutil.NewAuthlessReflectionInterceptor(srv))
|
|
func NewAuthlessReflectionInterceptor(srv reflection.GRPCServer) reflection.GRPCServer {
|
|
return interceptingRegistrar{srv}
|
|
}
|
|
|
|
type interceptingRegistrar struct {
|
|
delegate reflection.GRPCServer
|
|
}
|
|
|
|
func (ir interceptingRegistrar) GetServiceInfo() map[string]grpc.ServiceInfo {
|
|
return ir.delegate.GetServiceInfo()
|
|
}
|
|
|
|
func (ir interceptingRegistrar) RegisterService(desc *grpc.ServiceDesc, impl interface{}) {
|
|
reflectionSrv := impl.(rpb.ServerReflectionServer)
|
|
ir.delegate.RegisterService(desc, &authlessReflection{ServerReflectionServer: reflectionSrv})
|
|
}
|
|
|
|
type authlessReflection struct {
|
|
IgnoreAuthMixin
|
|
|
|
rpb.ServerReflectionServer
|
|
}
|