Support some configurations about data partition base on database level

Author: zerolbsony

iotdb-core/antlr/src/main/antlr4/org/apache/iotdb/db/qp/sql/IoTDBSqlParser.g4

@@ -118,6 +118,7 @@ databaseAttributeClause
 databaseAttributeKey
     : TTL
     | TIME_PARTITION_INTERVAL
+    | TIME_PARTITION_ORIGIN
     | SCHEMA_REGION_GROUP_NUM
     | DATA_REGION_GROUP_NUM
     ;

iotdb-core/antlr/src/main/antlr4/org/apache/iotdb/db/qp/sql/SqlLexer.g4

@@ -1211,6 +1211,10 @@ TIME_PARTITION_INTERVAL
     : T I M E '_' P A R T I T I O N '_' I N T E R V A L
     ;
 
+TIME_PARTITION_ORIGIN
+    : T I M E '_' P A R T I T I O N '_' O R I G I N
+    ;
+
 SCHEMA_REGION_GROUP_NUM
     : S C H E M A '_' R E G I O N '_' G R O U P '_' N U M
     ;

iotdb-core/confignode/src/main/java/org/apache/iotdb/confignode/persistence/schema/ClusterSchemaInfo.java

@@ -38,6 +38,7 @@
 import org.apache.iotdb.commons.utils.PathUtils;
 import org.apache.iotdb.commons.utils.StatusUtils;
 import org.apache.iotdb.commons.utils.TestOnly;
+import org.apache.iotdb.commons.utils.TimePartitionUtils;
 import org.apache.iotdb.confignode.consensus.request.read.database.CountDatabasePlan;
 import org.apache.iotdb.confignode.consensus.request.read.database.GetDatabasePlan;
 import org.apache.iotdb.confignode.consensus.request.read.table.DescTablePlan;
@@ -196,6 +197,10 @@ public TSStatus createDatabase(final DatabaseSchemaPlan plan) {
       final TDatabaseSchema databaseSchema = plan.getSchema();
       final PartialPath partialPathName = getQualifiedDatabasePartialPath(databaseSchema.getName());
 
+      // Update TimePartitionUtils cache with database-specific time partition settings
+      TimePartitionUtils.updateDatabaseTimePartitionConfig(
+          databaseSchema.getName(), databaseSchema);
+
       final ConfigMTree mTree = databaseSchema.isIsTableModel() ? tableModelMTree : treeModelMTree;
       mTree.setStorageGroup(partialPathName);
 
@@ -280,6 +285,9 @@ public TSStatus alterDatabase(final DatabaseSchemaPlan plan) {
           .getAsMNode()
           .setDatabaseSchema(currentSchema);
 
+      // Update TimePartitionUtils cache with new time partition settings
+      TimePartitionUtils.updateDatabaseTimePartitionConfig(currentSchema.getName(), currentSchema);
+
       result.setCode(TSStatusCode.SUCCESS_STATUS.getStatusCode());
     } catch (final MetadataException e) {
       LOGGER.error(ERROR_NAME, e);
@@ -305,6 +313,9 @@ public TSStatus deleteDatabase(final DeleteDatabasePlan plan) {
       (isTableModel ? tableModelMTree : treeModelMTree)
           .deleteDatabase(getQualifiedDatabasePartialPath(plan.getName()));
 
+      // Remove database-specific time partition configuration from cache
+      TimePartitionUtils.removeDatabaseTimePartitionConfig(plan.getName());
+
       result.setCode(TSStatusCode.SUCCESS_STATUS.getStatusCode());
     } catch (final MetadataException e) {
       LOGGER.warn("Database not exist", e);

iotdb-core/confignode/src/main/java/org/apache/iotdb/confignode/service/thrift/ConfigNodeRPCServiceProcessor.java

@@ -446,19 +446,20 @@ public TSStatus alterDatabase(final TDatabaseSchema databaseSchema) {
                   "Failed to alter database. Doesn't support ALTER DataReplicationFactor yet.");
     }
 
-    if (databaseSchema.isSetTimePartitionOrigin()) {
-      errorResp =
-          new TSStatus(TSStatusCode.DATABASE_CONFIG_ERROR.getStatusCode())
-              .setMessage(
-                  "Failed to alter database. Doesn't support ALTER TimePartitionOrigin yet.");
-    }
-
-    if (databaseSchema.isSetTimePartitionInterval()) {
-      errorResp =
-          new TSStatus(TSStatusCode.DATABASE_CONFIG_ERROR.getStatusCode())
-              .setMessage(
-                  "Failed to alter database. Doesn't support ALTER TimePartitionInterval yet.");
-    }
+    // Time partition settings are now supported for database-level configuration
+    // if (databaseSchema.isSetTimePartitionOrigin()) {
+    //   errorResp =
+    //       new TSStatus(TSStatusCode.DATABASE_CONFIG_ERROR.getStatusCode())
+    //           .setMessage(
+    //               "Failed to alter database. Doesn't support ALTER TimePartitionOrigin yet.");
+    // }
+
+    // if (databaseSchema.isSetTimePartitionInterval()) {
+    //   errorResp =
+    //       new TSStatus(TSStatusCode.DATABASE_CONFIG_ERROR.getStatusCode())
+    //           .setMessage(
+    //               "Failed to alter database. Doesn't support ALTER TimePartitionInterval yet.");
+    // }
 
     if (errorResp != null) {
       LOGGER.warn("Execute AlterDatabase: {} with result: {}", databaseSchema, errorResp);

iotdb-core/node-commons/src/main/java/org/apache/iotdb/commons/partition/DataPartition.java

@@ -109,7 +109,10 @@ public List<List<TTimePartitionSlot>> getTimePartitionRange(
         dataPartitionMap.get(storageGroup).get(seriesPartitionSlot);
     List<TTimePartitionSlot> timePartitionSlotList =
         map.keySet().stream()
-            .filter(key -> TimePartitionUtils.satisfyPartitionStartTime(timeFilter, key.startTime))
+            .filter(
+                key ->
+                    TimePartitionUtils.satisfyPartitionStartTime(
+                        timeFilter, key.startTime, storageGroup))
             .sorted(Comparator.comparingLong(TTimePartitionSlot::getStartTime))
             .collect(toList());
 
@@ -152,7 +155,8 @@ public List<TRegionReplicaSet> getDataRegionReplicaSetWithTimeFilter(
     return regionReplicaSetMap.entrySet().stream()
         .filter(
             entry ->
-                TimePartitionUtils.satisfyPartitionStartTime(timeFilter, entry.getKey().startTime))
+                TimePartitionUtils.satisfyPartitionStartTime(
+                    timeFilter, entry.getKey().startTime, storageGroup))
         .flatMap(entry -> entry.getValue().stream())
         .distinct()
         .collect(toList());
@@ -175,7 +179,8 @@ public List<TRegionReplicaSet> getDataRegionReplicaSetWithTimeFilter(
     return dataPartitionMap.get(database).get(seriesPartitionSlot).entrySet().stream()
         .filter(
             entry ->
-                TimePartitionUtils.satisfyPartitionStartTime(timeFilter, entry.getKey().startTime))
+                TimePartitionUtils.satisfyPartitionStartTime(
+                    timeFilter, entry.getKey().startTime, database))
         .flatMap(entry -> entry.getValue().stream())
         .distinct()
         .collect(toList());