Add ctx as first arg

internal/util.go

@@ -2,21 +2,28 @@ package internal
 
 import (
 	"context"
+	"reflect"
 	"time"
 
 	"github.com/go-redis/redis/v7/internal/util"
+	"go.opentelemetry.io/otel/api/core"
+	"go.opentelemetry.io/otel/api/global"
+	"go.opentelemetry.io/otel/api/trace"
+	"google.golang.org/grpc/codes"
 )
 
 func Sleep(ctx context.Context, dur time.Duration) error {
-	t := time.NewTimer(dur)
-	defer t.Stop()
+	return WithSpan(ctx, "sleep", func(ctx context.Context) error {
+		t := time.NewTimer(dur)
+		defer t.Stop()
 
-	select {
-	case <-t.C:
-		return nil
-	case <-ctx.Done():
-		return ctx.Err()
-	}
+		select {
+		case <-t.C:
+			return nil
+		case <-ctx.Done():
+			return ctx.Err()
+		}
+	})
 }
 
 func ToLower(s string) string {
@@ -54,3 +61,27 @@ func Unwrap(err error) error {
 	}
 	return u.Unwrap()
 }
+
+var (
+	logTypeKey    = core.Key("log.type")
+	logMessageKey = core.Key("log.message")
+)
+
+func WithSpan(ctx context.Context, name string, fn func(context.Context) error) error {
+	if !trace.SpanFromContext(ctx).IsRecording() {
+		return fn(ctx)
+	}
+
+	ctx, span := global.TraceProvider().Tracer("go-redis").Start(ctx, name)
+	defer span.End()
+
+	if err := fn(ctx); err != nil {
+		span.SetStatus(codes.Internal)
+		span.AddEvent(ctx, "error",
+			logTypeKey.String(reflect.TypeOf(err).String()),
+			logMessageKey.String(err.Error()),
+		)
+		return err
+	}
+	return nil
+}