fix: syncLeases redis wrong scaning

pubg · Aug 14, 2021 · 6f6b07c · 6f6b07c
1 parent c10de69
commit 6f6b07c
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/redis/redis-checkpoint.go b/redis/redis-checkpoint.go
@@ -363,12 +363,14 @@ func (checkpointer *RedisCheckpoint) syncLeases(shardStatus map[string]*par.Shar
 	}
 
 	checkpointer.lastLeaseSync = time.Now()
-	var cursor uint64
 
-	iter := checkpointer.svc.Scan(ctx, cursor, checkpointer.shardPrefix+"*", 1).Iterator()
+	keys, _, err := checkpointer.svc.Scan(ctx, 0, checkpointer.shardPrefix+"*", 0).Result()
 
-	for iter.Next(ctx) {
-		key := iter.Val()
+	if err != nil {
+		return fmt.Errorf("syncLeases Scan Error: %s", err)
+	}
+
+	for _, key := range keys {
 		j, err := checkpointer.svc.Get(ctx, key).Result()
 
 		if err != nil { // just logging