feat(taiko-client): adjust zk error handing to increase the zk coverage (#21173)

Author: YoGhurt111

packages/taiko-client/prover/proof_producer/proof_buffer.go

@@ -25,8 +25,8 @@ type ProofBuffer struct {
 // NewProofBuffer creates a new ProofBuffer instance.
 func NewProofBuffer(maxLength uint64) *ProofBuffer {
 	return &ProofBuffer{
-		buffer:      make([]*ProofResponse, 0, maxLength),
-		MaxLength:   maxLength,
+		buffer:    make([]*ProofResponse, 0, maxLength),
+		MaxLength: maxLength,
 	}
 }
 

packages/taiko-client/prover/proof_submitter/proof_submitter.go

@@ -190,21 +190,22 @@ func (s *ProofSubmitterPacaya) RequestProof(ctx context.Context, meta metadata.T
 					meta,
 					startAt,
 				); err != nil {
-					if errors.Is(err, proofProducer.ErrProofInProgress) || errors.Is(err, proofProducer.ErrRetry) {
-						if time.Since(startAt) > maxProofRequestTimeout {
-							log.Warn("Retry timeout exceeded maxProofRequestTimeout, switching to SGX proof as fallback")
+					if time.Since(startAt) > maxProofRequestTimeout {
+						log.Warn("Retry timeout exceeded maxProofRequestTimeout, switching to SGX proof as fallback")
+						useZK = false
+						startAt = time.Now()
+					} else {
+						if errors.Is(err, proofProducer.ErrZkAnyNotDrawn) {
+							log.Debug(
+								"ZK proof was not chosen, attempting to request SGX proof",
+								"batchID", opts.BatchID,
+								"err", err,
+							)
 							useZK = false
 							startAt = time.Now()
-						} else {
-							return fmt.Errorf("zk proof is WIP, status: %w", err)
 						}
-					} else {
-						log.Debug(
-							"ZK proof was not chosen or got unexpected error, attempting to request SGX proof",
-							"batchID", opts.BatchID,
-						)
-						useZK = false
-						startAt = time.Now()
+						log.Debug("Got error, retrying", "err", err)
+						return err
 					}
 				}
 			}

packages/taiko-client/prover/proof_submitter/proof_submitter_shasta.go

@@ -185,21 +185,22 @@ func (s *ProofSubmitterShasta) RequestProof(ctx context.Context, meta metadata.T
 				meta,
 				startAt,
 			); err != nil {
-				if errors.Is(err, proofProducer.ErrProofInProgress) || errors.Is(err, proofProducer.ErrRetry) {
-					if time.Since(startAt) > maxProofRequestTimeout {
-						log.Warn("Retry timeout exceeded maxProofRequestTimeout, switching to SGX proof as fallback")
+				if time.Since(startAt) > maxProofRequestTimeout {
+					log.Warn("Retry timeout exceeded maxProofRequestTimeout, switching to SGX proof as fallback")
+					useZK = false
+					startAt = time.Now()
+				} else {
+					if errors.Is(err, proofProducer.ErrZkAnyNotDrawn) {
+						log.Debug(
+							"ZK proof was not chosen, attempting to request SGX proof",
+							"proposalID", opts.ProposalID,
+							"err", err,
+						)
 						useZK = false
 						startAt = time.Now()
-					} else {
-						return fmt.Errorf("zk proof is WIP, status: %w", err)
 					}
-				} else {
-					log.Debug(
-						"ZK proof was not chosen or got unexpected error, attempting to request SGX proof",
-						"proposalID", opts.ProposalID,
-					)
-					useZK = false
-					startAt = time.Now()
+					log.Debug("Got error, retrying", "err", err)
+					return err
 				}
 			}
 		}