From d4dd8ef88733ef4ab09533fc99e40659d1ccb117 Mon Sep 17 00:00:00 2001
From: taobo <realtaobo@outlook.com>
Date: Mon, 3 Jun 2024 17:21:13 +0800
Subject: [PATCH] feat: return early in fast_round if the recover_quorum
 condition is not met

Signed-off-by: taobo <realtaobo@outlook.com>
---
 crates/curp/src/client/unary.rs | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/crates/curp/src/client/unary.rs b/crates/curp/src/client/unary.rs
index d0c7de17d..fdccfdf61 100644
--- a/crates/curp/src/client/unary.rs
+++ b/crates/curp/src/client/unary.rs
@@ -9,7 +9,7 @@ use tracing::{debug, warn};
 use super::{state::State, ClientApi, LeaderStateUpdate, ProposeResponse, RepeatableClientApi};
 use crate::{
     members::ServerId,
-    quorum,
+    quorum, recover_quorum,
     rpc::{
         connect::ConnectApi, ConfChange, CurpError, FetchClusterRequest, FetchClusterResponse,
         FetchReadStateRequest, Member, MoveLeaderRequest, ProposeConfChangeRequest, ProposeId,
@@ -97,12 +97,17 @@ impl<C: Command> Unary<C> {
             })
             .await;
         let super_quorum = super_quorum(responses.len());
+        let recover_quorum = recover_quorum(responses.len());
 
         let mut err: Option<CurpError> = None;
         let mut execute_result: Option<C::ER> = None;
-        let mut ok_cnt = 0;
+        let (mut ok_cnt, mut key_conflict_cnt) = (0, 0);
 
         while let Some((id, resp)) = responses.next().await {
+            if key_conflict_cnt >= recover_quorum {
+                return Err(CurpError::KeyConflict(()));
+            }
+
             let resp = match resp {
                 Ok(resp) => resp.into_inner(),
                 Err(e) => {
@@ -110,6 +115,9 @@ impl<C: Command> Unary<C> {
                     if e.should_abort_fast_round() {
                         return Err(e);
                     }
+                    if matches!(e, CurpError::KeyConflict(())) {
+                        key_conflict_cnt.add_assign(1);
+                    }
                     if let Some(old_err) = err.as_ref() {
                         if old_err.priority() <= e.priority() {
                             err = Some(e);