Generates a random traceparent when the tracer is from a NoopTracerPr…

…ovider (#7)

* Generates a random traceparent when the tracer is from a NoopTracerProvider

* Fix test

* no ci lint cache

.github/workflows/main.yml

@@ -43,3 +43,4 @@ jobs:
         uses: golangci/golangci-lint-action@v4
         with:
           version: v1.57
+          skip-cache: true

observability/observability.go

@@ -2,6 +2,8 @@ package observability
 
 import (
 	"context"
+	"crypto/rand"
+	"fmt"
 	"log/slog"
 
 	adapter "github.com/axiomhq/axiom-go/adapters/slog"
@@ -68,12 +70,41 @@ func tracerProvider(serviceName, serviceVersion string, exporters []trace.SpanEx
 	return trace.NewTracerProvider(opts...)
 }
 
+// generateTraceParent generates a random traceparent.
+// ex: 00-51651128eabd3c6f7695b9e6ee0e337d-03e60705fdf5efe0-01
+func generateTraceParent() (string, error) {
+	traceID := oteltrace.TraceID{}
+	_, err := rand.Read(traceID[:])
+	if err != nil {
+		return "", err
+	}
+
+	spanID := oteltrace.SpanID{}
+	_, err = rand.Read(spanID[:])
+	if err != nil {
+		return "", err
+	}
+
+	return fmt.Sprintf("00-%s-%s-01", traceID, spanID), nil
+}
+
 func GetTraceParentOfCurrentSpan(ctx context.Context) string {
 	carrier := propagation.MapCarrier{}
 
 	propagator.Inject(ctx, carrier)
 
-	return carrier.Get("traceparent")
+	traceparent := carrier.Get("traceparent")
+	if traceparent == "" {
+		// If the tracer is from a NoopTracerProvider, it will not generate a traceparent.
+		// In this case, we generate a random one.
+		traceparent, err := generateTraceParent()
+		if err != nil {
+			slog.ErrorContext(ctx, "failed to generate traceparent", "error", err)
+			return ""
+		}
+		return traceparent
+	}
+	return traceparent
 }
 
 func StartJobSpan[Result any](ctx context.Context, tracer oteltrace.Tracer, spanName string, job *workflowsv1.Job, f func(ctx context.Context) (Result, error)) (Result, error) {

workflows/v1/jobs_test.go

@@ -81,9 +81,13 @@ func TestJobService_Submit(t *testing.T) {
 			if len(client.reqs) != 1 {
 				t.Fatalf("Submit() expected 1 request, got %d", len(client.reqs))
 			}
-			if !reflect.DeepEqual(client.reqs[0], tt.wantReq) {
+			if !reflect.DeepEqual(client.reqs[0].GetTasks(), tt.wantReq.GetTasks()) {
 				t.Errorf("Submit() request = %v, want %v", client.reqs[0], tt.wantReq)
 			}
+			if !reflect.DeepEqual(client.reqs[0].GetJobName(), tt.wantReq.GetJobName()) {
+				t.Errorf("Submit() request = %v, want %v", client.reqs[0], tt.wantReq)
+			}
+			// We don't compare traceparent because it's generated randomly
 		})
 	}
 }