Use new OpenTelemetry SpanProcessor and SpanExporter

Signed-off-by: Aleksander Mistewicz <[email protected]>

Author: AwesomePatrol

server/embed/config_tracing.go

@@ -17,6 +17,7 @@ package embed
 import (
 	"context"
 	"fmt"
+	"time"
 
 	"go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc"
 	"go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracegrpc"
@@ -76,6 +77,7 @@ func newTracingExporter(ctx context.Context, cfg *Config) (*tracingExporter, err
 	}
 
 	traceProvider := tracesdk.NewTracerProvider(
+		tracesdk.WithSpanProcessor(traceutil.NewLongSpanProcessor(traceutil.NewLogExporter(cfg.logger), 100*time.Millisecond)),
 		tracesdk.WithBatcher(exporter),
 		tracesdk.WithResource(res),
 		tracesdk.WithSampler(

server/embed/etcd.go

@@ -32,6 +32,7 @@ import (
 	"time"
 
 	"github.com/soheilhy/cmux"
+	tracesdk "go.opentelemetry.io/otel/sdk/trace"
 	"go.uber.org/zap"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/credentials/insecure"
@@ -43,6 +44,7 @@ import (
 	"go.etcd.io/etcd/client/v3/credentials"
 	"go.etcd.io/etcd/pkg/v3/debugutil"
 	runtimeutil "go.etcd.io/etcd/pkg/v3/runtime"
+	"go.etcd.io/etcd/pkg/v3/traceutil"
 	"go.etcd.io/etcd/server/v3/config"
 	"go.etcd.io/etcd/server/v3/etcdserver"
 	"go.etcd.io/etcd/server/v3/etcdserver/api/etcdhttp"
@@ -251,6 +253,12 @@ func StartEtcd(inCfg *Config) (e *Etcd, err error) {
 		e.cfg.logger.Info(
 			"distributed tracing setup enabled",
 		)
+	} else {
+		// Initialize local trace logging even if distributed tracing is disabled.
+		traceProvider := tracesdk.NewTracerProvider(
+			tracesdk.WithSpanProcessor(traceutil.NewLongSpanProcessor(traceutil.NewLogExporter(e.cfg.logger), 100*time.Millisecond)),
+		)
+		traceutil.Init(traceProvider)
 	}
 
 	srvcfg.PeerTLSInfo.LocalAddr = srvcfg.LocalAddress