foomo/keel
1
0
Fork 0
mirror of https://github.com/foomo/keel.git synced 2025-10-16 12:35:34 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

wip: cleanup

Browse Source
This commit is contained in:
franklin 2023-08-11 17:43:16 +02:00
parent 5f01366d8c
commit 20c71407db
1 changed files with 9 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
integration/gotsrpc/net/middleware/telemetry.go
View File

@ -117,7 +117,7 @@ func Telemetry(opts ...TelemetryOption) middleware.Middleware {
// TelemetryWithOptions middleware
func TelemetryWithOptions(opts TelemetryOptions) middleware.Middleware {
observe := func(l *zap.Logger, r *http.Request, metric prometheus.ObserverVec, stats *gotsrpc.CallStats, operation string) {
observe := func(r *http.Request, metric prometheus.ObserverVec, stats *gotsrpc.CallStats, operation string) {
observer := metric.WithLabelValues(
stats.Func,
stats.Service,
@ -135,24 +135,12 @@ func TelemetryWithOptions(opts TelemetryOptions) middleware.Middleware {
duration = stats.Execution
}
spanCtx := trace.SpanContextFromContext(r.Context())
if v, ok := observer.(prometheus.ExemplarObserver); ok && opts.Exemplars && spanCtx.HasTraceID() {
l.Info("==> exemplar")
v.ObserveWithExemplar(duration.Seconds(), prometheus.Labels{
if exemplarObserver, ok := observer.(prometheus.ExemplarObserver); ok && opts.Exemplars && spanCtx.HasTraceID() && spanCtx.IsSampled() {
exemplarObserver.ObserveWithExemplar(duration.Seconds(), prometheus.Labels{
"traceID": spanCtx.TraceID().String(),
})
return
}
l.Info("==> no exemplar")
if !opts.Exemplars {
l.Info("==> not exemplar")
}
if !spanCtx.HasTraceID() {
l.Info("==> not traceID")
}
if !spanCtx.IsSampled() {
l.Info("==> not sampled")
}
observer.Observe(duration.Seconds())
}
return func(l *zap.Logger, name string, next http.Handler) http.Handler {
@ -164,16 +152,16 @@ func TelemetryWithOptions(opts TelemetryOptions) middleware.Middleware {
if stats, ok := gotsrpc.GetStatsForRequest(r); ok {
// create custom metics
if opts.Marshalling {
observe(l, r, gotsrpcRequestDurationSummary, stats, "marshalling")
observe(l, r, gotsrpcRequestDurationHistogram, stats, "marshalling")
observe(r, gotsrpcRequestDurationSummary, stats, "marshalling")
observe(r, gotsrpcRequestDurationHistogram, stats, "marshalling")
}
if opts.Unmarshalling {
observe(l, r, gotsrpcRequestDurationSummary, stats, "unmarshalling")
observe(l, r, gotsrpcRequestDurationHistogram, stats, "unmarshalling")
observe(r, gotsrpcRequestDurationSummary, stats, "unmarshalling")
observe(r, gotsrpcRequestDurationHistogram, stats, "unmarshalling")
}
if opts.Execution {
observe(l, r, gotsrpcRequestDurationSummary, stats, "execution")
observe(l, r, gotsrpcRequestDurationHistogram, stats, "execution")
observe(r, gotsrpcRequestDurationSummary, stats, "execution")
observe(r, gotsrpcRequestDurationHistogram, stats, "execution")
}
// enrich logger