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 fdc3ba22bda8ae260159744f30154af94cafccf5
Author: Steve Yurong Su <[email protected]>
AuthorDate: Fri Jun 21 11:53:21 2024 +0800

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

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 320b03334df..ea9ff2126b9 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
@@ -402,6 +402,14 @@ public class ImportTsFile extends AbstractTsFileTool {
           loadFileFailedNum.increment();
           ioTPrinter.println("Failed to import [ " + filePath + " ] file: " + 
e.getMessage());
 
+          // Reject because of memory controls
+          if (Objects.nonNull(e.getMessage()) && 
e.getMessage().contains("memory")) {
+            ioTPrinter.println(
+                "Rejecting file [ " + filePath + " ] due to memory 
constraints, will retry later.");
+            tsfileQueue.put(filePath);
+            continue;
+          }
+
           try {
             processingFile(filePath, failDir, failOperation);
             processingLoadFailedFileSuccessfulNum.increment();

Reply via email to