@itwin/tree-widget-react 4.0.0-alpha.25 → 4.0.0-alpha.26

package/lib/esm/tree-widget-react/components/trees/common/internal/caches/ElementChildrenCache.js

@@ -0,0 +1,151 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { defaultIfEmpty, defer, EMPTY, firstValueFrom, from, map, mergeMap, reduce, tap } from "rxjs";
+import { Guid, Id64 } from "@itwin/core-bentley";
+import { catchBeSQLiteInterrupts } from "../UseErrorState.js";
+import { getOptimalBatchSize } from "../Utils.js";
+/** @internal */
+export class ElementChildrenCache {
+    #queryExecutor;
+    #elementClassName;
+    #componentId;
+    #componentName;
+    #childElementsMap;
+    /** Stores element ids which have children query scheduled to execute. */
+    #childElementsLoadingMap;
+    constructor(props) {
+        this.#queryExecutor = props.queryExecutor;
+        this.#elementClassName = props.elementClassName;
+        this.#componentId = props.componentId;
+        this.#componentName = "ElementChildrenCache";
+        this.#childElementsMap = new Map();
+        this.#childElementsLoadingMap = new Map();
+    }
+    queryChildren({ elementIds }) {
+        if (elementIds.length === 0) {
+            return EMPTY;
+        }
+        return defer(() => {
+            const ctes = [
+                `
+          ElementChildren(id, parentId) AS (
+            SELECT ECInstanceId id, Parent.Id parentId
+            FROM ${this.#elementClassName}
+            WHERE Parent.Id IN (${elementIds.join(", ")})
+            UNION ALL
+            SELECT c.ECInstanceId id, c.Parent.Id
+            FROM ${this.#elementClassName} c
+            JOIN ElementChildren p ON c.Parent.Id = p.id
+          )
+        `,
+            ];
+            const ecsql = `
+        SELECT id, parentId
+        FROM ElementChildren
+      `;
+            return this.#queryExecutor.createQueryReader({ ecsql, ctes }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded", restartToken: `${this.#componentName}/${this.#componentId}/children/${Guid.createValue()}` });
+        }).pipe(catchBeSQLiteInterrupts, map((row) => {
+            return { id: row.id, parentId: row.parentId };
+        }));
+    }
+    getChildElementsTreeFromMap({ elementIds }) {
+        const result = new Map();
+        if (Id64.sizeOf(elementIds) === 0 || this.#childElementsMap.size === 0) {
+            return result;
+        }
+        for (const elementId of Id64.iterable(elementIds)) {
+            const entry = this.#childElementsMap.get(elementId);
+            if (!entry?.children) {
+                continue;
+            }
+            const elementChildrenTree = new Map();
+            result.set(elementId, { children: elementChildrenTree });
+            for (const childId of entry.children) {
+                const childrenTreeOfChild = this.getChildElementsTreeFromMap({ elementIds: childId });
+                // Need to add children tree created from childId. This tree includes childId as root element
+                // If child does not have children, children tree won't be created. Need to add childId with undefined children
+                elementChildrenTree.set(childId, { children: childrenTreeOfChild.size > 0 ? childrenTreeOfChild : undefined });
+            }
+        }
+        return result;
+    }
+    getChildElementsCountMap({ elementIds }) {
+        const result = new Map();
+        for (const elementId of Id64.iterable(elementIds)) {
+            const entry = this.#childElementsMap.get(elementId);
+            if (entry?.children) {
+                let totalChildrenCount = entry.children.length;
+                for (const childrenOfChildCount of this.getChildElementsCountMap({ elementIds: entry.children }).values()) {
+                    totalChildrenCount += childrenOfChildCount;
+                }
+                result.set(elementId, totalChildrenCount);
+            }
+        }
+        return result;
+    }
+    /**
+     * Populates #childElementsLoadingMap with promises. When these promises resolve, they will populate #childElementsMap with values and delete entries from #childElementsLoadingMap.
+     */
+    createChildElementsLoadingMapEntries({ elementsToQuery }) {
+        const getElementsToQueryPromise = async (batchedElementsToQuery) => firstValueFrom(this.queryChildren({ elementIds: batchedElementsToQuery }).pipe(
+        // Want to have void at the end instead of void[], so using reduce
+        reduce((acc, { parentId, id }) => {
+            let entry = this.#childElementsMap.get(parentId);
+            if (!entry) {
+                entry = { children: [] };
+                this.#childElementsMap.set(parentId, entry);
+            }
+            if (!entry.children) {
+                entry.children = [];
+            }
+            // Add child to parent's entry and add child to children map
+            entry.children.push(id);
+            if (!this.#childElementsMap.has(id)) {
+                this.#childElementsMap.set(id, { children: undefined });
+            }
+            return acc;
+        }, (() => { })()), tap({ complete: () => batchedElementsToQuery.forEach((elementId) => this.#childElementsLoadingMap.delete(elementId)) }), defaultIfEmpty((() => { })())));
+        const maximumBatchSize = 1000;
+        const totalSize = elementsToQuery.length;
+        const optimalBatchSize = getOptimalBatchSize({ totalSize, maximumBatchSize });
+        const loadingMapEntries = new Array();
+        // Don't want to slice if its not necessary
+        if (totalSize <= maximumBatchSize) {
+            loadingMapEntries.push(getElementsToQueryPromise(elementsToQuery));
+        }
+        else {
+            for (let i = 0; i < elementsToQuery.length; i += optimalBatchSize) {
+                loadingMapEntries.push(getElementsToQueryPromise(elementsToQuery.slice(i, i + optimalBatchSize)));
+            }
+        }
+        elementsToQuery.forEach((elementId, index) => this.#childElementsLoadingMap.set(elementId, loadingMapEntries[Math.floor(index / optimalBatchSize)]));
+        return { loadingMapEntries: Promise.all(loadingMapEntries).then(() => { }) };
+    }
+    createChildElementsMapEntries({ elementIds }) {
+        return from(Id64.iterable(elementIds)).pipe(reduce((acc, elementId) => {
+            if (this.#childElementsMap.has(elementId)) {
+                return acc;
+            }
+            const loadingPromise = this.#childElementsLoadingMap.get(elementId);
+            if (loadingPromise) {
+                acc.existingPromises.push(loadingPromise);
+            }
+            else {
+                acc.elementsToQuery.push(elementId);
+            }
+            return acc;
+        }, { existingPromises: new Array(), elementsToQuery: new Array() }), mergeMap(async ({ elementsToQuery, existingPromises }) => {
+            existingPromises.push(this.createChildElementsLoadingMapEntries({ elementsToQuery }).loadingMapEntries);
+            return Promise.all(existingPromises);
+        }));
+    }
+    getChildElementsTree({ elementIds }) {
+        return this.createChildElementsMapEntries({ elementIds }).pipe(map(() => this.getChildElementsTreeFromMap({ elementIds })));
+    }
+    getAllChildElementsCount({ elementIds }) {
+        return this.createChildElementsMapEntries({ elementIds }).pipe(map(() => this.getChildElementsCountMap({ elementIds })));
+    }
+}
+//# sourceMappingURL=ElementChildrenCache.js.map

package/lib/esm/tree-widget-react/components/trees/common/internal/caches/ElementChildrenCache.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ElementChildrenCache.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/common/internal/caches/ElementChildrenCache.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,cAAc,EAAE,KAAK,EAAE,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,MAAM,CAAC;AACtG,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,EAAE,uBAAuB,EAAE,MAAM,qBAAqB,CAAC;AAC9D,OAAO,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC;AAgBlD,gBAAgB;AAChB,MAAM,OAAO,oBAAoB;IAC/B,cAAc,CAA6B;IAClC,iBAAiB,CAAS;IACnC,YAAY,CAAa;IACzB,cAAc,CAAS;IACvB,iBAAiB,CAAmB;IACpC,yEAAyE;IACzE,wBAAwB,CAA0B;IAElD,YAAY,KAAgC;QAC1C,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC,aAAa,CAAC;QAC1C,IAAI,CAAC,iBAAiB,GAAG,KAAK,CAAC,gBAAgB,CAAC;QAChD,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC,WAAW,CAAC;QACtC,IAAI,CAAC,cAAc,GAAG,sBAAsB,CAAC;QAC7C,IAAI,CAAC,iBAAiB,GAAG,IAAI,GAAG,EAAE,CAAC;QACnC,IAAI,CAAC,wBAAwB,GAAG,IAAI,GAAG,EAAE,CAAC;IAC5C,CAAC;IAEO,aAAa,CAAC,EAAE,UAAU,EAA6B;QAC7D,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,OAAO,KAAK,CAAC;QACf,CAAC;QAED,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,IAAI,GAAG;gBACX;;;mBAGW,IAAI,CAAC,iBAAiB;kCACP,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC;;;mBAGpC,IAAI,CAAC,iBAAiB;;;SAGhC;aACF,CAAC;YACF,MAAM,KAAK,GAAG;;;OAGb,CAAC;YACF,OAAO,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC1C,EAAE,KAAK,EAAE,IAAI,EAAE,EACf,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,aAAa,IAAI,CAAC,WAAW,EAAE,EAAE,EAAE,CACpJ,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,uBAAuB,EACvB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACV,OAAO,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,CAAC;QAChD,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,2BAA2B,CAAC,EAAE,UAAU,EAA2B;QACzE,MAAM,MAAM,GAAiB,IAAI,GAAG,EAAE,CAAC;QACvC,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YACvE,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;YAClD,MAAM,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YACpD,IAAI,CAAC,KAAK,EAAE,QAAQ,EAAE,CAAC;gBACrB,SAAS;YACX,CAAC;YACD,MAAM,mBAAmB,GAAiB,IAAI,GAAG,EAAE,CAAC;YACpD,MAAM,CAAC,GAAG,CAAC,SAAS,EAAE,EAAE,QAAQ,EAAE,mBAAmB,EAAE,CAAC,CAAC;YACzD,KAAK,MAAM,OAAO,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;gBACrC,MAAM,mBAAmB,GAAG,IAAI,CAAC,2BAA2B,CAAC,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC,CAAC;gBACtF,6FAA6F;gBAC7F,+GAA+G;gBAC/G,mBAAmB,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,mBAAmB,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC;YACjH,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,wBAAwB,CAAC,EAAE,UAAU,EAA2B;QACtE,MAAM,MAAM,GAAG,IAAI,GAAG,EAAsB,CAAC;QAC7C,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;YAClD,MAAM,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YACpD,IAAI,KAAK,EAAE,QAAQ,EAAE,CAAC;gBACpB,IAAI,kBAAkB,GAAG,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC;gBAC/C,KAAK,MAAM,oBAAoB,IAAI,IAAI,CAAC,wBAAwB,CAAC,EAAE,UAAU,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;oBAC1G,kBAAkB,IAAI,oBAAoB,CAAC;gBAC7C,CAAC;gBACD,MAAM,CAAC,GAAG,CAAC,SAAS,EAAE,kBAAkB,CAAC,CAAC;YAC5C,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACK,oCAAoC,CAAC,EAAE,eAAe,EAAkC;QAC9F,MAAM,yBAAyB,GAAG,KAAK,EAAE,sBAAiC,EAAE,EAAE,CAC5E,cAAc,CACZ,IAAI,CAAC,aAAa,CAAC,EAAE,UAAU,EAAE,sBAAsB,EAAE,CAAC,CAAC,IAAI;QAC7D,kEAAkE;QAClE,MAAM,CACJ,CAAC,GAAG,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,EAAE,EAAE;YACxB,IAAI,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACjD,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,KAAK,GAAG,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;gBACzB,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;YAC9C,CAAC;YACD,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;gBACpB,KAAK,CAAC,QAAQ,GAAG,EAAE,CAAC;YACtB,CAAC;YACD,4DAA4D;YAC5D,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACxB,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;gBACpC,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;YAC1D,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EACD,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,EAAE,CACb,EACD,GAAG,CAAC,EAAE,QAAQ,EAAE,GAAG,EAAE,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,IAAI,CAAC,wBAAwB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,EACvH,cAAc,CAAC,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,EAAE,CAAC,CAC7B,CACF,CAAC;QACJ,MAAM,gBAAgB,GAAG,IAAI,CAAC;QAC9B,MAAM,SAAS,GAAG,eAAe,CAAC,MAAM,CAAC;QACzC,MAAM,gBAAgB,GAAG,mBAAmB,CAAC,EAAE,SAAS,EAAE,gBAAgB,EAAE,CAAC,CAAC;QAC9E,MAAM,iBAAiB,GAAG,IAAI,KAAK,EAAiB,CAAC;QACrD,2CAA2C;QAC3C,IAAI,SAAS,IAAI,gBAAgB,EAAE,CAAC;YAClC,iBAAiB,CAAC,IAAI,CAAC,yBAAyB,CAAC,eAAe,CAAC,CAAC,CAAC;QACrE,CAAC;aAAM,CAAC;YACN,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,eAAe,CAAC,MAAM,EAAE,CAAC,IAAI,gBAAgB,EAAE,CAAC;gBAClE,iBAAiB,CAAC,IAAI,CAAC,yBAAyB,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,gBAAgB,CAAC,CAAC,CAAC,CAAC;YACpG,CAAC;QACH,CAAC;QAED,eAAe,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,SAAS,EAAE,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC;QACrJ,OAAO,EAAE,iBAAiB,EAAE,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,EAAE,CAAC;IAC9E,CAAC;IAEO,6BAA6B,CAAC,EAAE,UAAU,EAA2B;QAC3E,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CACzC,MAAM,CACJ,CAAC,GAAG,EAAE,SAAS,EAAE,EAAE;YACjB,IAAI,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC1C,OAAO,GAAG,CAAC;YACb,CAAC;YACD,MAAM,cAAc,GAAG,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YACpE,IAAI,cAAc,EAAE,CAAC;gBACnB,GAAG,CAAC,gBAAgB,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAC5C,CAAC;iBAAM,CAAC;gBACN,GAAG,CAAC,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACtC,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EACD,EAAE,gBAAgB,EAAE,IAAI,KAAK,EAAiB,EAAE,eAAe,EAAE,IAAI,KAAK,EAAc,EAAE,CAC3F,EACD,QAAQ,CAAC,KAAK,EAAE,EAAE,eAAe,EAAE,gBAAgB,EAAE,EAAE,EAAE;YACvD,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,oCAAoC,CAAC,EAAE,eAAe,EAAE,CAAC,CAAC,iBAAiB,CAAC,CAAC;YACxG,OAAO,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QACvC,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEM,oBAAoB,CAAC,EAAE,UAAU,EAA2B;QACjE,OAAO,IAAI,CAAC,6BAA6B,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,2BAA2B,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;IAC9H,CAAC;IAEM,wBAAwB,CAAC,EAAE,UAAU,EAA2B;QACrE,OAAO,IAAI,CAAC,6BAA6B,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,wBAAwB,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;IAC3H,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { defaultIfEmpty, defer, EMPTY, firstValueFrom, from, map, mergeMap, reduce, tap } from \"rxjs\";\nimport { Guid, Id64 } from \"@itwin/core-bentley\";\nimport { catchBeSQLiteInterrupts } from \"../UseErrorState.js\";\nimport { getOptimalBatchSize } from \"../Utils.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { GuidString, Id64Arg, Id64Array, Id64String } from \"@itwin/core-bentley\";\nimport type { LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\nimport type { ChildrenTree } from \"../Utils.js\";\n\ntype ChildElementsMap = Map<Id64String, { children: Id64Array | undefined }>;\ntype ChildElementsLoadingMap = Map<Id64String, Promise<void>>;\n\ninterface ElementChildrenCacheProps {\n  queryExecutor: LimitingECSqlQueryExecutor;\n  elementClassName: string;\n  componentId: GuidString;\n}\n\n/** @internal */\nexport class ElementChildrenCache {\n  #queryExecutor: LimitingECSqlQueryExecutor;\n  readonly #elementClassName: string;\n  #componentId: GuidString;\n  #componentName: string;\n  #childElementsMap: ChildElementsMap;\n  /** Stores element ids which have children query scheduled to execute. */\n  #childElementsLoadingMap: ChildElementsLoadingMap;\n\n  constructor(props: ElementChildrenCacheProps) {\n    this.#queryExecutor = props.queryExecutor;\n    this.#elementClassName = props.elementClassName;\n    this.#componentId = props.componentId;\n    this.#componentName = \"ElementChildrenCache\";\n    this.#childElementsMap = new Map();\n    this.#childElementsLoadingMap = new Map();\n  }\n\n  private queryChildren({ elementIds }: { elementIds: Id64Array }): Observable<{ id: Id64String; parentId: Id64String }> {\n    if (elementIds.length === 0) {\n      return EMPTY;\n    }\n\n    return defer(() => {\n      const ctes = [\n        `\n          ElementChildren(id, parentId) AS (\n            SELECT ECInstanceId id, Parent.Id parentId\n            FROM ${this.#elementClassName}\n            WHERE Parent.Id IN (${elementIds.join(\", \")})\n            UNION ALL\n            SELECT c.ECInstanceId id, c.Parent.Id\n            FROM ${this.#elementClassName} c\n            JOIN ElementChildren p ON c.Parent.Id = p.id\n          )\n        `,\n      ];\n      const ecsql = `\n        SELECT id, parentId\n        FROM ElementChildren\n      `;\n      return this.#queryExecutor.createQueryReader(\n        { ecsql, ctes },\n        { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: `${this.#componentName}/${this.#componentId}/children/${Guid.createValue()}` },\n      );\n    }).pipe(\n      catchBeSQLiteInterrupts,\n      map((row) => {\n        return { id: row.id, parentId: row.parentId };\n      }),\n    );\n  }\n\n  private getChildElementsTreeFromMap({ elementIds }: { elementIds: Id64Arg }): ChildrenTree {\n    const result: ChildrenTree = new Map();\n    if (Id64.sizeOf(elementIds) === 0 || this.#childElementsMap.size === 0) {\n      return result;\n    }\n    for (const elementId of Id64.iterable(elementIds)) {\n      const entry = this.#childElementsMap.get(elementId);\n      if (!entry?.children) {\n        continue;\n      }\n      const elementChildrenTree: ChildrenTree = new Map();\n      result.set(elementId, { children: elementChildrenTree });\n      for (const childId of entry.children) {\n        const childrenTreeOfChild = this.getChildElementsTreeFromMap({ elementIds: childId });\n        // Need to add children tree created from childId. This tree includes childId as root element\n        // If child does not have children, children tree won't be created. Need to add childId with undefined children\n        elementChildrenTree.set(childId, { children: childrenTreeOfChild.size > 0 ? childrenTreeOfChild : undefined });\n      }\n    }\n    return result;\n  }\n\n  private getChildElementsCountMap({ elementIds }: { elementIds: Id64Arg }): Map<Id64String, number> {\n    const result = new Map<Id64String, number>();\n    for (const elementId of Id64.iterable(elementIds)) {\n      const entry = this.#childElementsMap.get(elementId);\n      if (entry?.children) {\n        let totalChildrenCount = entry.children.length;\n        for (const childrenOfChildCount of this.getChildElementsCountMap({ elementIds: entry.children }).values()) {\n          totalChildrenCount += childrenOfChildCount;\n        }\n        result.set(elementId, totalChildrenCount);\n      }\n    }\n    return result;\n  }\n\n  /**\n   * Populates #childElementsLoadingMap with promises. When these promises resolve, they will populate #childElementsMap with values and delete entries from #childElementsLoadingMap.\n   */\n  private createChildElementsLoadingMapEntries({ elementsToQuery }: { elementsToQuery: Id64Array }): { loadingMapEntries: Promise<void> } {\n    const getElementsToQueryPromise = async (batchedElementsToQuery: Id64Array) =>\n      firstValueFrom(\n        this.queryChildren({ elementIds: batchedElementsToQuery }).pipe(\n          // Want to have void at the end instead of void[], so using reduce\n          reduce(\n            (acc, { parentId, id }) => {\n              let entry = this.#childElementsMap.get(parentId);\n              if (!entry) {\n                entry = { children: [] };\n                this.#childElementsMap.set(parentId, entry);\n              }\n              if (!entry.children) {\n                entry.children = [];\n              }\n              // Add child to parent's entry and add child to children map\n              entry.children.push(id);\n              if (!this.#childElementsMap.has(id)) {\n                this.#childElementsMap.set(id, { children: undefined });\n              }\n              return acc;\n            },\n            (() => {})(),\n          ),\n          tap({ complete: () => batchedElementsToQuery.forEach((elementId) => this.#childElementsLoadingMap.delete(elementId)) }),\n          defaultIfEmpty((() => {})()),\n        ),\n      );\n    const maximumBatchSize = 1000;\n    const totalSize = elementsToQuery.length;\n    const optimalBatchSize = getOptimalBatchSize({ totalSize, maximumBatchSize });\n    const loadingMapEntries = new Array<Promise<void>>();\n    // Don't want to slice if its not necessary\n    if (totalSize <= maximumBatchSize) {\n      loadingMapEntries.push(getElementsToQueryPromise(elementsToQuery));\n    } else {\n      for (let i = 0; i < elementsToQuery.length; i += optimalBatchSize) {\n        loadingMapEntries.push(getElementsToQueryPromise(elementsToQuery.slice(i, i + optimalBatchSize)));\n      }\n    }\n\n    elementsToQuery.forEach((elementId, index) => this.#childElementsLoadingMap.set(elementId, loadingMapEntries[Math.floor(index / optimalBatchSize)]));\n    return { loadingMapEntries: Promise.all(loadingMapEntries).then(() => {}) };\n  }\n\n  private createChildElementsMapEntries({ elementIds }: { elementIds: Id64Arg }): Observable<void[]> {\n    return from(Id64.iterable(elementIds)).pipe(\n      reduce(\n        (acc, elementId) => {\n          if (this.#childElementsMap.has(elementId)) {\n            return acc;\n          }\n          const loadingPromise = this.#childElementsLoadingMap.get(elementId);\n          if (loadingPromise) {\n            acc.existingPromises.push(loadingPromise);\n          } else {\n            acc.elementsToQuery.push(elementId);\n          }\n          return acc;\n        },\n        { existingPromises: new Array<Promise<void>>(), elementsToQuery: new Array<Id64String>() },\n      ),\n      mergeMap(async ({ elementsToQuery, existingPromises }) => {\n        existingPromises.push(this.createChildElementsLoadingMapEntries({ elementsToQuery }).loadingMapEntries);\n        return Promise.all(existingPromises);\n      }),\n    );\n  }\n\n  public getChildElementsTree({ elementIds }: { elementIds: Id64Arg }): Observable<ChildrenTree> {\n    return this.createChildElementsMapEntries({ elementIds }).pipe(map(() => this.getChildElementsTreeFromMap({ elementIds })));\n  }\n\n  public getAllChildElementsCount({ elementIds }: { elementIds: Id64Arg }): Observable<Map<Id64String, number>> {\n    return this.createChildElementsMapEntries({ elementIds }).pipe(map(() => this.getChildElementsCountMap({ elementIds })));\n  }\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/common/internal/caches/ElementModelCategoriesCache.d.ts

@@ -0,0 +1,32 @@
+import type { Observable } from "rxjs";
+import type { GuidString, Id64Array, Id64Set, Id64String } from "@itwin/core-bentley";
+import type { LimitingECSqlQueryExecutor } from "@itwin/presentation-hierarchies";
+import type { ModelId } from "../Types.js";
+import type { ModeledElementsCache } from "./ModeledElementsCache.js";
+interface ElementModelCategoriesCacheProps {
+    queryExecutor: LimitingECSqlQueryExecutor;
+    componentId: GuidString;
+    elementClassName: string;
+    modeledElementsCache: ModeledElementsCache;
+}
+/** @internal */
+export declare class ElementModelCategoriesCache {
+    #private;
+    constructor(props: ElementModelCategoriesCacheProps);
+    private queryElementModelCategories;
+    private getModelsCategoriesInfo;
+    getCategoryElementModels(props: {
+        categoryId: Id64String;
+        includeSubModels?: boolean;
+        includeOnlyIfCategoryOfTopMostElement?: boolean;
+    }): Observable<Array<ModelId>>;
+    getModelCategoryIds({ modelId, includeOnlyIfCategoryOfTopMostElement, }: {
+        modelId: Id64String;
+        includeOnlyIfCategoryOfTopMostElement?: boolean;
+    }): Observable<Id64Set>;
+    getCategoriesOfModelsTopMostElements(modelIds: Id64Array): Observable<Id64Set>;
+    getAllCategoriesOfElements(): Observable<Id64Set>;
+    getAllModels(): Observable<Array<ModelId>>;
+}
+export {};
+//# sourceMappingURL=ElementModelCategoriesCache.d.ts.map

package/lib/esm/tree-widget-react/components/trees/common/internal/caches/ElementModelCategoriesCache.js

@@ -0,0 +1,106 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { defer, forkJoin, from, map, mergeMap, reduce, shareReplay } from "rxjs";
+import { CLASS_NAME_Model } from "../ClassNameDefinitions.js";
+import { catchBeSQLiteInterrupts } from "../UseErrorState.js";
+/** @internal */
+export class ElementModelCategoriesCache {
+    #queryExecutor;
+    #componentId;
+    #componentName;
+    #elementClassName;
+    #modeledElementsCache;
+    #modelsCategoriesInfo;
+    constructor(props) {
+        this.#queryExecutor = props.queryExecutor;
+        this.#elementClassName = props.elementClassName;
+        this.#modeledElementsCache = props.modeledElementsCache;
+        this.#componentId = props.componentId;
+        this.#componentName = "ElementModelCategoriesCache";
+    }
+    queryElementModelCategories() {
+        return defer(() => {
+            const query = `
+          SELECT * FROM (
+            SELECT
+              this.Model.Id modelId,
+              this.Category.Id categoryId,
+              MAX(IIF(this.Parent.Id IS NULL, 1, 0)) isTopMostElementCategory
+            FROM ${CLASS_NAME_Model} m
+            JOIN ${this.#elementClassName} this ON m.ECInstanceId = this.Model.Id
+            WHERE m.IsPrivate = false
+            GROUP BY modelId, categoryId
+          )
+        `;
+            return this.#queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded", restartToken: `${this.#componentName}/${this.#componentId}/element-models-and-categories` });
+        }).pipe(catchBeSQLiteInterrupts, map((row) => {
+            return { modelId: row.modelId, categoryId: row.categoryId, isTopMostElementCategory: !!row.isTopMostElementCategory };
+        }));
+    }
+    getModelsCategoriesInfo() {
+        this.#modelsCategoriesInfo ??= forkJoin({
+            modelCategories: this.queryElementModelCategories().pipe(reduce((acc, queriedCategory) => {
+                let modelEntry = acc.get(queriedCategory.modelId);
+                if (modelEntry === undefined) {
+                    modelEntry = { categoriesOfTopMostElements: new Set(), allCategories: new Set() };
+                    acc.set(queriedCategory.modelId, modelEntry);
+                }
+                modelEntry.allCategories.add(queriedCategory.categoryId);
+                if (queriedCategory.isTopMostElementCategory) {
+                    modelEntry.categoriesOfTopMostElements.add(queriedCategory.categoryId);
+                }
+                return acc;
+            }, new Map())),
+            allSubModels: this.#modeledElementsCache.getModeledElementsInfo().pipe(map(({ allSubModels }) => allSubModels)),
+        }).pipe(map(({ modelCategories, allSubModels }) => {
+            const result = new Map();
+            for (const [modelId, { categoriesOfTopMostElements, allCategories }] of modelCategories) {
+                const isSubModel = allSubModels.has(modelId);
+                result.set(modelId, {
+                    categoriesOfTopMostElements,
+                    allCategories,
+                    isSubModel,
+                });
+            }
+            return result;
+        }), shareReplay());
+        return this.#modelsCategoriesInfo;
+    }
+    getCategoryElementModels(props) {
+        const { categoryId, includeSubModels } = props;
+        return this.getModelsCategoriesInfo().pipe(map((modelCategories) => {
+            const categoryModels = new Array();
+            for (const [modelId, { allCategories, categoriesOfTopMostElements, isSubModel }] of modelCategories) {
+                if ((includeSubModels || !isSubModel) &&
+                    (props.includeOnlyIfCategoryOfTopMostElement ? categoriesOfTopMostElements.has(categoryId) : allCategories.has(categoryId))) {
+                    categoryModels.push(modelId);
+                }
+            }
+            return categoryModels;
+        }));
+    }
+    getModelCategoryIds({ modelId, includeOnlyIfCategoryOfTopMostElement, }) {
+        return this.getModelsCategoriesInfo().pipe(map((modelCategories) => (includeOnlyIfCategoryOfTopMostElement ? modelCategories.get(modelId)?.categoriesOfTopMostElements : modelCategories.get(modelId)?.allCategories) ??
+            new Set()));
+    }
+    getCategoriesOfModelsTopMostElements(modelIds) {
+        return this.getModelsCategoriesInfo().pipe(mergeMap((modelCategories) => from(modelIds).pipe(mergeMap((modelId) => modelCategories.get(modelId)?.categoriesOfTopMostElements ?? []))), reduce((acc, categoryId) => {
+            acc.add(categoryId);
+            return acc;
+        }, new Set()));
+    }
+    getAllCategoriesOfElements() {
+        return this.getModelsCategoriesInfo().pipe(mergeMap((modelCategories) => modelCategories.values()), reduce((acc, { allCategories }) => {
+            for (const categoryId of allCategories) {
+                acc.add(categoryId);
+            }
+            return acc;
+        }, new Set()));
+    }
+    getAllModels() {
+        return this.getModelsCategoriesInfo().pipe(map((modelCategories) => [...modelCategories.keys()]));
+    }
+}
+//# sourceMappingURL=ElementModelCategoriesCache.js.map

package/lib/esm/tree-widget-react/components/trees/common/internal/caches/ElementModelCategoriesCache.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ElementModelCategoriesCache.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/common/internal/caches/ElementModelCategoriesCache.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,MAAM,CAAC;AACjF,OAAO,EAAE,gBAAgB,EAAE,MAAM,4BAA4B,CAAC;AAC9D,OAAO,EAAE,uBAAuB,EAAE,MAAM,qBAAqB,CAAC;AAe9D,gBAAgB;AAChB,MAAM,OAAO,2BAA2B;IACtC,cAAc,CAA6B;IAC3C,YAAY,CAAa;IACzB,cAAc,CAAS;IACvB,iBAAiB,CAAS;IAC1B,qBAAqB,CAAuB;IAC5C,qBAAqB,CAEP;IAEd,YAAY,KAAuC;QACjD,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC,aAAa,CAAC;QAC1C,IAAI,CAAC,iBAAiB,GAAG,KAAK,CAAC,gBAAgB,CAAC;QAChD,IAAI,CAAC,qBAAqB,GAAG,KAAK,CAAC,oBAAoB,CAAC;QACxD,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC,WAAW,CAAC;QACtC,IAAI,CAAC,cAAc,GAAG,6BAA6B,CAAC;IACtD,CAAC;IAEO,2BAA2B;QAKjC,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,KAAK,GAAG;;;;;;mBAMD,gBAAgB;mBAChB,IAAI,CAAC,iBAAiB;;;;SAIhC,CAAC;YACJ,OAAO,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC1C,EAAE,KAAK,EAAE,KAAK,EAAE,EAChB,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,gCAAgC,EAAE,CACnJ,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,uBAAuB,EACvB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACV,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,wBAAwB,EAAE,CAAC,CAAC,GAAG,CAAC,wBAAwB,EAAE,CAAC;QACxH,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,uBAAuB;QAC7B,IAAI,CAAC,qBAAqB,KAAK,QAAQ,CAAC;YACtC,eAAe,EAAE,IAAI,CAAC,2BAA2B,EAAE,CAAC,IAAI,CACtD,MAAM,CAAC,CAAC,GAAG,EAAE,eAAe,EAAE,EAAE;gBAC9B,IAAI,UAAU,GAAG,GAAG,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;gBAClD,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;oBAC7B,UAAU,GAAG,EAAE,2BAA2B,EAAE,IAAI,GAAG,EAAE,EAAE,aAAa,EAAE,IAAI,GAAG,EAAE,EAAE,CAAC;oBAClF,GAAG,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;gBAC/C,CAAC;gBACD,UAAU,CAAC,aAAa,CAAC,GAAG,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;gBACzD,IAAI,eAAe,CAAC,wBAAwB,EAAE,CAAC;oBAC7C,UAAU,CAAC,2BAA2B,CAAC,GAAG,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;gBACzE,CAAC;gBACD,OAAO,GAAG,CAAC;YACb,CAAC,EAAE,IAAI,GAAG,EAA6F,CAAC,CACzG;YACD,YAAY,EAAE,IAAI,CAAC,qBAAqB,CAAC,sBAAsB,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,YAAY,CAAC,CAAC;SAChH,CAAC,CAAC,IAAI,CACL,GAAG,CAAC,CAAC,EAAE,eAAe,EAAE,YAAY,EAAE,EAAE,EAAE;YACxC,MAAM,MAAM,GAAG,IAAI,GAAG,EAAkH,CAAC;YACzI,KAAK,MAAM,CAAC,OAAO,EAAE,EAAE,2BAA2B,EAAE,aAAa,EAAE,CAAC,IAAI,eAAe,EAAE,CAAC;gBACxF,MAAM,UAAU,GAAG,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBAC7C,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE;oBAClB,2BAA2B;oBAC3B,aAAa;oBACb,UAAU;iBACX,CAAC,CAAC;YACL,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,EACF,WAAW,EAAE,CACd,CAAC;QACF,OAAO,IAAI,CAAC,qBAAqB,CAAC;IACpC,CAAC;IAEM,wBAAwB,CAAC,KAI/B;QACC,MAAM,EAAE,UAAU,EAAE,gBAAgB,EAAE,GAAG,KAAK,CAAC;QAC/C,OAAO,IAAI,CAAC,uBAAuB,EAAE,CAAC,IAAI,CACxC,GAAG,CAAC,CAAC,eAAe,EAAE,EAAE;YACtB,MAAM,cAAc,GAAG,IAAI,KAAK,EAAW,CAAC;YAC5C,KAAK,MAAM,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,2BAA2B,EAAE,UAAU,EAAE,CAAC,IAAI,eAAe,EAAE,CAAC;gBACpG,IACE,CAAC,gBAAgB,IAAI,CAAC,UAAU,CAAC;oBACjC,CAAC,KAAK,CAAC,qCAAqC,CAAC,CAAC,CAAC,2BAA2B,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,EAC3H,CAAC;oBACD,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC/B,CAAC;YACH,CAAC;YACD,OAAO,cAAc,CAAC;QACxB,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEM,mBAAmB,CAAC,EACzB,OAAO,EACP,qCAAqC,GAItC;QACC,OAAO,IAAI,CAAC,uBAAuB,EAAE,CAAC,IAAI,CACxC,GAAG,CACD,CAAC,eAAe,EAAE,EAAE,CAClB,CAAC,qCAAqC,CAAC,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,2BAA2B,CAAC,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,aAAa,CAAC;YACjJ,IAAI,GAAG,EAAE,CACZ,CACF,CAAC;IACJ,CAAC;IAEM,oCAAoC,CAAC,QAAmB;QAC7D,OAAO,IAAI,CAAC,uBAAuB,EAAE,CAAC,IAAI,CACxC,QAAQ,CAAC,CAAC,eAAe,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,2BAA2B,IAAI,EAAE,CAAC,CAAC,CAAC,EAC1I,MAAM,CAAC,CAAC,GAAG,EAAE,UAAU,EAAE,EAAE;YACzB,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YACpB,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,IAAI,GAAG,EAAc,CAAC,CAC1B,CAAC;IACJ,CAAC;IAEM,0BAA0B;QAC/B,OAAO,IAAI,CAAC,uBAAuB,EAAE,CAAC,IAAI,CACxC,QAAQ,CAAC,CAAC,eAAe,EAAE,EAAE,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC,EACvD,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE;YAChC,KAAK,MAAM,UAAU,IAAI,aAAa,EAAE,CAAC;gBACvC,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YACtB,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,IAAI,GAAG,EAAc,CAAC,CAC1B,CAAC;IACJ,CAAC;IAEM,YAAY;QACjB,OAAO,IAAI,CAAC,uBAAuB,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC,GAAG,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IACpG,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { defer, forkJoin, from, map, mergeMap, reduce, shareReplay } from \"rxjs\";\nimport { CLASS_NAME_Model } from \"../ClassNameDefinitions.js\";\nimport { catchBeSQLiteInterrupts } from \"../UseErrorState.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { GuidString, Id64Array, Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport type { LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\nimport type { CategoryId, ModelId } from \"../Types.js\";\nimport type { ModeledElementsCache } from \"./ModeledElementsCache.js\";\n\ninterface ElementModelCategoriesCacheProps {\n  queryExecutor: LimitingECSqlQueryExecutor;\n  componentId: GuidString;\n  elementClassName: string;\n  modeledElementsCache: ModeledElementsCache;\n}\n\n/** @internal */\nexport class ElementModelCategoriesCache {\n  #queryExecutor: LimitingECSqlQueryExecutor;\n  #componentId: GuidString;\n  #componentName: string;\n  #elementClassName: string;\n  #modeledElementsCache: ModeledElementsCache;\n  #modelsCategoriesInfo:\n    | Observable<Map<ModelId, { categoriesOfTopMostElements: Set<CategoryId>; allCategories: Set<CategoryId>; isSubModel: boolean }>>\n    | undefined;\n\n  constructor(props: ElementModelCategoriesCacheProps) {\n    this.#queryExecutor = props.queryExecutor;\n    this.#elementClassName = props.elementClassName;\n    this.#modeledElementsCache = props.modeledElementsCache;\n    this.#componentId = props.componentId;\n    this.#componentName = \"ElementModelCategoriesCache\";\n  }\n\n  private queryElementModelCategories(): Observable<{\n    modelId: Id64String;\n    categoryId: Id64String;\n    isTopMostElementCategory: boolean;\n  }> {\n    return defer(() => {\n      const query = `\n          SELECT * FROM (\n            SELECT\n              this.Model.Id modelId,\n              this.Category.Id categoryId,\n              MAX(IIF(this.Parent.Id IS NULL, 1, 0)) isTopMostElementCategory\n            FROM ${CLASS_NAME_Model} m\n            JOIN ${this.#elementClassName} this ON m.ECInstanceId = this.Model.Id\n            WHERE m.IsPrivate = false\n            GROUP BY modelId, categoryId\n          )\n        `;\n      return this.#queryExecutor.createQueryReader(\n        { ecsql: query },\n        { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: `${this.#componentName}/${this.#componentId}/element-models-and-categories` },\n      );\n    }).pipe(\n      catchBeSQLiteInterrupts,\n      map((row) => {\n        return { modelId: row.modelId, categoryId: row.categoryId, isTopMostElementCategory: !!row.isTopMostElementCategory };\n      }),\n    );\n  }\n\n  private getModelsCategoriesInfo() {\n    this.#modelsCategoriesInfo ??= forkJoin({\n      modelCategories: this.queryElementModelCategories().pipe(\n        reduce((acc, queriedCategory) => {\n          let modelEntry = acc.get(queriedCategory.modelId);\n          if (modelEntry === undefined) {\n            modelEntry = { categoriesOfTopMostElements: new Set(), allCategories: new Set() };\n            acc.set(queriedCategory.modelId, modelEntry);\n          }\n          modelEntry.allCategories.add(queriedCategory.categoryId);\n          if (queriedCategory.isTopMostElementCategory) {\n            modelEntry.categoriesOfTopMostElements.add(queriedCategory.categoryId);\n          }\n          return acc;\n        }, new Map<ModelId, { categoriesOfTopMostElements: Set<CategoryId>; allCategories: Set<CategoryId> }>()),\n      ),\n      allSubModels: this.#modeledElementsCache.getModeledElementsInfo().pipe(map(({ allSubModels }) => allSubModels)),\n    }).pipe(\n      map(({ modelCategories, allSubModels }) => {\n        const result = new Map<ModelId, { categoriesOfTopMostElements: Set<CategoryId>; allCategories: Set<CategoryId>; isSubModel: boolean }>();\n        for (const [modelId, { categoriesOfTopMostElements, allCategories }] of modelCategories) {\n          const isSubModel = allSubModels.has(modelId);\n          result.set(modelId, {\n            categoriesOfTopMostElements,\n            allCategories,\n            isSubModel,\n          });\n        }\n        return result;\n      }),\n      shareReplay(),\n    );\n    return this.#modelsCategoriesInfo;\n  }\n\n  public getCategoryElementModels(props: {\n    categoryId: Id64String;\n    includeSubModels?: boolean;\n    includeOnlyIfCategoryOfTopMostElement?: boolean;\n  }): Observable<Array<ModelId>> {\n    const { categoryId, includeSubModels } = props;\n    return this.getModelsCategoriesInfo().pipe(\n      map((modelCategories) => {\n        const categoryModels = new Array<ModelId>();\n        for (const [modelId, { allCategories, categoriesOfTopMostElements, isSubModel }] of modelCategories) {\n          if (\n            (includeSubModels || !isSubModel) &&\n            (props.includeOnlyIfCategoryOfTopMostElement ? categoriesOfTopMostElements.has(categoryId) : allCategories.has(categoryId))\n          ) {\n            categoryModels.push(modelId);\n          }\n        }\n        return categoryModels;\n      }),\n    );\n  }\n\n  public getModelCategoryIds({\n    modelId,\n    includeOnlyIfCategoryOfTopMostElement,\n  }: {\n    modelId: Id64String;\n    includeOnlyIfCategoryOfTopMostElement?: boolean;\n  }): Observable<Id64Set> {\n    return this.getModelsCategoriesInfo().pipe(\n      map(\n        (modelCategories) =>\n          (includeOnlyIfCategoryOfTopMostElement ? modelCategories.get(modelId)?.categoriesOfTopMostElements : modelCategories.get(modelId)?.allCategories) ??\n          new Set(),\n      ),\n    );\n  }\n\n  public getCategoriesOfModelsTopMostElements(modelIds: Id64Array): Observable<Id64Set> {\n    return this.getModelsCategoriesInfo().pipe(\n      mergeMap((modelCategories) => from(modelIds).pipe(mergeMap((modelId) => modelCategories.get(modelId)?.categoriesOfTopMostElements ?? []))),\n      reduce((acc, categoryId) => {\n        acc.add(categoryId);\n        return acc;\n      }, new Set<Id64String>()),\n    );\n  }\n\n  public getAllCategoriesOfElements(): Observable<Id64Set> {\n    return this.getModelsCategoriesInfo().pipe(\n      mergeMap((modelCategories) => modelCategories.values()),\n      reduce((acc, { allCategories }) => {\n        for (const categoryId of allCategories) {\n          acc.add(categoryId);\n        }\n        return acc;\n      }, new Set<CategoryId>()),\n    );\n  }\n\n  public getAllModels(): Observable<Array<ModelId>> {\n    return this.getModelsCategoriesInfo().pipe(map((modelCategories) => [...modelCategories.keys()]));\n  }\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/common/internal/{ModelCategoryElementsCountCache.d.ts → caches/ModelCategoryElementsCountCache.d.ts}

@@ -4,9 +4,16 @@ import type { LimitingECSqlQueryExecutor } from "@itwin/presentation-hierarchies
 /** @internal */
 export declare class ModelCategoryElementsCountCache implements Disposable {
     #private;
-    constructor(queryExecutor: LimitingECSqlQueryExecutor, elementsClassNames: string[], componentId: GuidString);
+    constructor(props: {
+        queryExecutor: LimitingECSqlQueryExecutor;
+        elementClassName: string;
+        componentId: GuidString;
+    });
     private queryCategoryElementCounts;
     [Symbol.dispose](): void;
-    getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Observable<number>;
+    getCategoryElementsCount({ modelId, categoryId }: {
+        modelId: Id64String;
+        categoryId: Id64String;
+    }): Observable<number>;
 }
 //# sourceMappingURL=ModelCategoryElementsCountCache.d.ts.map

package/lib/esm/tree-widget-react/components/trees/common/internal/{ModelCategoryElementsCountCache.js → caches/ModelCategoryElementsCountCache.js}

@@ -4,21 +4,21 @@
  *--------------------------------------------------------------------------------------------*/
 import { bufferCount, bufferTime, defer, filter, from, map, mergeAll, mergeMap, reduce, ReplaySubject, Subject, take, toArray } from "rxjs";
 import { assert, Guid } from "@itwin/core-bentley";
-import { catchBeSQLiteInterrupts } from "./UseErrorState.js";
-import { releaseMainThreadOnItemsCount } from "./Utils.js";
+import { catchBeSQLiteInterrupts } from "../UseErrorState.js";
+import { releaseMainThreadOnItemsCount } from "../Utils.js";
 /** @internal */
 export class ModelCategoryElementsCountCache {
     #cache = new Map();
     #requestsStream = new Subject();
     #subscription;
     #queryExecutor;
-    #elementsClassNames;
+    #elementClassName;
     #componentId;
     #componentName;
-    constructor(queryExecutor, elementsClassNames, componentId) {
-        this.#componentId = componentId;
-        this.#queryExecutor = queryExecutor;
-        this.#elementsClassNames = elementsClassNames;
+    constructor(props) {
+        this.#componentId = props.componentId;
+        this.#queryExecutor = props.queryExecutor;
+        this.#elementClassName = props.elementClassName;
         this.#componentName = "ModelCategoryElementsCountCache";
         this.#subscription = this.#requestsStream
             .pipe(bufferTime(20), filter((requests) => requests.length > 0), mergeMap((requests) => this.queryCategoryElementCounts(requests)), mergeAll())
@@ -44,10 +44,11 @@ export class ModelCategoryElementsCountCache {
         // we may have thousands of where clauses here, and sending a single query with all of them could take a
         // long time - instead, split it into smaller chunks
         bufferCount(100), mergeMap((whereClauses) => defer(() => this.#queryExecutor.createQueryReader({
-            ctes: this.#elementsClassNames.map((elementsClassName, index) => `
-                  CategoryElements${index}(id, modelId, categoryId) AS (
+            ctes: [
+                `
+                  CategoryElements(id, modelId, categoryId) AS (
                     SELECT ECInstanceId, Model.Id, Category.Id
-                    FROM ${elementsClassName}
+                    FROM ${this.#elementClassName}
                     WHERE
                       Parent.Id IS NULL
                       AND (
@@ -57,19 +58,14 @@ export class ModelCategoryElementsCountCache {
                     UNION ALL
 
                     SELECT c.ECInstanceId, p.modelId, p.categoryId
-                    FROM ${elementsClassName} c
-                    JOIN CategoryElements${index} p ON c.Parent.Id = p.id
+                    FROM ${this.#elementClassName} c
+                    JOIN CategoryElements p ON c.Parent.Id = p.id
                   )
-                `),
+                `,
+            ],
             ecsql: `
                 SELECT modelId, categoryId, COUNT(id) elementsCount
-                FROM (
-                  ${this.#elementsClassNames
-                .map((_, index) => `
-                      SELECT * FROM CategoryElements${index}
-                    `)
-                .join(" UNION ALL ")}
-                )
+                FROM (SELECT * FROM CategoryElements)
                 GROUP BY modelId, categoryId
               `,
         }, {
@@ -98,7 +94,7 @@ export class ModelCategoryElementsCountCache {
     [Symbol.dispose]() {
         this.#subscription.unsubscribe();
     }
-    getCategoryElementsCount(modelId, categoryId) {
+    getCategoryElementsCount({ modelId, categoryId }) {
         const cacheKey = `${modelId}-${categoryId}`;
         let result = this.#cache.get(cacheKey);
         if (result !== undefined) {

package/lib/esm/tree-widget-react/components/trees/common/internal/caches/ModelCategoryElementsCountCache.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ModelCategoryElementsCountCache.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/common/internal/caches/ModelCategoryElementsCountCache.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,aAAa,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AAC5I,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,uBAAuB,EAAE,MAAM,qBAAqB,CAAC;AAC9D,OAAO,EAAE,6BAA6B,EAAE,MAAM,aAAa,CAAC;AAS5D,gBAAgB;AAChB,MAAM,OAAO,+BAA+B;IAC1C,MAAM,GAAG,IAAI,GAAG,EAAqC,CAAC;IACtD,eAAe,GAAG,IAAI,OAAO,EAAmD,CAAC;IACjF,aAAa,CAAe;IAC5B,cAAc,CAA6B;IAC3C,iBAAiB,CAAS;IAC1B,YAAY,CAAa;IACzB,cAAc,CAAS;IAEvB,YAAmB,KAAuG;QACxH,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC,WAAW,CAAC;QACtC,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC,aAAa,CAAC;QAC1C,IAAI,CAAC,iBAAiB,GAAG,KAAK,CAAC,gBAAgB,CAAC;QAChD,IAAI,CAAC,cAAc,GAAG,iCAAiC,CAAC;QACxD,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,eAAe;aACtC,IAAI,CACH,UAAU,CAAC,EAAE,CAAC,EACd,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,EACzC,QAAQ,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,IAAI,CAAC,0BAA0B,CAAC,QAAQ,CAAC,CAAC,EACjE,QAAQ,EAAE,CACX;aACA,SAAS,CAAC;YACT,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,aAAa,EAAE,EAAE,EAAE;gBAC/C,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC,CAAC;gBAC5D,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;gBAClB,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC9B,CAAC;SACF,CAAC,CAAC;IACP,CAAC;IAEO,0BAA0B,CAChC,KAA6D;QAE7D,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CACrB,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,EAAE,EAAE;YACtC,MAAM,KAAK,GAAG,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC/B,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;YAC1C,CAAC;iBAAM,CAAC;gBACN,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YACxB,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,IAAI,GAAG,EAAuB,CAAC,EAClC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,EAAE,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC,EAC1D,GAAG,CAAC,CAAC,CAAC,OAAO,EAAE,WAAW,CAAC,EAAE,EAAE,CAAC,cAAc,OAAO,wBAAwB,CAAC,GAAG,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;QAC5G,wGAAwG;QACxG,oDAAoD;QACpD,WAAW,CAAC,GAAG,CAAC,EAChB,QAAQ,CAAC,CAAC,YAAY,EAAE,EAAE,CACxB,KAAK,CAAC,GAAG,EAAE,CACT,IAAI,CAAC,cAAc,CAAC,iBAAiB,CACnC;YACE,IAAI,EAAE;gBACJ;;;2BAGW,IAAI,CAAC,iBAAiB;;;;0BAIvB,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC;;;;;;2BAMxB,IAAI,CAAC,iBAAiB;;;iBAGhC;aACF;YACD,KAAK,EAAE;;;;eAIN;SACF,EACD;YACE,SAAS,EAAE,oBAAoB;YAC/B,KAAK,EAAE,WAAW;YAClB,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,4BAA4B,IAAI,CAAC,WAAW,EAAE,EAAE;SAC1G,CACF,CACF,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAChC,EACD,6BAA6B,CAAC,GAAG,CAAC,EAClC,MAAM,CACJ,CAAC,EAAE,GAAG,EAAE,SAAS,EAAE,EAAE,GAAG,EAAE,EAAE;YAC1B,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC,EAAE;gBACvE,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,UAAU,EAAE,GAAG,CAAC,UAAU;gBAC1B,aAAa,EAAE,GAAG,CAAC,aAAa;aACjC,CAAC,CAAC;YACH,OAAO,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC;QAC5B,CAAC,EACD;YACE,GAAG,EAAE,IAAI,GAAG,EAAkF;YAC9F,SAAS,EAAE,CAAC,QAAyD,EAAE,EAAE,CAAC,GAAG,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,UAAU,EAAE;SACvH,CACF,EACD,QAAQ,CAAC,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE;YACtC,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,EAAE,EAAE;gBACxC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC,EAAE,CAAC;oBACpD,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,EAAE,EAAE,UAAU,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC,CAAC;gBAC5F,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;QAC/B,CAAC,CAAC,EACF,OAAO,EAAE,CACV,CAAC;IACJ,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,CAAC;IACnC,CAAC;IAEM,wBAAwB,CAAC,EAAE,OAAO,EAAE,UAAU,EAAmD;QACtG,MAAM,QAAQ,GAAqB,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC;QAC9D,IAAI,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACvC,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACzB,OAAO,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QACpC,CAAC;QAED,MAAM,GAAG,IAAI,aAAa,CAAC,CAAC,CAAC,CAAC;QAC9B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAClC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACnD,OAAO,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IACpC,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { bufferCount, bufferTime, defer, filter, from, map, mergeAll, mergeMap, reduce, ReplaySubject, Subject, take, toArray } from \"rxjs\";\nimport { assert, Guid } from \"@itwin/core-bentley\";\nimport { catchBeSQLiteInterrupts } from \"../UseErrorState.js\";\nimport { releaseMainThreadOnItemsCount } from \"../Utils.js\";\n\nimport type { Observable, Subscription } from \"rxjs\";\nimport type { GuidString, Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport type { LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\nimport type { CategoryId, ModelId } from \"../Types.js\";\n\ntype ModelCategoryKey = `${ModelId}-${CategoryId}`;\n\n/** @internal */\nexport class ModelCategoryElementsCountCache implements Disposable {\n  #cache = new Map<ModelCategoryKey, Subject<number>>();\n  #requestsStream = new Subject<{ modelId: Id64String; categoryId: Id64String }>();\n  #subscription: Subscription;\n  #queryExecutor: LimitingECSqlQueryExecutor;\n  #elementClassName: string;\n  #componentId: GuidString;\n  #componentName: string;\n\n  public constructor(props: { queryExecutor: LimitingECSqlQueryExecutor; elementClassName: string; componentId: GuidString }) {\n    this.#componentId = props.componentId;\n    this.#queryExecutor = props.queryExecutor;\n    this.#elementClassName = props.elementClassName;\n    this.#componentName = \"ModelCategoryElementsCountCache\";\n    this.#subscription = this.#requestsStream\n      .pipe(\n        bufferTime(20),\n        filter((requests) => requests.length > 0),\n        mergeMap((requests) => this.queryCategoryElementCounts(requests)),\n        mergeAll(),\n      )\n      .subscribe({\n        next: ({ modelId, categoryId, elementsCount }) => {\n          const subject = this.#cache.get(`${modelId}-${categoryId}`);\n          assert(!!subject);\n          subject.next(elementsCount);\n        },\n      });\n  }\n\n  private queryCategoryElementCounts(\n    input: Array<{ modelId: Id64String; categoryId: Id64String }>,\n  ): Observable<Array<{ modelId: Id64String; categoryId: Id64String; elementsCount: number }>> {\n    return from(input).pipe(\n      reduce((acc, { modelId, categoryId }) => {\n        const entry = acc.get(modelId);\n        if (!entry) {\n          acc.set(modelId, new Set([categoryId]));\n        } else {\n          entry.add(categoryId);\n        }\n        return acc;\n      }, new Map<Id64String, Id64Set>()),\n      mergeMap((modelCategoryMap) => modelCategoryMap.entries()),\n      map(([modelId, categoryIds]) => `Model.Id = ${modelId} AND Category.Id IN (${[...categoryIds].join(\", \")})`),\n      // we may have thousands of where clauses here, and sending a single query with all of them could take a\n      // long time - instead, split it into smaller chunks\n      bufferCount(100),\n      mergeMap((whereClauses) =>\n        defer(() =>\n          this.#queryExecutor.createQueryReader(\n            {\n              ctes: [\n                `\n                  CategoryElements(id, modelId, categoryId) AS (\n                    SELECT ECInstanceId, Model.Id, Category.Id\n                    FROM ${this.#elementClassName}\n                    WHERE\n                      Parent.Id IS NULL\n                      AND (\n                        ${whereClauses.join(\" OR \")}\n                      )\n\n                    UNION ALL\n\n                    SELECT c.ECInstanceId, p.modelId, p.categoryId\n                    FROM ${this.#elementClassName} c\n                    JOIN CategoryElements p ON c.Parent.Id = p.id\n                  )\n                `,\n              ],\n              ecsql: `\n                SELECT modelId, categoryId, COUNT(id) elementsCount\n                FROM (SELECT * FROM CategoryElements)\n                GROUP BY modelId, categoryId\n              `,\n            },\n            {\n              rowFormat: \"ECSqlPropertyNames\",\n              limit: \"unbounded\",\n              restartToken: `${this.#componentName}/${this.#componentId}/category-element-counts/${Guid.createValue()}`,\n            },\n          ),\n        ).pipe(catchBeSQLiteInterrupts),\n      ),\n      releaseMainThreadOnItemsCount(500),\n      reduce(\n        ({ acc, createKey }, row) => {\n          acc.set(createKey({ modelId: row.modelId, categoryId: row.categoryId }), {\n            modelId: row.modelId,\n            categoryId: row.categoryId,\n            elementsCount: row.elementsCount,\n          });\n          return { acc, createKey };\n        },\n        {\n          acc: new Map<string, { modelId: Id64String; categoryId: Id64String; elementsCount: number }>(),\n          createKey: (keyProps: { modelId: Id64String; categoryId: Id64String }) => `${keyProps.modelId}-${keyProps.categoryId}`,\n        },\n      ),\n      mergeMap(({ acc: result, createKey }) => {\n        input.forEach(({ modelId, categoryId }) => {\n          if (!result.has(createKey({ modelId, categoryId }))) {\n            result.set(createKey({ modelId, categoryId }), { categoryId, modelId, elementsCount: 0 });\n          }\n        });\n\n        return from(result.values());\n      }),\n      toArray(),\n    );\n  }\n\n  public [Symbol.dispose]() {\n    this.#subscription.unsubscribe();\n  }\n\n  public getCategoryElementsCount({ modelId, categoryId }: { modelId: Id64String; categoryId: Id64String }): Observable<number> {\n    const cacheKey: ModelCategoryKey = `${modelId}-${categoryId}`;\n    let result = this.#cache.get(cacheKey);\n    if (result !== undefined) {\n      return from(result).pipe(take(1));\n    }\n\n    result = new ReplaySubject(1);\n    this.#cache.set(cacheKey, result);\n    this.#requestsStream.next({ modelId, categoryId });\n    return from(result).pipe(take(1));\n  }\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/common/internal/caches/ModeledElementsCache.d.ts

@@ -0,0 +1,27 @@
+import type { Observable } from "rxjs";
+import type { GuidString, Id64Array, Id64String } from "@itwin/core-bentley";
+import type { LimitingECSqlQueryExecutor } from "@itwin/presentation-hierarchies";
+import type { CategoryId, ElementId, ModelId } from "../Types.js";
+interface ModeledElementsCacheProps {
+    queryExecutor: LimitingECSqlQueryExecutor;
+    componentId: GuidString;
+    elementClassName: string;
+}
+/** @internal */
+export declare class ModeledElementsCache {
+    #private;
+    constructor(props: ModeledElementsCacheProps);
+    private queryModeledElements;
+    getModeledElementsInfo(): Observable<{
+        modelWithCategoryModeledElements: Map<ModelId, Map<CategoryId, Set<ElementId>>>;
+        allSubModels: Set<ElementId>;
+        childSubModels: Map<ElementId, Set<ElementId>>;
+    }>;
+    getSubModelsUnderElement(elementId: Id64String): Observable<Id64Array>;
+    getCategoryModeledElements({ modelId, categoryId }: {
+        modelId: Id64String;
+        categoryId: Id64String;
+    }): Observable<Id64String>;
+}
+export {};
+//# sourceMappingURL=ModeledElementsCache.d.ts.map

package/lib/esm/tree-widget-react/components/trees/common/internal/caches/ModeledElementsCache.js

@@ -0,0 +1,106 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { defer, map, mergeMap, reduce, shareReplay } from "rxjs";
+import { CLASS_NAME_Model } from "../ClassNameDefinitions.js";
+import { catchBeSQLiteInterrupts } from "../UseErrorState.js";
+/** @internal */
+export class ModeledElementsCache {
+    #queryExecutor;
+    #componentId;
+    #componentName;
+    #elementClassName;
+    // ElementId here is also a ModelId, since those elements are sub models.
+    #modeledElementsInfo;
+    constructor(props) {
+        this.#queryExecutor = props.queryExecutor;
+        this.#componentId = props.componentId;
+        this.#elementClassName = props.elementClassName;
+        this.#componentName = "ModeledElementsCache";
+    }
+    queryModeledElements() {
+        return defer(() => {
+            const query = `
+        SELECT
+          me.ECInstanceId modeledElementId,
+          me.Category.Id categoryId,
+          me.Model.Id modelId,
+          IIF(me.Parent.Id IS NULL,
+            '',
+            (
+              WITH RECURSIVE ModeledElementParents(parentId, parentPath) AS (
+                SELECT p.Parent.Id, CAST(IdToHex(p.ECInstanceId) AS TEXT)
+                FROM ${this.#elementClassName} p
+                WHERE p.ECInstanceId = me.Parent.Id
+                UNION ALL
+                SELECT pOfp.Parent.Id, CAST(IdToHex(pOfp.ECInstanceId) AS TEXT) || ';' || c.parentPath
+                FROM ${this.#elementClassName} pOfp
+                JOIN ModeledElementParents c ON c.parentId = pOfp.ECInstanceId
+              )
+              SELECT parentPath
+              FROM ModeledElementParents
+              WHERE parentId IS NULL
+            )
+          ) parentElements
+        FROM ${CLASS_NAME_Model} m
+        JOIN ${this.#elementClassName} me ON me.ECInstanceId = m.ModeledElement.Id
+        WHERE
+          m.IsPrivate = false
+          AND m.ECInstanceId IN (SELECT Model.Id FROM ${this.#elementClassName})
+      `;
+            return this.#queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded", restartToken: `${this.#componentName}/${this.#componentId}/modeled-elements` });
+        }).pipe(catchBeSQLiteInterrupts, map((row) => {
+            return { modelId: row.modelId, categoryId: row.categoryId, modeledElementId: row.modeledElementId, parentElements: row.parentElements.split(";") };
+        }));
+    }
+    getModeledElementsInfo() {
+        this.#modeledElementsInfo ??= this.queryModeledElements().pipe(reduce((acc, { modelId, categoryId, modeledElementId, parentElements }) => {
+            let modelEntry = acc.modelWithCategoryModeledElements.get(modelId);
+            if (!modelEntry) {
+                modelEntry = new Map();
+                acc.modelWithCategoryModeledElements.set(modelId, modelEntry);
+            }
+            const categoryEntry = modelEntry.get(categoryId);
+            if (!categoryEntry) {
+                modelEntry.set(categoryId, new Set([modeledElementId]));
+            }
+            else {
+                categoryEntry.add(modeledElementId);
+            }
+            acc.allSubModels.add(modeledElementId);
+            parentElements.forEach((parentElementId) => {
+                const entry = acc.childSubModels.get(parentElementId);
+                if (!entry) {
+                    acc.childSubModels.set(parentElementId, new Set([modeledElementId]));
+                }
+                else {
+                    entry.add(modeledElementId);
+                }
+            });
+            return acc;
+        }, {
+            modelWithCategoryModeledElements: new Map(),
+            allSubModels: new Set(),
+            childSubModels: new Map(),
+        }), shareReplay());
+        return this.#modeledElementsInfo;
+    }
+    getSubModelsUnderElement(elementId) {
+        return this.getModeledElementsInfo().pipe(map(({ allSubModels, childSubModels }) => {
+            const subModels = new Array();
+            if (allSubModels.has(elementId)) {
+                subModels.push(elementId);
+            }
+            const elementEntry = childSubModels.get(elementId);
+            for (const childSubModelId of elementEntry ?? []) {
+                subModels.push(childSubModelId);
+            }
+            return subModels;
+        }));
+    }
+    getCategoryModeledElements({ modelId, categoryId }) {
+        return this.getModeledElementsInfo().pipe(mergeMap(({ modelWithCategoryModeledElements }) => modelWithCategoryModeledElements.get(modelId)?.get(categoryId) ?? new Set()));
+    }
+}
+//# sourceMappingURL=ModeledElementsCache.js.map

package/lib/esm/tree-widget-react/components/trees/common/internal/caches/ModeledElementsCache.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ModeledElementsCache.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/common/internal/caches/ModeledElementsCache.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,QAAQ,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,MAAM,CAAC;AACjE,OAAO,EAAE,gBAAgB,EAAE,MAAM,4BAA4B,CAAC;AAC9D,OAAO,EAAE,uBAAuB,EAAE,MAAM,qBAAqB,CAAC;AAa9D,gBAAgB;AAChB,MAAM,OAAO,oBAAoB;IAC/B,cAAc,CAA6B;IAC3C,YAAY,CAAa;IACzB,cAAc,CAAS;IACvB,iBAAiB,CAAS;IAC1B,yEAAyE;IACzE,oBAAoB,CAMN;IAEd,YAAY,KAAgC;QAC1C,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC,aAAa,CAAC;QAC1C,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC,WAAW,CAAC;QACtC,IAAI,CAAC,iBAAiB,GAAG,KAAK,CAAC,gBAAgB,CAAC;QAChD,IAAI,CAAC,cAAc,GAAG,sBAAsB,CAAC;IAC/C,CAAC;IAEO,oBAAoB;QAM1B,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,KAAK,GAAG;;;;;;;;;;uBAUG,IAAI,CAAC,iBAAiB;;;;uBAItB,IAAI,CAAC,iBAAiB;;;;;;;;eAQ9B,gBAAgB;eAChB,IAAI,CAAC,iBAAiB;;;wDAGmB,IAAI,CAAC,iBAAiB;OACvE,CAAC;YACF,OAAO,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC1C,EAAE,KAAK,EAAE,KAAK,EAAE,EAChB,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,mBAAmB,EAAE,CACtI,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,uBAAuB,EACvB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACV,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,gBAAgB,EAAE,GAAG,CAAC,gBAAgB,EAAE,cAAc,EAAE,GAAG,CAAC,cAAc,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;QACrJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEM,sBAAsB;QAC3B,IAAI,CAAC,oBAAoB,KAAK,IAAI,CAAC,oBAAoB,EAAE,CAAC,IAAI,CAC5D,MAAM,CACJ,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,cAAc,EAAE,EAAE,EAAE;YACjE,IAAI,UAAU,GAAG,GAAG,CAAC,gCAAgC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACnE,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,UAAU,GAAG,IAAI,GAAG,EAAE,CAAC;gBACvB,GAAG,CAAC,gCAAgC,CAAC,GAAG,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;YAChE,CAAC;YACD,MAAM,aAAa,GAAG,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YACjD,IAAI,CAAC,aAAa,EAAE,CAAC;gBACnB,UAAU,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,GAAG,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC;YAC1D,CAAC;iBAAM,CAAC;gBACN,aAAa,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;YACtC,CAAC;YACD,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;YACvC,cAAc,CAAC,OAAO,CAAC,CAAC,eAAe,EAAE,EAAE;gBACzC,MAAM,KAAK,GAAG,GAAG,CAAC,cAAc,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;gBACtD,IAAI,CAAC,KAAK,EAAE,CAAC;oBACX,GAAG,CAAC,cAAc,CAAC,GAAG,CAAC,eAAe,EAAE,IAAI,GAAG,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC;gBACvE,CAAC;qBAAM,CAAC;oBACN,KAAK,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;gBAC9B,CAAC;YACH,CAAC,CAAC,CAAC;YACH,OAAO,GAAG,CAAC;QACb,CAAC,EACD;YACE,gCAAgC,EAAE,IAAI,GAAG,EAA4C;YACrF,YAAY,EAAE,IAAI,GAAG,EAAa;YAClC,cAAc,EAAE,IAAI,GAAG,EAA6B;SACrD,CACF,EACD,WAAW,EAAE,CACd,CAAC;QACF,OAAO,IAAI,CAAC,oBAAoB,CAAC;IACnC,CAAC;IAEM,wBAAwB,CAAC,SAAqB;QACnD,OAAO,IAAI,CAAC,sBAAsB,EAAE,CAAC,IAAI,CACvC,GAAG,CAAC,CAAC,EAAE,YAAY,EAAE,cAAc,EAAE,EAAE,EAAE;YACvC,MAAM,SAAS,GAAG,IAAI,KAAK,EAAa,CAAC;YACzC,IAAI,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;gBAChC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC5B,CAAC;YACD,MAAM,YAAY,GAAG,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YACnD,KAAK,MAAM,eAAe,IAAI,YAAY,IAAI,EAAE,EAAE,CAAC;gBACjD,SAAS,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAClC,CAAC;YACD,OAAO,SAAS,CAAC;QACnB,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEM,0BAA0B,CAAC,EAAE,OAAO,EAAE,UAAU,EAAmD;QACxG,OAAO,IAAI,CAAC,sBAAsB,EAAE,CAAC,IAAI,CACvC,QAAQ,CAAC,CAAC,EAAE,gCAAgC,EAAE,EAAE,EAAE,CAAC,gCAAgC,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,GAAG,CAAC,UAAU,CAAC,IAAI,IAAI,GAAG,EAAc,CAAC,CAC5I,CAAC;IACJ,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { defer, map, mergeMap, reduce, shareReplay } from \"rxjs\";\nimport { CLASS_NAME_Model } from \"../ClassNameDefinitions.js\";\nimport { catchBeSQLiteInterrupts } from \"../UseErrorState.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { GuidString, Id64Array, Id64String } from \"@itwin/core-bentley\";\nimport type { LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\nimport type { CategoryId, ElementId, ModelId } from \"../Types.js\";\n\ninterface ModeledElementsCacheProps {\n  queryExecutor: LimitingECSqlQueryExecutor;\n  componentId: GuidString;\n  elementClassName: string;\n}\n\n/** @internal */\nexport class ModeledElementsCache {\n  #queryExecutor: LimitingECSqlQueryExecutor;\n  #componentId: GuidString;\n  #componentName: string;\n  #elementClassName: string;\n  // ElementId here is also a ModelId, since those elements are sub models.\n  #modeledElementsInfo:\n    | Observable<{\n        modelWithCategoryModeledElements: Map<ModelId, Map<CategoryId, Set<ElementId>>>;\n        allSubModels: Set<ElementId>;\n        childSubModels: Map<ElementId, Set<ElementId>>;\n      }>\n    | undefined;\n\n  constructor(props: ModeledElementsCacheProps) {\n    this.#queryExecutor = props.queryExecutor;\n    this.#componentId = props.componentId;\n    this.#elementClassName = props.elementClassName;\n    this.#componentName = \"ModeledElementsCache\";\n  }\n\n  private queryModeledElements(): Observable<{\n    modelId: Id64String;\n    modeledElementId: Id64String;\n    categoryId: Id64String;\n    parentElements: Id64Array;\n  }> {\n    return defer(() => {\n      const query = `\n        SELECT\n          me.ECInstanceId modeledElementId,\n          me.Category.Id categoryId,\n          me.Model.Id modelId,\n          IIF(me.Parent.Id IS NULL,\n            '',\n            (\n              WITH RECURSIVE ModeledElementParents(parentId, parentPath) AS (\n                SELECT p.Parent.Id, CAST(IdToHex(p.ECInstanceId) AS TEXT)\n                FROM ${this.#elementClassName} p\n                WHERE p.ECInstanceId = me.Parent.Id\n                UNION ALL\n                SELECT pOfp.Parent.Id, CAST(IdToHex(pOfp.ECInstanceId) AS TEXT) || ';' || c.parentPath\n                FROM ${this.#elementClassName} pOfp\n                JOIN ModeledElementParents c ON c.parentId = pOfp.ECInstanceId\n              )\n              SELECT parentPath\n              FROM ModeledElementParents\n              WHERE parentId IS NULL\n            )\n          ) parentElements\n        FROM ${CLASS_NAME_Model} m\n        JOIN ${this.#elementClassName} me ON me.ECInstanceId = m.ModeledElement.Id\n        WHERE\n          m.IsPrivate = false\n          AND m.ECInstanceId IN (SELECT Model.Id FROM ${this.#elementClassName})\n      `;\n      return this.#queryExecutor.createQueryReader(\n        { ecsql: query },\n        { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: `${this.#componentName}/${this.#componentId}/modeled-elements` },\n      );\n    }).pipe(\n      catchBeSQLiteInterrupts,\n      map((row) => {\n        return { modelId: row.modelId, categoryId: row.categoryId, modeledElementId: row.modeledElementId, parentElements: row.parentElements.split(\";\") };\n      }),\n    );\n  }\n\n  public getModeledElementsInfo() {\n    this.#modeledElementsInfo ??= this.queryModeledElements().pipe(\n      reduce(\n        (acc, { modelId, categoryId, modeledElementId, parentElements }) => {\n          let modelEntry = acc.modelWithCategoryModeledElements.get(modelId);\n          if (!modelEntry) {\n            modelEntry = new Map();\n            acc.modelWithCategoryModeledElements.set(modelId, modelEntry);\n          }\n          const categoryEntry = modelEntry.get(categoryId);\n          if (!categoryEntry) {\n            modelEntry.set(categoryId, new Set([modeledElementId]));\n          } else {\n            categoryEntry.add(modeledElementId);\n          }\n          acc.allSubModels.add(modeledElementId);\n          parentElements.forEach((parentElementId) => {\n            const entry = acc.childSubModels.get(parentElementId);\n            if (!entry) {\n              acc.childSubModels.set(parentElementId, new Set([modeledElementId]));\n            } else {\n              entry.add(modeledElementId);\n            }\n          });\n          return acc;\n        },\n        {\n          modelWithCategoryModeledElements: new Map<ModelId, Map<CategoryId, Set<ElementId>>>(),\n          allSubModels: new Set<ElementId>(),\n          childSubModels: new Map<ElementId, Set<ElementId>>(),\n        },\n      ),\n      shareReplay(),\n    );\n    return this.#modeledElementsInfo;\n  }\n\n  public getSubModelsUnderElement(elementId: Id64String): Observable<Id64Array> {\n    return this.getModeledElementsInfo().pipe(\n      map(({ allSubModels, childSubModels }) => {\n        const subModels = new Array<ElementId>();\n        if (allSubModels.has(elementId)) {\n          subModels.push(elementId);\n        }\n        const elementEntry = childSubModels.get(elementId);\n        for (const childSubModelId of elementEntry ?? []) {\n          subModels.push(childSubModelId);\n        }\n        return subModels;\n      }),\n    );\n  }\n\n  public getCategoryModeledElements({ modelId, categoryId }: { modelId: Id64String; categoryId: Id64String }): Observable<Id64String> {\n    return this.getModeledElementsInfo().pipe(\n      mergeMap(({ modelWithCategoryModeledElements }) => modelWithCategoryModeledElements.get(modelId)?.get(categoryId) ?? new Set<Id64String>()),\n    );\n  }\n}\n"]}