fix

pubg · Jul 2, 2021 · 9a4e474 · 9a4e474
1 parent 1204196
commit 9a4e474
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/redis/redis-checkpoint.go b/redis/redis-checkpoint.go
@@ -182,13 +182,21 @@ func (checkpointer *RedisCheckpoint) CheckpointSequence(shard *par.ShardStatus)
 		defer checkpointer.unlock(shard.ID, mutex)
 	}
 
+	claimRequest := ""
+
+	if checkpointer.kclConfig.EnableLeaseStealing {
+		if checkpoint, err := checkpointer.getItem(shard.ID); err != nil && checkpoint.Checkpoint != "" {
+			claimRequest = checkpoint.Checkpoint
+		}
+	}
+
 	leaseTimeout := shard.GetLeaseTimeout().UTC().Format(time.RFC3339)
 	newCheckpoint := &ShardCheckpoint{
 		ShardID:      shard.ID,
 		Checkpoint:   shard.GetCheckpoint(),
 		AssignedTo:   shard.GetLeaseOwner(),
 		LeaseTimeout: leaseTimeout,
-		ClaimRequest: shard.ClaimRequest,
+		ClaimRequest: claimRequest,
 	}
 
 	if len(shard.ParentShardId) > 0 {