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

feat: add env configuration

Browse Source
This commit is contained in:
franklin 2023-08-14 13:21:49 +02:00
parent 0a682498d4
commit ab382f801a
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -10,6 +10,7 @@ import (
"time"
"github.com/foomo/gotsrpc/v2"
"github.com/foomo/keel/env"
httplog "github.com/foomo/keel/net/http/log"
"github.com/foomo/keel/telemetry"
"github.com/prometheus/client_golang/prometheus"
@ -36,7 +37,7 @@ const (
type (
TelemetryOptions struct {
Exemplars bool
ExemplarsDisabled bool
ObserveExecution bool
ObserveMarshalling bool
ObserveUnmarshalling bool
@ -77,18 +78,18 @@ var (
// DefaultTelemetryOptions returns the default options
func DefaultTelemetryOptions() TelemetryOptions {
return TelemetryOptions{
Exemplars: false,
ObserveExecution: true,
ObserveMarshalling: false,
ObserveUnmarshalling: false,
PayloadAttributeDisabled: true,
PayloadAttributeDisabled: env.GetBool("OTEL_GOTSRPC_PAYLOAD_ATTRIBUTE_DISABLED", true),
ExemplarsDisabled: env.GetBool("OTEL_GOTSRPC_EXEMPLARS_DISABLED", false),
}
}
// TelemetryWithExemplars middleware option
func TelemetryWithExemplars(v bool) TelemetryOption {
// TelemetryWithExemplarsDisabled middleware option
func TelemetryWithExemplarsDisabled(v bool) TelemetryOption {
return func(o *TelemetryOptions) {
o.Exemplars = v
o.ExemplarsDisabled = v
}
}
@ -150,7 +151,7 @@ func TelemetryWithOptions(opts TelemetryOptions) middleware.Middleware {
case "execution":
duration = stats.Execution
}
if exemplarObserver, ok := observer.(prometheus.ExemplarObserver); ok && opts.Exemplars && spanCtx.HasTraceID() && spanCtx.IsSampled() {
if exemplarObserver, ok := observer.(prometheus.ExemplarObserver); ok && opts.ExemplarsDisabled && spanCtx.HasTraceID() && spanCtx.IsSampled() {
exemplarObserver.ObserveWithExemplar(duration.Seconds(), prometheus.Labels{
"traceID": spanCtx.TraceID().String(),
})