mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-14 11:58:02 +00:00
87 lines
2.9 KiB
Go
87 lines
2.9 KiB
Go
|
package middleware
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"strings"
|
||
|
|
||
|
"github.com/grpc-ecosystem/grpc-gateway/runtime"
|
||
|
"google.golang.org/grpc"
|
||
|
"google.golang.org/grpc/status"
|
||
|
|
||
|
_ "github.com/caos/zitadel/internal/statik"
|
||
|
"github.com/caos/zitadel/internal/telemetry/metrics"
|
||
|
)
|
||
|
|
||
|
const (
|
||
|
GrpcMethod = "grpc_method"
|
||
|
ReturnCode = "return_code"
|
||
|
GrpcRequestCounter = "grpc.server.request_counter"
|
||
|
GrpcRequestCounterDescription = "Grpc request counter"
|
||
|
TotalGrpcRequestCounter = "grpc.server.total_request_counter"
|
||
|
TotalGrpcRequestCounterDescription = "Total grpc request counter"
|
||
|
GrpcStatusCodeCounter = "grpc.server.grpc_status_code"
|
||
|
GrpcStatusCodeCounterDescription = "Grpc status code counter"
|
||
|
)
|
||
|
|
||
|
func MetricsHandler(metricTypes []metrics.MetricType, ignoredMethodSuffixes ...string) grpc.UnaryServerInterceptor {
|
||
|
return func(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
|
||
|
return RegisterMetrics(ctx, req, info, handler, metricTypes, ignoredMethodSuffixes...)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func RegisterMetrics(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler, metricTypes []metrics.MetricType, ignoredMethodSuffixes ...string) (_ interface{}, err error) {
|
||
|
if len(metricTypes) == 0 {
|
||
|
return handler(ctx, req)
|
||
|
}
|
||
|
|
||
|
for _, ignore := range ignoredMethodSuffixes {
|
||
|
if strings.HasSuffix(info.FullMethod, ignore) {
|
||
|
return handler(ctx, req)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
resp, err := handler(ctx, req)
|
||
|
if containsMetricsMethod(metrics.MetricTypeRequestCount, metricTypes) {
|
||
|
RegisterGrpcRequestCounter(ctx, info)
|
||
|
}
|
||
|
if containsMetricsMethod(metrics.MetricTypeTotalCount, metricTypes) {
|
||
|
RegisterGrpcTotalRequestCounter(ctx)
|
||
|
}
|
||
|
if containsMetricsMethod(metrics.MetricTypeStatusCode, metricTypes) {
|
||
|
RegisterGrpcRequestCodeCounter(ctx, info, err)
|
||
|
}
|
||
|
return resp, err
|
||
|
}
|
||
|
|
||
|
func RegisterGrpcRequestCounter(ctx context.Context, info *grpc.UnaryServerInfo) {
|
||
|
var labels = map[string]interface{}{
|
||
|
GrpcMethod: info.FullMethod,
|
||
|
}
|
||
|
metrics.RegisterCounter(GrpcRequestCounter, GrpcRequestCounterDescription)
|
||
|
metrics.AddCount(ctx, GrpcRequestCounter, 1, labels)
|
||
|
}
|
||
|
|
||
|
func RegisterGrpcTotalRequestCounter(ctx context.Context) {
|
||
|
metrics.RegisterCounter(TotalGrpcRequestCounter, TotalGrpcRequestCounterDescription)
|
||
|
metrics.AddCount(ctx, TotalGrpcRequestCounter, 1, nil)
|
||
|
}
|
||
|
|
||
|
func RegisterGrpcRequestCodeCounter(ctx context.Context, info *grpc.UnaryServerInfo, err error) {
|
||
|
statusCode := status.Code(err)
|
||
|
var labels = map[string]interface{}{
|
||
|
GrpcMethod: info.FullMethod,
|
||
|
ReturnCode: runtime.HTTPStatusFromCode(statusCode),
|
||
|
}
|
||
|
metrics.RegisterCounter(GrpcStatusCodeCounter, GrpcStatusCodeCounterDescription)
|
||
|
metrics.AddCount(ctx, GrpcStatusCodeCounter, 1, labels)
|
||
|
}
|
||
|
|
||
|
func containsMetricsMethod(metricType metrics.MetricType, metricTypes []metrics.MetricType) bool {
|
||
|
for _, m := range metricTypes {
|
||
|
if m == metricType {
|
||
|
return true
|
||
|
}
|
||
|
}
|
||
|
return false
|
||
|
}
|