From 11c76800a2316a2552fd4d0865aa1df6500acabf Mon Sep 17 00:00:00 2001
From: Ho <fan@scroll.io>
Date: Mon, 2 Mar 2026 14:14:28 +0900
Subject: [PATCH] fix: recover should be added if chunk do updated

---
 .../internal/logic/provertask/batch_prover_task.go     | 10 +++++-----
 .../internal/logic/provertask/bundle_prover_task.go    |  9 ++++-----
 .../internal/logic/provertask/chunk_prover_task.go     |  8 ++++----
 3 files changed, 13 insertions(+), 14 deletions(-)

diff --git a/coordinator/internal/logic/provertask/batch_prover_task.go b/coordinator/internal/logic/provertask/batch_prover_task.go
index 810dc13a2..d3d0d0f93 100644
--- a/coordinator/internal/logic/provertask/batch_prover_task.go
+++ b/coordinator/internal/logic/provertask/batch_prover_task.go
@@ -163,17 +163,17 @@ func (bp *BatchProverTask) Assign(ctx *gin.Context, getTaskParameter *coordinato
 				log.Error("failed to update batch attempts", "height", getTaskParameter.ProverHeight, "err", updateAttemptsErr)
 				return nil, ErrCoordinatorInternalFailure
 			}
+			if rowsAffected == 0 {
+				time.Sleep(100 * time.Millisecond)
+				continue
+			}
+
 			defer func(batchTask *orm.Batch) {
 				if retErr != nil {
 					bp.recoverActiveAttempts(ctx, batchTask)
 					log.Debug("recover active attempts", "batch task_id", batchTask.Hash)
 				}
 			}(tmpBatchTask)
-
-			if rowsAffected == 0 {
-				time.Sleep(100 * time.Millisecond)
-				continue
-			}
 		}
 
 		batchTask = tmpBatchTask
diff --git a/coordinator/internal/logic/provertask/bundle_prover_task.go b/coordinator/internal/logic/provertask/bundle_prover_task.go
index b4ffd5e3f..25f810c74 100644
--- a/coordinator/internal/logic/provertask/bundle_prover_task.go
+++ b/coordinator/internal/logic/provertask/bundle_prover_task.go
@@ -161,17 +161,16 @@ func (bp *BundleProverTask) Assign(ctx *gin.Context, getTaskParameter *coordinat
 				log.Error("failed to update bundle attempts", "height", getTaskParameter.ProverHeight, "err", updateAttemptsErr)
 				return nil, ErrCoordinatorInternalFailure
 			}
+			if rowsAffected == 0 {
+				time.Sleep(100 * time.Millisecond)
+				continue
+			}
 			defer func(bundleTask *orm.Bundle) {
 				if retErr != nil {
 					bp.recoverActiveAttempts(ctx, bundleTask)
 					log.Debug("recover active attempts", "bundle task_id", bundleTask.Hash)
 				}
 			}(tmpBundleTask)
-
-			if rowsAffected == 0 {
-				time.Sleep(100 * time.Millisecond)
-				continue
-			}
 		}
 		bundleTask = tmpBundleTask
 		break
diff --git a/coordinator/internal/logic/provertask/chunk_prover_task.go b/coordinator/internal/logic/provertask/chunk_prover_task.go
index ce876a175..52f137f86 100644
--- a/coordinator/internal/logic/provertask/chunk_prover_task.go
+++ b/coordinator/internal/logic/provertask/chunk_prover_task.go
@@ -166,6 +166,10 @@ func (cp *ChunkProverTask) Assign(ctx *gin.Context, getTaskParameter *coordinato
 				log.Error("failed to update chunk attempts", "height", getTaskParameter.ProverHeight, "err", updateAttemptsErr)
 				return nil, ErrCoordinatorInternalFailure
 			}
+			if rowsAffected == 0 {
+				time.Sleep(100 * time.Millisecond)
+				continue
+			}
 			defer func(chunkTask *orm.Chunk) {
 				if retErr != nil {
 					cp.recoverActiveAttempts(ctx, chunkTask)
@@ -173,10 +177,6 @@ func (cp *ChunkProverTask) Assign(ctx *gin.Context, getTaskParameter *coordinato
 				}
 			}(tmpChunkTask)
 
-			if rowsAffected == 0 {
-				time.Sleep(100 * time.Millisecond)
-				continue
-			}
 		}
 		chunkTask = tmpChunkTask
 		break