| /* |
| * 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.iotdb.db.storageengine.dataregion.memtable; |
| |
| import org.apache.iotdb.commons.path.AlignedPath; |
| import org.apache.iotdb.commons.path.PartialPath; |
| import org.apache.iotdb.commons.path.PathPatternUtil; |
| import org.apache.iotdb.db.storageengine.dataregion.wal.buffer.IWALByteBufferView; |
| import org.apache.iotdb.tsfile.utils.BitMap; |
| import org.apache.iotdb.tsfile.utils.Pair; |
| import org.apache.iotdb.tsfile.write.schema.IMeasurementSchema; |
| |
| import java.io.DataInputStream; |
| import java.io.IOException; |
| import java.util.ArrayList; |
| import java.util.Collections; |
| import java.util.List; |
| import java.util.Map; |
| import java.util.Set; |
| |
| public class AlignedWritableMemChunkGroup implements IWritableMemChunkGroup { |
| |
| private AlignedWritableMemChunk memChunk; |
| |
| public AlignedWritableMemChunkGroup(List<IMeasurementSchema> schemaList) { |
| memChunk = new AlignedWritableMemChunk(schemaList); |
| } |
| |
| private AlignedWritableMemChunkGroup() { |
| // Empty constructor |
| } |
| |
| @Override |
| public boolean writeValuesWithFlushCheck( |
| long[] times, |
| Object[] columns, |
| BitMap[] bitMaps, |
| List<IMeasurementSchema> schemaList, |
| int start, |
| int end) { |
| return memChunk.writeAlignedValuesWithFlushCheck( |
| times, columns, bitMaps, schemaList, start, end); |
| } |
| |
| @Override |
| public void release() { |
| memChunk.release(); |
| } |
| |
| @Override |
| public long count() { |
| return memChunk.count(); |
| } |
| |
| /** |
| * Check whether this MemChunkGroup contains a measurement. If a VECTOR_PLACEHOLDER passed from |
| * outer, always return true because AlignedMemChunkGroup existing. |
| */ |
| @Override |
| public boolean contains(String measurement) { |
| // used for calculate memtable size |
| if (AlignedPath.VECTOR_PLACEHOLDER.equals(measurement)) { |
| return true; |
| } |
| return memChunk.containsMeasurement(measurement); |
| } |
| |
| @Override |
| public boolean writeWithFlushCheck( |
| long insertTime, Object[] objectValue, List<IMeasurementSchema> schemaList) { |
| return memChunk.writeAlignedValueWithFlushCheck(insertTime, objectValue, schemaList); |
| } |
| |
| @Override |
| public Map<String, IWritableMemChunk> getMemChunkMap() { |
| if (memChunk.count() == 0) { |
| return Collections.emptyMap(); |
| } |
| return Collections.singletonMap("", memChunk); |
| } |
| |
| @SuppressWarnings("squid:S3776") |
| @Override |
| public int delete( |
| PartialPath originalPath, PartialPath devicePath, long startTimestamp, long endTimestamp) { |
| int deletedPointsNumber = 0; |
| Set<String> measurements = memChunk.getAllMeasurements(); |
| List<String> columnsToBeRemoved = new ArrayList<>(); |
| String targetMeasurement = originalPath.getMeasurement(); |
| if (PathPatternUtil.hasWildcard(targetMeasurement)) { |
| for (String measurement : measurements) { |
| if (!PathPatternUtil.isNodeMatch(targetMeasurement, measurement)) { |
| continue; |
| } |
| Pair<Integer, Boolean> deleteInfo = |
| memChunk.deleteDataFromAColumn(startTimestamp, endTimestamp, measurement); |
| deletedPointsNumber += deleteInfo.left; |
| if (Boolean.TRUE.equals(deleteInfo.right)) { |
| columnsToBeRemoved.add(measurement); |
| } |
| } |
| } else { |
| if (measurements.contains(targetMeasurement)) { |
| Pair<Integer, Boolean> deleteInfo = |
| memChunk.deleteDataFromAColumn(startTimestamp, endTimestamp, targetMeasurement); |
| deletedPointsNumber += deleteInfo.left; |
| if (Boolean.TRUE.equals(deleteInfo.right)) { |
| columnsToBeRemoved.add(targetMeasurement); |
| } |
| } |
| } |
| |
| for (String columnToBeRemoved : columnsToBeRemoved) { |
| memChunk.removeColumn(columnToBeRemoved); |
| } |
| return deletedPointsNumber; |
| } |
| |
| @Override |
| public long getCurrentTVListSize(String measurement) { |
| return memChunk.getTVList().rowCount(); |
| } |
| |
| @Override |
| public long getMaxTime() { |
| return memChunk.getMaxTime(); |
| } |
| |
| public AlignedWritableMemChunk getAlignedMemChunk() { |
| return memChunk; |
| } |
| |
| @Override |
| public int serializedSize() { |
| return memChunk.serializedSize(); |
| } |
| |
| @Override |
| public void serializeToWAL(IWALByteBufferView buffer) { |
| memChunk.serializeToWAL(buffer); |
| } |
| |
| public static AlignedWritableMemChunkGroup deserialize(DataInputStream stream) |
| throws IOException { |
| AlignedWritableMemChunkGroup memChunkGroup = new AlignedWritableMemChunkGroup(); |
| memChunkGroup.memChunk = AlignedWritableMemChunk.deserialize(stream); |
| return memChunkGroup; |
| } |
| } |