JackieTien97 commented on code in PR #17169:
URL: https://github.com/apache/iotdb/pull/17169#discussion_r2938526486
##########
iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/execution/operator/source/relational/InformationSchemaContentSupplierFactory.java:
##########
@@ -1165,7 +1215,297 @@ public boolean hasNext() {
}
}
- private abstract static class TsBlockSupplier implements Iterator<TsBlock> {
+ private static class TableDiskUsageSupplier implements
IInformationSchemaContentSupplier {
+ private final List<TSDataType> dataTypes;
+ private final Map<String, List<TTableInfo>> databaseTableInfoMap;
+ private final Filter pushDownFilter;
+ private final PaginationController paginationController;
+ private final OperatorContext operatorContext;
+
+ private DataRegion currentDataRegion;
+ private boolean currentDatabaseOnlyHasOneTable;
+
+ private TableDiskUsageCacheReader currentDataRegionCacheReader;
+ private DataRegionTableSizeQueryContext
currentDataRegionTableSizeQueryContext;
+
+ private final StorageEngineTimePartitionIterator dataRegionIterator;
+
+ private long prepareCacheReaderCostInNS = 0;
+ private long loadObjectFileCostInNS = 0;
+ private long prepareCachedTsFileIDCostInNS = 0;
+ private long checkAllFilesInTsFileManagerCostInNS = 0;
+ private long readTsFileCacheValueFilesCostInNS = 0;
+
+ private TableDiskUsageSupplier(
+ final List<TSDataType> dataTypes,
+ final UserEntity userEntity,
+ final Filter pushDownFilter,
+ final PaginationController paginationController,
+ final OperatorContext operatorContext,
+ final List<Integer> regionsForCurrentSubTask)
+ throws TException, ClientManagerException {
+ this.dataTypes = dataTypes;
+ this.pushDownFilter = pushDownFilter;
+ this.paginationController = paginationController;
+ this.operatorContext = operatorContext;
+
AuthorityChecker.getAccessControl().checkUserGlobalSysPrivilege(userEntity);
+ this.databaseTableInfoMap =
+
operatorContext.getInstanceContext().getDataNodeQueryContext().getDatabaseTableInfoMap();
+ Set<Integer> regions = new HashSet<>(regionsForCurrentSubTask);
+ operatorContext.recordSpecifiedInfo(
+ PlanGraphPrinter.REGIONS_OF_CURRENT_SUB_TASK,
regionsForCurrentSubTask.toString());
+ this.dataRegionIterator =
+ new StorageEngineTimePartitionIterator(
+ Optional.of(
+ dataRegion -> {
+ List<TTableInfo> tTableInfos =
+ databaseTableInfoMap.get(dataRegion.getDatabaseName());
+ if (tTableInfos == null || tTableInfos.isEmpty()) {
+ return false;
+ }
+ if (!regions.contains(dataRegion.getDataRegionId())) {
+ return false;
+ }
+ currentDataRegionTableSizeQueryContext =
+ new DataRegionTableSizeQueryContext(
+ false, operatorContext.getInstanceContext());
+ return true;
+ }),
+ Optional.empty());
Review Comment:
add some comments about this weird codes.
--
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]
