[VOL-3199] Added support for dynamic enable/disable of Trace Publishing
Change-Id: Iefa126a82dc9ad127ae361b79b1ada429b609f33
diff --git a/internal/pkg/ofagent/changeEvent.go b/internal/pkg/ofagent/changeEvent.go
index 3f5bb7f..6a79d42 100644
--- a/internal/pkg/ofagent/changeEvent.go
+++ b/internal/pkg/ofagent/changeEvent.go
@@ -43,7 +43,7 @@
return
}
opt := grpc.EmptyCallOption{}
- streamCtx, streamDone := context.WithCancel(context.Background())
+ streamCtx, streamDone := context.WithCancel(log.WithSpanFromContext(context.Background(), ctx))
defer streamDone()
stream, err := ofa.volthaClient.Get().ReceiveChangeEvents(streamCtx, &empty.Empty{}, opt)
if err != nil {
diff --git a/internal/pkg/ofagent/connection.go b/internal/pkg/ofagent/connection.go
index 4fe0de3..a518638 100644
--- a/internal/pkg/ofagent/connection.go
+++ b/internal/pkg/ofagent/connection.go
@@ -26,7 +26,6 @@
"github.com/opencord/voltha-lib-go/v3/pkg/log"
"github.com/opencord/voltha-lib-go/v3/pkg/probe"
"github.com/opencord/voltha-protos/v3/go/voltha"
- "github.com/opentracing/opentracing-go"
"google.golang.org/grpc"
)
@@ -47,10 +46,10 @@
conn, err := grpc.Dial(ofa.VolthaApiEndPoint,
grpc.WithInsecure(),
grpc.WithStreamInterceptor(grpc_middleware.ChainStreamClient(
- grpc_opentracing.StreamClientInterceptor(grpc_opentracing.WithTracer(opentracing.GlobalTracer())),
+ grpc_opentracing.StreamClientInterceptor(grpc_opentracing.WithTracer(log.ActiveTracerProxy{})),
)),
grpc.WithUnaryInterceptor(grpc_middleware.ChainUnaryClient(
- grpc_opentracing.UnaryClientInterceptor(grpc_opentracing.WithTracer(opentracing.GlobalTracer())),
+ grpc_opentracing.UnaryClientInterceptor(grpc_opentracing.WithTracer(log.ActiveTracerProxy{})),
)))
if err == nil {
svc := voltha.NewVolthaServiceClient(conn)
diff --git a/internal/pkg/ofagent/packetOut.go b/internal/pkg/ofagent/packetOut.go
index c8bd299..6fe6546 100644
--- a/internal/pkg/ofagent/packetOut.go
+++ b/internal/pkg/ofagent/packetOut.go
@@ -39,8 +39,8 @@
return
}
opt := grpc.EmptyCallOption{}
- streamCtx, streamDone := context.WithCancel(log.WithSpanFromContext(context.Background(), ctx))
- outClient, err := ofa.volthaClient.Get().StreamPacketsOut(streamCtx, opt)
+ streamCtx, streamDone := context.WithCancel(context.Background())
+ outClient, err := ofa.volthaClient.Get().StreamPacketsOut(log.WithSpanFromContext(streamCtx, ctx), opt)
defer streamDone()
if err != nil {
logger.Errorw(ctx, "streamPacketOut Error creating packetout stream ", log.Fields{"error": err})