mirror of
https://github.com/pomerium/pomerium.git
synced 2025-05-02 03:46:29 +02:00
3e17befff7
- Update envoy bootstrap config to protobufs - Reorganize tracing config to avoid cyclic import - Push down zipkin config to Envoy - Update tracing options to provide sample rate
80 lines
2.4 KiB
Go
80 lines
2.4 KiB
Go
package trace
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"contrib.go.opencensus.io/exporter/jaeger"
|
|
ocZipkin "contrib.go.opencensus.io/exporter/zipkin"
|
|
"github.com/openzipkin/zipkin-go"
|
|
zipkinHTTP "github.com/openzipkin/zipkin-go/reporter/http"
|
|
"go.opencensus.io/trace"
|
|
|
|
"github.com/pomerium/pomerium/config"
|
|
"github.com/pomerium/pomerium/internal/log"
|
|
)
|
|
|
|
// RegisterTracing creates a new trace exporter from TracingOptions.
|
|
func RegisterTracing(opts *config.TracingOptions) (trace.Exporter, error) {
|
|
var exporter trace.Exporter
|
|
var err error
|
|
switch opts.Provider {
|
|
case config.JaegerTracingProviderName:
|
|
exporter, err = registerJaeger(opts)
|
|
case config.ZipkinTracingProviderName:
|
|
exporter, err = registerZipkin(opts)
|
|
default:
|
|
return nil, fmt.Errorf("telemetry/trace: provider %s unknown", opts.Provider)
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
trace.ApplyConfig(trace.Config{DefaultSampler: trace.ProbabilitySampler(opts.SampleRate)})
|
|
|
|
log.Debug().Interface("Opts", opts).Msg("telemetry/trace: exporter created")
|
|
return exporter, nil
|
|
}
|
|
|
|
// UnregisterTracing unregisters a trace exporter.
|
|
func UnregisterTracing(exporter trace.Exporter) {
|
|
trace.UnregisterExporter(exporter)
|
|
}
|
|
|
|
func registerJaeger(opts *config.TracingOptions) (trace.Exporter, error) {
|
|
jOpts := jaeger.Options{
|
|
ServiceName: opts.Service,
|
|
AgentEndpoint: opts.JaegerAgentEndpoint,
|
|
}
|
|
if opts.JaegerCollectorEndpoint != nil {
|
|
jOpts.CollectorEndpoint = opts.JaegerCollectorEndpoint.String()
|
|
}
|
|
jex, err := jaeger.NewExporter(jOpts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
trace.RegisterExporter(jex)
|
|
return jex, nil
|
|
}
|
|
|
|
func registerZipkin(opts *config.TracingOptions) (trace.Exporter, error) {
|
|
localEndpoint, err := zipkin.NewEndpoint(opts.Service, "")
|
|
if err != nil {
|
|
return nil, fmt.Errorf("telemetry/trace: could not create local endpoint: %w", err)
|
|
}
|
|
|
|
reporter := zipkinHTTP.NewReporter(opts.ZipkinEndpoint.String())
|
|
|
|
exporter := ocZipkin.NewExporter(reporter, localEndpoint)
|
|
trace.RegisterExporter(exporter)
|
|
|
|
return exporter, nil
|
|
}
|
|
|
|
// StartSpan starts a new child span of the current span in the context. If
|
|
// there is no span in the context, creates a new trace and span.
|
|
//
|
|
// Returned context contains the newly created span. You can use it to
|
|
// propagate the returned span in process.
|
|
func StartSpan(ctx context.Context, name string, o ...trace.StartOption) (context.Context, *trace.Span) {
|
|
return trace.StartSpan(ctx, name, o...)
|
|
}
|