@itwin/tree-widget-react 3.15.0 → 3.16.0

package/lib/cjs/tree-widget-react/components/trees/models-tree/ModelsTreeDefinition.js

@@ -6,6 +6,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ModelsTreeDefinition = exports.ModelsTreeInstanceKeyPathsProps = exports.defaultHierarchyConfiguration = void 0;
 const rxjs_1 = require("rxjs");
+const core_bentley_1 = require("@itwin/core-bentley");
 const core_common_1 = require("@itwin/core-common");
 const presentation_hierarchies_1 = require("@itwin/presentation-hierarchies");
 const presentation_shared_1 = require("@itwin/presentation-shared");
@@ -31,18 +32,20 @@ var ModelsTreeInstanceKeyPathsProps;
     ModelsTreeInstanceKeyPathsProps.isLabelProps = isLabelProps;
 })(ModelsTreeInstanceKeyPathsProps || (exports.ModelsTreeInstanceKeyPathsProps = ModelsTreeInstanceKeyPathsProps = {}));
 class ModelsTreeDefinition {
-    _impl;
-    _idsCache;
-    _hierarchyConfig;
-    _selectQueryFactory;
-    _nodeLabelSelectClauseFactory;
-    _queryExecutor;
-    _isSupported;
+    #impl;
+    #idsCache;
+    #hierarchyConfig;
+    #selectQueryFactory;
+    #nodeLabelSelectClauseFactory;
+    #queryExecutor;
+    #isSupported;
+    static #componentName = "ModelsTreeDefinition";
+    #componentId;
     constructor(props) {
-        this._impl = (0, presentation_hierarchies_1.createPredicateBasedHierarchyDefinition)({
+        this.#impl = (0, presentation_hierarchies_1.createPredicateBasedHierarchyDefinition)({
             classHierarchyInspector: props.imodelAccess,
             hierarchy: {
-                rootNodes: async (requestProps) => this.createSubjectChildrenQuery({ ...requestProps, parentNodeInstanceIds: this._hierarchyConfig.hideRootSubject ? [core_common_1.IModel.rootSubjectId] : [] }),
+                rootNodes: async (requestProps) => this.createSubjectChildrenQuery({ ...requestProps, parentNodeInstanceIds: this.#hierarchyConfig.hideRootSubject ? [core_common_1.IModel.rootSubjectId] : [] }),
                 childNodes: [
                     {
                         parentInstancesNodePredicate: "BisCore.Subject",
@@ -67,24 +70,48 @@ class ModelsTreeDefinition {
                 ],
             },
         });
-        this._idsCache = props.idsCache;
-        this._queryExecutor = props.imodelAccess;
-        this._hierarchyConfig = props.hierarchyConfig;
-        this._nodeLabelSelectClauseFactory = (0, presentation_shared_1.createBisInstanceLabelSelectClauseFactory)({ classHierarchyInspector: props.imodelAccess });
-        this._selectQueryFactory = (0, presentation_hierarchies_1.createNodesQueryClauseFactory)({
+        this.#componentId = props.componentId ?? core_bentley_1.Guid.createValue();
+        this.#idsCache = props.idsCache;
+        this.#queryExecutor = props.imodelAccess;
+        this.#hierarchyConfig = props.hierarchyConfig;
+        this.#nodeLabelSelectClauseFactory = (0, presentation_shared_1.createBisInstanceLabelSelectClauseFactory)({ classHierarchyInspector: props.imodelAccess });
+        this.#selectQueryFactory = (0, presentation_hierarchies_1.createNodesQueryClauseFactory)({
             imodelAccess: props.imodelAccess,
-            instanceLabelSelectClauseFactory: this._nodeLabelSelectClauseFactory,
+            instanceLabelSelectClauseFactory: this.#nodeLabelSelectClauseFactory,
         });
     }
     async postProcessNode(node) {
         if (presentation_hierarchies_1.ProcessedHierarchyNode.isGroupingNode(node)) {
+            let hasDirectNonFilteredTargets = false;
+            let hasFilterTargetAncestor = false;
+            for (const child of node.children) {
+                if (child.filtering) {
+                    if (child.filtering.hasFilterTargetAncestor) {
+                        hasFilterTargetAncestor = true;
+                        break;
+                    }
+                    if (!child.filtering.isFilterTarget) {
+                        hasDirectNonFilteredTargets = true;
+                        break;
+                    }
+                }
+            }
             return {
                 ...node,
-                label: this._hierarchyConfig.elementClassGrouping === "enableWithCounts" ? `${node.label} (${node.children.length})` : node.label,
+                ...(hasFilterTargetAncestor
+                    ? {
+                        filtering: {
+                            ...(node.filtering ?? {}),
+                            hasFilterTargetAncestor,
+                        },
+                    }
+                    : {}),
+                label: this.#hierarchyConfig.elementClassGrouping === "enableWithCounts" ? `${node.label} (${node.children.length})` : node.label,
                 extendedData: {
                     ...node.extendedData,
                     // add `modelId` and `categoryId` from the first grouped element
                     ...node.children[0].extendedData,
+                    ...(hasDirectNonFilteredTargets ? { hasDirectNonFilteredTargets } : {}),
                     // `imageId` is assigned to instance nodes at query time, but grouping ones need to
                     // be handled during post-processing
                     imageId: "icon-ec-class",
@@ -94,27 +121,27 @@ class ModelsTreeDefinition {
         return node;
     }
     async defineHierarchyLevel(props) {
-        if (this._isSupported === undefined) {
-            this._isSupported = this.isSupported();
+        if (this.#isSupported === undefined) {
+            this.#isSupported = this.isSupported();
         }
-        if ((await this._isSupported) === false) {
+        if ((await this.#isSupported) === false) {
             return [];
         }
-        return this._impl.defineHierarchyLevel(props);
+        return this.#impl.defineHierarchyLevel(props);
     }
     async createSubjectChildrenQuery({ parentNodeInstanceIds: parentSubjectIds, instanceFilter, }) {
         const [subjectFilterClauses, modelFilterClauses] = await Promise.all([
-            this._selectQueryFactory.createFilterClauses({
+            this.#selectQueryFactory.createFilterClauses({
                 filter: instanceFilter,
                 contentClass: { fullName: "BisCore.Subject", alias: "this" },
             }),
-            this._selectQueryFactory.createFilterClauses({
+            this.#selectQueryFactory.createFilterClauses({
                 filter: instanceFilter,
                 contentClass: { fullName: "BisCore.GeometricModel3d", alias: "this" },
             }),
         ]);
         const [childSubjectIds, childModelIds] = parentSubjectIds.length
-            ? await Promise.all([this._idsCache.getChildSubjectIds(parentSubjectIds), this._idsCache.getChildSubjectModelIds(parentSubjectIds)])
+            ? await Promise.all([this.#idsCache.getChildSubjectIds(parentSubjectIds), this.#idsCache.getChildSubjectModelIds(parentSubjectIds)])
             : [[core_common_1.IModel.rootSubjectId], []];
         const defs = new Array();
         childSubjectIds.length &&
@@ -123,11 +150,11 @@ class ModelsTreeDefinition {
                 query: {
                     ecsql: `
             SELECT
-              ${await this._selectQueryFactory.createSelectClause({
+              ${await this.#selectQueryFactory.createSelectClause({
                         ecClassId: { selector: "this.ECClassId" },
                         ecInstanceId: { selector: "this.ECInstanceId" },
                         nodeLabel: {
-                            selector: await this._nodeLabelSelectClauseFactory.createSelectClause({
+                            selector: await this.#nodeLabelSelectClauseFactory.createSelectClause({
                                 classAlias: "this",
                                 className: "BisCore.Subject",
                             }),
@@ -149,7 +176,7 @@ class ModelsTreeDefinition {
               ${subjectFilterClauses.where ? `AND ${subjectFilterClauses.where}` : ""}
           `,
                     bindings: [
-                        { type: "idset", value: await this._idsCache.getParentSubjectIds() },
+                        { type: "idset", value: await this.#idsCache.getParentSubjectIds() },
                         ...childSubjectIds.map((id) => ({ type: "id", value: id })),
                     ],
                 },
@@ -162,11 +189,11 @@ class ModelsTreeDefinition {
             SELECT model.ECInstanceId AS ECInstanceId, model.*
             FROM (
               SELECT
-                ${await this._selectQueryFactory.createSelectClause({
+                ${await this.#selectQueryFactory.createSelectClause({
                         ecClassId: { selector: "m.ECClassId" },
                         ecInstanceId: { selector: "m.ECInstanceId" },
                         nodeLabel: {
-                            selector: await this._nodeLabelSelectClauseFactory.createSelectClause({
+                            selector: await this.#nodeLabelSelectClauseFactory.createSelectClause({
                                 classAlias: "partition",
                                 className: "BisCore.InformationPartitionElement",
                             }),
@@ -182,12 +209,12 @@ class ModelsTreeDefinition {
                       END
                     `,
                         },
-                        hasChildren: this._hierarchyConfig.showEmptyModels
+                        hasChildren: this.#hierarchyConfig.showEmptyModels
                             ? {
                                 selector: `
                           IFNULL((
                             SELECT 1
-                            FROM ${this._hierarchyConfig.elementClassSpecification} e
+                            FROM ${this.#hierarchyConfig.elementClassSpecification} e
                             WHERE e.Model.Id = m.ECInstanceId
                             LIMIT 1
                           ), 0)
@@ -223,7 +250,7 @@ class ModelsTreeDefinition {
                 query: {
                     ecsql: `
             SELECT
-              ${await this._selectQueryFactory.createSelectClause({
+              ${await this.#selectQueryFactory.createSelectClause({
                         ecClassId: { selector: "this.ECClassId" },
                         ecInstanceId: { selector: "this.ECInstanceId" },
                         nodeLabel: "", // doesn't matter - the node is always hidden
@@ -233,7 +260,7 @@ class ModelsTreeDefinition {
             WHERE
               this.ModeledElement.Id IN (${elementIds.map(() => "?").join(",")})
               AND NOT this.IsPrivate
-              AND this.ECInstanceId IN (SELECT Model.Id FROM ${this._hierarchyConfig.elementClassSpecification})
+              AND this.ECInstanceId IN (SELECT Model.Id FROM ${this.#hierarchyConfig.elementClassSpecification})
           `,
                     bindings: [...elementIds.map((id) => ({ type: "id", value: id }))],
                 },
@@ -241,7 +268,7 @@ class ModelsTreeDefinition {
         ];
     }
     async createGeometricModel3dChildrenQuery({ parentNodeInstanceIds: modelIds, instanceFilter, }) {
-        const instanceFilterClauses = await this._selectQueryFactory.createFilterClauses({
+        const instanceFilterClauses = await this.#selectQueryFactory.createFilterClauses({
             filter: instanceFilter,
             contentClass: { fullName: "BisCore.SpatialCategory", alias: "this" },
         });
@@ -251,11 +278,11 @@ class ModelsTreeDefinition {
                 query: {
                     ecsql: `
             SELECT
-              ${await this._selectQueryFactory.createSelectClause({
+              ${await this.#selectQueryFactory.createSelectClause({
                         ecClassId: { selector: "this.ECClassId" },
                         ecInstanceId: { selector: "this.ECInstanceId" },
                         nodeLabel: {
-                            selector: await this._nodeLabelSelectClauseFactory.createSelectClause({
+                            selector: await this.#nodeLabelSelectClauseFactory.createSelectClause({
                                 classAlias: "this",
                                 className: "BisCore.SpatialCategory",
                             }),
@@ -274,7 +301,7 @@ class ModelsTreeDefinition {
             WHERE
               EXISTS (
                 SELECT 1
-                FROM ${this._hierarchyConfig.elementClassSpecification} element
+                FROM ${this.#hierarchyConfig.elementClassSpecification} element
                 WHERE
                   element.Model.Id IN (${modelIds.map(() => "?").join(",")})
                   AND element.Category.Id = +this.ECInstanceId
@@ -292,30 +319,30 @@ class ModelsTreeDefinition {
         if (modelIds.length === 0) {
             throw new Error(`Invalid category node "${parentNode.label}" - missing model information.`);
         }
-        const instanceFilterClauses = await this._selectQueryFactory.createFilterClauses({
+        const instanceFilterClauses = await this.#selectQueryFactory.createFilterClauses({
             filter: instanceFilter,
-            contentClass: { fullName: this._hierarchyConfig.elementClassSpecification, alias: "this" },
+            contentClass: { fullName: this.#hierarchyConfig.elementClassSpecification, alias: "this" },
         });
-        const modeledElements = await (0, rxjs_1.firstValueFrom)((0, rxjs_1.from)(modelIds).pipe((0, rxjs_1.mergeMap)(async (modelId) => this._idsCache.getCategoriesModeledElements(modelId, categoryIds)), (0, rxjs_1.reduce)((acc, foundModeledElements) => {
+        const modeledElements = await (0, rxjs_1.firstValueFrom)((0, rxjs_1.from)(modelIds).pipe((0, rxjs_1.mergeMap)(async (modelId) => this.#idsCache.getCategoriesModeledElements(modelId, categoryIds)), (0, rxjs_1.reduce)((acc, foundModeledElements) => {
             return acc.concat(foundModeledElements);
         }, new Array())));
         return [
             {
-                fullClassName: this._hierarchyConfig.elementClassSpecification,
+                fullClassName: this.#hierarchyConfig.elementClassSpecification,
                 query: {
                     ecsql: `
             SELECT
-              ${await this._selectQueryFactory.createSelectClause({
+              ${await this.#selectQueryFactory.createSelectClause({
                         ecClassId: { selector: "this.ECClassId" },
                         ecInstanceId: { selector: "this.ECInstanceId" },
                         nodeLabel: {
-                            selector: await this._nodeLabelSelectClauseFactory.createSelectClause({
+                            selector: await this.#nodeLabelSelectClauseFactory.createSelectClause({
                                 classAlias: "this",
-                                className: this._hierarchyConfig.elementClassSpecification,
+                                className: this.#hierarchyConfig.elementClassSpecification,
                             }),
                         },
                         grouping: {
-                            byClass: this._hierarchyConfig.elementClassGrouping !== "disable",
+                            byClass: this.#hierarchyConfig.elementClassGrouping !== "disable",
                         },
                         hasChildren: {
                             selector: `
@@ -324,7 +351,7 @@ class ModelsTreeDefinition {
                       1,
                       IFNULL((
                         SELECT 1
-                        FROM ${this._hierarchyConfig.elementClassSpecification} ce
+                        FROM ${this.#hierarchyConfig.elementClassSpecification} ce
                         WHERE ce.Parent.Id = this.ECInstanceId
                         LIMIT 1
                       ), 0)
@@ -352,33 +379,33 @@ class ModelsTreeDefinition {
         ];
     }
     async createGeometricElement3dChildrenQuery({ parentNodeInstanceIds: elementIds, instanceFilter, }) {
-        const instanceFilterClauses = await this._selectQueryFactory.createFilterClauses({
+        const instanceFilterClauses = await this.#selectQueryFactory.createFilterClauses({
             filter: instanceFilter,
-            contentClass: { fullName: this._hierarchyConfig.elementClassSpecification, alias: "this" },
+            contentClass: { fullName: this.#hierarchyConfig.elementClassSpecification, alias: "this" },
         });
         return [
             {
-                fullClassName: this._hierarchyConfig.elementClassSpecification,
+                fullClassName: this.#hierarchyConfig.elementClassSpecification,
                 query: {
                     ecsql: `
             SELECT
-              ${await this._selectQueryFactory.createSelectClause({
+              ${await this.#selectQueryFactory.createSelectClause({
                         ecClassId: { selector: "this.ECClassId" },
                         ecInstanceId: { selector: "this.ECInstanceId" },
                         nodeLabel: {
-                            selector: await this._nodeLabelSelectClauseFactory.createSelectClause({
+                            selector: await this.#nodeLabelSelectClauseFactory.createSelectClause({
                                 classAlias: "this",
-                                className: this._hierarchyConfig.elementClassSpecification,
+                                className: this.#hierarchyConfig.elementClassSpecification,
                             }),
                         },
                         grouping: {
-                            byClass: this._hierarchyConfig.elementClassGrouping !== "disable",
+                            byClass: this.#hierarchyConfig.elementClassGrouping !== "disable",
                         },
                         hasChildren: {
                             selector: `
                     IFNULL((
                       SELECT 1
-                      FROM ${this._hierarchyConfig.elementClassSpecification} ce
+                      FROM ${this.#hierarchyConfig.elementClassSpecification} ce
                       JOIN BisCore.Model m ON ce.Model.Id = m.ECInstanceId
                       WHERE ce.Parent.Id = this.ECInstanceId OR (ce.Model.Id = this.ECInstanceId AND m.IsPrivate = false)
                       LIMIT 1
@@ -405,20 +432,25 @@ class ModelsTreeDefinition {
     }
     static async createInstanceKeyPaths(props) {
         return (0, rxjs_1.lastValueFrom)((0, rxjs_1.defer)(() => {
+            const componentInfo = { componentId: props.componentId ?? core_bentley_1.Guid.createValue(), componentName: this.#componentName };
             if (ModelsTreeInstanceKeyPathsProps.isLabelProps(props)) {
                 const labelsFactory = (0, presentation_shared_1.createBisInstanceLabelSelectClauseFactory)({ classHierarchyInspector: props.imodelAccess });
-                return createInstanceKeyPathsFromInstanceLabelObs({ ...props, labelsFactory });
+                return createInstanceKeyPathsFromInstanceLabelObs({
+                    ...props,
+                    ...componentInfo,
+                    labelsFactory,
+                });
             }
-            return createInstanceKeyPathsFromTargetItemsObs(props);
+            return createInstanceKeyPathsFromTargetItemsObs({ ...props, ...componentInfo });
         }).pipe(props.abortSignal ? (0, rxjs_1.takeUntil)((0, rxjs_1.fromEvent)(props.abortSignal, "abort")) : rxjs_1.identity, (0, rxjs_1.defaultIfEmpty)([])));
     }
     supportsFiltering() {
-        return this._hierarchyConfig.hierarchyLevelFiltering === "enable";
+        return this.#hierarchyConfig.hierarchyLevelFiltering === "enable";
     }
     async isSupported() {
-        const [schemaName, className] = this._hierarchyConfig.elementClassSpecification.split(/[\.:]/);
+        const [schemaName, className] = this.#hierarchyConfig.elementClassSpecification.split(/[\.:]/);
         if (!schemaName || !className) {
-            throw new Error(`Provided class specification ${this._hierarchyConfig.elementClassSpecification} should be in format {SchemaName}:{ClassName} or {SchemaName}.{ClassName}`);
+            throw new Error(`Provided class specification ${this.#hierarchyConfig.elementClassSpecification} should be in format {SchemaName}:{ClassName} or {SchemaName}.{ClassName}`);
         }
         const query = {
             ecsql: `
@@ -432,14 +464,17 @@ class ModelsTreeDefinition {
                 { type: "string", value: className },
             ],
         };
-        for await (const _row of this._queryExecutor.createQueryReader(query)) {
+        for await (const _row of this.#queryExecutor.createQueryReader(query, {
+            restartToken: `${ModelsTreeDefinition.#componentName}/${this.#componentId}/is-class-supported`,
+        })) {
             return true;
         }
         return false;
     }
 }
 exports.ModelsTreeDefinition = ModelsTreeDefinition;
-function createGeometricElementInstanceKeyPaths(imodelAccess, idsCache, hierarchyConfig, targetItems) {
+function createGeometricElementInstanceKeyPaths(props) {
+    const { targetItems, chunkIndex, componentId, componentName, hierarchyConfig, idsCache, imodelAccess } = props;
     const elementIds = targetItems.filter((info) => typeof info === "string");
     const groupInfos = targetItems.filter((info) => typeof info !== "string");
     const separator = ";";
@@ -499,7 +534,7 @@ function createGeometricElementInstanceKeyPaths(imodelAccess, idsCache, hierarch
       FROM ModelsCategoriesElementsHierarchy mce
       WHERE mce.ParentId IS NULL
     `;
-        return imodelAccess.createQueryReader({ ctes, ecsql }, { rowFormat: "Indexes", limit: "unbounded" });
+        return imodelAccess.createQueryReader({ ctes, ecsql }, { rowFormat: "Indexes", limit: "unbounded", restartToken: `${componentName}/${componentId}/geometric-element-paths/${chunkIndex}` });
     }).pipe((0, Utils_js_2.releaseMainThreadOnItemsCount)(300), (0, rxjs_1.map)((row) => parseQueryRow(row, groupInfos, separator, hierarchyConfig.elementClassSpecification)), (0, rxjs_1.mergeMap)(({ modelId, elementHierarchyPath, groupingNode }) => (0, rxjs_1.from)(idsCache.createModelInstanceKeyPaths(modelId)).pipe((0, rxjs_1.mergeAll)(), (0, rxjs_1.map)((modelPath) => {
         // We don't want to modify the original path, we create a copy that we can modify
         const newModelPath = [...modelPath];
@@ -541,7 +576,7 @@ function parseQueryRow(row, groupInfos, separator, elementClassName) {
         groupingNode: row[2] === -1 ? undefined : groupInfos[row[2]].groupingNode,
     };
 }
-function createInstanceKeyPathsFromTargetItemsObs({ targetItems, imodelAccess, hierarchyConfig, idsCache, limit, }) {
+function createInstanceKeyPathsFromTargetItemsObs({ targetItems, imodelAccess, hierarchyConfig, idsCache, limit, componentId, componentName, }) {
     if (limit !== "unbounded" && targetItems.length > (limit ?? MAX_FILTERING_INSTANCE_KEY_COUNT)) {
         throw new TreeErrors_js_1.FilterLimitExceededError(limit ?? MAX_FILTERING_INSTANCE_KEY_COUNT);
     }
@@ -581,7 +616,15 @@ function createInstanceKeyPathsFromTargetItemsObs({ targetItems, imodelAccess, h
         elements: new Array(),
     }), (0, rxjs_1.switchMap)(async (ids) => {
         const elementsLength = ids.elements.length;
-        return (0, Rxjs_js_1.collect)((0, rxjs_1.merge)((0, rxjs_1.from)(ids.subjects).pipe((0, rxjs_1.mergeMap)((id) => (0, rxjs_1.from)(idsCache.createSubjectInstanceKeysPath(id)).pipe((0, rxjs_1.map)(presentation_hierarchies_1.HierarchyFilteringPath.normalize)))), (0, rxjs_1.from)(ids.models).pipe((0, rxjs_1.mergeMap)((id) => (0, rxjs_1.from)(idsCache.createModelInstanceKeyPaths(id)).pipe((0, rxjs_1.mergeAll)(), (0, rxjs_1.map)(presentation_hierarchies_1.HierarchyFilteringPath.normalize)))), (0, rxjs_1.from)(ids.categories).pipe((0, rxjs_1.mergeMap)((id) => (0, rxjs_1.from)(idsCache.createCategoryInstanceKeyPaths(id)).pipe((0, rxjs_1.mergeAll)(), (0, rxjs_1.map)(presentation_hierarchies_1.HierarchyFilteringPath.normalize)))), (0, rxjs_1.from)(ids.elements).pipe((0, rxjs_1.bufferCount)(Math.ceil(elementsLength / Math.ceil(elementsLength / 5000))), (0, Utils_js_2.releaseMainThreadOnItemsCount)(1), (0, rxjs_1.mergeMap)((block) => createGeometricElementInstanceKeyPaths(imodelAccess, idsCache, hierarchyConfig, block), 10))));
+        return (0, Rxjs_js_1.collect)((0, rxjs_1.merge)((0, rxjs_1.from)(ids.subjects).pipe((0, rxjs_1.mergeMap)((id) => (0, rxjs_1.from)(idsCache.createSubjectInstanceKeysPath(id)).pipe((0, rxjs_1.map)(presentation_hierarchies_1.HierarchyFilteringPath.normalize)))), (0, rxjs_1.from)(ids.models).pipe((0, rxjs_1.mergeMap)((id) => (0, rxjs_1.from)(idsCache.createModelInstanceKeyPaths(id)).pipe((0, rxjs_1.mergeAll)(), (0, rxjs_1.map)(presentation_hierarchies_1.HierarchyFilteringPath.normalize)))), (0, rxjs_1.from)(ids.categories).pipe((0, rxjs_1.mergeMap)((id) => (0, rxjs_1.from)(idsCache.createCategoryInstanceKeyPaths(id)).pipe((0, rxjs_1.mergeAll)(), (0, rxjs_1.map)(presentation_hierarchies_1.HierarchyFilteringPath.normalize)))), (0, rxjs_1.from)(ids.elements).pipe((0, rxjs_1.bufferCount)(Math.ceil(elementsLength / Math.ceil(elementsLength / 5000))), (0, Utils_js_2.releaseMainThreadOnItemsCount)(1), (0, rxjs_1.mergeMap)((block, chunkIndex) => createGeometricElementInstanceKeyPaths({
+            imodelAccess,
+            idsCache,
+            hierarchyConfig,
+            targetItems: block,
+            componentId,
+            componentName,
+            chunkIndex,
+        }), 10))));
     }));
 }
 function createInstanceKeyPathsFromInstanceLabelObs(props) {
@@ -624,7 +667,7 @@ function createInstanceKeyPathsFromInstanceLabelObs(props) {
     }).pipe((0, rxjs_1.mergeMap)((queryProps) => {
         return imodelAccess.createQueryReader(queryProps, {
             rowFormat: "Indexes",
-            restartToken: "tree-widget/models-tree/filter-by-label-query",
+            restartToken: `${props.componentName}/${props.componentId}/filter-by-label`,
             limit,
         });
     }), (0, rxjs_1.map)((row) => ({ className: row[0], id: row[1] })), (0, rxjs_1.toArray)(), (0, rxjs_1.mergeMap)((targetKeys) => createInstanceKeyPathsFromTargetItemsObs({ ...props, targetItems: targetKeys })));