From b9f17a251b47d9fa4047b78a9e2a7b1ec4d95ebf Mon Sep 17 00:00:00 2001
From: YC27 <76414902+YC27@users.noreply.github.com>
Date: Wed, 25 Sep 2024 16:50:12 +0800
Subject: [PATCH] Load: Fix the source parameter of the load tsfile script
 cannot specify a file (#13619)

(cherry picked from commit 59475a6a6031c07e74bf5ab25f991e4077dbf450)
---
 .../org/apache/iotdb/tool/tsfile/ImportTsFileScanTool.java    | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/iotdb-client/cli/src/main/java/org/apache/iotdb/tool/tsfile/ImportTsFileScanTool.java b/iotdb-client/cli/src/main/java/org/apache/iotdb/tool/tsfile/ImportTsFileScanTool.java
index 4ec9f2290848..4891f4254861 100644
--- a/iotdb-client/cli/src/main/java/org/apache/iotdb/tool/tsfile/ImportTsFileScanTool.java
+++ b/iotdb-client/cli/src/main/java/org/apache/iotdb/tool/tsfile/ImportTsFileScanTool.java
@@ -86,7 +86,9 @@ public static void setSourceFullPath(final String sourceFullPath) {
   }
 
   public static int getSourceFullPathLength() {
-    return ImportTsFileScanTool.sourceFullPath.length();
+    return new File(sourceFullPath).isDirectory()
+        ? ImportTsFileScanTool.sourceFullPath.length()
+        : new File(ImportTsFileScanTool.sourceFullPath).getParent().length();
   }
 
   public static int getTsFileQueueSize() {