Fix compatibility overloads for partition APIs

Author: zerolbsony

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/storageengine/load/splitter/TsFileSplitter.java

@@ -93,6 +93,11 @@ public TsFileSplitter(File tsFile, TsFileDataConsumer consumer) {
     this.consumer = consumer;
   }
 
+  public void splitTsFileByDataPartition()
+      throws IOException, LoadFileException, IllegalStateException {
+    splitTsFileByDataPartition(null);
+  }
+
   @SuppressWarnings({"squid:S3776", "squid:S6541"})
   public void splitTsFileByDataPartition(String database)
       throws IOException, LoadFileException, IllegalStateException {

iotdb-core/node-commons/src/main/java/org/apache/iotdb/commons/utils/TimePartitionUtils.java

@@ -214,22 +214,42 @@ public static TTimePartitionSlot getTimePartitionSlot(long time, String database
     return timePartitionSlot;
   }
 
+  public static TTimePartitionSlot getTimePartitionSlot(long time) {
+    return getTimePartitionSlot(time, null);
+  }
+
   public static long getTimePartitionInterval(String database) {
     return getDatabaseConfig(database).getTimePartitionInterval();
   }
 
+  public static long getTimePartitionInterval() {
+    return timePartitionInterval;
+  }
+
   public static long getTimePartitionOrigin(String database) {
     return getDatabaseConfig(database).getTimePartitionOrigin();
   }
 
+  public static long getTimePartitionOrigin() {
+    return timePartitionOrigin;
+  }
+
   public static long getTimePartitionLowerBound(long time, String database) {
     return getTimePartitionLowerBoundInternal(time, getDatabaseConfig(database));
   }
 
+  public static long getTimePartitionLowerBound(long time) {
+    return getTimePartitionLowerBound(time, null);
+  }
+
   public static long getTimePartitionUpperBound(long time, String database) {
     return getTimePartitionUpperBoundInternal(time, getDatabaseConfig(database));
   }
 
+  public static long getTimePartitionUpperBound(long time) {
+    return getTimePartitionUpperBound(time, null);
+  }
+
   public static long getTimePartitionId(long time, String database) {
     DatabaseTimePartitionConfig config = getDatabaseConfig(database);
     time -= config.getTimePartitionOrigin();
@@ -238,11 +258,22 @@ public static long getTimePartitionId(long time, String database) {
         : time / config.getTimePartitionInterval() - 1;
   }
 
+  public static long getTimePartitionId(long time) {
+    time -= timePartitionOrigin;
+    return time > 0 || time % timePartitionInterval == 0
+        ? time / timePartitionInterval
+        : time / timePartitionInterval - 1;
+  }
+
   public static long getStartTimeByPartitionId(long partitionId, String database) {
     DatabaseTimePartitionConfig config = getDatabaseConfig(database);
     return (partitionId * config.getTimePartitionInterval()) + config.getTimePartitionOrigin();
   }
 
+  public static long getStartTimeByPartitionId(long partitionId) {
+    return (partitionId * timePartitionInterval) + timePartitionOrigin;
+  }
+
   public static boolean satisfyPartitionId(
       long startTime, long endTime, long partitionId, String database) {
     DatabaseTimePartitionConfig config = getDatabaseConfig(database);
@@ -261,6 +292,20 @@ public static boolean satisfyPartitionId(
     return startPartition <= partitionId && endPartition >= partitionId;
   }
 
+  public static boolean satisfyPartitionId(long startTime, long endTime, long partitionId) {
+    long startPartition =
+        originMayCauseOverflow
+            ? getTimePartitionIdWithoutOverflow(
+                startTime, timePartitionOrigin, timePartitionInterval)
+            : getTimePartitionId(startTime);
+    long endPartition =
+        originMayCauseOverflow
+            ? getTimePartitionIdWithoutOverflow(
+                endTime, timePartitionOrigin, timePartitionInterval)
+            : getTimePartitionId(endTime);
+    return startPartition <= partitionId && endPartition >= partitionId;
+  }
+
   public static boolean satisfyPartitionStartTime(
       Filter timeFilter, long partitionStartTime, String database) {
     if (timeFilter == null) {
@@ -274,6 +319,17 @@ public static boolean satisfyPartitionStartTime(
     return timeFilter.satisfyStartEndTime(partitionStartTime, partitionEndTime);
   }
 
+  public static boolean satisfyPartitionStartTime(Filter timeFilter, long partitionStartTime) {
+    if (timeFilter == null) {
+      return true;
+    }
+    long partitionEndTime =
+        partitionStartTime >= timePartitionLowerBoundWithoutOverflow
+            ? Long.MAX_VALUE
+            : (partitionStartTime + timePartitionInterval - 1);
+    return timeFilter.satisfyStartEndTime(partitionStartTime, partitionEndTime);
+  }
+
   public static boolean satisfyTimePartition(Filter timeFilter, long partitionId, String database) {
     DatabaseTimePartitionConfig config = getDatabaseConfig(database);
     long partitionStartTime;
@@ -290,6 +346,20 @@ public static boolean satisfyTimePartition(Filter timeFilter, long partitionId,
     return satisfyPartitionStartTime(timeFilter, partitionStartTime, database);
   }
 
+  public static boolean satisfyTimePartition(Filter timeFilter, long partitionId) {
+    long partitionStartTime;
+    if (originMayCauseOverflow) {
+      partitionStartTime =
+          BigInteger.valueOf(partitionId)
+              .multiply(bigTimePartitionInterval)
+              .add(bigTimePartitionOrigin)
+              .longValue();
+    } else {
+      partitionStartTime = partitionId * timePartitionInterval + timePartitionOrigin;
+    }
+    return satisfyPartitionStartTime(timeFilter, partitionStartTime);
+  }
+
   public static long getEstimateTimePartitionSize(long startTime, long endTime, String database) {
     DatabaseTimePartitionConfig config = getDatabaseConfig(database);
     if (endTime > 0 && startTime < 0) {
@@ -302,6 +372,17 @@ public static long getEstimateTimePartitionSize(long startTime, long endTime, St
     return (endTime - startTime) / config.getTimePartitionInterval() + 1;
   }
 
+  public static long getEstimateTimePartitionSize(long startTime, long endTime) {
+    if (endTime > 0 && startTime < 0) {
+      return BigInteger.valueOf(endTime)
+              .subtract(BigInteger.valueOf(startTime))
+              .divide(bigTimePartitionInterval)
+              .longValue()
+          + 1;
+    }
+    return (endTime - startTime) / timePartitionInterval + 1;
+  }
+
   // Helper methods for database-specific calculations
   private static long getTimePartitionLowerBoundInternal(
       long time, DatabaseTimePartitionConfig config) {