modify the deal strategy for metadata lease exception in DataDriver and ErrorHandlingUtils (#18438)

Author: alpass163gmail

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/execution/driver/DataDriver.java

@@ -21,7 +21,9 @@
 
 import org.apache.iotdb.calc.exception.QueryProcessException;
 import org.apache.iotdb.calc.execution.operator.Operator;
+import org.apache.iotdb.commons.exception.MetadataLeaseFencedException;
 import org.apache.iotdb.db.i18n.DataNodeQueryMessages;
+import org.apache.iotdb.db.i18n.DataNodeSchemaMessages;
 import org.apache.iotdb.db.queryengine.execution.operator.source.DataSourceOperator;
 import org.apache.iotdb.db.queryengine.plan.planner.plan.FragmentInstance;
 import org.apache.iotdb.db.storageengine.dataregion.read.IQueryDataSource;
@@ -65,21 +67,29 @@ protected boolean init(SettableFuture<?> blockedFuture) {
         } else {
           return true;
         }
+      } catch (MetadataLeaseFencedException e) {
+        LOGGER.info(
+            DataNodeSchemaMessages.METADATA_LEASE_IS_FENCED, driverContext.getDriverTaskID(), e);
+        handleInitFailure(e, blockedFuture);
       } catch (Throwable t) {
         LOGGER.error(
             DataNodeQueryMessages.FAILED_TO_DO_THE_INITIALIZATION_FOR_DRIVER_ARG,
             driverContext.getDriverTaskID(),
             t);
-        driverContext.failed(t);
-        blockedFuture.setException(t);
-        throwIfUnchecked(t);
-        // should never happen
-        throw new AssertionError(t);
+        handleInitFailure(t, blockedFuture);
       }
     }
     return true;
   }
 
+  private void handleInitFailure(Throwable t, SettableFuture<?> blockedFuture) {
+    driverContext.failed(t);
+    blockedFuture.setException(t);
+    throwIfUnchecked(t);
+    // should never happen
+    throw new AssertionError(t);
+  }
+
   @Override
   public boolean isInit() {
     return init;

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/utils/ErrorHandlingUtils.java

@@ -135,7 +135,10 @@ public static TSStatus onQueryException(Exception e, String operation, TSStatusC
             || status.getCode() == TSStatusCode.PATH_ALREADY_EXIST.getStatusCode()
             || status.getCode() == TSStatusCode.PIPE_NOT_EXIST_ERROR.getStatusCode()
             || status.getCode() == TSStatusCode.QUERY_TIMEOUT.getStatusCode()
-            || status.getCode() == TSStatusCode.CANNOT_READ_TSFILE.getStatusCode()) {
+            || status.getCode() == TSStatusCode.CANNOT_READ_TSFILE.getStatusCode()
+            || status.getCode() == TSStatusCode.METADATA_LEASE_FENCED.getStatusCode()
+            || status.getCode()
+                == TSStatusCode.METADATA_LEASE_FENCED_RETRY_REQUIRED.getStatusCode()) {
           LOGGER.info(message);
         } else {
           LOGGER.warn(message, e);