[fix][dingo-sdk] Added preWrite to the problem of leader switching, reporting an error instead of SDK side retry

Author: githubgxll

java/dingo-sdk/src/main/java/io/dingodb/sdk/common/utils/ErrorCodeUtils.java

@@ -90,5 +90,10 @@ public static Strategy errorToStrategy(int code) {
         return Strategy.FAILED;
     }
 
+    public static boolean isPreWriteRequestFailed(int code, boolean isPreWrite) {
+        return isPreWrite && (code == ERAFT_NOTLEADER_VALUE || code == ERAFT_COMMITLOG_VALUE);
+    }
+
+
     public static final Function<Integer, Strategy> errorToStrategyFunc = ErrorCodeUtils::errorToStrategy;
 }

java/dingo-sdk/src/main/java/io/dingodb/sdk/service/caller/ServiceCaller.java

@@ -3,6 +3,7 @@
 import io.dingodb.sdk.common.DingoClientException.ExhaustedRetryException;
 import io.dingodb.sdk.common.DingoClientException.InvalidRouteTableException;
 import io.dingodb.sdk.common.DingoClientException.RequestErrorException;
+import io.dingodb.sdk.common.utils.ErrorCodeUtils;
 import io.dingodb.sdk.service.JsonMessageUtils;
 import io.dingodb.sdk.service.Caller;
 import io.dingodb.sdk.service.ChannelProvider;
@@ -12,6 +13,7 @@
 import io.dingodb.sdk.service.entity.Message.Response;
 import io.dingodb.sdk.service.entity.error.Errno;
 import io.dingodb.sdk.service.entity.error.Error;
+import io.dingodb.sdk.service.entity.store.TxnPrewriteRequest;
 import io.grpc.CallOptions;
 import io.grpc.Channel;
 import io.grpc.MethodDescriptor;
@@ -25,6 +27,7 @@
 import java.util.function.Function;
 
 import static io.dingodb.sdk.common.utils.ErrorCodeUtils.errorToStrategy;
+import static io.dingodb.sdk.common.utils.ErrorCodeUtils.isPreWriteRequestFailed;
 
 @Slf4j
 public class ServiceCaller<S extends Service<S>> implements Caller<S> {
@@ -96,8 +99,10 @@ public <REQ extends Request, RES extends Response> RES call(
                     errMsgs.compute(
                         channel.authority() + ">>" + error.getErrmsg(), (k, v) -> v == null ? 1 : v + 1
                     );
+                    boolean isPreWriteFailed = isPreWriteFailed(request);
                     switch (handler.onErrStrategy(
-                        errorToStrategy(errCode),
+                            isPreWriteRequestFailed(errCode, isPreWriteFailed) ?
+                        ErrorCodeUtils.Strategy.FAILED : errorToStrategy(errCode),
                         this.retry, retry, request, response, options, channel.authority(), requestId
                     )) {
                         case RETRY:
@@ -141,6 +146,13 @@ public <REQ extends Request, RES extends Response> RES call(
         throw generateException(methodName, requestId, lastRequest, connected, errMsgs, handler);
     }
 
+    private static <REQ extends Request> boolean isPreWriteFailed(REQ request) {
+        if (request instanceof TxnPrewriteRequest) {
+            return (((TxnPrewriteRequest) request).isTryOnePc() || ((TxnPrewriteRequest) request).isUseAsyncCommit());
+        }
+        return false;
+    }
+
     private void waitRetry() {
         LockSupport.parkNanos(TimeUnit.SECONDS.toNanos(1));
     }