JackieTien97 commented on code in PR #12539:
URL: https://github.com/apache/iotdb/pull/12539#discussion_r1604781686
##########
iotdb-core/datanode/src/main/java/org/apache/iotdb/db/storageengine/dataregion/memtable/AbstractMemTable.java:
##########
@@ -465,6 +481,278 @@ public ReadOnlyMemChunk query(
.getReadOnlyMemChunkFromMemTable(context, this, modsToMemtable,
ttlLowerBound);
}
+ @Override
+ public void queryForSeriesRegionScan(
+ PartialPath fullPath,
+ long ttlLowerBound,
+ Map<String, List<IChunkMetadata>> chunkMetaDataMap,
+ Map<String, List<IChunkHandle>> memChunkHandleMap,
+ List<Pair<Modification, IMemTable>> modsToMemTabled) {
+
+ IDeviceID deviceID =
DeviceIDFactory.getInstance().getDeviceID(fullPath.getDevicePath());
+ String measurementId = fullPath.getMeasurement();
+ Map<IDeviceID, IWritableMemChunkGroup> memTableMap = getMemTableMap();
+
+ // check If MemTable Contains this path
+ if (!memTableMap.containsKey(deviceID) ||
!memTableMap.get(deviceID).contains(measurementId)) {
+ return;
+ }
+
+ if (fullPath instanceof MeasurementPath) {
+ List<TimeRange> deletionList = new ArrayList<>();
+ if (modsToMemTabled != null) {
+ deletionList =
+ ModificationUtils.constructDeletionList(
+ (MeasurementPath) fullPath, this, modsToMemTabled,
ttlLowerBound);
+ }
+ getMemChunkHandleFromMemTable(
+ deviceID, measurementId, chunkMetaDataMap, memChunkHandleMap,
deletionList);
+ } else {
+ List<List<TimeRange>> deletionList = new ArrayList<>();
+ if (modsToMemTabled != null) {
+ deletionList =
+ ModificationUtils.constructDeletionList(
+ (AlignedPath) fullPath, this, modsToMemTabled, ttlLowerBound);
+ }
+
+ getMemAlignedChunkHandleFromMemTable(
+ deviceID,
+ ((AlignedPath) fullPath).getSchemaList(),
+ chunkMetaDataMap,
+ memChunkHandleMap,
+ deletionList);
+ }
+ }
+
+ @Override
+ public void queryForDeviceRegionScan(
+ IDeviceID deviceID,
+ boolean isAligned,
+ long ttlLowerBound,
+ Map<String, List<IChunkMetadata>> chunkMetadataMap,
+ Map<String, List<IChunkHandle>> memChunkHandleMap,
+ List<Pair<Modification, IMemTable>> modsToMemTabled)
+ throws MetadataException {
+
+ Map<IDeviceID, IWritableMemChunkGroup> memTableMap = getMemTableMap();
+
+ // check If MemTable Contains this path
+ if (!memTableMap.containsKey(deviceID)) {
+ return;
+ }
+
+ IWritableMemChunkGroup writableMemChunkGroup = memTableMap.get(deviceID);
+ if (isAligned) {
+ getMemAlignedChunkHandleFromMemTable(
+ ((PlainDeviceID) deviceID).toStringID(),
+ (AlignedWritableMemChunkGroup) writableMemChunkGroup,
+ chunkMetadataMap,
+ memChunkHandleMap,
+ ttlLowerBound,
+ modsToMemTabled);
+ } else {
+ getMemChunkHandleFromMemTable(
+ ((PlainDeviceID) deviceID).toStringID(),
+ (WritableMemChunkGroup) writableMemChunkGroup,
+ chunkMetadataMap,
+ memChunkHandleMap,
+ ttlLowerBound,
+ modsToMemTabled);
+ }
+ }
+
+ private void getMemChunkHandleFromMemTable(
+ IDeviceID deviceID,
+ String measurementId,
+ Map<String, List<IChunkMetadata>> chunkMetadataMap,
+ Map<String, List<IChunkHandle>> memChunkHandleMap,
+ List<TimeRange> deletionList) {
+
+ IWritableMemChunk memChunk =
memTableMap.get(deviceID).getMemChunkMap().get(measurementId);
+
+ TVList tvListCopy = memChunk.getSortedTvListForQuery();
+ long[] timestamps = filterDeletedTimestamp(tvListCopy, deletionList);
+
+ chunkMetadataMap
+ .computeIfAbsent(measurementId, k -> new ArrayList<>())
+ .add(
+ buildChunkMetaDataForMemoryChunk(
+ measurementId,
+ timestamps[0],
+ timestamps[timestamps.length - 1],
+ Collections.emptyList()));
+ memChunkHandleMap
+ .computeIfAbsent(measurementId, k -> new ArrayList<>())
+ .add(new MemChunkHandleImpl(timestamps));
+ }
+
+ private void getMemAlignedChunkHandleFromMemTable(
+ IDeviceID deviceID,
+ List<IMeasurementSchema> schemaList,
+ Map<String, List<IChunkMetadata>> chunkMetadataList,
+ Map<String, List<IChunkHandle>> memChunkHandleMap,
+ List<List<TimeRange>> deletionList) {
+
+ AlignedWritableMemChunk alignedMemChunk =
+ ((AlignedWritableMemChunkGroup)
memTableMap.get(deviceID)).getAlignedMemChunk();
+ AlignedTVList alignedTVListCopy =
+ (AlignedTVList) alignedMemChunk.getSortedTvListForQuery(schemaList);
+
+ buildAlignedMemChunkHandle(
+ alignedTVListCopy, deletionList, schemaList, chunkMetadataList,
memChunkHandleMap);
+ }
+
+ private void getMemAlignedChunkHandleFromMemTable(
+ String deviceID,
+ AlignedWritableMemChunkGroup writableMemChunkGroup,
+ Map<String, List<IChunkMetadata>> chunkMetadataList,
+ Map<String, List<IChunkHandle>> memChunkHandleMap,
+ long ttlLowerBound,
+ List<Pair<Modification, IMemTable>> modsToMemTabled) {
+
+ AlignedWritableMemChunk memChunk =
writableMemChunkGroup.getAlignedMemChunk();
+ List<IMeasurementSchema> schemaList = memChunk.getSchemaList();
+
+ AlignedTVList alignedTVListCopy = (AlignedTVList)
memChunk.getSortedTvListForQuery(schemaList);
+
+ List<List<TimeRange>> deletionList = new ArrayList<>();
+ for (IMeasurementSchema schema : schemaList) {
Review Comment:
what if modsToMemTabled is null or empty
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
