Merge pull request #6 from runloopai/dines/fix-parent-id

Fix parent id for traces

Author: dines-rl

cmd/overlaybd-snapshotter/main.go

@@ -51,6 +51,7 @@ func requestIDInterceptor(ctx context.Context, req interface{}, info *grpc.Unary
 	requestID := tracing.GetRequestID(ctx)
 	if requestID == "" {
 		requestID = mylog.GenerateRequestID()
+		ctx = tracing.SetRequestID(ctx, requestID)
 	}
 
 	ctx = mylog.WithRequestID(ctx, requestID)
@@ -143,7 +144,7 @@ func main() {
 		tracing.WithServerTracing(),
 		grpc.UnaryInterceptor(requestIDInterceptor),
 	)
-	snapshotsapi.RegisterSnapshotsServer(srv, snapshotservice.FromSnapshotter(sn))
+	snapshotsapi.RegisterSnapshotsServer(srv, tracing.WithTracing(snapshotservice.FromSnapshotter(sn)))
 
 	address := strings.TrimSpace(pconfig.Address)
 

pkg/convertor/convertor.go

@@ -678,6 +678,7 @@ func (c *overlaybdConvertor) applyOCIV1LayerInObd(
 	afterApply func(root string) error, // do something after apply tar stream
 ) (string, error) {
 	startTime := time.Now()
+	
 	ctx, span := tracer.Start(ctx, "applyOCIV1LayerInObd",
 		trace.WithAttributes(
 			attribute.String("parent_id", parentID),