Add TableDeviceChunkMetadata

Author: JackieTien97

java/tsfile/src/main/java/org/apache/tsfile/file/metadata/AbstractAlignedChunkMetadata.java

@@ -0,0 +1,195 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+package org.apache.tsfile.file.metadata;
+
+import org.apache.tsfile.enums.TSDataType;
+import org.apache.tsfile.file.metadata.statistics.Statistics;
+import org.apache.tsfile.read.common.TimeRange;
+import org.apache.tsfile.read.controller.IChunkLoader;
+
+import java.io.OutputStream;
+import java.io.Serializable;
+import java.util.List;
+import java.util.Optional;
+
+public abstract class AbstractAlignedChunkMetadata implements IChunkMetadata {
+
+  // ChunkMetadata for time column
+  protected final IChunkMetadata timeChunkMetadata;
+  // ChunkMetadata for all subSensors in the vector
+  protected final List<IChunkMetadata> valueChunkMetadataList;
+
+  /** ChunkLoader of metadata, used to create IChunkReader */
+  private IChunkLoader chunkLoader;
+
+  AbstractAlignedChunkMetadata(
+      IChunkMetadata timeChunkMetadata, List<IChunkMetadata> valueChunkMetadataList) {
+    this.timeChunkMetadata = timeChunkMetadata;
+    this.valueChunkMetadataList = valueChunkMetadataList;
+  }
+
+  @Override
+  public Statistics<? extends Serializable> getTimeStatistics() {
+    return timeChunkMetadata.getStatistics();
+  }
+
+  @Override
+  public Optional<Statistics<? extends Serializable>> getMeasurementStatistics(
+      int measurementIndex) {
+    IChunkMetadata chunkMetadata = valueChunkMetadataList.get(measurementIndex);
+    return Optional.ofNullable(chunkMetadata == null ? null : chunkMetadata.getStatistics());
+  }
+
+  @Override
+  public boolean hasNullValue(int measurementIndex) {
+    long rowCount = getTimeStatistics().getCount();
+    Optional<Statistics<? extends Serializable>> statistics =
+        getMeasurementStatistics(measurementIndex);
+    return statistics.map(stat -> stat.hasNullValue(rowCount)).orElse(true);
+  }
+
+  @Override
+  public int getMeasurementCount() {
+    return valueChunkMetadataList.size();
+  }
+
+  @Override
+  public boolean isModified() {
+    return timeChunkMetadata.isModified();
+  }
+
+  @Override
+  public void setModified(boolean modified) {
+    timeChunkMetadata.setModified(modified);
+    for (IChunkMetadata v : valueChunkMetadataList) {
+      if (v != null) {
+        v.setModified(modified);
+      }
+    }
+  }
+
+  @Override
+  public boolean isSeq() {
+    return timeChunkMetadata.isSeq();
+  }
+
+  @Override
+  public void setSeq(boolean seq) {
+    timeChunkMetadata.setSeq(seq);
+    for (IChunkMetadata v : valueChunkMetadataList) {
+      if (v != null) {
+        v.setSeq(seq);
+      }
+    }
+  }
+
+  @Override
+  public long getVersion() {
+    return timeChunkMetadata.getVersion();
+  }
+
+  @Override
+  public void setVersion(long version) {
+    timeChunkMetadata.setVersion(version);
+    for (IChunkMetadata valueChunkMetadata : valueChunkMetadataList) {
+      if (valueChunkMetadata != null) {
+        valueChunkMetadata.setVersion(version);
+      }
+    }
+  }
+
+  @Override
+  public long getOffsetOfChunkHeader() {
+    return timeChunkMetadata.getOffsetOfChunkHeader();
+  }
+
+  @Override
+  public long getStartTime() {
+    return timeChunkMetadata.getStartTime();
+  }
+
+  @Override
+  public long getEndTime() {
+    return timeChunkMetadata.getEndTime();
+  }
+
+  @Override
+  public IChunkLoader getChunkLoader() {
+    return chunkLoader;
+  }
+
+  @Override
+  public boolean needSetChunkLoader() {
+    return chunkLoader == null;
+  }
+
+  @Override
+  public void setChunkLoader(IChunkLoader chunkLoader) {
+    this.chunkLoader = chunkLoader;
+  }
+
+  @Override
+  public void setClosed(boolean closed) {
+    timeChunkMetadata.setClosed(closed);
+    for (IChunkMetadata chunkMetadata : valueChunkMetadataList) {
+      if (chunkMetadata != null) {
+        chunkMetadata.setClosed(closed);
+      }
+    }
+  }
+
+  @Override
+  public TSDataType getDataType() {
+    return timeChunkMetadata.getDataType();
+  }
+
+  @Override
+  public String getMeasurementUid() {
+    return timeChunkMetadata.getMeasurementUid();
+  }
+
+  @Override
+  public void insertIntoSortedDeletions(TimeRange timeRange) {
+    throw new UnsupportedOperationException();
+  }
+
+  @Override
+  public List<TimeRange> getDeleteIntervalList() {
+    throw new UnsupportedOperationException();
+  }
+
+  @Override
+  public int serializeTo(OutputStream outputStream, boolean serializeStatistic) {
+    throw new UnsupportedOperationException("VectorChunkMetadata doesn't support serial method");
+  }
+
+  @Override
+  public byte getMask() {
+    return 0;
+  }
+
+  public IChunkMetadata getTimeChunkMetadata() {
+    return timeChunkMetadata;
+  }
+
+  public List<IChunkMetadata> getValueChunkMetadataList() {
+    return valueChunkMetadataList;
+  }
+}

java/tsfile/src/main/java/org/apache/tsfile/file/metadata/AbstractAlignedTimeSeriesMetadata.java

@@ -112,7 +112,7 @@ public List<IChunkMetadata> loadChunkMetadataList() {
     return chunkMetadataLoader.loadChunkMetadataList(this);
   }
 
-  public List<AlignedChunkMetadata> getCopiedChunkMetadataList() {
+  public List<AbstractAlignedChunkMetadata> getCopiedChunkMetadataList() {
     List<IChunkMetadata> timeChunkMetadata = timeseriesMetadata.getCopiedChunkMetadataList();
     List<List<IChunkMetadata>> valueChunkMetadataList = new ArrayList<>();
     for (TimeseriesMetadata metadata : valueTimeseriesMetadataList) {
@@ -122,7 +122,7 @@ public List<AlignedChunkMetadata> getCopiedChunkMetadataList() {
     return getAlignedChunkMetadata(timeChunkMetadata, valueChunkMetadataList);
   }
 
-  public List<AlignedChunkMetadata> getChunkMetadataList() {
+  public List<AbstractAlignedChunkMetadata> getChunkMetadataList() {
     List<IChunkMetadata> timeChunkMetadata = timeseriesMetadata.getChunkMetadataList();
     List<List<IChunkMetadata>> valueChunkMetadataList = new ArrayList<>();
     for (TimeseriesMetadata metadata : valueTimeseriesMetadataList) {
@@ -133,9 +133,9 @@ public List<AlignedChunkMetadata> getChunkMetadataList() {
   }
 
   /** Notice: if all the value chunks is empty chunk, then return empty list. */
-  private List<AlignedChunkMetadata> getAlignedChunkMetadata(
+  private List<AbstractAlignedChunkMetadata> getAlignedChunkMetadata(
       List<IChunkMetadata> timeChunkMetadata, List<List<IChunkMetadata>> valueChunkMetadataList) {
-    List<AlignedChunkMetadata> res = new ArrayList<>();
+    List<AbstractAlignedChunkMetadata> res = new ArrayList<>();
     for (int i = 0; i < timeChunkMetadata.size(); i++) {
       // only need time column
       if (valueTimeseriesMetadataList.isEmpty()) {
@@ -160,11 +160,14 @@ private List<AlignedChunkMetadata> getAlignedChunkMetadata(
   }
 
   abstract void constructAlignedChunkMetadata(
-      List<AlignedChunkMetadata> res,
+      List<AbstractAlignedChunkMetadata> res,
       IChunkMetadata timeChunkMetadata,
       List<IChunkMetadata> chunkMetadataList,
       boolean exits);
 
+  abstract AbstractAlignedChunkMetadata constructOnlyTimeChunkMetadata(
+      IChunkMetadata timeChunkMetadata);
+
   @Override
   public void setChunkMetadataLoader(IChunkMetadataLoader chunkMetadataLoader) {
     this.chunkMetadataLoader = chunkMetadataLoader;