rmcguire/go-app
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

support runtime.ServeMux opts for grpc-gateway

Browse Source
This commit is contained in:
Ryan McGuire 2025-03-25 10:49:58 -04:00
parent adfedc9239
commit b44b6a331c
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pkg/srv/grpc/gateway.go
View File

@ -18,7 +18,7 @@ func (a *appGRPCServer) registerServiceGatewayHandlers(ctx context.Context, serv
if len(service.GwRegistrationFuncs) < 1 { if len(service.GwRegistrationFuncs) < 1 {
return return
} else if a.gatewayMux == nil { } else if a.gatewayMux == nil {
a.gatewayMux = runtime.NewServeMux() a.gatewayMux = runtime.NewServeMux(a.opts.GRPCGatewayOpts...)
} }
ctx, span := a.tracer.Start(ctx, "appgrpc.init.prepare.service.gwHandlers", trace.WithAttributes( ctx, span := a.tracer.Start(ctx, "appgrpc.init.prepare.service.gwHandlers", trace.WithAttributes(
@ -50,6 +50,7 @@ func (a *appGRPCServer) GetClientConn(ctx context.Context) *grpc.ClientConn {
if err != nil { if err != nil {
span.RecordError(err) span.RecordError(err)
span.SetStatus(codes.Error, err.Error()) span.SetStatus(codes.Error, err.Error())
panic(err)
} }
return clientConn return clientConn

3
pkg/srv/grpc/opts/config.go
View File

@ -19,7 +19,8 @@ type AppGRPC struct {
UnaryInterceptors []grpc.UnaryServerInterceptor UnaryInterceptors []grpc.UnaryServerInterceptor
StreamInterceptors []grpc.StreamServerInterceptor StreamInterceptors []grpc.StreamServerInterceptor
GRPCOpts []grpc.ServerOption GRPCOpts []grpc.ServerOption
GRPCDialOpts []grpc.DialOption // Map ServerOptions to DialOpts for GRPC Gateway support GRPCDialOpts []grpc.DialOption // Map ServerOptions to DialOpts for GRPC Gateway support
GRPCGatewayOpts []runtime.ServeMuxOption // Configure grpc-gateway ServeMux
GRPCDone <-chan error GRPCDone <-chan error
gatewayMux *runtime.ServeMux gatewayMux *runtime.ServeMux
} }