Load: Fix the source parameter of the load tsfile script cannot speci…

…fy a file (#13619) (cherry picked from commit 59475a6)
apache · Sep 25, 2024 · b9f17a2 · b9f17a2
1 parent 8e77c20
commit b9f17a2
Showing 1 changed file with 3 additions and 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
@@ -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() {