Fix flaky LastFlushTimeMapTest (apache#14225)

Author: HTHou

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/storageengine/dataregion/memtable/TsFileProcessor.java

@@ -1311,6 +1311,7 @@ public Future<?> asyncClose() {
    * TODO if the flushing thread is too fast, the tmpMemTable.wait() may never wakeup Tips: I am
    * trying to solve this issue by checking whether the table exist before wait()
    */
+  @TestOnly
   public void syncFlush() throws IOException {
     IMemTable tmpMemTable;
     flushQueryLock.writeLock().lock();

iotdb-core/datanode/src/test/java/org/apache/iotdb/db/storageengine/dataregion/DataRegionTest.java

@@ -197,10 +197,7 @@ record = new TSRecord(j, deviceId);
       record.addTuple(DataPoint.getDataPoint(TSDataType.INT32, measurementId, String.valueOf(j)));
       dataRegion.insert(buildInsertRowNodeByTSRecord(record));
     }
-
-    for (TsFileProcessor tsfileProcessor : dataRegion.getWorkUnsequenceTsFileProcessors()) {
-      tsfileProcessor.syncFlush();
-    }
+    dataRegion.syncCloseWorkingTsFileProcessors(false);
 
     for (int j = 11; j <= 20; j++) {
       record = new TSRecord(j, deviceId);
@@ -753,10 +750,6 @@ public void testDisableSeparateDataForInsertRowPlan()
       dataRegion.syncCloseAllWorkingTsFileProcessors();
     }
 
-    for (TsFileProcessor tsfileProcessor : dataRegion.getWorkUnsequenceTsFileProcessors()) {
-      tsfileProcessor.syncFlush();
-    }
-
     QueryDataSource queryDataSource =
         dataRegion.query(
             Collections.singletonList(nonAlignedFullPath), device, context, null, null);
@@ -838,10 +831,6 @@ public void testDisableSeparateDataForInsertTablet1()
     dataRegion.insertTablet(insertTabletNode2);
     dataRegion.syncCloseAllWorkingTsFileProcessors();
 
-    for (TsFileProcessor tsfileProcessor : dataRegion.getWorkUnsequenceTsFileProcessors()) {
-      tsfileProcessor.syncFlush();
-    }
-
     QueryDataSource queryDataSource =
         dataRegion.query(
             Collections.singletonList(nonAlignedFullPath), device, context, null, null);
@@ -922,10 +911,6 @@ public void testDisableSeparateDataForInsertTablet2()
     dataRegion.insertTablet(insertTabletNode2);
     dataRegion.syncCloseAllWorkingTsFileProcessors();
 
-    for (TsFileProcessor tsfileProcessor : dataRegion.getWorkUnsequenceTsFileProcessors()) {
-      tsfileProcessor.syncFlush();
-    }
-
     QueryDataSource queryDataSource =
         dataRegion.query(
             Collections.singletonList(nonAlignedFullPath), device, context, null, null);
@@ -1006,10 +991,6 @@ public void testDisableSeparateDataForInsertTablet3()
     dataRegion.insertTablet(insertTabletNode2);
     dataRegion.syncCloseAllWorkingTsFileProcessors();
 
-    for (TsFileProcessor tsfileProcessor : dataRegion.getWorkUnsequenceTsFileProcessors()) {
-      tsfileProcessor.syncFlush();
-    }
-
     QueryDataSource queryDataSource =
         dataRegion.query(
             Collections.singletonList(nonAlignedFullPath), device, context, null, null);
@@ -1084,10 +1065,6 @@ public void testSmallReportProportionInsertRow()
       dataRegion1.syncCloseAllWorkingTsFileProcessors();
     }
     dataRegion1.syncCloseAllWorkingTsFileProcessors();
-
-    for (TsFileProcessor tsfileProcessor : dataRegion1.getWorkUnsequenceTsFileProcessors()) {
-      tsfileProcessor.syncFlush();
-    }
     IDeviceID tmpDeviceId = IDeviceID.Factory.DEFAULT_FACTORY.create("root.ln22");
     QueryDataSource queryDataSource =
         dataRegion1.query(
@@ -1733,9 +1710,7 @@ public void testDeleteDataDirectlyUnseqWriteModsOrDeleteFiles()
       dataRegion.insert(buildInsertRowNodeByTSRecord(record));
     }
 
-    for (TsFileProcessor processor : dataRegion.getWorkSequenceTsFileProcessors()) {
-      processor.syncFlush();
-    }
+    dataRegion.syncCloseWorkingTsFileProcessors(true);
     TsFileResource tsFileResourceUnSeq = dataRegion.getTsFileManager().getTsFileList(false).get(0);
 
     Assert.assertTrue(tsFileResourceSeq.getTsFile().exists());

iotdb-core/datanode/src/test/java/org/apache/iotdb/db/storageengine/dataregion/LastFlushTimeMapTest.java

@@ -25,7 +25,6 @@
 import org.apache.iotdb.db.exception.WriteProcessException;
 import org.apache.iotdb.db.storageengine.StorageEngine;
 import org.apache.iotdb.db.storageengine.dataregion.compaction.schedule.CompactionTaskManager;
-import org.apache.iotdb.db.storageengine.dataregion.memtable.TsFileProcessor;
 import org.apache.iotdb.db.storageengine.dataregion.tsfile.TsFileResource;
 import org.apache.iotdb.db.storageengine.dataregion.tsfile.timeindex.ArrayDeviceTimeIndex;
 import org.apache.iotdb.db.utils.EnvironmentUtils;
@@ -89,9 +88,7 @@ record = new TSRecord(j, "root.vehicle.d0");
       dataRegion.insert(DataRegionTest.buildInsertRowNodeByTSRecord(record));
     }
 
-    for (TsFileProcessor tsfileProcessor : dataRegion.getWorkUnsequenceTsFileProcessors()) {
-      tsfileProcessor.syncFlush();
-    }
+    dataRegion.syncCloseWorkingTsFileProcessors(false);
     Assert.assertEquals(
         10000,
         dataRegion
@@ -117,10 +114,6 @@ record = new TSRecord(j, "root.vehicle.d0");
       record.addTuple(DataPoint.getDataPoint(TSDataType.INT32, measurementId, String.valueOf(j)));
       dataRegion.insert(DataRegionTest.buildInsertRowNodeByTSRecord(record));
     }
-
-    for (TsFileProcessor tsfileProcessor : dataRegion.getWorkUnsequenceTsFileProcessors()) {
-      tsfileProcessor.syncFlush();
-    }
     dataRegion.syncCloseAllWorkingTsFileProcessors();
     Assert.assertEquals(
         10000,
@@ -164,10 +157,6 @@ record = new TSRecord(j, "root.vehicle.d0");
       record.addTuple(DataPoint.getDataPoint(TSDataType.INT32, measurementId, String.valueOf(j)));
       dataRegion.insert(DataRegionTest.buildInsertRowNodeByTSRecord(record));
     }
-
-    for (TsFileProcessor tsfileProcessor : dataRegion.getWorkUnsequenceTsFileProcessors()) {
-      tsfileProcessor.syncFlush();
-    }
     dataRegion.syncCloseAllWorkingTsFileProcessors();
     Assert.assertEquals(
         604_800_000,