mirror of
https://github.com/hanzoai/telemetry.git
synced 2026-10-30 20:01:49 +00:00
Joshua MacDonaldandGitHub
593537a94c
#### Description API change in extensionmiddleware, an unstable package to allow middleware access to the start context. HTTP only. gRPC is treated in #14536. This is technically a breaking change in configmiddleware and for all implementations of this extension. We believe no one is using this feature outside of this repository, see memorylimiter is fixed here. #### Link to tracking issue Part of https://github.com/open-telemetry/opentelemetry-collector/issues/14523 #### Documentation N/A
54 lines
1.6 KiB
Go
54 lines
1.6 KiB
Go
// Copyright The OpenTelemetry Authors
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package extensionmiddleware // import "go.opentelemetry.io/collector/extension/extensionmiddleware"
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
// HTTPServer defines the interface for HTTP server middleware extensions.
|
|
type HTTPServer interface {
|
|
// GetHTTPHandler wraps the provided base http.Handler.
|
|
GetHTTPHandler(_ context.Context) (WrapHTTPHandlerFunc, error)
|
|
}
|
|
|
|
// GRPCServer defines the interface for gRPC server middleware extensions.
|
|
type GRPCServer interface {
|
|
// GetGRPCServerOptions returns options for a gRPC server.
|
|
GetGRPCServerOptions(context.Context) ([]grpc.ServerOption, error)
|
|
}
|
|
|
|
var _ HTTPServer = (*GetHTTPHandlerFunc)(nil)
|
|
|
|
// GetHTTPHandlerFunc is a function that implements HTTPServer.
|
|
type GetHTTPHandlerFunc func(_ context.Context) (WrapHTTPHandlerFunc, error)
|
|
|
|
func (f GetHTTPHandlerFunc) GetHTTPHandler(ctx context.Context) (WrapHTTPHandlerFunc, error) {
|
|
if f == nil {
|
|
return func(_ context.Context, h http.Handler) (http.Handler, error) {
|
|
return h, nil
|
|
}, nil
|
|
}
|
|
return f(ctx)
|
|
}
|
|
|
|
var _ GRPCServer = (*GetGRPCServerOptionsFunc)(nil)
|
|
|
|
// GetGRPCServerOptionsFunc is a function that implements GRPCServer.
|
|
type GetGRPCServerOptionsFunc func(context.Context) ([]grpc.ServerOption, error)
|
|
|
|
func (f GetGRPCServerOptionsFunc) GetGRPCServerOptions(ctx context.Context) ([]grpc.ServerOption, error) {
|
|
if f == nil {
|
|
return nil, nil
|
|
}
|
|
return f(ctx)
|
|
}
|
|
|
|
// WrapHTTPHandlerFunc is called to initialize a new instance of
|
|
// HTTP server middleware at runtime.
|
|
type WrapHTTPHandlerFunc = func(context.Context, http.Handler) (http.Handler, error)
|