Bump Redis client version to v8.11.4
[ric-plt/sdlgo.git] / internal / sdlgoredis / sdlgosentinel.go
index 014d4ba..8a1b331 100644 (file)
 package sdlgoredis
 
 import (
-       "github.com/go-redis/redis/v7"
+       "context"
+       "fmt"
+       "github.com/go-redis/redis/v8"
+       "strconv"
 )
 
 type Sentinel struct {
+       ctx context.Context
        IredisSentinelClient
        Cfg *Config
 }
 
 type IredisSentinelClient interface {
-       Master(name string) *redis.StringStringMapCmd
-       Slaves(name string) *redis.SliceCmd
-       Sentinels(name string) *redis.SliceCmd
+       Master(ctx context.Context, name string) *redis.StringStringMapCmd
+       Slaves(ctx context.Context, name string) *redis.SliceCmd
+       Sentinels(ctx context.Context, name string) *redis.SliceCmd
 }
 
 type RedisSentinelCreateCb func(cfg *Config, addr string) *Sentinel
@@ -42,6 +46,7 @@ type RedisSentinelCreateCb func(cfg *Config, addr string) *Sentinel
 func newRedisSentinel(cfg *Config, addr string) *Sentinel {
        redisAddress := addr + ":" + cfg.sentinelPort
        return &Sentinel{
+               ctx: context.Background(),
                IredisSentinelClient: redis.NewSentinelClient(&redis.Options{
                        Addr:       redisAddress,
                        Password:   "", // no password set
@@ -55,14 +60,22 @@ func newRedisSentinel(cfg *Config, addr string) *Sentinel {
 
 func (s *Sentinel) GetDbState() (*DbState, error) {
        state := new(DbState)
-       mState, mErr := s.getMasterDbState()
+       pState, pErr := s.getPrimaryDbState()
        rState, rErr := s.getReplicasState()
        sState, sErr := s.getSentinelsState()
-       state.MasterDbState = *mState
+       state.PrimaryDbState = *pState
        state.ReplicasDbState = rState
        state.SentinelsDbState = sState
-       if mErr != nil {
-               return state, mErr
+
+       cnt, err := strconv.Atoi(s.Cfg.nodeCnt)
+       if err != nil {
+               state.Err = fmt.Errorf("Sentinel DBAAS_NODE_COUNT configuration value '%s' conversion to integer failed", s.Cfg.nodeCnt)
+               return state, state.Err
+       }
+       state.ConfigNodeCnt = cnt
+
+       if pErr != nil {
+               return state, pErr
        }
        if rErr != nil {
                return state, rErr
@@ -70,9 +83,9 @@ func (s *Sentinel) GetDbState() (*DbState, error) {
        return state, sErr
 }
 
-func (s *Sentinel) getMasterDbState() (*MasterDbState, error) {
-       state := new(MasterDbState)
-       redisVal, redisErr := s.Master(s.Cfg.masterName).Result()
+func (s *Sentinel) getPrimaryDbState() (*PrimaryDbState, error) {
+       state := new(PrimaryDbState)
+       redisVal, redisErr := s.Master(s.ctx, s.Cfg.masterName).Result()
        if redisErr == nil {
                state.Fields.Ip = redisVal["ip"]
                state.Fields.Port = redisVal["port"]
@@ -87,10 +100,10 @@ func (s *Sentinel) getReplicasState() (*ReplicasDbState, error) {
        states := new(ReplicasDbState)
        states.States = make([]*ReplicaDbState, 0)
 
-       redisVal, redisErr := s.Slaves(s.Cfg.masterName).Result()
+       redisVal, redisErr := s.Slaves(s.ctx, s.Cfg.masterName).Result()
        if redisErr == nil {
-               for _, redisSlave := range redisVal {
-                       replicaState := readReplicaState(redisSlave.([]interface{}))
+               for _, redisReplica := range redisVal {
+                       replicaState := readReplicaState(redisReplica.([]interface{}))
                        states.States = append(states.States, replicaState)
                }
        }
@@ -98,19 +111,19 @@ func (s *Sentinel) getReplicasState() (*ReplicasDbState, error) {
        return states, redisErr
 }
 
-func readReplicaState(redisSlaves []interface{}) *ReplicaDbState {
+func readReplicaState(redisReplicas []interface{}) *ReplicaDbState {
        state := new(ReplicaDbState)
-       for i := 0; i < len(redisSlaves); i += 2 {
-               if redisSlaves[i].(string) == "ip" {
-                       state.Fields.Ip = redisSlaves[i+1].(string)
-               } else if redisSlaves[i].(string) == "port" {
-                       state.Fields.Port = redisSlaves[i+1].(string)
-               } else if redisSlaves[i].(string) == "flags" {
-                       state.Fields.Flags = redisSlaves[i+1].(string)
-               } else if redisSlaves[i].(string) == "role-reported" {
-                       state.Fields.Role = redisSlaves[i+1].(string)
-               } else if redisSlaves[i].(string) == "master-link-status" {
-                       state.Fields.MasterLinkStatus = redisSlaves[i+1].(string)
+       for i := 0; i < len(redisReplicas); i += 2 {
+               if redisReplicas[i].(string) == "ip" {
+                       state.Fields.Ip = redisReplicas[i+1].(string)
+               } else if redisReplicas[i].(string) == "port" {
+                       state.Fields.Port = redisReplicas[i+1].(string)
+               } else if redisReplicas[i].(string) == "flags" {
+                       state.Fields.Flags = redisReplicas[i+1].(string)
+               } else if redisReplicas[i].(string) == "role-reported" {
+                       state.Fields.Role = redisReplicas[i+1].(string)
+               } else if redisReplicas[i].(string) == "master-link-status" {
+                       state.Fields.PrimaryLinkStatus = redisReplicas[i+1].(string)
                }
        }
        return state
@@ -120,11 +133,15 @@ func (s *Sentinel) getSentinelsState() (*SentinelsDbState, error) {
        states := new(SentinelsDbState)
        states.States = make([]*SentinelDbState, 0)
 
-       redisVal, redisErr := s.Sentinels(s.Cfg.masterName).Result()
+       redisVal, redisErr := s.Sentinels(s.ctx, s.Cfg.masterName).Result()
        if redisErr == nil {
                for _, redisSentinel := range redisVal {
                        sentinelState := readSentinelState(redisSentinel.([]interface{}))
-                       states.States = append(states.States, sentinelState)
+                       // Ignore a sentinel entry with zero port, because missing of fix
+                       // for the Redis Bug #9240.
+                       if sentinelState.Fields.Port != "0" {
+                               states.States = append(states.States, sentinelState)
+                       }
                }
        }
        states.Err = redisErr