redis/go-redis
1
0
Fork 0
mirror of https://github.com/redis/go-redis.git synced 2025-07-18 00:20:57 +03:00
Code Activity

Merge branch 'master' into ndyakov/keyless-commands-tx-pipeline

Browse Source
This commit is contained in:
Nedyalko Dyakov
2025-06-19 11:42:16 +03:00
committed by GitHub
parent 2a32680c8a f4358acffc
commit fd2f7044cb
1 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
osscluster.go
View File

@ -340,6 +340,7 @@ type clusterNode struct {
latency uint32 // atomic latency uint32 // atomic
generation uint32 // atomic generation uint32 // atomic
failing uint32 // atomic failing uint32 // atomic
loaded uint32 // atomic
// last time the latency measurement was performed for the node, stored in nanoseconds // last time the latency measurement was performed for the node, stored in nanoseconds
// from epoch // from epoch
@ -406,6 +407,7 @@ func (n *clusterNode) Latency() time.Duration {
func (n *clusterNode) MarkAsFailing() { func (n *clusterNode) MarkAsFailing() {
atomic.StoreUint32(&n.failing, uint32(time.Now().Unix())) atomic.StoreUint32(&n.failing, uint32(time.Now().Unix()))
atomic.StoreUint32(&n.loaded, 0)
} }
func (n *clusterNode) Failing() bool { func (n *clusterNode) Failing() bool {
@ -449,11 +451,21 @@ func (n *clusterNode) SetLastLatencyMeasurement(t time.Time) {
} }
func (n *clusterNode) Loading() bool { func (n *clusterNode) Loading() bool {
loaded := atomic.LoadUint32(&n.loaded)
if loaded == 1 {
return false
}
// check if the node is loading
ctx, cancel := context.WithTimeout(context.Background(), 100*time.Millisecond) ctx, cancel := context.WithTimeout(context.Background(), 100*time.Millisecond)
defer cancel() defer cancel()
err := n.Client.Ping(ctx).Err() err := n.Client.Ping(ctx).Err()
return err != nil && isLoadingError(err) loading := err != nil && isLoadingError(err)
if !loading {
atomic.StoreUint32(&n.loaded, 1)
}
return loading
} }
//------------------------------------------------------------------------------ //------------------------------------------------------------------------------