mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-26 17:47:48 +00:00
fa9f581d56
* chore: move to new org * logging * fix: org rename caos -> zitadel Co-authored-by: adlerhurst <silvan.reusser@gmail.com>
34 lines
851 B
Go
34 lines
851 B
Go
package middleware
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
grpc_utils "github.com/zitadel/zitadel/internal/api/grpc"
|
|
grpc_trace "go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
type GRPCMethod string
|
|
|
|
func DefaultTracingServer() grpc.UnaryServerInterceptor {
|
|
return TracingServer(grpc_utils.Healthz, grpc_utils.Readiness, grpc_utils.Validation)
|
|
}
|
|
|
|
func TracingServer(ignoredMethods ...GRPCMethod) grpc.UnaryServerInterceptor {
|
|
return func(
|
|
ctx context.Context,
|
|
req interface{},
|
|
info *grpc.UnaryServerInfo,
|
|
handler grpc.UnaryHandler,
|
|
) (interface{}, error) {
|
|
|
|
for _, ignoredMethod := range ignoredMethods {
|
|
if strings.HasSuffix(info.FullMethod, string(ignoredMethod)) {
|
|
return handler(ctx, req)
|
|
}
|
|
}
|
|
return grpc_trace.UnaryServerInterceptor()(ctx, req, info, handler)
|
|
}
|
|
}
|