mirror of
https://github.com/hanzoai/deploy.git
synced 2026-10-07 08:44:50 +00:00
790464f5d8
Signed-off-by: Justin Marquis <[email protected]> Co-authored-by: Soumya Ghosh Dastidar <[email protected]>
34 lines
1.1 KiB
Go
34 lines
1.1 KiB
Go
package grpc
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
var (
|
|
otelUnaryInterceptor grpc.UnaryClientInterceptor
|
|
otelStreamInterceptor grpc.StreamClientInterceptor
|
|
interceptorsInitialized = sync.Once{}
|
|
)
|
|
|
|
// otel interceptors must be created once to avoid memory leak
|
|
// see https://github.com/open-telemetry/opentelemetry-go-contrib/issues/4226 for details
|
|
func ensureInitialized() {
|
|
interceptorsInitialized.Do(func() {
|
|
otelUnaryInterceptor = otelgrpc.UnaryClientInterceptor() //nolint:staticcheck // TODO: ignore SA1019 for depreciation: see https://github.com/argoproj/argo-cd/issues/18258
|
|
otelStreamInterceptor = otelgrpc.StreamClientInterceptor() //nolint:staticcheck // TODO: ignore SA1019 for depreciation: see https://github.com/argoproj/argo-cd/issues/18258
|
|
})
|
|
}
|
|
|
|
func OTELUnaryClientInterceptor() grpc.UnaryClientInterceptor {
|
|
ensureInitialized()
|
|
return otelUnaryInterceptor
|
|
}
|
|
|
|
func OTELStreamClientInterceptor() grpc.StreamClientInterceptor {
|
|
ensureInitialized()
|
|
return otelStreamInterceptor
|
|
}
|