replace go.opencensus.io with opentelemetry in idp
This commit is contained in:
@@ -57,6 +57,7 @@ func Server(opts ...Option) (http.Service, error) {
|
|||||||
svc.Logger(options.Logger),
|
svc.Logger(options.Logger),
|
||||||
svc.Config(options.Config),
|
svc.Config(options.Config),
|
||||||
svc.Middleware(
|
svc.Middleware(
|
||||||
|
middleware.TraceContext,
|
||||||
middleware.RealIP,
|
middleware.RealIP,
|
||||||
middleware.RequestID,
|
middleware.RequestID,
|
||||||
middleware.NoCache,
|
middleware.NoCache,
|
||||||
|
|||||||
+40
-69
@@ -1,87 +1,58 @@
|
|||||||
package tracing
|
package tracing
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"time"
|
|
||||||
|
|
||||||
"contrib.go.opencensus.io/exporter/jaeger"
|
|
||||||
"contrib.go.opencensus.io/exporter/ocagent"
|
|
||||||
"contrib.go.opencensus.io/exporter/zipkin"
|
|
||||||
openzipkin "github.com/openzipkin/zipkin-go"
|
|
||||||
zipkinhttp "github.com/openzipkin/zipkin-go/reporter/http"
|
|
||||||
"github.com/owncloud/ocis/idp/pkg/config"
|
"github.com/owncloud/ocis/idp/pkg/config"
|
||||||
"github.com/owncloud/ocis/ocis-pkg/log"
|
"github.com/owncloud/ocis/ocis-pkg/log"
|
||||||
"go.opencensus.io/stats/view"
|
"go.opentelemetry.io/otel/exporters/jaeger"
|
||||||
"go.opencensus.io/trace"
|
"go.opentelemetry.io/otel/propagation"
|
||||||
|
"go.opentelemetry.io/otel/sdk/resource"
|
||||||
|
sdktrace "go.opentelemetry.io/otel/sdk/trace"
|
||||||
|
semconv "go.opentelemetry.io/otel/semconv/v1.4.0"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
// Propagator ensures the entire module uses the same trace propagation strategy.
|
||||||
|
Propagator propagation.TextMapPropagator
|
||||||
|
|
||||||
|
// TraceProvider is the global trace provider for the proxy service.
|
||||||
|
TraceProvider *sdktrace.TracerProvider
|
||||||
)
|
)
|
||||||
|
|
||||||
func Configure(cfg *config.Config, logger log.Logger) error {
|
func Configure(cfg *config.Config, logger log.Logger) error {
|
||||||
if cfg.Tracing.Enabled {
|
if cfg.Tracing.Enabled {
|
||||||
switch t := cfg.Tracing.Type; t {
|
switch t := cfg.Tracing.Type; t {
|
||||||
case "agent":
|
|
||||||
exporter, err := ocagent.NewExporter(
|
|
||||||
ocagent.WithReconnectionPeriod(5*time.Second),
|
|
||||||
ocagent.WithAddress(cfg.Tracing.Endpoint),
|
|
||||||
ocagent.WithServiceName(cfg.Tracing.Service),
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
logger.Error().
|
|
||||||
Err(err).
|
|
||||||
Str("endpoint", cfg.Tracing.Endpoint).
|
|
||||||
Str("collector", cfg.Tracing.Collector).
|
|
||||||
Msg("Failed to create agent tracing")
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
trace.RegisterExporter(exporter)
|
|
||||||
view.RegisterExporter(exporter)
|
|
||||||
case "jaeger":
|
case "jaeger":
|
||||||
exporter, err := jaeger.NewExporter(
|
{
|
||||||
jaeger.Options{
|
exp, err := jaeger.New(
|
||||||
AgentEndpoint: cfg.Tracing.Endpoint,
|
jaeger.WithCollectorEndpoint(
|
||||||
CollectorEndpoint: cfg.Tracing.Collector,
|
jaeger.WithEndpoint(cfg.Tracing.Collector),
|
||||||
Process: jaeger.Process{
|
|
||||||
ServiceName: cfg.Tracing.Service,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
logger.Error().
|
|
||||||
Err(err).
|
|
||||||
Str("endpoint", cfg.Tracing.Endpoint).
|
|
||||||
Str("collector", cfg.Tracing.Collector).
|
|
||||||
Msg("Failed to create jaeger tracing")
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
trace.RegisterExporter(exporter)
|
|
||||||
case "zipkin":
|
|
||||||
endpoint, err := openzipkin.NewEndpoint(
|
|
||||||
cfg.Tracing.Service,
|
|
||||||
cfg.Tracing.Endpoint,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
logger.Error().
|
|
||||||
Err(err).
|
|
||||||
Str("endpoint", cfg.Tracing.Endpoint).
|
|
||||||
Str("collector", cfg.Tracing.Collector).
|
|
||||||
Msg("Failed to create zipkin tracing")
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
exporter := zipkin.NewExporter(
|
|
||||||
zipkinhttp.NewReporter(
|
|
||||||
cfg.Tracing.Collector,
|
|
||||||
),
|
),
|
||||||
endpoint,
|
|
||||||
)
|
)
|
||||||
trace.RegisterExporter(exporter)
|
if err != nil {
|
||||||
default:
|
panic(err)
|
||||||
logger.Warn().
|
|
||||||
Str("type", t).
|
|
||||||
Msg("Unknown tracing backend")
|
|
||||||
}
|
}
|
||||||
trace.ApplyConfig(
|
|
||||||
trace.Config{
|
// set package level trace provider and propagator.
|
||||||
DefaultSampler: trace.AlwaysSample(),
|
TraceProvider = sdktrace.NewTracerProvider(
|
||||||
},
|
sdktrace.WithBatcher(exp),
|
||||||
|
sdktrace.WithResource(resource.NewWithAttributes(
|
||||||
|
semconv.SchemaURL,
|
||||||
|
semconv.ServiceNameKey.String("idp")),
|
||||||
|
),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
Propagator = propagation.NewCompositeTextMapPropagator(
|
||||||
|
propagation.Baggage{},
|
||||||
|
propagation.TraceContext{},
|
||||||
|
)
|
||||||
|
}
|
||||||
|
case "agent":
|
||||||
|
fallthrough
|
||||||
|
case "zipkin":
|
||||||
|
fallthrough
|
||||||
|
default:
|
||||||
|
logger.Warn().Str("type", t).Msg("Unknown tracing backend")
|
||||||
|
}
|
||||||
} else {
|
} else {
|
||||||
logger.Debug().
|
logger.Debug().
|
||||||
Msg("Tracing is not enabled")
|
Msg("Tracing is not enabled")
|
||||||
|
|||||||
Reference in New Issue
Block a user