add grpc support

This commit is contained in:
2025-03-06 17:17:59 -05:00
4 changed files with 35 additions and 10 deletions

View File

@ -201,7 +201,7 @@ func (s *settings) newMeterProvider(ctx context.Context) (*metric.MeterProvider,
host, set := os.LookupEnv("OTEL_EXPORTER_OTLP_ENDPOINT")
var otlpExporter *otlpmetricgrpc.Exporter
if set && host != "" {
if exp, err := otlpmetricgrpc.New(ctx, otlpmetricgrpc.WithInsecure()); err != nil {
if exp, err := otlpmetricgrpc.New(ctx); err != nil {
return nil, fmt.Errorf("otlpmetricgrpc.New: %w", err)
} else {
otlpExporter = exp

View File

@ -37,6 +37,7 @@ type HTTPServerOpts struct {
HandleFuncs []HTTPFunc
Middleware []http.Handler
HealthCheckFuncs []HealthCheckFunc
CustomListener net.Listener
}
func prepHTTPServer(opts *HTTPServerOpts) *http.Server {
@ -151,16 +152,26 @@ func initHTTPServer(opts *HTTPServerOpts) (
server = prepHTTPServer(opts)
go func() {
l.Debug().Msg("HTTP Server Started")
err := server.ListenAndServe()
var err error
if opts.CustomListener != nil {
err = server.Serve(opts.CustomListener)
} else {
err = server.ListenAndServe()
}
if err != nil && err != http.ErrServerClosed {
l.Err(err).Msg("HTTP server error")
} else {
l.Info().Msg("HTTP server shut down")
}
// Notify app initiator
doneChan <- nil
}()
l.Debug().Msg("HTTP Server Started")
// Shut down http server with a deadline
return func(shutdownCtx context.Context) error {
l.Debug().Msg("stopping http server")