Restore the ability to reset current span in context nil

gocontext.go

@@ -7,10 +7,12 @@ type contextKey struct{}
 var activeSpanKey = contextKey{}
 
 // ContextWithSpan returns a new `context.Context` that holds a reference to
-// `span`'s SpanContext.
+// the span. If span is nil, a new context without an active span is returned.
 func ContextWithSpan(ctx context.Context, span Span) context.Context {
-	if tracerWithHook, ok := span.Tracer().(TracerContextWithSpanExtension); ok {
-		ctx = tracerWithHook.ContextWithSpanHook(ctx, span)
+	if span != nil {
+		if tracerWithHook, ok := span.Tracer().(TracerContextWithSpanExtension); ok {
+			ctx = tracerWithHook.ContextWithSpanHook(ctx, span)
+		}
 	}
 	return context.WithValue(ctx, activeSpanKey, span)
 }

gocontext_test.go

@@ -27,6 +27,11 @@ func TestContextWithSpan(t *testing.T) {
 	if span != span2 {
 		t.Errorf("Not the same span returned from context, expected=%+v, actual=%+v", span, span2)
 	}
+
+	ctx = ContextWithSpan(ctx, nil)
+	if s := SpanFromContext(ctx); s != nil {
+		t.Errorf("Not able to reset span in context, expected=nil, actual=%+v", s)
+	}
 }
 
 type noopExtTracer struct {