Fix PerfBenchmarkRunner to skip the tmp dir (#9298)

apache · Aug 30, 2022 · d1a71d8 · d1a71d8
1 parent 6da31ab
commit d1a71d8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pinot-tools/src/main/java/org/apache/pinot/tools/perf/PerfBenchmarkRunner.java b/pinot-tools/src/main/java/org/apache/pinot/tools/perf/PerfBenchmarkRunner.java
@@ -168,7 +168,8 @@ public static void loadTable(PerfBenchmarkDriver driver, String dataDir, String
       List<String> invertedIndexColumns, List<String> bloomFilterColumns)
       throws Exception {
     boolean tableConfigured = false;
-    File[] segments = new File(dataDir, tableName).listFiles();
+    // Skip BaseTableDataManager._resourceTmpDir
+    File[] segments = new File(dataDir, tableName).listFiles((dir, name) -> !name.equals("tmp"));
     Preconditions.checkNotNull(segments);
     for (File segment : segments) {
       if (segment.isDirectory()) {