b

Author: Caideyipi

iotdb-core/node-commons/src/main/java/org/apache/iotdb/commons/schema/table/TsTable.java

@@ -25,6 +25,7 @@
 import org.apache.iotdb.commons.schema.table.column.AttributeColumnSchema;
 import org.apache.iotdb.commons.schema.table.column.FieldColumnSchema;
 import org.apache.iotdb.commons.schema.table.column.TagColumnSchema;
+import org.apache.iotdb.commons.schema.table.column.TimeColumnSchema;
 import org.apache.iotdb.commons.schema.table.column.TsTableColumnCategory;
 import org.apache.iotdb.commons.schema.table.column.TsTableColumnSchema;
 import org.apache.iotdb.commons.schema.table.column.TsTableColumnSchemaUtil;
@@ -93,6 +94,9 @@ public class TsTable {
   private transient int tagNums = 0;
   private transient int fieldNum = 0;
 
+  // Initiated during creation and never changed the reference
+  private transient TsTableColumnSchema timeColumnSchema;
+
   public TsTable(final String tableName) {
     this.tableName = tableName;
   }
@@ -101,7 +105,12 @@ public TsTable(final String tableName) {
   public TsTable(String tableName, ImmutableList<TsTableColumnSchema> columnSchemas) {
     this.tableName = tableName;
     columnSchemas.forEach(
-        columnSchema -> columnSchemaMap.put(columnSchema.getColumnName(), columnSchema));
+        columnSchema -> {
+          columnSchemaMap.put(columnSchema.getColumnName(), columnSchema);
+          if (columnSchema instanceof TimeColumnSchema) {
+            timeColumnSchema = columnSchema;
+          }
+        });
   }
 
   public TsTable(TsTable origin) {
@@ -146,10 +155,10 @@ public TsTableColumnSchema getColumnSchema(final String columnName) {
   public TsTableColumnSchema getTimeColumnSchema() {
     if (Objects.isNull(timeColumnSchema)) {
       timeColumnSchema =
-              columnSchemaMap.values().stream()
-                      .filter(column -> column instanceof TimeColumnSchema)
-                      .findFirst()
-                      .orElse(null);
+          columnSchemaMap.values().stream()
+              .filter(column -> column instanceof TimeColumnSchema)
+              .findFirst()
+              .orElse(null);
     }
     return timeColumnSchema;
   }