refactor: add trace

sandwich-go · Dec 3, 2024 · 4ff748c · 4ff748c
1 parent c5bea22
commit 4ff748c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/mux.go b/mux.go
@@ -267,7 +267,10 @@ func (m *mux) blockingMulti(ctx context.Context, cmd []Completed) (resp *redisre
 
 func (m *mux) pipeline(ctx context.Context, cmd Completed) (resp RedisResult) {
 	slot := slotfn(len(m.wire), cmd.Slot(), cmd.NoReply())
+	var finishPick func(error)
+	ctx, finishPick = StartTrace(ctx, "cluster-client.cluster.pick")
 	wire := m.pipe(slot)
+	finishPick(nil)
 	if resp = wire.Do(ctx, cmd); isBroken(resp.NonRedisError(), wire) {
 		m.wire[slot].CompareAndSwap(wire, m.init)
 	}