Optimize Otel instrumentation

internal/pool/conn.go

@@ -9,7 +9,6 @@ import (
 
 	"github.com/go-redis/redis/v8/internal"
 	"github.com/go-redis/redis/v8/internal/proto"
-	"go.opentelemetry.io/otel/trace"
 )
 
 var noDeadline = time.Time{}
@@ -66,7 +65,9 @@ func (cn *Conn) RemoteAddr() net.Addr {
 }
 
 func (cn *Conn) WithReader(ctx context.Context, timeout time.Duration, fn func(rd *proto.Reader) error) error {
-	return internal.WithSpan(ctx, "redis.with_reader", func(ctx context.Context, span trace.Span) error {
+	ctx, span := internal.StartSpan(ctx, "redis.with_reader")
+	defer span.End()
+
 	if err := cn.netConn.SetReadDeadline(cn.deadline(ctx, timeout)); err != nil {
 		return internal.RecordError(ctx, span, err)
 	}
@@ -74,13 +75,14 @@ func (cn *Conn) WithReader(ctx context.Context, timeout time.Duration, fn func(r
 		return internal.RecordError(ctx, span, err)
 	}
 	return nil
-	})
 }
 
 func (cn *Conn) WithWriter(
 	ctx context.Context, timeout time.Duration, fn func(wr *proto.Writer) error,
 ) error {
-	return internal.WithSpan(ctx, "redis.with_writer", func(ctx context.Context, span trace.Span) error {
+	ctx, span := internal.StartSpan(ctx, "redis.with_writer")
+	defer span.End()
+
 	if err := cn.netConn.SetWriteDeadline(cn.deadline(ctx, timeout)); err != nil {
 		return internal.RecordError(ctx, span, err)
 	}
@@ -100,7 +102,6 @@ func (cn *Conn) WithWriter(
 	internal.WritesCounter.Add(ctx, 1)
 
 	return nil
-	})
 }
 
 func (cn *Conn) Close() error {

internal/util.go

@@ -11,7 +11,9 @@ import (
 )
 
 func Sleep(ctx context.Context, dur time.Duration) error {
-	return WithSpan(ctx, "time.Sleep", func(ctx context.Context, span trace.Span) error {
+	_, span := StartSpan(ctx, "time.Sleep")
+	defer span.End()
+
 	t := time.NewTimer(dur)
 	defer t.Stop()
 
@@ -21,7 +23,6 @@ func Sleep(ctx context.Context, dur time.Duration) error {
 	case <-ctx.Done():
 		return ctx.Err()
 	}
-	})
 }
 
 func ToLower(s string) string {
@@ -54,15 +55,11 @@ func isLower(s string) bool {
 
 var tracer = otel.Tracer("github.com/go-redis/redis")
 
-func WithSpan(ctx context.Context, name string, fn func(context.Context, trace.Span) error) error {
+func StartSpan(ctx context.Context, name string) (context.Context, trace.Span) {
 	if span := trace.SpanFromContext(ctx); !span.IsRecording() {
-		return fn(ctx, span)
+		return ctx, span
 	}
-
+	return tracer.Start(ctx, name)
-	ctx, span := tracer.Start(ctx, name)
-	defer span.End()
-
-	return fn(ctx, span)
 }
 
 func RecordError(ctx context.Context, span trace.Span, err error) error {

options.go

@@ -15,7 +15,6 @@ import (
 	"github.com/go-redis/redis/v8/internal"
 	"github.com/go-redis/redis/v8/internal/pool"
 	"go.opentelemetry.io/otel/attribute"
-	"go.opentelemetry.io/otel/trace"
 )
 
 // Limiter is the interface of a rate limiter or a circuit breaker.
@@ -292,20 +291,21 @@ func getUserPassword(u *url.URL) (string, string) {
 func newConnPool(opt *Options) *pool.ConnPool {
 	return pool.NewConnPool(&pool.Options{
 		Dialer: func(ctx context.Context) (net.Conn, error) {
-			var conn net.Conn
+			ctx, span := internal.StartSpan(ctx, "redis.dial")
-			err := internal.WithSpan(ctx, "redis.dial", func(ctx context.Context, span trace.Span) error {
+			defer span.End()
+
+			if span.IsRecording() {
 				span.SetAttributes(
 					attribute.String("db.connection_string", opt.Addr),
 				)
-
-				var err error
-				conn, err = opt.Dialer(ctx, opt.Network, opt.Addr)
-				if err != nil {
-					_ = internal.RecordError(ctx, span, err)
 			}
-				return err
+
-			})
+			cn, err := opt.Dialer(ctx, opt.Network, opt.Addr)
-			return conn, err
+			if err != nil {
+				return nil, internal.RecordError(ctx, span, err)
+			}
+
+			return cn, nil
 		},
 		PoolSize:           opt.PoolSize,
 		MinIdleConns:       opt.MinIdleConns,

redis.go

@@ -11,7 +11,6 @@ import (
 	"github.com/go-redis/redis/v8/internal/pool"
 	"github.com/go-redis/redis/v8/internal/proto"
 	"go.opentelemetry.io/otel/attribute"
-	"go.opentelemetry.io/otel/trace"
 )
 
 // Nil reply returned by Redis when key does not exist.
@@ -214,10 +213,7 @@ func (c *baseClient) _getConn(ctx context.Context) (*pool.Conn, error) {
 		return cn, nil
 	}
 
-	err = internal.WithSpan(ctx, "redis.init_conn", func(ctx context.Context, span trace.Span) error {
+	if err := c.initConn(ctx, cn); err != nil {
-		return c.initConn(ctx, cn)
-	})
-	if err != nil {
 		c.connPool.Remove(ctx, cn, err)
 		if err := errors.Unwrap(err); err != nil {
 			return nil, err
@@ -241,6 +237,9 @@ func (c *baseClient) initConn(ctx context.Context, cn *pool.Conn) error {
 		return nil
 	}
 
+	ctx, span := internal.StartSpan(ctx, "redis.init_conn")
+	defer span.End()
+
 	connPool := pool.NewSingleConnPool(c.connPool, cn)
 	conn := newConn(ctx, c.opt, connPool)
 
@@ -288,7 +287,9 @@ func (c *baseClient) releaseConn(ctx context.Context, cn *pool.Conn, err error)
 func (c *baseClient) withConn(
 	ctx context.Context, fn func(context.Context, *pool.Conn) error,
 ) error {
-	return internal.WithSpan(ctx, "redis.with_conn", func(ctx context.Context, span trace.Span) error {
+	ctx, span := internal.StartSpan(ctx, "redis.with_conn")
+	defer span.End()
+
 	cn, err := c.getConn(ctx)
 	if err != nil {
 		return err
@@ -324,7 +325,6 @@ func (c *baseClient) withConn(
 	case err = <-errc:
 		return err
 	}
-	})
 }
 
 func (c *baseClient) process(ctx context.Context, cmd Cmder) error {
@@ -332,11 +332,20 @@ func (c *baseClient) process(ctx context.Context, cmd Cmder) error {
 	for attempt := 0; attempt <= c.opt.MaxRetries; attempt++ {
 		attempt := attempt
 
-		var retry bool
+		retry, err := c._process(ctx, cmd, attempt)
-		err := internal.WithSpan(ctx, "redis.process", func(ctx context.Context, span trace.Span) error {
+		if err == nil || !retry {
+			return err
+		}
+
+		lastErr = err
+	}
+	return lastErr
+}
+
+func (c *baseClient) _process(ctx context.Context, cmd Cmder, attempt int) (bool, error) {
 	if attempt > 0 {
 		if err := internal.Sleep(ctx, c.retryBackoff(attempt)); err != nil {
-					return err
+			return false, err
 		}
 	}
 
@@ -360,17 +369,11 @@ func (c *baseClient) process(ctx context.Context, cmd Cmder) error {
 		return nil
 	})
 	if err == nil {
-				return nil
+		return false, nil
 	}
-			retry = shouldRetry(err, atomic.LoadUint32(&retryTimeout) == 1)
+
-			return err
+	retry := shouldRetry(err, atomic.LoadUint32(&retryTimeout) == 1)
-		})
+	return retry, err
-		if err == nil || !retry {
-			return err
-		}
-		lastErr = err
-	}
-	return lastErr
 }
 
 func (c *baseClient) retryBackoff(attempt int) time.Duration {