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

implement grpc gateway and addl handler support

Browse Source
This commit is contained in:
Ryan McGuire 2025-03-24 16:39:11 -04:00
parent 262f6a4232
commit 7f60a59d42
7 changed files with 15 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pkg/app/run.go
View File

@ -9,9 +9,6 @@ import (
"gitea.libretechconsulting.com/rmcguire/go-app/pkg/config"
)
// TODO: Make Configurable
const GRPC_GW_API_PATH = "/api"
func (a *App) Done() <-chan any {
return a.appDone
}

2
pkg/app/run_grpc.go
View File

@ -25,7 +25,7 @@ func (a *App) runGRPC(ctx context.Context) {
}
a.HTTP.Handlers = append(a.HTTP.Handlers, opts.HTTPHandler{
Prefix: GRPC_GW_API_PATH,
Prefix: a.cfg.GRPC.GRPCGatewayPath,
StripPrefix: true,
Handler: a.GRPC.GetGatewayMux(),
})

2
pkg/config/types_grpc.go
View File

@ -15,4 +15,6 @@ type GRPCConfig struct {
LogRequests bool `yaml:"logRequests" env:"APP_GRPC_LOG_REQUESTS" json:"logRequests,omitempty"`
EnableReflection bool `yaml:"enableReflection" env:"APP_GRPC_ENABLE_REFLECTION" json:"enableReflection,omitempty"`
EnableInstrumentation bool `yaml:"enableInstrumentation" env:"APP_GRPC_ENABLE_INSTRUMENTATION" json:"enableInstrumentation,omitempty"` // requires OTEL
EnableGRPCGateway bool `yaml:"enableGRPCGateway" json:"enableGRPCGateway,omitempty" env:"APP_GRPC_ENABLE_GATEWAY" default:"true"`
GRPCGatewayPath string `yaml:"grpcGatewayPath" json:"grpcGatewayPath,omitempty" env:"APP_GRPC_GATEWAY_PATH" default:"/grpc-api"`
}

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

@ -5,7 +5,7 @@ import (
"fmt"
"net"
"github.com/grpc-ecosystem/grpc-gateway/runtime"
"github.com/grpc-ecosystem/grpc-gateway/v2/runtime"
"go.opentelemetry.io/otel/attribute"
"go.opentelemetry.io/otel/codes"
"go.opentelemetry.io/otel/trace"

2
pkg/srv/grpc/grpc.go
View File

@ -4,7 +4,7 @@ import (
"context"
"strings"
"github.com/grpc-ecosystem/grpc-gateway/runtime"
"github.com/grpc-ecosystem/grpc-gateway/v2/runtime"
"github.com/rs/zerolog"
"go.opentelemetry.io/otel/attribute"
"go.opentelemetry.io/otel/codes"

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

@ -3,7 +3,7 @@ package opts
import (
"context"
"github.com/grpc-ecosystem/grpc-gateway/runtime"
"github.com/grpc-ecosystem/grpc-gateway/v2/runtime"
"google.golang.org/grpc"
"gitea.libretechconsulting.com/rmcguire/go-app/pkg/config"
@ -25,12 +25,14 @@ type AppGRPC struct {
}
type GRPCService struct {
Name string // Descriptive name of service
Type *grpc.ServiceDesc // Type (from protoc generated code)
Service any // Implementation of GRPCService.Type (ptr)
GwRegistrationFuncs []func(context.Context, *runtime.ServeMux, *grpc.ClientConn) error // Gateway regustration handler funcs
Name string // Descriptive name of service
Type *grpc.ServiceDesc // Type (from protoc generated code)
Service any // Implementation of GRPCService.Type (ptr)
GwRegistrationFuncs []GwRegistrationFunc // Gateway regustration handler funcs
}
type GwRegistrationFunc func(context.Context, *runtime.ServeMux, *grpc.ClientConn) error
func (a *AppGRPC) SetGatewayMux(mux *runtime.ServeMux) {
a.gatewayMux = mux
}

4
pkg/srv/grpc/services.go
View File

@ -19,7 +19,9 @@ func (a *appGRPCServer) registerGRPCServices(ctx context.Context) {
for _, service := range a.opts.Services {
a.registerGRPCService(ctx, service)
a.registerServiceGatewayHandlers(ctx, service)
if a.opts.EnableGRPCGateway {
a.registerServiceGatewayHandlers(ctx, service)
}
}
span.SetStatus(codes.Ok, "")