mirror of https://github.com/go-redis/redis.git
Compare commits
1 Commits
a2d86d85a0
...
f1ed74fa2c
Author | SHA1 | Date |
---|---|---|
Andreas Bergmeier | f1ed74fa2c |
|
@ -8,7 +8,7 @@ jobs:
|
||||||
- name: Checkout
|
- name: Checkout
|
||||||
uses: actions/checkout@v4
|
uses: actions/checkout@v4
|
||||||
- name: Check Spelling
|
- name: Check Spelling
|
||||||
uses: rojopolis/spellcheck-github-actions@0.45.0
|
uses: rojopolis/spellcheck-github-actions@0.40.0
|
||||||
with:
|
with:
|
||||||
config_path: .github/spellcheck-settings.yml
|
config_path: .github/spellcheck-settings.yml
|
||||||
task_name: Markdown
|
task_name: Markdown
|
||||||
|
|
|
@ -167,8 +167,6 @@ func (cmd *baseCmd) stringArg(pos int) string {
|
||||||
switch v := arg.(type) {
|
switch v := arg.(type) {
|
||||||
case string:
|
case string:
|
||||||
return v
|
return v
|
||||||
case []byte:
|
|
||||||
return string(v)
|
|
||||||
default:
|
default:
|
||||||
// TODO: consider using appendArg
|
// TODO: consider using appendArg
|
||||||
return fmt.Sprint(v)
|
return fmt.Sprint(v)
|
||||||
|
|
|
@ -21,10 +21,6 @@ import (
|
||||||
"github.com/redis/go-redis/v9/internal/rand"
|
"github.com/redis/go-redis/v9/internal/rand"
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
|
||||||
minLatencyMeasurementInterval = 10 * time.Second
|
|
||||||
)
|
|
||||||
|
|
||||||
var errClusterNoNodes = fmt.Errorf("redis: cluster has no nodes")
|
var errClusterNoNodes = fmt.Errorf("redis: cluster has no nodes")
|
||||||
|
|
||||||
// ClusterOptions are used to configure a cluster client and should be
|
// ClusterOptions are used to configure a cluster client and should be
|
||||||
|
@ -320,10 +316,6 @@ type clusterNode struct {
|
||||||
latency uint32 // atomic
|
latency uint32 // atomic
|
||||||
generation uint32 // atomic
|
generation uint32 // atomic
|
||||||
failing uint32 // atomic
|
failing uint32 // atomic
|
||||||
|
|
||||||
// last time the latency measurement was performed for the node, stored in nanoseconds
|
|
||||||
// from epoch
|
|
||||||
lastLatencyMeasurement int64 // atomic
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func newClusterNode(clOpt *ClusterOptions, addr string) *clusterNode {
|
func newClusterNode(clOpt *ClusterOptions, addr string) *clusterNode {
|
||||||
|
@ -376,7 +368,6 @@ func (n *clusterNode) updateLatency() {
|
||||||
latency = float64(dur) / float64(successes)
|
latency = float64(dur) / float64(successes)
|
||||||
}
|
}
|
||||||
atomic.StoreUint32(&n.latency, uint32(latency+0.5))
|
atomic.StoreUint32(&n.latency, uint32(latency+0.5))
|
||||||
n.SetLastLatencyMeasurement(time.Now())
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (n *clusterNode) Latency() time.Duration {
|
func (n *clusterNode) Latency() time.Duration {
|
||||||
|
@ -406,10 +397,6 @@ func (n *clusterNode) Generation() uint32 {
|
||||||
return atomic.LoadUint32(&n.generation)
|
return atomic.LoadUint32(&n.generation)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (n *clusterNode) LastLatencyMeasurement() int64 {
|
|
||||||
return atomic.LoadInt64(&n.lastLatencyMeasurement)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (n *clusterNode) SetGeneration(gen uint32) {
|
func (n *clusterNode) SetGeneration(gen uint32) {
|
||||||
for {
|
for {
|
||||||
v := atomic.LoadUint32(&n.generation)
|
v := atomic.LoadUint32(&n.generation)
|
||||||
|
@ -419,15 +406,6 @@ func (n *clusterNode) SetGeneration(gen uint32) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (n *clusterNode) SetLastLatencyMeasurement(t time.Time) {
|
|
||||||
for {
|
|
||||||
v := atomic.LoadInt64(&n.lastLatencyMeasurement)
|
|
||||||
if t.UnixNano() < v || atomic.CompareAndSwapInt64(&n.lastLatencyMeasurement, v, t.UnixNano()) {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------
|
||||||
|
|
||||||
type clusterNodes struct {
|
type clusterNodes struct {
|
||||||
|
@ -515,11 +493,10 @@ func (c *clusterNodes) GC(generation uint32) {
|
||||||
c.mu.Lock()
|
c.mu.Lock()
|
||||||
|
|
||||||
c.activeAddrs = c.activeAddrs[:0]
|
c.activeAddrs = c.activeAddrs[:0]
|
||||||
now := time.Now()
|
|
||||||
for addr, node := range c.nodes {
|
for addr, node := range c.nodes {
|
||||||
if node.Generation() >= generation {
|
if node.Generation() >= generation {
|
||||||
c.activeAddrs = append(c.activeAddrs, addr)
|
c.activeAddrs = append(c.activeAddrs, addr)
|
||||||
if c.opt.RouteByLatency && node.LastLatencyMeasurement() < now.Add(-minLatencyMeasurementInterval).UnixNano() {
|
if c.opt.RouteByLatency {
|
||||||
go node.updateLatency()
|
go node.updateLatency()
|
||||||
}
|
}
|
||||||
continue
|
continue
|
||||||
|
|
|
@ -653,32 +653,6 @@ var _ = Describe("ClusterClient", func() {
|
||||||
Expect(client.Close()).NotTo(HaveOccurred())
|
Expect(client.Close()).NotTo(HaveOccurred())
|
||||||
})
|
})
|
||||||
|
|
||||||
It("determines hash slots correctly for generic commands", func() {
|
|
||||||
opt := redisClusterOptions()
|
|
||||||
opt.MaxRedirects = -1
|
|
||||||
client := cluster.newClusterClient(ctx, opt)
|
|
||||||
|
|
||||||
err := client.Do(ctx, "GET", "A").Err()
|
|
||||||
Expect(err).To(Equal(redis.Nil))
|
|
||||||
|
|
||||||
err = client.Do(ctx, []byte("GET"), []byte("A")).Err()
|
|
||||||
Expect(err).To(Equal(redis.Nil))
|
|
||||||
|
|
||||||
Eventually(func() error {
|
|
||||||
return client.SwapNodes(ctx, "A")
|
|
||||||
}, 30*time.Second).ShouldNot(HaveOccurred())
|
|
||||||
|
|
||||||
err = client.Do(ctx, "GET", "A").Err()
|
|
||||||
Expect(err).To(HaveOccurred())
|
|
||||||
Expect(err.Error()).To(ContainSubstring("MOVED"))
|
|
||||||
|
|
||||||
err = client.Do(ctx, []byte("GET"), []byte("A")).Err()
|
|
||||||
Expect(err).To(HaveOccurred())
|
|
||||||
Expect(err.Error()).To(ContainSubstring("MOVED"))
|
|
||||||
|
|
||||||
Expect(client.Close()).NotTo(HaveOccurred())
|
|
||||||
})
|
|
||||||
|
|
||||||
It("follows node redirection immediately", func() {
|
It("follows node redirection immediately", func() {
|
||||||
// Configure retry backoffs far in excess of the expected duration of redirection
|
// Configure retry backoffs far in excess of the expected duration of redirection
|
||||||
opt := redisClusterOptions()
|
opt := redisClusterOptions()
|
||||||
|
|
2
redis.go
2
redis.go
|
@ -176,6 +176,8 @@ func (hs *hooksMixin) withProcessPipelineHook(
|
||||||
}
|
}
|
||||||
|
|
||||||
func (hs *hooksMixin) dialHook(ctx context.Context, network, addr string) (net.Conn, error) {
|
func (hs *hooksMixin) dialHook(ctx context.Context, network, addr string) (net.Conn, error) {
|
||||||
|
hs.hooksMu.Lock()
|
||||||
|
defer hs.hooksMu.Unlock()
|
||||||
return hs.current.dial(ctx, network, addr)
|
return hs.current.dial(ctx, network, addr)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -6,7 +6,6 @@ import (
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"net"
|
"net"
|
||||||
"sync"
|
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
@ -634,67 +633,3 @@ var _ = Describe("Hook with MinIdleConns", func() {
|
||||||
}))
|
}))
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|
||||||
var _ = Describe("Dialer connection timeouts", func() {
|
|
||||||
var client *redis.Client
|
|
||||||
|
|
||||||
const dialSimulatedDelay = 1 * time.Second
|
|
||||||
|
|
||||||
BeforeEach(func() {
|
|
||||||
options := redisOptions()
|
|
||||||
options.Dialer = func(ctx context.Context, network, addr string) (net.Conn, error) {
|
|
||||||
// Simulated slow dialer.
|
|
||||||
// Note that the following sleep is deliberately not context-aware.
|
|
||||||
time.Sleep(dialSimulatedDelay)
|
|
||||||
return net.Dial("tcp", options.Addr)
|
|
||||||
}
|
|
||||||
options.MinIdleConns = 1
|
|
||||||
client = redis.NewClient(options)
|
|
||||||
})
|
|
||||||
|
|
||||||
AfterEach(func() {
|
|
||||||
err := client.Close()
|
|
||||||
Expect(err).NotTo(HaveOccurred())
|
|
||||||
})
|
|
||||||
|
|
||||||
It("does not contend on connection dial for concurrent commands", func() {
|
|
||||||
var wg sync.WaitGroup
|
|
||||||
|
|
||||||
const concurrency = 10
|
|
||||||
|
|
||||||
durations := make(chan time.Duration, concurrency)
|
|
||||||
errs := make(chan error, concurrency)
|
|
||||||
|
|
||||||
start := time.Now()
|
|
||||||
wg.Add(concurrency)
|
|
||||||
|
|
||||||
for i := 0; i < concurrency; i++ {
|
|
||||||
go func() {
|
|
||||||
defer wg.Done()
|
|
||||||
|
|
||||||
start := time.Now()
|
|
||||||
err := client.Ping(ctx).Err()
|
|
||||||
durations <- time.Since(start)
|
|
||||||
errs <- err
|
|
||||||
}()
|
|
||||||
}
|
|
||||||
|
|
||||||
wg.Wait()
|
|
||||||
close(durations)
|
|
||||||
close(errs)
|
|
||||||
|
|
||||||
// All commands should eventually succeed, after acquiring a connection.
|
|
||||||
for err := range errs {
|
|
||||||
Expect(err).NotTo(HaveOccurred())
|
|
||||||
}
|
|
||||||
|
|
||||||
// Each individual command should complete within the simulated dial duration bound.
|
|
||||||
for duration := range durations {
|
|
||||||
Expect(duration).To(BeNumerically("<", 2*dialSimulatedDelay))
|
|
||||||
}
|
|
||||||
|
|
||||||
// Due to concurrent execution, the entire test suite should also complete within
|
|
||||||
// the same dial duration bound applied for individual commands.
|
|
||||||
Expect(time.Since(start)).To(BeNumerically("<", 2*dialSimulatedDelay))
|
|
||||||
})
|
|
||||||
})
|
|
||||||
|
|
Loading…
Reference in New Issue