Sanket SudakeandGitHub
33473a4528
* Capture important open telemetry events with span in fission * Add context to missing HTTP calls * Add Trace ID in logs * capture trace id in the proxy handler function * Always registry tracer to get traceID Signed-off-by: Sanket Sudake <sanketsudake@gmail.com>
57 lines
1.3 KiB
Go
57 lines
1.3 KiB
Go
package tracing
|
|
|
|
import (
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"contrib.go.opencensus.io/exporter/jaeger"
|
|
"go.opencensus.io/trace"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
func TracingEnabled(logger *zap.Logger) bool {
|
|
openTracingEnabled, err := strconv.ParseBool(os.Getenv("OPENTRACING_ENABLED"))
|
|
if err != nil {
|
|
if logger != nil {
|
|
logger.Error("Error parsing OpenTracing enabled flag", zap.Error(err))
|
|
}
|
|
return false
|
|
}
|
|
return openTracingEnabled
|
|
}
|
|
|
|
func RegisterTraceExporter(logger *zap.Logger, collectorEndpoint, serviceName string) error {
|
|
if len(collectorEndpoint) == 0 {
|
|
logger.Info("skipping trace exporter registration")
|
|
return nil
|
|
}
|
|
|
|
exporter, err := jaeger.NewExporter(jaeger.Options{
|
|
CollectorEndpoint: collectorEndpoint,
|
|
Process: jaeger.Process{
|
|
ServiceName: serviceName,
|
|
Tags: []jaeger.Tag{
|
|
jaeger.BoolTag("fission", true),
|
|
},
|
|
},
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
trace.RegisterExporter(exporter)
|
|
|
|
if strings.EqualFold(serviceName, "Fission-Fetcher") {
|
|
trace.ApplyConfig(trace.Config{DefaultSampler: trace.AlwaysSample()})
|
|
} else {
|
|
samplingRate, err := strconv.ParseFloat(os.Getenv("TRACING_SAMPLING_RATE"), 32)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
trace.ApplyConfig(trace.Config{DefaultSampler: trace.ProbabilitySampler(samplingRate)})
|
|
}
|
|
|
|
return nil
|
|
}
|