fix

taikoxyz · Sep 30, 2024 · 9c2271c · 9c2271c
1 parent fc667be
commit 9c2271c
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/packages/taiko-client/prover/proof_producer/sgx_producer.go b/packages/taiko-client/prover/proof_producer/sgx_producer.go
@@ -308,6 +308,13 @@ func (s *SGXProofProducer) requestBatchProof(
 		return nil, fmt.Errorf("failed to get batch proof, msg: %s", output.ErrorMessage)
 	}
 
+	if output.Data.Status == ErrProofInProgress.Error() {
+		return nil, ErrProofInProgress
+	}
+	if output.Data.Status == StatusRegistered {
+		return nil, ErrRetry
+	}
+
 	if len(output.Data.Proof.Proof) == 0 {
 		return nil, errEmptyProof
 	}

diff --git a/packages/taiko-client/prover/proof_producer/zkvm_producer.go b/packages/taiko-client/prover/proof_producer/zkvm_producer.go
@@ -435,6 +435,12 @@ func (s *ZKvmProofProducer) requestBatchProof(
 	if len(output.ErrorMessage) > 0 {
 		return nil, fmt.Errorf("failed to get batch proof, msg: %s", output.ErrorMessage)
 	}
+	if output.Data.Status == ErrProofInProgress.Error() {
+		return nil, ErrProofInProgress
+	}
+	if output.Data.Status == StatusRegistered {
+		return nil, ErrRetry
+	}
 
 	if len(output.Data.Proof.Proof) == 0 {
 		return nil, errEmptyProof