(iotdb) 03/03: Update ImportTsFile.java

[rong]

This is an automated email from the ASF dual-hosted git repository.

rong pushed a commit to branch load-oom-retry
in repository https://gitbox.apache.org/repos/asf/iotdb.git

commit fb64bef780159a0de6d9ba93e8693c7b15b0d885
Author: Steve Yurong Su <r...@apache.org>
AuthorDate: Fri Jun 21 11:57:48 2024 +0800

    Update ImportTsFile.java
---
 .../cli/src/main/java/org/apache/iotdb/tool/ImportTsFile.java    | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git 
a/iotdb-client/cli/src/main/java/org/apache/iotdb/tool/ImportTsFile.java 
b/iotdb-client/cli/src/main/java/org/apache/iotdb/tool/ImportTsFile.java
index 7f82fca4afe..d69b4c1731e 100644
--- a/iotdb-client/cli/src/main/java/org/apache/iotdb/tool/ImportTsFile.java
+++ b/iotdb-client/cli/src/main/java/org/apache/iotdb/tool/ImportTsFile.java
@@ -386,6 +386,7 @@ public class ImportTsFile extends AbstractTsFileTool {
 
         try {
           sessionPool.executeNonQueryStatement(sql);
+
           loadFileSuccessfulNum.increment();
           ioTPrinter.println("Imported [ " + filePath + " ] file 
successfully!");
 
@@ -401,10 +402,7 @@ public class ImportTsFile extends AbstractTsFileTool {
                     + processSuccessException.getMessage());
           }
         } catch (Exception e) {
-          loadFileFailedNum.increment();
-          ioTPrinter.println("Failed to import [ " + filePath + " ] file: " + 
e.getMessage());
-
-          // Reject because of memory controls
+          // Reject because of memory controls, do retry later
           if (Objects.nonNull(e.getMessage()) && 
e.getMessage().contains("memory")) {
             ioTPrinter.println(
                 "Rejecting file [ " + filePath + " ] due to memory 
constraints, will retry later.");
@@ -412,6 +410,9 @@ public class ImportTsFile extends AbstractTsFileTool {
             continue;
           }
 
+          loadFileFailedNum.increment();
+          ioTPrinter.println("Failed to import [ " + filePath + " ] file: " + 
e.getMessage());
+
           try {
             processingFile(filePath, failDir, failOperation);
             processingLoadFailedFileSuccessfulNum.increment();