@itwin/imodel-transformer 2.0.0-dev.41 → 2.0.0-dev.42

package/lib/cjs/ElementAspectExportProcessor.d.ts

@@ -14,13 +14,9 @@ export interface ElementAspectExportProcessorHandler {
  * @internal
  */
 export declare class ElementAspectExportProcessor {
-    private readonly _aspectClasses;
     /** ElementAspect classes excluded from source queries. */
     private readonly _excludedElementAspectClassFullNames;
-    /** ECClassIds excluded from source queries: every excluded class plus all of its
-     * declared subclasses.
-     */
-    private _excludedClassIds;
+    private _hasExportableAspects;
     private _aspectChanges;
     /** ElementAspect class names excluded from source queries. */
     get excludedElementAspectClassFullNames(): ReadonlySet<string>;
@@ -37,19 +33,11 @@ export declare class ElementAspectExportProcessor {
     resetCaches(): void;
     /** Excludes an ElementAspect class from subsequent queries and export callbacks. */
     excludeElementAspectClass(classFullName: string): void;
-    private exportAspectsLoop;
+    private getExcludedClassFilter;
+    private getHasExportableAspects;
+    private queryHasExportableAspects;
+    private getAspectProps;
     private queryAspects;
-    private getAspectClasses;
-    private queryAspectClasses;
-    private getExcludedClassIds;
-    /** Resolves every excluded class full name to the full set of ECClassIds it covers:
-     * the class itself plus every declared subclass. Reuses the same
-     * ECDbMeta.ClassHasAllBaseClasses shape as {@link queryAspectClasses}, rooted at
-     * each excluded class instead of at ElementUniqueAspect/ElementMultiAspect.
-     */
-    private queryExcludedClassIds;
-    /** Scans which concrete subclasses of the given base aspect class have rows for the supplied owners. */
-    private queryPopulatedAspectClassIds;
 }
 export {};
 //# sourceMappingURL=ElementAspectExportProcessor.d.ts.map

package/lib/cjs/ElementAspectExportProcessor.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ElementAspectExportProcessor.d.ts","sourceRoot":"","sources":["../../src/ElementAspectExportProcessor.ts"],"names":[],"mappings":"AAKA,OAAO,EACL,aAAa,EACb,kBAAkB,EAClB,mBAAmB,EACnB,QAAQ,EACT,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAW,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAQ1D,UAAU,aAAa;IACrB,QAAQ,CAAC,SAAS,EAAE,WAAW,CAAC,UAAU,CAAC,CAAC;IAC5C,QAAQ,CAAC,SAAS,EAAE,WAAW,CAAC,UAAU,CAAC,CAAC;CAC7C;AAED,MAAM,WAAW,mCAAmC;IAClD,yBAAyB,CAAC,MAAM,EAAE,aAAa,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IACnE,2BAA2B,CACzB,YAAY,EAAE,mBAAmB,EACjC,QAAQ,CAAC,EAAE,OAAO,GAAG,SAAS,GAC7B,OAAO,CAAC,IAAI,CAAC,CAAC;IACjB,2BAA2B,CACzB,YAAY,EAAE,kBAAkB,EAAE,GACjC,OAAO,CAAC,IAAI,CAAC,CAAC;IACjB,aAAa,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;CACpC;AAED;;GAEG;AACH,qBAAa,4BAA4B;IACvC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAG3B;IACJ,0DAA0D;IAC1D,OAAO,CAAC,QAAQ,CAAC,oCAAoC,CAAqB;IAC1E;;OAEG;IACH,OAAO,CAAC,iBAAiB,CAA+C;IACxE,OAAO,CAAC,cAAc,CAA4B;IAElD,8DAA8D;IAC9D,IAAW,mCAAmC,IAAI,WAAW,CAAC,MAAM,CAAC,CAEpE;IACD,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAW;IACrC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAsC;gBAG7D,QAAQ,EAAE,QAAQ,EAClB,OAAO,EAAE,mCAAmC;IAM9C;;OAEG;IACU,uBAAuB,CAClC,eAAe,EAAE,WAAW,CAAC,UAAU,CAAC,GACvC,OAAO,CAAC,IAAI,CAAC;IAqChB,4GAA4G;IACrG,gBAAgB,CAAC,aAAa,CAAC,EAAE,aAAa,GAAG,IAAI;IAI5D,iFAAiF;IAC1E,WAAW,IAAI,IAAI;IAK1B,oFAAoF;IAC7E,yBAAyB,CAAC,aAAa,EAAE,MAAM,GAAG,IAAI;YAK/C,iBAAiB;YAgBhB,YAAY;YAgDb,gBAAgB;YAahB,kBAAkB;YAqClB,mBAAmB;IAOjC;;;;OAIG;YACW,qBAAqB;IAsBnC,wGAAwG;YAC1F,4BAA4B;CAqB3C"}
+{"version":3,"file":"ElementAspectExportProcessor.d.ts","sourceRoot":"","sources":["../../src/ElementAspectExportProcessor.ts"],"names":[],"mappings":"AAKA,OAAO,EACL,aAAa,EACb,kBAAkB,EAClB,mBAAmB,EACnB,QAAQ,EACT,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,UAAU,EAAa,MAAM,qBAAqB,CAAC;AAS5D,UAAU,aAAa;IACrB,QAAQ,CAAC,SAAS,EAAE,WAAW,CAAC,UAAU,CAAC,CAAC;IAC5C,QAAQ,CAAC,SAAS,EAAE,WAAW,CAAC,UAAU,CAAC,CAAC;CAC7C;AAED,MAAM,WAAW,mCAAmC;IAClD,yBAAyB,CAAC,MAAM,EAAE,aAAa,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IACnE,2BAA2B,CACzB,YAAY,EAAE,mBAAmB,EACjC,QAAQ,CAAC,EAAE,OAAO,GAAG,SAAS,GAC7B,OAAO,CAAC,IAAI,CAAC,CAAC;IACjB,2BAA2B,CACzB,YAAY,EAAE,kBAAkB,EAAE,GACjC,OAAO,CAAC,IAAI,CAAC,CAAC;IACjB,aAAa,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;CACpC;AAED;;GAEG;AACH,qBAAa,4BAA4B;IACvC,0DAA0D;IAC1D,OAAO,CAAC,QAAQ,CAAC,oCAAoC,CAAqB;IAC1E,OAAO,CAAC,qBAAqB,CAA+B;IAC5D,OAAO,CAAC,cAAc,CAA4B;IAElD,8DAA8D;IAC9D,IAAW,mCAAmC,IAAI,WAAW,CAAC,MAAM,CAAC,CAEpE;IACD,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAW;IACrC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAsC;gBAG7D,QAAQ,EAAE,QAAQ,EAClB,OAAO,EAAE,mCAAmC;IAM9C;;OAEG;IACU,uBAAuB,CAClC,eAAe,EAAE,WAAW,CAAC,UAAU,CAAC,GACvC,OAAO,CAAC,IAAI,CAAC;IA6ChB,4GAA4G;IACrG,gBAAgB,CAAC,aAAa,CAAC,EAAE,aAAa,GAAG,IAAI;IAI5D,iFAAiF;IAC1E,WAAW,IAAI,IAAI;IAI1B,oFAAoF;IAC7E,yBAAyB,CAAC,aAAa,EAAE,MAAM,GAAG,IAAI;IAM7D,OAAO,CAAC,sBAAsB;YAkBhB,uBAAuB;YAKvB,yBAAyB;IAuBvC,OAAO,CAAC,cAAc;YA4BP,YAAY;CAqC5B"}

package/lib/cjs/ElementAspectExportProcessor.js

@@ -6,19 +6,16 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ElementAspectExportProcessor = void 0;
 const core_backend_1 = require("@itwin/core-backend");
+const core_bentley_1 = require("@itwin/core-bentley");
 const ECSqlReaderAsyncIterableIteratorAdapter_1 = require("./ECSqlReaderAsyncIterableIteratorAdapter");
 const core_common_1 = require("@itwin/core-common");
 /** Queries and exports ElementAspects for accepted source owners, applying class and handler filters and grouping multi-aspects by owner.
  * @internal
  */
 class ElementAspectExportProcessor {
-    _aspectClasses = new Map();
     /** ElementAspect classes excluded from source queries. */
     _excludedElementAspectClassFullNames = new Set();
-    /** ECClassIds excluded from source queries: every excluded class plus all of its
-     * declared subclasses.
-     */
-    _excludedClassIds;
+    _hasExportableAspects;
     _aspectChanges;
     /** ElementAspect class names excluded from source queries. */
     get excludedElementAspectClassFullNames() {
@@ -34,31 +31,43 @@ class ElementAspectExportProcessor {
      * Multi-aspects are emitted in one callback group per owner.
      */
     async exportAllElementAspects(ownerElementIds) {
-        if (ownerElementIds.size === 0)
+        if (ownerElementIds.size === 0 || !(await this.getHasExportableAspects()))
             return;
-        await this.exportAspectsLoop(core_backend_1.ElementUniqueAspect.classFullName, async (uniqueAspect) => {
-            const isUpdate = this._aspectChanges?.updateIds.has(uniqueAspect.id)
-                ? true
-                : this._aspectChanges?.insertIds.has(uniqueAspect.id)
-                    ? false
-                    : undefined;
-            await this._handler.onExportElementUniqueAspect(uniqueAspect, isUpdate);
+        let multiAspectOwnerId;
+        let multiAspects = [];
+        const exportMultiAspects = async () => {
+            if (multiAspects.length === 0)
+                return;
+            await this._handler.onExportElementMultiAspects(multiAspects);
             await this._handler.trackProgress();
-        }, ownerElementIds);
-        const multiAspectsByOwner = new Map();
-        await this.exportAspectsLoop(core_backend_1.ElementMultiAspect.classFullName, async (multiAspect) => {
-            const ownerAspects = multiAspectsByOwner.get(multiAspect.element.id);
-            if (ownerAspects === undefined) {
-                multiAspectsByOwner.set(multiAspect.element.id, [multiAspect]);
+            multiAspects = [];
+            multiAspectOwnerId = undefined;
+        };
+        for await (const aspect of this.queryAspects(ownerElementIds)) {
+            if (!(await this._handler.shouldExportElementAspect(aspect)))
+                continue;
+            if (multiAspectOwnerId !== undefined &&
+                multiAspectOwnerId !== aspect.element.id) {
+                await exportMultiAspects();
+            }
+            if (aspect instanceof core_backend_1.ElementUniqueAspect) {
+                const isUpdate = this._aspectChanges?.updateIds.has(aspect.id)
+                    ? true
+                    : this._aspectChanges?.insertIds.has(aspect.id)
+                        ? false
+                        : undefined;
+                await this._handler.onExportElementUniqueAspect(aspect, isUpdate);
+                await this._handler.trackProgress();
+            }
+            else if (aspect instanceof core_backend_1.ElementMultiAspect) {
+                multiAspectOwnerId = aspect.element.id;
+                multiAspects.push(aspect);
             }
             else {
-                ownerAspects.push(multiAspect);
+                throw new Error(`Unexpected ElementAspect type ${aspect.classFullName}`);
             }
-        }, ownerElementIds);
-        for (const multiAspects of multiAspectsByOwner.values()) {
-            await this._handler.onExportElementMultiAspects(multiAspects);
-            await this._handler.trackProgress();
         }
+        await exportMultiAspects();
     }
     /** Sets the aspect changes used to distinguish inserted and updated unique aspects during change export. */
     setAspectChanges(aspectChanges) {
@@ -66,120 +75,92 @@ class ElementAspectExportProcessor {
     }
     /** Clears source schema and content caches at an outer export-scope boundary. */
     resetCaches() {
-        this._aspectClasses.clear();
-        this._excludedClassIds = undefined;
+        this._hasExportableAspects = undefined;
     }
     /** Excludes an ElementAspect class from subsequent queries and export callbacks. */
     excludeElementAspectClass(classFullName) {
-        this._excludedElementAspectClassFullNames.add(classFullName);
-        this._excludedClassIds = undefined;
-    }
-    async exportAspectsLoop(baseAspectClass, exportAspect, ownerElementIds) {
-        for await (const aspect of this.queryAspects(baseAspectClass, ownerElementIds)) {
-            if (!(await this._handler.shouldExportElementAspect(aspect))) {
-                continue;
-            }
-            await exportAspect(aspect);
-        }
-    }
-    async *queryAspects(baseElementAspectClassFullName, ownerElementIds) {
-        const queryOwnerElementIds = new Set(ownerElementIds);
-        const populatedClassIds = await this.queryPopulatedAspectClassIds(baseElementAspectClassFullName, queryOwnerElementIds);
-        if (populatedClassIds.size === 0)
+        if (this._excludedElementAspectClassFullNames.has(classFullName))
             return;
-        const aspectClassNameIdMap = await this.getAspectClasses(baseElementAspectClassFullName);
-        const excludedClassIds = await this.getExcludedClassIds();
-        for (const [classId, { schemaName, className }] of aspectClassNameIdMap) {
-            // Skip classes with no rows in the current owner scope: most schemas declare
-            // far more concrete aspect subclasses than a batch actually populates.
-            if (!populatedClassIds.has(classId))
-                continue;
-            // excludedClassIds is the excluded class plus every declared subclass.
-            if (excludedClassIds.has(classId))
-                continue;
-            const classFullName = `${schemaName}:${className}`;
-            const queryParams = new core_common_1.QueryBinder()
-                .bindId("classId", classId)
-                .bindIdSet("ownerElementIds", queryOwnerElementIds);
-            const aspectQueryReader = this._sourceDb.createQueryReader(`SELECT aspect.* FROM [${schemaName}]:[${className}] aspect
-         INNER JOIN IdSet(:ownerElementIds) ids ON ids.id = aspect.Element.Id
-         WHERE aspect.ECClassId = :classId
-         ORDER BY aspect.Element.Id, aspect.ECInstanceId OPTIONS ENABLE_EXPERIMENTAL_FEATURES`, queryParams, 
-            // eslint-disable-next-line @typescript-eslint/no-deprecated
-            { rowFormat: core_common_1.QueryRowFormat.UseJsPropertyNames, usePrimaryConn: true });
-            for await (const rowProxy of (0, ECSqlReaderAsyncIterableIteratorAdapter_1.ensureECSqlReaderIsAsyncIterableIterator)(aspectQueryReader)) {
-                const { className: _className, ...aspectProps } = rowProxy.toRow();
-                aspectProps.classFullName = classFullName;
-                yield this._sourceDb.constructEntity(aspectProps);
-            }
-        }
+        this._excludedElementAspectClassFullNames.add(classFullName);
+        this.resetCaches();
     }
-    async getAspectClasses(baseElementAspectClassFullName) {
-        let aspectClasses = this._aspectClasses.get(baseElementAspectClassFullName);
-        if (aspectClasses === undefined) {
-            aspectClasses = this.queryAspectClasses(baseElementAspectClassFullName);
-            this._aspectClasses.set(baseElementAspectClassFullName, aspectClasses);
+    getExcludedClassFilter(queryParams) {
+        const excludedClassParameters = [];
+        let excludedClassIndex = 0;
+        for (const classFullName of this._excludedElementAspectClassFullNames) {
+            const parameterName = `excludedAspectClass${excludedClassIndex++}`;
+            excludedClassParameters.push(`ec_classid(:${parameterName})`);
+            queryParams.bindString(parameterName, classFullName.replace(".", ":"));
         }
-        return aspectClasses;
+        return excludedClassParameters.length === 0
+            ? ""
+            : `AND NOT EXISTS (
+          SELECT 1 FROM meta.ClassHasAllBaseClasses excluded
+          WHERE excluded.SourceECInstanceId = aspect.ECClassId
+            AND excluded.TargetECInstanceId IN (${excludedClassParameters.join(",")})
+        )`;
     }
-    async queryAspectClasses(baseElementAspectClassFullName) {
-        const aspectClassNameIdMap = new Map();
-        const aspectClassesQueryReader = this._sourceDb.createQueryReader(`
-        SELECT c.ECInstanceId as classId,
-          (ec_className(c.ECInstanceId, 's')) as schemaName,
-          (ec_className(c.ECInstanceId, 'c')) as className
-        FROM ECDbMeta.ClassHasAllBaseClasses r
-        JOIN ECDbMeta.ECClassDef c ON c.ECInstanceId = r.SourceECInstanceId
-        WHERE r.TargetECInstanceId = ec_classId(:baseClassName)
-        ORDER BY schemaName, className
-      `, new core_common_1.QueryBinder().bindString("baseClassName", baseElementAspectClassFullName), { usePrimaryConn: true });
-        for await (const rowProxy of (0, ECSqlReaderAsyncIterableIteratorAdapter_1.ensureECSqlReaderIsAsyncIterableIterator)(aspectClassesQueryReader)) {
-            const row = rowProxy.toRow();
-            aspectClassNameIdMap.set(row.classId, {
-                schemaName: row.schemaName,
-                className: row.className,
-            });
-        }
-        return aspectClassNameIdMap;
+    async getHasExportableAspects() {
+        this._hasExportableAspects ??= this.queryHasExportableAspects();
+        return this._hasExportableAspects;
     }
-    async getExcludedClassIds() {
-        if (this._excludedClassIds === undefined) {
-            this._excludedClassIds = this.queryExcludedClassIds();
-        }
-        return this._excludedClassIds;
+    async queryHasExportableAspects() {
+        const queryParams = new core_common_1.QueryBinder();
+        const excludedClassFilter = this.getExcludedClassFilter(queryParams);
+        const reader = this._sourceDb.createQueryReader(`SELECT ECInstanceId FROM (
+         SELECT aspect.ECInstanceId
+         FROM Bis.ElementMultiAspect aspect
+         WHERE TRUE ${excludedClassFilter}
+         UNION ALL
+         SELECT aspect.ECInstanceId
+         FROM Bis.ElementUniqueAspect aspect
+         WHERE TRUE ${excludedClassFilter}
+       )
+       LIMIT 1`, queryParams, { usePrimaryConn: true });
+        for await (const _row of (0, ECSqlReaderAsyncIterableIteratorAdapter_1.ensureECSqlReaderIsAsyncIterableIterator)(reader))
+            return true;
+        return false;
     }
-    /** Resolves every excluded class full name to the full set of ECClassIds it covers:
-     * the class itself plus every declared subclass. Reuses the same
-     * ECDbMeta.ClassHasAllBaseClasses shape as {@link queryAspectClasses}, rooted at
-     * each excluded class instead of at ElementUniqueAspect/ElementMultiAspect.
-     */
-    async queryExcludedClassIds() {
-        const excludedClassIds = new Set();
-        for (const classFullName of this._excludedElementAspectClassFullNames) {
-            const excludedClassesQueryReader = this._sourceDb.createQueryReader(`
-          SELECT c.ECInstanceId as classId
-          FROM ECDbMeta.ClassHasAllBaseClasses r
-          JOIN ECDbMeta.ECClassDef c ON c.ECInstanceId = r.SourceECInstanceId
-          WHERE r.TargetECInstanceId = ec_classId(:excludedClassName)
-        `, new core_common_1.QueryBinder().bindString("excludedClassName", classFullName), { usePrimaryConn: true });
-            for await (const rowProxy of (0, ECSqlReaderAsyncIterableIteratorAdapter_1.ensureECSqlReaderIsAsyncIterableIterator)(excludedClassesQueryReader)) {
-                excludedClassIds.add(rowProxy.toRow().classId);
-            }
+    getAspectProps(rawInstance) {
+        const parsedRow = typeof rawInstance === "string"
+            ? JSON.parse(rawInstance, core_common_1.Base64EncodedString.reviver)
+            : rawInstance;
+        if (!core_bentley_1.JsonUtils.isObject(parsedRow))
+            throw new Error("Expected an ElementAspect instance query to return an object");
+        const row = {};
+        for (const [key, value] of Object.entries(parsedRow)) {
+            const ecPropertyName = key.length === 0 ? key : key[0].toUpperCase() + key.substring(1);
+            row[core_common_1.ECJsNames.toJsName(ecPropertyName)] = value;
         }
-        return excludedClassIds;
+        const className = row.className;
+        if (typeof className !== "string")
+            throw new Error("Expected an ElementAspect instance query to return a className");
+        row.classFullName = className.replace(".", ":");
+        delete row.className;
+        return row;
     }
-    /** Scans which concrete subclasses of the given base aspect class have rows for the supplied owners. */
-    async queryPopulatedAspectClassIds(baseElementAspectClassFullName, ownerElementIds) {
-        const [schemaName, className] = baseElementAspectClassFullName.split(":");
-        const populatedClassIds = new Set();
-        const populatedClassesQueryReader = this._sourceDb.createQueryReader(`SELECT DISTINCT aspect.ECClassId as classId
-       FROM [${schemaName}]:[${className}] aspect
-       INNER JOIN IdSet(:ownerElementIds) ids ON ids.id = aspect.Element.Id
-       OPTIONS ENABLE_EXPERIMENTAL_FEATURES`, new core_common_1.QueryBinder().bindIdSet("ownerElementIds", ownerElementIds), { usePrimaryConn: true });
-        for await (const rowProxy of (0, ECSqlReaderAsyncIterableIteratorAdapter_1.ensureECSqlReaderIsAsyncIterableIterator)(populatedClassesQueryReader)) {
-            populatedClassIds.add(rowProxy.toRow().classId);
+    async *queryAspects(ownerElementIds) {
+        const queryParams = new core_common_1.QueryBinder().bindIdSet("ownerElementIds", ownerElementIds);
+        const excludedClassFilter = this.getExcludedClassFilter(queryParams);
+        const aspectQueryReader = this._sourceDb.createQueryReader(`WITH OwnerIds AS (SELECT id FROM IdSet(:ownerElementIds))
+       SELECT $ FROM (
+         SELECT aspect.ECInstanceId, aspect.ECClassId,
+           aspect.Element.Id AS OwnerId, 1 AS AspectKind
+         FROM OwnerIds owners
+         CROSS JOIN Bis.ElementMultiAspect aspect
+         WHERE aspect.Element.Id = owners.id ${excludedClassFilter}
+         UNION ALL
+         SELECT aspect.ECInstanceId, aspect.ECClassId,
+           aspect.Element.Id AS OwnerId, 0 AS AspectKind
+         FROM OwnerIds owners
+         CROSS JOIN Bis.ElementUniqueAspect aspect
+         WHERE aspect.Element.Id = owners.id ${excludedClassFilter}
+       )
+       ORDER BY AspectKind, OwnerId, ECClassId, ECInstanceId
+       OPTIONS USE_JS_PROP_NAMES DO_NOT_TRUNCATE_BLOB`, queryParams, { usePrimaryConn: true });
+        for await (const rowProxy of (0, ECSqlReaderAsyncIterableIteratorAdapter_1.ensureECSqlReaderIsAsyncIterableIterator)(aspectQueryReader)) {
+            yield this._sourceDb.constructEntity(this.getAspectProps(rowProxy[0]));
         }
-        return populatedClassIds;
     }
 }
 exports.ElementAspectExportProcessor = ElementAspectExportProcessor;

package/lib/cjs/ElementAspectExportProcessor.js.map

@@ -1 +1 @@
-{"version":3,"file":"ElementAspectExportProcessor.js","sourceRoot":"","sources":["../../src/ElementAspectExportProcessor.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG,sDAK6B;AAE7B,uGAAqG;AACrG,oDAI4B;AAmB5B;;GAEG;AACH,MAAa,4BAA4B;IACtB,cAAc,GAAG,IAAI,GAAG,EAGtC,CAAC;IACJ,0DAA0D;IACzC,oCAAoC,GAAG,IAAI,GAAG,EAAU,CAAC;IAC1E;;OAEG;IACK,iBAAiB,CAA+C;IAChE,cAAc,CAA4B;IAElD,8DAA8D;IAC9D,IAAW,mCAAmC;QAC5C,OAAO,IAAI,CAAC,oCAAoC,CAAC;IACnD,CAAC;IACgB,SAAS,CAAW;IACpB,QAAQ,CAAsC;IAE/D,YACE,QAAkB,EAClB,OAA4C;QAE5C,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAC1B,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;IAC1B,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,uBAAuB,CAClC,eAAwC;QAExC,IAAI,eAAe,CAAC,IAAI,KAAK,CAAC;YAAE,OAAO;QAEvC,MAAM,IAAI,CAAC,iBAAiB,CAC1B,kCAAmB,CAAC,aAAa,EACjC,KAAK,EAAE,YAAY,EAAE,EAAE;YACrB,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,EAAE,SAAS,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC;gBAClE,CAAC,CAAC,IAAI;gBACN,CAAC,CAAC,IAAI,CAAC,cAAc,EAAE,SAAS,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC;oBACnD,CAAC,CAAC,KAAK;oBACP,CAAC,CAAC,SAAS,CAAC;YAChB,MAAM,IAAI,CAAC,QAAQ,CAAC,2BAA2B,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;YACxE,MAAM,IAAI,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;QACtC,CAAC,EACD,eAAe,CAChB,CAAC;QAEF,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAAoC,CAAC;QACxE,MAAM,IAAI,CAAC,iBAAiB,CAC1B,iCAAkB,CAAC,aAAa,EAChC,KAAK,EAAE,WAAW,EAAE,EAAE;YACpB,MAAM,YAAY,GAAG,mBAAmB,CAAC,GAAG,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACrE,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;gBAC/B,mBAAmB,CAAC,GAAG,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC;YACjE,CAAC;iBAAM,CAAC;gBACN,YAAY,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACjC,CAAC;QACH,CAAC,EACD,eAAe,CAChB,CAAC;QAEF,KAAK,MAAM,YAAY,IAAI,mBAAmB,CAAC,MAAM,EAAE,EAAE,CAAC;YACxD,MAAM,IAAI,CAAC,QAAQ,CAAC,2BAA2B,CAAC,YAAY,CAAC,CAAC;YAC9D,MAAM,IAAI,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;QACtC,CAAC;IACH,CAAC;IAED,4GAA4G;IACrG,gBAAgB,CAAC,aAA6B;QACnD,IAAI,CAAC,cAAc,GAAG,aAAa,CAAC;IACtC,CAAC;IAED,iFAAiF;IAC1E,WAAW;QAChB,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAC;QAC5B,IAAI,CAAC,iBAAiB,GAAG,SAAS,CAAC;IACrC,CAAC;IAED,oFAAoF;IAC7E,yBAAyB,CAAC,aAAqB;QACpD,IAAI,CAAC,oCAAoC,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QAC7D,IAAI,CAAC,iBAAiB,GAAG,SAAS,CAAC;IACrC,CAAC;IAEO,KAAK,CAAC,iBAAiB,CAC7B,eAAuB,EACvB,YAA0C,EAC1C,eAAwC;QAExC,IAAI,KAAK,EAAE,MAAM,MAAM,IAAI,IAAI,CAAC,YAAY,CAC1C,eAAe,EACf,eAAe,CAChB,EAAE,CAAC;YACF,IAAI,CAAC,CAAC,MAAM,IAAI,CAAC,QAAQ,CAAC,yBAAyB,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC;gBAC7D,SAAS;YACX,CAAC;YACD,MAAM,YAAY,CAAC,MAAM,CAAC,CAAC;QAC7B,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,YAAY,CACzB,8BAAsC,EACtC,eAAwC;QAExC,MAAM,oBAAoB,GAAG,IAAI,GAAG,CAAC,eAAe,CAAY,CAAC;QACjE,MAAM,iBAAiB,GAAG,MAAM,IAAI,CAAC,4BAA4B,CAC/D,8BAA8B,EAC9B,oBAAoB,CACrB,CAAC;QACF,IAAI,iBAAiB,CAAC,IAAI,KAAK,CAAC;YAAE,OAAO;QAEzC,MAAM,oBAAoB,GAAG,MAAM,IAAI,CAAC,gBAAgB,CACtD,8BAA8B,CAC/B,CAAC;QACF,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAC1D,KAAK,MAAM,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,IAAI,oBAAoB,EAAE,CAAC;YACxE,6EAA6E;YAC7E,uEAAuE;YACvE,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,OAAO,CAAC;gBAAE,SAAS;YAC9C,uEAAuE;YACvE,IAAI,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC;gBAAE,SAAS;YAE5C,MAAM,aAAa,GAAG,GAAG,UAAU,IAAI,SAAS,EAAE,CAAC;YACnD,MAAM,WAAW,GAAG,IAAI,yBAAW,EAAE;iBAClC,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC;iBAC1B,SAAS,CAAC,iBAAiB,EAAE,oBAAoB,CAAC,CAAC;YACtD,MAAM,iBAAiB,GAAG,IAAI,CAAC,SAAS,CAAC,iBAAiB,CACxD,yBAAyB,UAAU,MAAM,SAAS;;;8FAGoC,EACtF,WAAW;YACX,4DAA4D;YAC5D,EAAE,SAAS,EAAE,4BAAc,CAAC,kBAAkB,EAAE,cAAc,EAAE,IAAI,EAAE,CACvE,CAAC;YACF,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,IAAA,kFAAwC,EACnE,iBAAiB,CAClB,EAAE,CAAC;gBACF,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,GAAG,WAAW,EAAE,GAC7C,QAAQ,CAAC,KAAK,EAEb,CAAC;gBACJ,WAAW,CAAC,aAAa,GAAG,aAAa,CAAC;gBAC1C,MAAM,IAAI,CAAC,SAAS,CAAC,eAAe,CAAI,WAAW,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAC5B,8BAAsC;QAItC,IAAI,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC;QAC5E,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;YAChC,aAAa,GAAG,IAAI,CAAC,kBAAkB,CAAC,8BAA8B,CAAC,CAAC;YACxE,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,8BAA8B,EAAE,aAAa,CAAC,CAAC;QACzE,CAAC;QACD,OAAO,aAAa,CAAC;IACvB,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAC9B,8BAAsC;QAItC,MAAM,oBAAoB,GAAG,IAAI,GAAG,EAGjC,CAAC;QACJ,MAAM,wBAAwB,GAAG,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAC/D;;;;;;;;OAQC,EACD,IAAI,yBAAW,EAAE,CAAC,UAAU,CAC1B,eAAe,EACf,8BAA8B,CAC/B,EACD,EAAE,cAAc,EAAE,IAAI,EAAE,CACzB,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,IAAA,kFAAwC,EACnE,wBAAwB,CACzB,EAAE,CAAC;YACF,MAAM,GAAG,GAAG,QAAQ,CAAC,KAAK,EAAE,CAAC;YAC7B,oBAAoB,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE;gBACpC,UAAU,EAAE,GAAG,CAAC,UAAU;gBAC1B,SAAS,EAAE,GAAG,CAAC,SAAS;aACzB,CAAC,CAAC;QACL,CAAC;QACD,OAAO,oBAAoB,CAAC;IAC9B,CAAC;IAEO,KAAK,CAAC,mBAAmB;QAC/B,IAAI,IAAI,CAAC,iBAAiB,KAAK,SAAS,EAAE,CAAC;YACzC,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC,qBAAqB,EAAE,CAAC;QACxD,CAAC;QACD,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAChC,CAAC;IAED;;;;OAIG;IACK,KAAK,CAAC,qBAAqB;QACjC,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAc,CAAC;QAC/C,KAAK,MAAM,aAAa,IAAI,IAAI,CAAC,oCAAoC,EAAE,CAAC;YACtE,MAAM,0BAA0B,GAAG,IAAI,CAAC,SAAS,CAAC,iBAAiB,CACjE;;;;;SAKC,EACD,IAAI,yBAAW,EAAE,CAAC,UAAU,CAAC,mBAAmB,EAAE,aAAa,CAAC,EAChE,EAAE,cAAc,EAAE,IAAI,EAAE,CACzB,CAAC;YACF,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,IAAA,kFAAwC,EACnE,0BAA0B,CAC3B,EAAE,CAAC;gBACF,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,CAAC;YACjD,CAAC;QACH,CAAC;QACD,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IAED,wGAAwG;IAChG,KAAK,CAAC,4BAA4B,CACxC,8BAAsC,EACtC,eAAwB;QAExB,MAAM,CAAC,UAAU,EAAE,SAAS,CAAC,GAAG,8BAA8B,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC1E,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAc,CAAC;QAChD,MAAM,2BAA2B,GAAG,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAClE;eACS,UAAU,MAAM,SAAS;;4CAEI,EACtC,IAAI,yBAAW,EAAE,CAAC,SAAS,CAAC,iBAAiB,EAAE,eAAe,CAAC,EAC/D,EAAE,cAAc,EAAE,IAAI,EAAE,CACzB,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,IAAA,kFAAwC,EACnE,2BAA2B,CAC5B,EAAE,CAAC;YACF,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC,OAAO,CAAC,CAAC;QAClD,CAAC;QACD,OAAO,iBAAiB,CAAC;IAC3B,CAAC;CACF;AAjQD,oEAiQC","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 {\n  ElementAspect,\n  ElementMultiAspect,\n  ElementUniqueAspect,\n  IModelDb,\n} from \"@itwin/core-backend\";\nimport { Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport { ensureECSqlReaderIsAsyncIterableIterator } from \"./ECSqlReaderAsyncIterableIteratorAdapter\";\nimport {\n  ElementAspectProps,\n  QueryBinder,\n  QueryRowFormat,\n} from \"@itwin/core-common\";\n\ninterface AspectChanges {\n  readonly insertIds: ReadonlySet<Id64String>;\n  readonly updateIds: ReadonlySet<Id64String>;\n}\n\nexport interface ElementAspectExportProcessorHandler {\n  shouldExportElementAspect(aspect: ElementAspect): Promise<boolean>;\n  onExportElementUniqueAspect(\n    uniqueAspect: ElementUniqueAspect,\n    isUpdate?: boolean | undefined\n  ): Promise<void>;\n  onExportElementMultiAspects(\n    multiAspects: ElementMultiAspect[]\n  ): Promise<void>;\n  trackProgress: () => Promise<void>;\n}\n\n/** Queries and exports ElementAspects for accepted source owners, applying class and handler filters and grouping multi-aspects by owner.\n * @internal\n */\nexport class ElementAspectExportProcessor {\n  private readonly _aspectClasses = new Map<\n    string,\n    Promise<ReadonlyMap<Id64String, { schemaName: string; className: string }>>\n  >();\n  /** ElementAspect classes excluded from source queries. */\n  private readonly _excludedElementAspectClassFullNames = new Set<string>();\n  /** ECClassIds excluded from source queries: every excluded class plus all of its\n   * declared subclasses.\n   */\n  private _excludedClassIds: Promise<ReadonlySet<Id64String>> | undefined;\n  private _aspectChanges: AspectChanges | undefined;\n\n  /** ElementAspect class names excluded from source queries. */\n  public get excludedElementAspectClassFullNames(): ReadonlySet<string> {\n    return this._excludedElementAspectClassFullNames;\n  }\n  private readonly _sourceDb: IModelDb;\n  private readonly _handler: ElementAspectExportProcessorHandler;\n\n  public constructor(\n    sourceDb: IModelDb,\n    handler: ElementAspectExportProcessorHandler\n  ) {\n    this._sourceDb = sourceDb;\n    this._handler = handler;\n  }\n\n  /** Exports ElementAspects owned by the supplied elements.\n   * Multi-aspects are emitted in one callback group per owner.\n   */\n  public async exportAllElementAspects(\n    ownerElementIds: ReadonlySet<Id64String>\n  ): Promise<void> {\n    if (ownerElementIds.size === 0) return;\n\n    await this.exportAspectsLoop<ElementUniqueAspect>(\n      ElementUniqueAspect.classFullName,\n      async (uniqueAspect) => {\n        const isUpdate = this._aspectChanges?.updateIds.has(uniqueAspect.id)\n          ? true\n          : this._aspectChanges?.insertIds.has(uniqueAspect.id)\n            ? false\n            : undefined;\n        await this._handler.onExportElementUniqueAspect(uniqueAspect, isUpdate);\n        await this._handler.trackProgress();\n      },\n      ownerElementIds\n    );\n\n    const multiAspectsByOwner = new Map<Id64String, ElementMultiAspect[]>();\n    await this.exportAspectsLoop<ElementMultiAspect>(\n      ElementMultiAspect.classFullName,\n      async (multiAspect) => {\n        const ownerAspects = multiAspectsByOwner.get(multiAspect.element.id);\n        if (ownerAspects === undefined) {\n          multiAspectsByOwner.set(multiAspect.element.id, [multiAspect]);\n        } else {\n          ownerAspects.push(multiAspect);\n        }\n      },\n      ownerElementIds\n    );\n\n    for (const multiAspects of multiAspectsByOwner.values()) {\n      await this._handler.onExportElementMultiAspects(multiAspects);\n      await this._handler.trackProgress();\n    }\n  }\n\n  /** Sets the aspect changes used to distinguish inserted and updated unique aspects during change export. */\n  public setAspectChanges(aspectChanges?: AspectChanges): void {\n    this._aspectChanges = aspectChanges;\n  }\n\n  /** Clears source schema and content caches at an outer export-scope boundary. */\n  public resetCaches(): void {\n    this._aspectClasses.clear();\n    this._excludedClassIds = undefined;\n  }\n\n  /** Excludes an ElementAspect class from subsequent queries and export callbacks. */\n  public excludeElementAspectClass(classFullName: string): void {\n    this._excludedElementAspectClassFullNames.add(classFullName);\n    this._excludedClassIds = undefined;\n  }\n\n  private async exportAspectsLoop<T extends ElementAspect>(\n    baseAspectClass: string,\n    exportAspect: (aspect: T) => Promise<void>,\n    ownerElementIds: ReadonlySet<Id64String>\n  ): Promise<void> {\n    for await (const aspect of this.queryAspects<T>(\n      baseAspectClass,\n      ownerElementIds\n    )) {\n      if (!(await this._handler.shouldExportElementAspect(aspect))) {\n        continue;\n      }\n      await exportAspect(aspect);\n    }\n  }\n\n  private async *queryAspects<T extends ElementAspect>(\n    baseElementAspectClassFullName: string,\n    ownerElementIds: ReadonlySet<Id64String>\n  ) {\n    const queryOwnerElementIds = new Set(ownerElementIds) as Id64Set;\n    const populatedClassIds = await this.queryPopulatedAspectClassIds(\n      baseElementAspectClassFullName,\n      queryOwnerElementIds\n    );\n    if (populatedClassIds.size === 0) return;\n\n    const aspectClassNameIdMap = await this.getAspectClasses(\n      baseElementAspectClassFullName\n    );\n    const excludedClassIds = await this.getExcludedClassIds();\n    for (const [classId, { schemaName, className }] of aspectClassNameIdMap) {\n      // Skip classes with no rows in the current owner scope: most schemas declare\n      // far more concrete aspect subclasses than a batch actually populates.\n      if (!populatedClassIds.has(classId)) continue;\n      // excludedClassIds is the excluded class plus every declared subclass.\n      if (excludedClassIds.has(classId)) continue;\n\n      const classFullName = `${schemaName}:${className}`;\n      const queryParams = new QueryBinder()\n        .bindId(\"classId\", classId)\n        .bindIdSet(\"ownerElementIds\", queryOwnerElementIds);\n      const aspectQueryReader = this._sourceDb.createQueryReader(\n        `SELECT aspect.* FROM [${schemaName}]:[${className}] aspect\n         INNER JOIN IdSet(:ownerElementIds) ids ON ids.id = aspect.Element.Id\n         WHERE aspect.ECClassId = :classId\n         ORDER BY aspect.Element.Id, aspect.ECInstanceId OPTIONS ENABLE_EXPERIMENTAL_FEATURES`,\n        queryParams,\n        // eslint-disable-next-line @typescript-eslint/no-deprecated\n        { rowFormat: QueryRowFormat.UseJsPropertyNames, usePrimaryConn: true }\n      );\n      for await (const rowProxy of ensureECSqlReaderIsAsyncIterableIterator(\n        aspectQueryReader\n      )) {\n        const { className: _className, ...aspectProps } =\n          rowProxy.toRow() as ElementAspectProps & {\n            className?: string;\n          };\n        aspectProps.classFullName = classFullName;\n        yield this._sourceDb.constructEntity<T>(aspectProps);\n      }\n    }\n  }\n\n  private async getAspectClasses(\n    baseElementAspectClassFullName: string\n  ): Promise<\n    ReadonlyMap<Id64String, { schemaName: string; className: string }>\n  > {\n    let aspectClasses = this._aspectClasses.get(baseElementAspectClassFullName);\n    if (aspectClasses === undefined) {\n      aspectClasses = this.queryAspectClasses(baseElementAspectClassFullName);\n      this._aspectClasses.set(baseElementAspectClassFullName, aspectClasses);\n    }\n    return aspectClasses;\n  }\n\n  private async queryAspectClasses(\n    baseElementAspectClassFullName: string\n  ): Promise<\n    ReadonlyMap<Id64String, { schemaName: string; className: string }>\n  > {\n    const aspectClassNameIdMap = new Map<\n      Id64String,\n      { schemaName: string; className: string }\n    >();\n    const aspectClassesQueryReader = this._sourceDb.createQueryReader(\n      `\n        SELECT c.ECInstanceId as classId,\n          (ec_className(c.ECInstanceId, 's')) as schemaName,\n          (ec_className(c.ECInstanceId, 'c')) as className\n        FROM ECDbMeta.ClassHasAllBaseClasses r\n        JOIN ECDbMeta.ECClassDef c ON c.ECInstanceId = r.SourceECInstanceId\n        WHERE r.TargetECInstanceId = ec_classId(:baseClassName)\n        ORDER BY schemaName, className\n      `,\n      new QueryBinder().bindString(\n        \"baseClassName\",\n        baseElementAspectClassFullName\n      ),\n      { usePrimaryConn: true }\n    );\n    for await (const rowProxy of ensureECSqlReaderIsAsyncIterableIterator(\n      aspectClassesQueryReader\n    )) {\n      const row = rowProxy.toRow();\n      aspectClassNameIdMap.set(row.classId, {\n        schemaName: row.schemaName,\n        className: row.className,\n      });\n    }\n    return aspectClassNameIdMap;\n  }\n\n  private async getExcludedClassIds(): Promise<ReadonlySet<Id64String>> {\n    if (this._excludedClassIds === undefined) {\n      this._excludedClassIds = this.queryExcludedClassIds();\n    }\n    return this._excludedClassIds;\n  }\n\n  /** Resolves every excluded class full name to the full set of ECClassIds it covers:\n   * the class itself plus every declared subclass. Reuses the same\n   * ECDbMeta.ClassHasAllBaseClasses shape as {@link queryAspectClasses}, rooted at\n   * each excluded class instead of at ElementUniqueAspect/ElementMultiAspect.\n   */\n  private async queryExcludedClassIds(): Promise<ReadonlySet<Id64String>> {\n    const excludedClassIds = new Set<Id64String>();\n    for (const classFullName of this._excludedElementAspectClassFullNames) {\n      const excludedClassesQueryReader = this._sourceDb.createQueryReader(\n        `\n          SELECT c.ECInstanceId as classId\n          FROM ECDbMeta.ClassHasAllBaseClasses r\n          JOIN ECDbMeta.ECClassDef c ON c.ECInstanceId = r.SourceECInstanceId\n          WHERE r.TargetECInstanceId = ec_classId(:excludedClassName)\n        `,\n        new QueryBinder().bindString(\"excludedClassName\", classFullName),\n        { usePrimaryConn: true }\n      );\n      for await (const rowProxy of ensureECSqlReaderIsAsyncIterableIterator(\n        excludedClassesQueryReader\n      )) {\n        excludedClassIds.add(rowProxy.toRow().classId);\n      }\n    }\n    return excludedClassIds;\n  }\n\n  /** Scans which concrete subclasses of the given base aspect class have rows for the supplied owners. */\n  private async queryPopulatedAspectClassIds(\n    baseElementAspectClassFullName: string,\n    ownerElementIds: Id64Set\n  ): Promise<ReadonlySet<Id64String>> {\n    const [schemaName, className] = baseElementAspectClassFullName.split(\":\");\n    const populatedClassIds = new Set<Id64String>();\n    const populatedClassesQueryReader = this._sourceDb.createQueryReader(\n      `SELECT DISTINCT aspect.ECClassId as classId\n       FROM [${schemaName}]:[${className}] aspect\n       INNER JOIN IdSet(:ownerElementIds) ids ON ids.id = aspect.Element.Id\n       OPTIONS ENABLE_EXPERIMENTAL_FEATURES`,\n      new QueryBinder().bindIdSet(\"ownerElementIds\", ownerElementIds),\n      { usePrimaryConn: true }\n    );\n    for await (const rowProxy of ensureECSqlReaderIsAsyncIterableIterator(\n      populatedClassesQueryReader\n    )) {\n      populatedClassIds.add(rowProxy.toRow().classId);\n    }\n    return populatedClassIds;\n  }\n}\n"]}
+{"version":3,"file":"ElementAspectExportProcessor.js","sourceRoot":"","sources":["../../src/ElementAspectExportProcessor.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG,sDAK6B;AAC7B,sDAA4D;AAC5D,uGAAqG;AACrG,oDAK4B;AAmB5B;;GAEG;AACH,MAAa,4BAA4B;IACvC,0DAA0D;IACzC,oCAAoC,GAAG,IAAI,GAAG,EAAU,CAAC;IAClE,qBAAqB,CAA+B;IACpD,cAAc,CAA4B;IAElD,8DAA8D;IAC9D,IAAW,mCAAmC;QAC5C,OAAO,IAAI,CAAC,oCAAoC,CAAC;IACnD,CAAC;IACgB,SAAS,CAAW;IACpB,QAAQ,CAAsC;IAE/D,YACE,QAAkB,EAClB,OAA4C;QAE5C,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAC1B,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;IAC1B,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,uBAAuB,CAClC,eAAwC;QAExC,IAAI,eAAe,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,IAAI,CAAC,uBAAuB,EAAE,CAAC;YACvE,OAAO;QAET,IAAI,kBAA0C,CAAC;QAC/C,IAAI,YAAY,GAAyB,EAAE,CAAC;QAC5C,MAAM,kBAAkB,GAAG,KAAK,IAAI,EAAE;YACpC,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO;YACtC,MAAM,IAAI,CAAC,QAAQ,CAAC,2BAA2B,CAAC,YAAY,CAAC,CAAC;YAC9D,MAAM,IAAI,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;YACpC,YAAY,GAAG,EAAE,CAAC;YAClB,kBAAkB,GAAG,SAAS,CAAC;QACjC,CAAC,CAAC;QAEF,IAAI,KAAK,EAAE,MAAM,MAAM,IAAI,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,EAAE,CAAC;YAC9D,IAAI,CAAC,CAAC,MAAM,IAAI,CAAC,QAAQ,CAAC,yBAAyB,CAAC,MAAM,CAAC,CAAC;gBAAE,SAAS;YAEvE,IACE,kBAAkB,KAAK,SAAS;gBAChC,kBAAkB,KAAK,MAAM,CAAC,OAAO,CAAC,EAAE,EACxC,CAAC;gBACD,MAAM,kBAAkB,EAAE,CAAC;YAC7B,CAAC;YAED,IAAI,MAAM,YAAY,kCAAmB,EAAE,CAAC;gBAC1C,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,EAAE,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;oBAC5D,CAAC,CAAC,IAAI;oBACN,CAAC,CAAC,IAAI,CAAC,cAAc,EAAE,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC7C,CAAC,CAAC,KAAK;wBACP,CAAC,CAAC,SAAS,CAAC;gBAChB,MAAM,IAAI,CAAC,QAAQ,CAAC,2BAA2B,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;gBAClE,MAAM,IAAI,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;YACtC,CAAC;iBAAM,IAAI,MAAM,YAAY,iCAAkB,EAAE,CAAC;gBAChD,kBAAkB,GAAG,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;gBACvC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC5B,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,KAAK,CACb,iCAAiC,MAAM,CAAC,aAAa,EAAE,CACxD,CAAC;YACJ,CAAC;QACH,CAAC;QAED,MAAM,kBAAkB,EAAE,CAAC;IAC7B,CAAC;IAED,4GAA4G;IACrG,gBAAgB,CAAC,aAA6B;QACnD,IAAI,CAAC,cAAc,GAAG,aAAa,CAAC;IACtC,CAAC;IAED,iFAAiF;IAC1E,WAAW;QAChB,IAAI,CAAC,qBAAqB,GAAG,SAAS,CAAC;IACzC,CAAC;IAED,oFAAoF;IAC7E,yBAAyB,CAAC,aAAqB;QACpD,IAAI,IAAI,CAAC,oCAAoC,CAAC,GAAG,CAAC,aAAa,CAAC;YAAE,OAAO;QACzE,IAAI,CAAC,oCAAoC,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QAC7D,IAAI,CAAC,WAAW,EAAE,CAAC;IACrB,CAAC;IAEO,sBAAsB,CAAC,WAAwB;QACrD,MAAM,uBAAuB,GAAa,EAAE,CAAC;QAC7C,IAAI,kBAAkB,GAAG,CAAC,CAAC;QAC3B,KAAK,MAAM,aAAa,IAAI,IAAI,CAAC,oCAAoC,EAAE,CAAC;YACtE,MAAM,aAAa,GAAG,sBAAsB,kBAAkB,EAAE,EAAE,CAAC;YACnE,uBAAuB,CAAC,IAAI,CAAC,eAAe,aAAa,GAAG,CAAC,CAAC;YAC9D,WAAW,CAAC,UAAU,CAAC,aAAa,EAAE,aAAa,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QACzE,CAAC;QAED,OAAO,uBAAuB,CAAC,MAAM,KAAK,CAAC;YACzC,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC;;;kDAG0C,uBAAuB,CAAC,IAAI,CAAC,GAAG,CAAC;UACzE,CAAC;IACT,CAAC;IAEO,KAAK,CAAC,uBAAuB;QACnC,IAAI,CAAC,qBAAqB,KAAK,IAAI,CAAC,yBAAyB,EAAE,CAAC;QAChE,OAAO,IAAI,CAAC,qBAAqB,CAAC;IACpC,CAAC;IAEO,KAAK,CAAC,yBAAyB;QACrC,MAAM,WAAW,GAAG,IAAI,yBAAW,EAAE,CAAC;QACtC,MAAM,mBAAmB,GAAG,IAAI,CAAC,sBAAsB,CAAC,WAAW,CAAC,CAAC;QACrE,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAC7C;;;sBAGgB,mBAAmB;;;;sBAInB,mBAAmB;;eAE1B,EACT,WAAW,EACX,EAAE,cAAc,EAAE,IAAI,EAAE,CACzB,CAAC;QAEF,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,IAAA,kFAAwC,EAAC,MAAM,CAAC;YACvE,OAAO,IAAI,CAAC;QACd,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,cAAc,CAAC,WAAoB;QACzC,MAAM,SAAS,GACb,OAAO,WAAW,KAAK,QAAQ;YAC7B,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,iCAAmB,CAAC,OAAO,CAAC;YACtD,CAAC,CAAC,WAAW,CAAC;QAClB,IAAI,CAAC,wBAAS,CAAC,QAAQ,CAAC,SAAS,CAAC;YAChC,MAAM,IAAI,KAAK,CACb,8DAA8D,CAC/D,CAAC;QAEJ,MAAM,GAAG,GAA4B,EAAE,CAAC;QACxC,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;YACrD,MAAM,cAAc,GAClB,GAAG,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;YACnE,GAAG,CAAC,uBAAS,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,GAAG,KAAK,CAAC;QAClD,CAAC;QAED,MAAM,SAAS,GAAG,GAAG,CAAC,SAAS,CAAC;QAChC,IAAI,OAAO,SAAS,KAAK,QAAQ;YAC/B,MAAM,IAAI,KAAK,CACb,gEAAgE,CACjE,CAAC;QAEJ,GAAG,CAAC,aAAa,GAAG,SAAS,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAChD,OAAO,GAAG,CAAC,SAAS,CAAC;QACrB,OAAO,GAAoC,CAAC;IAC9C,CAAC;IAEO,KAAK,CAAC,CAAC,YAAY,CACzB,eAAwC;QAExC,MAAM,WAAW,GAAG,IAAI,yBAAW,EAAE,CAAC,SAAS,CAC7C,iBAAiB,EACjB,eAAe,CAChB,CAAC;QACF,MAAM,mBAAmB,GAAG,IAAI,CAAC,sBAAsB,CAAC,WAAW,CAAC,CAAC;QACrE,MAAM,iBAAiB,GAAG,IAAI,CAAC,SAAS,CAAC,iBAAiB,CACxD;;;;;;+CAMyC,mBAAmB;;;;;;+CAMnB,mBAAmB;;;sDAGZ,EAChD,WAAW,EACX,EAAE,cAAc,EAAE,IAAI,EAAE,CACzB,CAAC;QAEF,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,IAAA,kFAAwC,EACnE,iBAAiB,CAClB,EAAE,CAAC;YACF,MAAM,IAAI,CAAC,SAAS,CAAC,eAAe,CAClC,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CACjC,CAAC;QACJ,CAAC;IACH,CAAC;CACF;AAvMD,oEAuMC","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 {\n  ElementAspect,\n  ElementMultiAspect,\n  ElementUniqueAspect,\n  IModelDb,\n} from \"@itwin/core-backend\";\nimport { Id64String, JsonUtils } from \"@itwin/core-bentley\";\nimport { ensureECSqlReaderIsAsyncIterableIterator } from \"./ECSqlReaderAsyncIterableIteratorAdapter\";\nimport {\n  Base64EncodedString,\n  ECJsNames,\n  ElementAspectProps,\n  QueryBinder,\n} from \"@itwin/core-common\";\n\ninterface AspectChanges {\n  readonly insertIds: ReadonlySet<Id64String>;\n  readonly updateIds: ReadonlySet<Id64String>;\n}\n\nexport interface ElementAspectExportProcessorHandler {\n  shouldExportElementAspect(aspect: ElementAspect): Promise<boolean>;\n  onExportElementUniqueAspect(\n    uniqueAspect: ElementUniqueAspect,\n    isUpdate?: boolean | undefined\n  ): Promise<void>;\n  onExportElementMultiAspects(\n    multiAspects: ElementMultiAspect[]\n  ): Promise<void>;\n  trackProgress: () => Promise<void>;\n}\n\n/** Queries and exports ElementAspects for accepted source owners, applying class and handler filters and grouping multi-aspects by owner.\n * @internal\n */\nexport class ElementAspectExportProcessor {\n  /** ElementAspect classes excluded from source queries. */\n  private readonly _excludedElementAspectClassFullNames = new Set<string>();\n  private _hasExportableAspects: Promise<boolean> | undefined;\n  private _aspectChanges: AspectChanges | undefined;\n\n  /** ElementAspect class names excluded from source queries. */\n  public get excludedElementAspectClassFullNames(): ReadonlySet<string> {\n    return this._excludedElementAspectClassFullNames;\n  }\n  private readonly _sourceDb: IModelDb;\n  private readonly _handler: ElementAspectExportProcessorHandler;\n\n  public constructor(\n    sourceDb: IModelDb,\n    handler: ElementAspectExportProcessorHandler\n  ) {\n    this._sourceDb = sourceDb;\n    this._handler = handler;\n  }\n\n  /** Exports ElementAspects owned by the supplied elements.\n   * Multi-aspects are emitted in one callback group per owner.\n   */\n  public async exportAllElementAspects(\n    ownerElementIds: ReadonlySet<Id64String>\n  ): Promise<void> {\n    if (ownerElementIds.size === 0 || !(await this.getHasExportableAspects()))\n      return;\n\n    let multiAspectOwnerId: Id64String | undefined;\n    let multiAspects: ElementMultiAspect[] = [];\n    const exportMultiAspects = async () => {\n      if (multiAspects.length === 0) return;\n      await this._handler.onExportElementMultiAspects(multiAspects);\n      await this._handler.trackProgress();\n      multiAspects = [];\n      multiAspectOwnerId = undefined;\n    };\n\n    for await (const aspect of this.queryAspects(ownerElementIds)) {\n      if (!(await this._handler.shouldExportElementAspect(aspect))) continue;\n\n      if (\n        multiAspectOwnerId !== undefined &&\n        multiAspectOwnerId !== aspect.element.id\n      ) {\n        await exportMultiAspects();\n      }\n\n      if (aspect instanceof ElementUniqueAspect) {\n        const isUpdate = this._aspectChanges?.updateIds.has(aspect.id)\n          ? true\n          : this._aspectChanges?.insertIds.has(aspect.id)\n            ? false\n            : undefined;\n        await this._handler.onExportElementUniqueAspect(aspect, isUpdate);\n        await this._handler.trackProgress();\n      } else if (aspect instanceof ElementMultiAspect) {\n        multiAspectOwnerId = aspect.element.id;\n        multiAspects.push(aspect);\n      } else {\n        throw new Error(\n          `Unexpected ElementAspect type ${aspect.classFullName}`\n        );\n      }\n    }\n\n    await exportMultiAspects();\n  }\n\n  /** Sets the aspect changes used to distinguish inserted and updated unique aspects during change export. */\n  public setAspectChanges(aspectChanges?: AspectChanges): void {\n    this._aspectChanges = aspectChanges;\n  }\n\n  /** Clears source schema and content caches at an outer export-scope boundary. */\n  public resetCaches(): void {\n    this._hasExportableAspects = undefined;\n  }\n\n  /** Excludes an ElementAspect class from subsequent queries and export callbacks. */\n  public excludeElementAspectClass(classFullName: string): void {\n    if (this._excludedElementAspectClassFullNames.has(classFullName)) return;\n    this._excludedElementAspectClassFullNames.add(classFullName);\n    this.resetCaches();\n  }\n\n  private getExcludedClassFilter(queryParams: QueryBinder): string {\n    const excludedClassParameters: string[] = [];\n    let excludedClassIndex = 0;\n    for (const classFullName of this._excludedElementAspectClassFullNames) {\n      const parameterName = `excludedAspectClass${excludedClassIndex++}`;\n      excludedClassParameters.push(`ec_classid(:${parameterName})`);\n      queryParams.bindString(parameterName, classFullName.replace(\".\", \":\"));\n    }\n\n    return excludedClassParameters.length === 0\n      ? \"\"\n      : `AND NOT EXISTS (\n          SELECT 1 FROM meta.ClassHasAllBaseClasses excluded\n          WHERE excluded.SourceECInstanceId = aspect.ECClassId\n            AND excluded.TargetECInstanceId IN (${excludedClassParameters.join(\",\")})\n        )`;\n  }\n\n  private async getHasExportableAspects(): Promise<boolean> {\n    this._hasExportableAspects ??= this.queryHasExportableAspects();\n    return this._hasExportableAspects;\n  }\n\n  private async queryHasExportableAspects(): Promise<boolean> {\n    const queryParams = new QueryBinder();\n    const excludedClassFilter = this.getExcludedClassFilter(queryParams);\n    const reader = this._sourceDb.createQueryReader(\n      `SELECT ECInstanceId FROM (\n         SELECT aspect.ECInstanceId\n         FROM Bis.ElementMultiAspect aspect\n         WHERE TRUE ${excludedClassFilter}\n         UNION ALL\n         SELECT aspect.ECInstanceId\n         FROM Bis.ElementUniqueAspect aspect\n         WHERE TRUE ${excludedClassFilter}\n       )\n       LIMIT 1`,\n      queryParams,\n      { usePrimaryConn: true }\n    );\n\n    for await (const _row of ensureECSqlReaderIsAsyncIterableIterator(reader))\n      return true;\n    return false;\n  }\n\n  private getAspectProps(rawInstance: unknown): ElementAspectProps {\n    const parsedRow: unknown =\n      typeof rawInstance === \"string\"\n        ? JSON.parse(rawInstance, Base64EncodedString.reviver)\n        : rawInstance;\n    if (!JsonUtils.isObject(parsedRow))\n      throw new Error(\n        \"Expected an ElementAspect instance query to return an object\"\n      );\n\n    const row: Record<string, unknown> = {};\n    for (const [key, value] of Object.entries(parsedRow)) {\n      const ecPropertyName =\n        key.length === 0 ? key : key[0].toUpperCase() + key.substring(1);\n      row[ECJsNames.toJsName(ecPropertyName)] = value;\n    }\n\n    const className = row.className;\n    if (typeof className !== \"string\")\n      throw new Error(\n        \"Expected an ElementAspect instance query to return a className\"\n      );\n\n    row.classFullName = className.replace(\".\", \":\");\n    delete row.className;\n    return row as unknown as ElementAspectProps;\n  }\n\n  private async *queryAspects(\n    ownerElementIds: ReadonlySet<Id64String>\n  ): AsyncIterableIterator<ElementAspect> {\n    const queryParams = new QueryBinder().bindIdSet(\n      \"ownerElementIds\",\n      ownerElementIds\n    );\n    const excludedClassFilter = this.getExcludedClassFilter(queryParams);\n    const aspectQueryReader = this._sourceDb.createQueryReader(\n      `WITH OwnerIds AS (SELECT id FROM IdSet(:ownerElementIds))\n       SELECT $ FROM (\n         SELECT aspect.ECInstanceId, aspect.ECClassId,\n           aspect.Element.Id AS OwnerId, 1 AS AspectKind\n         FROM OwnerIds owners\n         CROSS JOIN Bis.ElementMultiAspect aspect\n         WHERE aspect.Element.Id = owners.id ${excludedClassFilter}\n         UNION ALL\n         SELECT aspect.ECInstanceId, aspect.ECClassId,\n           aspect.Element.Id AS OwnerId, 0 AS AspectKind\n         FROM OwnerIds owners\n         CROSS JOIN Bis.ElementUniqueAspect aspect\n         WHERE aspect.Element.Id = owners.id ${excludedClassFilter}\n       )\n       ORDER BY AspectKind, OwnerId, ECClassId, ECInstanceId\n       OPTIONS USE_JS_PROP_NAMES DO_NOT_TRUNCATE_BLOB`,\n      queryParams,\n      { usePrimaryConn: true }\n    );\n\n    for await (const rowProxy of ensureECSqlReaderIsAsyncIterableIterator(\n      aspectQueryReader\n    )) {\n      yield this._sourceDb.constructEntity<ElementAspect>(\n        this.getAspectProps(rowProxy[0])\n      );\n    }\n  }\n}\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@itwin/imodel-transformer",
-  "version": "2.0.0-dev.41",
+  "version": "2.0.0-dev.42",
   "description": "API for exporting an iModel's parts and also importing them into another iModel",
   "main": "lib/cjs/imodel-transformer.js",
   "typings": "lib/cjs/imodel-transformer",