@itwin/imodel-transformer 2.0.0-dev.33 → 2.0.0-dev.35

package/lib/cjs/ElementAspectCleanup.d.ts

@@ -0,0 +1,16 @@
+import { EditTxn, ElementAspect, IModelDb } from "@itwin/core-backend";
+import { Id64String } from "@itwin/core-bentley";
+/** Deletes replaceable ElementAspects for target owners while preserving excluded classes and transformer provenance aspects.
+ * @internal
+ */
+export declare class ElementAspectCleanup {
+    private readonly _targetDb;
+    private readonly _editTxn;
+    private readonly _deleteAspect;
+    constructor(_targetDb: IModelDb, _editTxn: EditTxn, _deleteAspect: (aspect: ElementAspect) => Promise<void>);
+    /** Deletes replaceable unique and multi-aspects owned by the supplied target elements.
+     * Excluded classes and transformer provenance aspects for `provenanceScopeId` are preserved. Each deletion is performed through the configured callback and requires the target `EditTxn` to be active.
+     */
+    delete(targetElementIds: ReadonlySet<Id64String>, excludedElementAspectClassFullNames: ReadonlySet<string>, provenanceScopeId?: Id64String, pageSize?: number): Promise<void>;
+}
+//# sourceMappingURL=ElementAspectCleanup.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ElementAspectCleanup.d.ts","sourceRoot":"","sources":["../../src/ElementAspectCleanup.ts"],"names":[],"mappings":"AAKA,OAAO,EACL,OAAO,EACP,aAAa,EAIb,QAAQ,EACT,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAW,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAG1D;;GAEG;AACH,qBAAa,oBAAoB;IAE7B,OAAO,CAAC,QAAQ,CAAC,SAAS;IAC1B,OAAO,CAAC,QAAQ,CAAC,QAAQ;IACzB,OAAO,CAAC,QAAQ,CAAC,aAAa;gBAFb,SAAS,EAAE,QAAQ,EACnB,QAAQ,EAAE,OAAO,EACjB,aAAa,EAAE,CAAC,MAAM,EAAE,aAAa,KAAK,OAAO,CAAC,IAAI,CAAC;IAG1E;;OAEG;IACU,MAAM,CACjB,gBAAgB,EAAE,WAAW,CAAC,UAAU,CAAC,EACzC,mCAAmC,EAAE,WAAW,CAAC,MAAM,CAAC,EACxD,iBAAiB,CAAC,EAAE,UAAU,EAC9B,QAAQ,SAAwB,GAC/B,OAAO,CAAC,IAAI,CAAC;CAiEjB"}

package/lib/cjs/ElementAspectCleanup.js

@@ -0,0 +1,83 @@
+"use strict";
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ElementAspectCleanup = void 0;
+const core_backend_1 = require("@itwin/core-backend");
+const core_common_1 = require("@itwin/core-common");
+/** Deletes replaceable ElementAspects for target owners while preserving excluded classes and transformer provenance aspects.
+ * @internal
+ */
+class ElementAspectCleanup {
+    _targetDb;
+    _editTxn;
+    _deleteAspect;
+    constructor(_targetDb, _editTxn, _deleteAspect) {
+        this._targetDb = _targetDb;
+        this._editTxn = _editTxn;
+        this._deleteAspect = _deleteAspect;
+    }
+    /** Deletes replaceable unique and multi-aspects owned by the supplied target elements.
+     * Excluded classes and transformer provenance aspects for `provenanceScopeId` are preserved. Each deletion is performed through the configured callback and requires the target `EditTxn` to be active.
+     */
+    async delete(targetElementIds, excludedElementAspectClassFullNames, provenanceScopeId, pageSize = core_backend_1.IModelDb.maxLimit - 1) {
+        if (!this._editTxn.isActive) {
+            throw new Error("The target EditTxn must be active when deleting ElementAspects.");
+        }
+        if (pageSize <= 0 || !Number.isSafeInteger(pageSize)) {
+            throw new Error("ElementAspect deletion pageSize must be a positive integer.");
+        }
+        if (targetElementIds.size === 0)
+            return;
+        const ids = new Set(targetElementIds);
+        const targetExcludedElementAspectClassFullNames = [
+            ...excludedElementAspectClassFullNames,
+        ].filter((classFullName) => this._targetDb.containsClass(classFullName));
+        for (const aspectClassFullName of [
+            core_backend_1.ElementUniqueAspect.classFullName,
+            core_backend_1.ElementMultiAspect.classFullName,
+        ]) {
+            while (true) {
+                const params = new core_common_1.QueryBinder().bindIdSet("elementIds", ids);
+                let whereClause = "TRUE";
+                if (provenanceScopeId !== undefined) {
+                    params.bindId("provenanceScopeId", provenanceScopeId);
+                    whereClause += ` AND ECInstanceId NOT IN (
+            SELECT ECInstanceId FROM ${core_backend_1.ExternalSourceAspect.classFullName}
+            WHERE Element.Id = :provenanceScopeId OR Scope.Id = :provenanceScopeId
+          )`;
+                }
+                if (targetExcludedElementAspectClassFullNames.length > 0) {
+                    whereClause += ` AND ECInstanceId NOT IN (
+            SELECT ECInstanceId FROM ${aspectClassFullName}
+            WHERE ECClassId IS (${[
+                        ...targetExcludedElementAspectClassFullNames,
+                    ].join(", ")})
+          )`;
+                }
+                const query = `SELECT aspect.ECInstanceId as id
+          FROM ${aspectClassFullName} aspect
+          INNER JOIN IdSet(:elementIds) ids ON ids.id = aspect.Element.Id
+          WHERE ${whereClause}
+          LIMIT ${pageSize}
+          OPTIONS ENABLE_EXPERIMENTAL_FEATURES`;
+                // Drain the full page before deleting: mutating a table while a reader is still
+                // scanning it is unsafe.
+                const candidateIds = [];
+                for await (const row of this._targetDb.createQueryReader(query, params, { usePrimaryConn: true })) {
+                    candidateIds.push(row.id);
+                }
+                if (candidateIds.length === 0)
+                    break;
+                for (const aspectId of candidateIds) {
+                    const aspect = this._targetDb.elements.getAspect(aspectId);
+                    await this._deleteAspect(aspect);
+                }
+            }
+        }
+    }
+}
+exports.ElementAspectCleanup = ElementAspectCleanup;
+//# sourceMappingURL=ElementAspectCleanup.js.map

package/lib/cjs/ElementAspectCleanup.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ElementAspectCleanup.js","sourceRoot":"","sources":["../../src/ElementAspectCleanup.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG,sDAO6B;AAE7B,oDAAiD;AAEjD;;GAEG;AACH,MAAa,oBAAoB;IAEZ;IACA;IACA;IAHnB,YACmB,SAAmB,EACnB,QAAiB,EACjB,aAAuD;QAFvD,cAAS,GAAT,SAAS,CAAU;QACnB,aAAQ,GAAR,QAAQ,CAAS;QACjB,kBAAa,GAAb,aAAa,CAA0C;IACvE,CAAC;IAEJ;;OAEG;IACI,KAAK,CAAC,MAAM,CACjB,gBAAyC,EACzC,mCAAwD,EACxD,iBAA8B,EAC9B,QAAQ,GAAG,uBAAQ,CAAC,QAAQ,GAAG,CAAC;QAEhC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CACb,iEAAiE,CAClE,CAAC;QACJ,CAAC;QACD,IAAI,QAAQ,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,QAAQ,CAAC,EAAE,CAAC;YACrD,MAAM,IAAI,KAAK,CACb,6DAA6D,CAC9D,CAAC;QACJ,CAAC;QACD,IAAI,gBAAgB,CAAC,IAAI,KAAK,CAAC;YAAE,OAAO;QAExC,MAAM,GAAG,GAAG,IAAI,GAAG,CAAa,gBAAgB,CAAY,CAAC;QAC7D,MAAM,yCAAyC,GAAG;YAChD,GAAG,mCAAmC;SACvC,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,aAAa,CAAC,CAAC,CAAC;QACzE,KAAK,MAAM,mBAAmB,IAAI;YAChC,kCAAmB,CAAC,aAAa;YACjC,iCAAkB,CAAC,aAAa;SACjC,EAAE,CAAC;YACF,OAAO,IAAI,EAAE,CAAC;gBACZ,MAAM,MAAM,GAAG,IAAI,yBAAW,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,GAAG,CAAC,CAAC;gBAC9D,IAAI,WAAW,GAAG,MAAM,CAAC;gBACzB,IAAI,iBAAiB,KAAK,SAAS,EAAE,CAAC;oBACpC,MAAM,CAAC,MAAM,CAAC,mBAAmB,EAAE,iBAAiB,CAAC,CAAC;oBACtD,WAAW,IAAI;uCACc,mCAAoB,CAAC,aAAa;;YAE7D,CAAC;gBACL,CAAC;gBACD,IAAI,yCAAyC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACzD,WAAW,IAAI;uCACc,mBAAmB;kCACxB;wBACpB,GAAG,yCAAyC;qBAC7C,CAAC,IAAI,CAAC,IAAI,CAAC;YACZ,CAAC;gBACL,CAAC;gBAED,MAAM,KAAK,GAAG;iBACL,mBAAmB;;kBAElB,WAAW;kBACX,QAAQ;+CACqB,CAAC;gBACxC,gFAAgF;gBAChF,yBAAyB;gBACzB,MAAM,YAAY,GAAiB,EAAE,CAAC;gBACtC,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,SAAS,CAAC,iBAAiB,CACtD,KAAK,EACL,MAAM,EACN,EAAE,cAAc,EAAE,IAAI,EAAE,CACzB,EAAE,CAAC;oBACF,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAC5B,CAAC;gBACD,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC;oBAAE,MAAM;gBAErC,KAAK,MAAM,QAAQ,IAAI,YAAY,EAAE,CAAC;oBACpC,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;oBAC3D,MAAM,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;gBACnC,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;CACF;AAhFD,oDAgFC","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  EditTxn,\n  ElementAspect,\n  ElementMultiAspect,\n  ElementUniqueAspect,\n  ExternalSourceAspect,\n  IModelDb,\n} from \"@itwin/core-backend\";\nimport { Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport { QueryBinder } from \"@itwin/core-common\";\n\n/** Deletes replaceable ElementAspects for target owners while preserving excluded classes and transformer provenance aspects.\n * @internal\n */\nexport class ElementAspectCleanup {\n  public constructor(\n    private readonly _targetDb: IModelDb,\n    private readonly _editTxn: EditTxn,\n    private readonly _deleteAspect: (aspect: ElementAspect) => Promise<void>\n  ) {}\n\n  /** Deletes replaceable unique and multi-aspects owned by the supplied target elements.\n   * Excluded classes and transformer provenance aspects for `provenanceScopeId` are preserved. Each deletion is performed through the configured callback and requires the target `EditTxn` to be active.\n   */\n  public async delete(\n    targetElementIds: ReadonlySet<Id64String>,\n    excludedElementAspectClassFullNames: ReadonlySet<string>,\n    provenanceScopeId?: Id64String,\n    pageSize = IModelDb.maxLimit - 1\n  ): Promise<void> {\n    if (!this._editTxn.isActive) {\n      throw new Error(\n        \"The target EditTxn must be active when deleting ElementAspects.\"\n      );\n    }\n    if (pageSize <= 0 || !Number.isSafeInteger(pageSize)) {\n      throw new Error(\n        \"ElementAspect deletion pageSize must be a positive integer.\"\n      );\n    }\n    if (targetElementIds.size === 0) return;\n\n    const ids = new Set<Id64String>(targetElementIds) as Id64Set;\n    const targetExcludedElementAspectClassFullNames = [\n      ...excludedElementAspectClassFullNames,\n    ].filter((classFullName) => this._targetDb.containsClass(classFullName));\n    for (const aspectClassFullName of [\n      ElementUniqueAspect.classFullName,\n      ElementMultiAspect.classFullName,\n    ]) {\n      while (true) {\n        const params = new QueryBinder().bindIdSet(\"elementIds\", ids);\n        let whereClause = \"TRUE\";\n        if (provenanceScopeId !== undefined) {\n          params.bindId(\"provenanceScopeId\", provenanceScopeId);\n          whereClause += ` AND ECInstanceId NOT IN (\n            SELECT ECInstanceId FROM ${ExternalSourceAspect.classFullName}\n            WHERE Element.Id = :provenanceScopeId OR Scope.Id = :provenanceScopeId\n          )`;\n        }\n        if (targetExcludedElementAspectClassFullNames.length > 0) {\n          whereClause += ` AND ECInstanceId NOT IN (\n            SELECT ECInstanceId FROM ${aspectClassFullName}\n            WHERE ECClassId IS (${[\n              ...targetExcludedElementAspectClassFullNames,\n            ].join(\", \")})\n          )`;\n        }\n\n        const query = `SELECT aspect.ECInstanceId as id\n          FROM ${aspectClassFullName} aspect\n          INNER JOIN IdSet(:elementIds) ids ON ids.id = aspect.Element.Id\n          WHERE ${whereClause}\n          LIMIT ${pageSize}\n          OPTIONS ENABLE_EXPERIMENTAL_FEATURES`;\n        // Drain the full page before deleting: mutating a table while a reader is still\n        // scanning it is unsafe.\n        const candidateIds: Id64String[] = [];\n        for await (const row of this._targetDb.createQueryReader(\n          query,\n          params,\n          { usePrimaryConn: true }\n        )) {\n          candidateIds.push(row.id);\n        }\n        if (candidateIds.length === 0) break;\n\n        for (const aspectId of candidateIds) {\n          const aspect = this._targetDb.elements.getAspect(aspectId);\n          await this._deleteAspect(aspect);\n        }\n      }\n    }\n  }\n}\n"]}

package/lib/cjs/ElementAspectExportCoordinator.d.ts

@@ -0,0 +1,44 @@
+import { Id64String } from "@itwin/core-bentley";
+export type ElementAspectExportPreparation = (excludedElementAspectClassFullNames: ReadonlySet<string>, ownerElementIds: ReadonlySet<Id64String>) => Promise<void>;
+/** Coordinates scoped batches of accepted ElementAspect owners, including preparation before aspect export.
+ * @internal
+ */
+export declare class ElementAspectExportCoordinator {
+    private readonly _defaultBatchSize;
+    private readonly _excludedClassFullNames;
+    private readonly _exportAspects;
+    private readonly _resetCaches;
+    private readonly _acceptedOwnerIds;
+    private readonly _processedOwnerIds;
+    private _batchSize;
+    private _depth;
+    private _prepare?;
+    constructor(_defaultBatchSize: number, _excludedClassFullNames: () => ReadonlySet<string>, _exportAspects: (ownerElementIds: ReadonlySet<Id64String>) => Promise<void>, _resetCaches?: () => void);
+    /** Whether an accepted-owner collection scope is active. */
+    get isActive(): boolean;
+    /** Sets the callback that prepares each accepted-owner group before its aspects are exported. */
+    setPreparation(prepare: ElementAspectExportPreparation): void;
+    /** Begins an accepted-owner collection scope.
+     * Nested scopes share the outer scope's state and batch size.
+     */
+    begin(batchSize?: number): void;
+    /** Ends an accepted-owner collection scope.
+     * Ending the outermost scope exports pending owners and resets all scope state.
+     */
+    end(): Promise<void>;
+    /** Discards the active scope and its pending owner state without exporting it. */
+    abort(): void;
+    /** Runs element export within an accepted-owner collection scope.
+     * A scope created by this call is aborted if element export fails; an existing outer scope remains responsible for its own failure handling.
+     */
+    run(exportElements: () => Promise<void>): Promise<void>;
+    /** Adds an accepted source owner to the current group and exports the group when it reaches the configured batch size. */
+    addAcceptedOwner(elementId: Id64String): Promise<void>;
+    /** Runs preparation and aspect export for an explicit owner set in bounded groups. */
+    exportOwners(ownerElementIds: ReadonlySet<Id64String>): Promise<void>;
+    private exportOwnerBatch;
+    private flush;
+    private resetScopeState;
+    private reset;
+}
+//# sourceMappingURL=ElementAspectExportCoordinator.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ElementAspectExportCoordinator.d.ts","sourceRoot":"","sources":["../../src/ElementAspectExportCoordinator.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAEjD,MAAM,MAAM,8BAA8B,GAAG,CAC3C,mCAAmC,EAAE,WAAW,CAAC,MAAM,CAAC,EACxD,eAAe,EAAE,WAAW,CAAC,UAAU,CAAC,KACrC,OAAO,CAAC,IAAI,CAAC,CAAC;AAEnB;;GAEG;AACH,qBAAa,8BAA8B;IAQvC,OAAO,CAAC,QAAQ,CAAC,iBAAiB;IAClC,OAAO,CAAC,QAAQ,CAAC,uBAAuB;IACxC,OAAO,CAAC,QAAQ,CAAC,cAAc;IAG/B,OAAO,CAAC,QAAQ,CAAC,YAAY;IAZ/B,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAyB;IAC3D,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAyB;IAC5D,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,MAAM,CAAK;IACnB,OAAO,CAAC,QAAQ,CAAC,CAAiC;gBAG/B,iBAAiB,EAAE,MAAM,EACzB,uBAAuB,EAAE,MAAM,WAAW,CAAC,MAAM,CAAC,EAClD,cAAc,EAAE,CAC/B,eAAe,EAAE,WAAW,CAAC,UAAU,CAAC,KACrC,OAAO,CAAC,IAAI,CAAC,EACD,YAAY,GAAE,MAAM,IAAe;IAKtD,4DAA4D;IAC5D,IAAW,QAAQ,IAAI,OAAO,CAE7B;IAED,iGAAiG;IAC1F,cAAc,CAAC,OAAO,EAAE,8BAA8B,GAAG,IAAI;IAIpE;;OAEG;IACI,KAAK,CAAC,SAAS,SAAyB,GAAG,IAAI;IAStD;;OAEG;IACU,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC;IAiBjC,kFAAkF;IAC3E,KAAK,IAAI,IAAI;IAIpB;;OAEG;IACU,GAAG,CAAC,cAAc,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAgBpE,0HAA0H;IAC7G,gBAAgB,CAAC,SAAS,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;IASnE,sFAAsF;IACzE,YAAY,CACvB,eAAe,EAAE,WAAW,CAAC,UAAU,CAAC,GACvC,OAAO,CAAC,IAAI,CAAC;YAgCF,gBAAgB;YAOhB,KAAK;IAMnB,OAAO,CAAC,eAAe;IAKvB,OAAO,CAAC,KAAK;CAKd"}

package/lib/cjs/ElementAspectExportCoordinator.js

@@ -0,0 +1,148 @@
+"use strict";
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ElementAspectExportCoordinator = void 0;
+/** Coordinates scoped batches of accepted ElementAspect owners, including preparation before aspect export.
+ * @internal
+ */
+class ElementAspectExportCoordinator {
+    _defaultBatchSize;
+    _excludedClassFullNames;
+    _exportAspects;
+    _resetCaches;
+    _acceptedOwnerIds = new Set();
+    _processedOwnerIds = new Set();
+    _batchSize;
+    _depth = 0;
+    _prepare;
+    constructor(_defaultBatchSize, _excludedClassFullNames, _exportAspects, _resetCaches = () => { }) {
+        this._defaultBatchSize = _defaultBatchSize;
+        this._excludedClassFullNames = _excludedClassFullNames;
+        this._exportAspects = _exportAspects;
+        this._resetCaches = _resetCaches;
+        this._batchSize = _defaultBatchSize;
+    }
+    /** Whether an accepted-owner collection scope is active. */
+    get isActive() {
+        return this._depth > 0;
+    }
+    /** Sets the callback that prepares each accepted-owner group before its aspects are exported. */
+    setPreparation(prepare) {
+        this._prepare = prepare;
+    }
+    /** Begins an accepted-owner collection scope.
+     * Nested scopes share the outer scope's state and batch size.
+     */
+    begin(batchSize = this._defaultBatchSize) {
+        if (this._depth === 0) {
+            this.resetScopeState();
+            this._resetCaches();
+            this._batchSize = batchSize;
+        }
+        this._depth++;
+    }
+    /** Ends an accepted-owner collection scope.
+     * Ending the outermost scope exports pending owners and resets all scope state.
+     */
+    async end() {
+        if (this._depth === 0) {
+            throw new Error("No scoped element export is active.");
+        }
+        this._depth--;
+        if (this._depth > 0)
+            return;
+        try {
+            if (this._acceptedOwnerIds.size > 0) {
+                await this.flush();
+            }
+        }
+        finally {
+            this.reset();
+        }
+    }
+    /** Discards the active scope and its pending owner state without exporting it. */
+    abort() {
+        this.reset();
+    }
+    /** Runs element export within an accepted-owner collection scope.
+     * A scope created by this call is aborted if element export fails; an existing outer scope remains responsible for its own failure handling.
+     */
+    async run(exportElements) {
+        if (this.isActive) {
+            await exportElements();
+            return;
+        }
+        this.begin();
+        try {
+            await exportElements();
+        }
+        catch (error) {
+            this.abort();
+            throw error;
+        }
+        await this.end();
+    }
+    /** Adds an accepted source owner to the current group and exports the group when it reaches the configured batch size. */
+    async addAcceptedOwner(elementId) {
+        if (!this.isActive || this._processedOwnerIds.has(elementId))
+            return;
+        this._acceptedOwnerIds.add(elementId);
+        if (this._acceptedOwnerIds.size >= this._batchSize) {
+            await this.flush();
+        }
+    }
+    /** Runs preparation and aspect export for an explicit owner set in bounded groups. */
+    async exportOwners(ownerElementIds) {
+        let ownerIds = ownerElementIds;
+        if (this.isActive && ownerElementIds.size > 0) {
+            const unprocessedOwnerIds = new Set();
+            for (const elementId of ownerElementIds) {
+                if (!this._processedOwnerIds.has(elementId)) {
+                    unprocessedOwnerIds.add(elementId);
+                    this._processedOwnerIds.add(elementId);
+                }
+            }
+            if (unprocessedOwnerIds.size === 0)
+                return;
+            ownerIds = unprocessedOwnerIds;
+        }
+        if (ownerIds.size <= this._defaultBatchSize) {
+            await this.exportOwnerBatch(ownerIds);
+            return;
+        }
+        let ownerBatch = new Set();
+        for (const elementId of ownerIds) {
+            ownerBatch.add(elementId);
+            if (ownerBatch.size === this._defaultBatchSize) {
+                await this.exportOwnerBatch(ownerBatch);
+                ownerBatch = new Set();
+            }
+        }
+        if (ownerBatch.size > 0) {
+            await this.exportOwnerBatch(ownerBatch);
+        }
+    }
+    async exportOwnerBatch(ownerElementIds) {
+        await this._prepare?.(this._excludedClassFullNames(), ownerElementIds);
+        await this._exportAspects(ownerElementIds);
+    }
+    async flush() {
+        const ownerElementIds = new Set(this._acceptedOwnerIds);
+        this._acceptedOwnerIds.clear();
+        await this.exportOwners(ownerElementIds);
+    }
+    resetScopeState() {
+        this._acceptedOwnerIds.clear();
+        this._processedOwnerIds.clear();
+    }
+    reset() {
+        this._depth = 0;
+        this.resetScopeState();
+        this._batchSize = this._defaultBatchSize;
+    }
+}
+exports.ElementAspectExportCoordinator = ElementAspectExportCoordinator;
+//# sourceMappingURL=ElementAspectExportCoordinator.js.map

package/lib/cjs/ElementAspectExportCoordinator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ElementAspectExportCoordinator.js","sourceRoot":"","sources":["../../src/ElementAspectExportCoordinator.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAShG;;GAEG;AACH,MAAa,8BAA8B;IAQtB;IACA;IACA;IAGA;IAZF,iBAAiB,GAAG,IAAI,GAAG,EAAc,CAAC;IAC1C,kBAAkB,GAAG,IAAI,GAAG,EAAc,CAAC;IACpD,UAAU,CAAS;IACnB,MAAM,GAAG,CAAC,CAAC;IACX,QAAQ,CAAkC;IAElD,YACmB,iBAAyB,EACzB,uBAAkD,EAClD,cAEC,EACD,eAA2B,GAAG,EAAE,GAAE,CAAC;QALnC,sBAAiB,GAAjB,iBAAiB,CAAQ;QACzB,4BAAuB,GAAvB,uBAAuB,CAA2B;QAClD,mBAAc,GAAd,cAAc,CAEb;QACD,iBAAY,GAAZ,YAAY,CAAuB;QAEpD,IAAI,CAAC,UAAU,GAAG,iBAAiB,CAAC;IACtC,CAAC;IAED,4DAA4D;IAC5D,IAAW,QAAQ;QACjB,OAAO,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;IACzB,CAAC;IAED,iGAAiG;IAC1F,cAAc,CAAC,OAAuC;QAC3D,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;IAC1B,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,iBAAiB;QAC7C,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtB,IAAI,CAAC,eAAe,EAAE,CAAC;YACvB,IAAI,CAAC,YAAY,EAAE,CAAC;YACpB,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;QAC9B,CAAC;QACD,IAAI,CAAC,MAAM,EAAE,CAAC;IAChB,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,GAAG;QACd,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;QACzD,CAAC;QAED,IAAI,CAAC,MAAM,EAAE,CAAC;QACd,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC;YAAE,OAAO;QAE5B,IAAI,CAAC;YACH,IAAI,IAAI,CAAC,iBAAiB,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;gBACpC,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;YACrB,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,KAAK,EAAE,CAAC;QACf,CAAC;IACH,CAAC;IAED,kFAAkF;IAC3E,KAAK;QACV,IAAI,CAAC,KAAK,EAAE,CAAC;IACf,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,GAAG,CAAC,cAAmC;QAClD,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,MAAM,cAAc,EAAE,CAAC;YACvB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,KAAK,EAAE,CAAC;QACb,IAAI,CAAC;YACH,MAAM,cAAc,EAAE,CAAC;QACzB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,KAAK,EAAE,CAAC;YACb,MAAM,KAAK,CAAC;QACd,CAAC;QACD,MAAM,IAAI,CAAC,GAAG,EAAE,CAAC;IACnB,CAAC;IAED,0HAA0H;IACnH,KAAK,CAAC,gBAAgB,CAAC,SAAqB;QACjD,IAAI,CAAC,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,SAAS,CAAC;YAAE,OAAO;QAErE,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACtC,IAAI,IAAI,CAAC,iBAAiB,CAAC,IAAI,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACnD,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;QACrB,CAAC;IACH,CAAC;IAED,sFAAsF;IAC/E,KAAK,CAAC,YAAY,CACvB,eAAwC;QAExC,IAAI,QAAQ,GAAG,eAAe,CAAC;QAC/B,IAAI,IAAI,CAAC,QAAQ,IAAI,eAAe,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YAC9C,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAAc,CAAC;YAClD,KAAK,MAAM,SAAS,IAAI,eAAe,EAAE,CAAC;gBACxC,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;oBAC5C,mBAAmB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;oBACnC,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;gBACzC,CAAC;YACH,CAAC;YACD,IAAI,mBAAmB,CAAC,IAAI,KAAK,CAAC;gBAAE,OAAO;YAC3C,QAAQ,GAAG,mBAAmB,CAAC;QACjC,CAAC;QAED,IAAI,QAAQ,CAAC,IAAI,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC5C,MAAM,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;YACtC,OAAO;QACT,CAAC;QAED,IAAI,UAAU,GAAG,IAAI,GAAG,EAAc,CAAC;QACvC,KAAK,MAAM,SAAS,IAAI,QAAQ,EAAE,CAAC;YACjC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAC1B,IAAI,UAAU,CAAC,IAAI,KAAK,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBAC/C,MAAM,IAAI,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;gBACxC,UAAU,GAAG,IAAI,GAAG,EAAc,CAAC;YACrC,CAAC;QACH,CAAC;QACD,IAAI,UAAU,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YACxB,MAAM,IAAI,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAC5B,eAAwC;QAExC,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC,IAAI,CAAC,uBAAuB,EAAE,EAAE,eAAe,CAAC,CAAC;QACvE,MAAM,IAAI,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;IAC7C,CAAC;IAEO,KAAK,CAAC,KAAK;QACjB,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACxD,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,CAAC;QAC/B,MAAM,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,CAAC;IAC3C,CAAC;IAEO,eAAe;QACrB,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,CAAC;QAC/B,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,CAAC;IAClC,CAAC;IAEO,KAAK;QACX,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;QAChB,IAAI,CAAC,eAAe,EAAE,CAAC;QACvB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,iBAAiB,CAAC;IAC3C,CAAC;CACF;AAxJD,wEAwJC","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 { Id64String } from \"@itwin/core-bentley\";\n\nexport type ElementAspectExportPreparation = (\n  excludedElementAspectClassFullNames: ReadonlySet<string>,\n  ownerElementIds: ReadonlySet<Id64String>\n) => Promise<void>;\n\n/** Coordinates scoped batches of accepted ElementAspect owners, including preparation before aspect export.\n * @internal\n */\nexport class ElementAspectExportCoordinator {\n  private readonly _acceptedOwnerIds = new Set<Id64String>();\n  private readonly _processedOwnerIds = new Set<Id64String>();\n  private _batchSize: number;\n  private _depth = 0;\n  private _prepare?: ElementAspectExportPreparation;\n\n  public constructor(\n    private readonly _defaultBatchSize: number,\n    private readonly _excludedClassFullNames: () => ReadonlySet<string>,\n    private readonly _exportAspects: (\n      ownerElementIds: ReadonlySet<Id64String>\n    ) => Promise<void>,\n    private readonly _resetCaches: () => void = () => {}\n  ) {\n    this._batchSize = _defaultBatchSize;\n  }\n\n  /** Whether an accepted-owner collection scope is active. */\n  public get isActive(): boolean {\n    return this._depth > 0;\n  }\n\n  /** Sets the callback that prepares each accepted-owner group before its aspects are exported. */\n  public setPreparation(prepare: ElementAspectExportPreparation): void {\n    this._prepare = prepare;\n  }\n\n  /** Begins an accepted-owner collection scope.\n   * Nested scopes share the outer scope's state and batch size.\n   */\n  public begin(batchSize = this._defaultBatchSize): void {\n    if (this._depth === 0) {\n      this.resetScopeState();\n      this._resetCaches();\n      this._batchSize = batchSize;\n    }\n    this._depth++;\n  }\n\n  /** Ends an accepted-owner collection scope.\n   * Ending the outermost scope exports pending owners and resets all scope state.\n   */\n  public async end(): Promise<void> {\n    if (this._depth === 0) {\n      throw new Error(\"No scoped element export is active.\");\n    }\n\n    this._depth--;\n    if (this._depth > 0) return;\n\n    try {\n      if (this._acceptedOwnerIds.size > 0) {\n        await this.flush();\n      }\n    } finally {\n      this.reset();\n    }\n  }\n\n  /** Discards the active scope and its pending owner state without exporting it. */\n  public abort(): void {\n    this.reset();\n  }\n\n  /** Runs element export within an accepted-owner collection scope.\n   * A scope created by this call is aborted if element export fails; an existing outer scope remains responsible for its own failure handling.\n   */\n  public async run(exportElements: () => Promise<void>): Promise<void> {\n    if (this.isActive) {\n      await exportElements();\n      return;\n    }\n\n    this.begin();\n    try {\n      await exportElements();\n    } catch (error) {\n      this.abort();\n      throw error;\n    }\n    await this.end();\n  }\n\n  /** Adds an accepted source owner to the current group and exports the group when it reaches the configured batch size. */\n  public async addAcceptedOwner(elementId: Id64String): Promise<void> {\n    if (!this.isActive || this._processedOwnerIds.has(elementId)) return;\n\n    this._acceptedOwnerIds.add(elementId);\n    if (this._acceptedOwnerIds.size >= this._batchSize) {\n      await this.flush();\n    }\n  }\n\n  /** Runs preparation and aspect export for an explicit owner set in bounded groups. */\n  public async exportOwners(\n    ownerElementIds: ReadonlySet<Id64String>\n  ): Promise<void> {\n    let ownerIds = ownerElementIds;\n    if (this.isActive && ownerElementIds.size > 0) {\n      const unprocessedOwnerIds = new Set<Id64String>();\n      for (const elementId of ownerElementIds) {\n        if (!this._processedOwnerIds.has(elementId)) {\n          unprocessedOwnerIds.add(elementId);\n          this._processedOwnerIds.add(elementId);\n        }\n      }\n      if (unprocessedOwnerIds.size === 0) return;\n      ownerIds = unprocessedOwnerIds;\n    }\n\n    if (ownerIds.size <= this._defaultBatchSize) {\n      await this.exportOwnerBatch(ownerIds);\n      return;\n    }\n\n    let ownerBatch = new Set<Id64String>();\n    for (const elementId of ownerIds) {\n      ownerBatch.add(elementId);\n      if (ownerBatch.size === this._defaultBatchSize) {\n        await this.exportOwnerBatch(ownerBatch);\n        ownerBatch = new Set<Id64String>();\n      }\n    }\n    if (ownerBatch.size > 0) {\n      await this.exportOwnerBatch(ownerBatch);\n    }\n  }\n\n  private async exportOwnerBatch(\n    ownerElementIds: ReadonlySet<Id64String>\n  ): Promise<void> {\n    await this._prepare?.(this._excludedClassFullNames(), ownerElementIds);\n    await this._exportAspects(ownerElementIds);\n  }\n\n  private async flush(): Promise<void> {\n    const ownerElementIds = new Set(this._acceptedOwnerIds);\n    this._acceptedOwnerIds.clear();\n    await this.exportOwners(ownerElementIds);\n  }\n\n  private resetScopeState(): void {\n    this._acceptedOwnerIds.clear();\n    this._processedOwnerIds.clear();\n  }\n\n  private reset(): void {\n    this._depth = 0;\n    this.resetScopeState();\n    this._batchSize = this._defaultBatchSize;\n  }\n}\n"]}

package/lib/cjs/ElementAspectExportProcessor.d.ts

@@ -0,0 +1,55 @@
+import { ElementAspect, ElementMultiAspect, ElementUniqueAspect, IModelDb } from "@itwin/core-backend";
+import { Id64String } from "@itwin/core-bentley";
+interface AspectChanges {
+    readonly insertIds: ReadonlySet<Id64String>;
+    readonly updateIds: ReadonlySet<Id64String>;
+}
+export interface ElementAspectExportProcessorHandler {
+    shouldExportElementAspect(aspect: ElementAspect): Promise<boolean>;
+    onExportElementUniqueAspect(uniqueAspect: ElementUniqueAspect, isUpdate?: boolean | undefined): Promise<void>;
+    onExportElementMultiAspects(multiAspects: ElementMultiAspect[]): Promise<void>;
+    trackProgress: () => Promise<void>;
+}
+/** Queries and exports ElementAspects for accepted source owners, applying class and handler filters and grouping multi-aspects by owner.
+ * @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 _aspectChanges;
+    /** ElementAspect class names excluded from source queries. */
+    get excludedElementAspectClassFullNames(): ReadonlySet<string>;
+    private readonly _sourceDb;
+    private readonly _handler;
+    constructor(sourceDb: IModelDb, handler: ElementAspectExportProcessorHandler);
+    /** Exports ElementAspects owned by the supplied elements.
+     * Multi-aspects are emitted in one callback group per owner.
+     */
+    exportAllElementAspects(ownerElementIds: ReadonlySet<Id64String>): Promise<void>;
+    /** Sets the aspect changes used to distinguish inserted and updated unique aspects during change export. */
+    setAspectChanges(aspectChanges?: AspectChanges): void;
+    /** Clears source schema and content caches at an outer export-scope boundary. */
+    resetCaches(): void;
+    /** Excludes an ElementAspect class from subsequent queries and export callbacks. */
+    excludeElementAspectClass(classFullName: string): void;
+    private exportAspectsLoop;
+    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

@@ -0,0 +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"}

package/lib/cjs/ElementAspectExportProcessor.js

@@ -0,0 +1,186 @@
+"use strict";
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ElementAspectExportProcessor = void 0;
+const core_backend_1 = require("@itwin/core-backend");
+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;
+    _aspectChanges;
+    /** ElementAspect class names excluded from source queries. */
+    get excludedElementAspectClassFullNames() {
+        return this._excludedElementAspectClassFullNames;
+    }
+    _sourceDb;
+    _handler;
+    constructor(sourceDb, handler) {
+        this._sourceDb = sourceDb;
+        this._handler = handler;
+    }
+    /** Exports ElementAspects owned by the supplied elements.
+     * Multi-aspects are emitted in one callback group per owner.
+     */
+    async exportAllElementAspects(ownerElementIds) {
+        if (ownerElementIds.size === 0)
+            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);
+            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]);
+            }
+            else {
+                ownerAspects.push(multiAspect);
+            }
+        }, ownerElementIds);
+        for (const multiAspects of multiAspectsByOwner.values()) {
+            await this._handler.onExportElementMultiAspects(multiAspects);
+            await this._handler.trackProgress();
+        }
+    }
+    /** Sets the aspect changes used to distinguish inserted and updated unique aspects during change export. */
+    setAspectChanges(aspectChanges) {
+        this._aspectChanges = aspectChanges;
+    }
+    /** Clears source schema and content caches at an outer export-scope boundary. */
+    resetCaches() {
+        this._aspectClasses.clear();
+        this._excludedClassIds = 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)
+            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);
+            }
+        }
+    }
+    async getAspectClasses(baseElementAspectClassFullName) {
+        let aspectClasses = this._aspectClasses.get(baseElementAspectClassFullName);
+        if (aspectClasses === undefined) {
+            aspectClasses = this.queryAspectClasses(baseElementAspectClassFullName);
+            this._aspectClasses.set(baseElementAspectClassFullName, aspectClasses);
+        }
+        return aspectClasses;
+    }
+    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 getExcludedClassIds() {
+        if (this._excludedClassIds === undefined) {
+            this._excludedClassIds = this.queryExcludedClassIds();
+        }
+        return this._excludedClassIds;
+    }
+    /** 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);
+            }
+        }
+        return excludedClassIds;
+    }
+    /** 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);
+        }
+        return populatedClassIds;
+    }
+}
+exports.ElementAspectExportProcessor = ElementAspectExportProcessor;
+//# sourceMappingURL=ElementAspectExportProcessor.js.map

package/lib/cjs/ElementAspectExportProcessor.js.map

@@ -0,0 +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"]}

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

@@ -1 +1 @@
-{"version":3,"file":"IModelCloneContext.d.ts","sourceRoot":"","sources":["../../src/IModelCloneContext.ts"],"names":[],"mappings":"AAQA,OAAO,EAAQ,UAAU,EAAU,MAAM,qBAAqB,CAAC;AAC/D,OAAO,EAKL,kBAAkB,EAClB,YAAY,EACZ,eAAe,EAKhB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAEL,OAAO,EACP,aAAa,EAEb,yBAAyB,EACzB,cAAc,EACf,MAAM,qBAAqB,CAAC;AAQ7B;;GAEG;AACH,qBAAa,kBAAmB,SAAQ,yBAAyB;IAC/D,OAAO,CAAC,cAAc,CAA+B;IACrD,OAAO,CAAC,iBAAiB,CAAqC;IAE9D,0HAA0H;IACpG,UAAU;IAIhC;;OAEG;IACmB,YAAY,CAChC,aAAa,EAAE,OAAO,EACtB,YAAY,CAAC,EAAE,cAAc,CAAC,mBAAmB,GAChD,OAAO,CAAC,YAAY,CAAC;IA2DxB,uGAAuG;IAChG,kBAAkB,CACvB,cAAc,EAAE,UAAU,EAC1B,cAAc,EAAE,UAAU,GACzB,IAAI;IAIP,mEAAmE;IAC5D,mBAAmB,CAAC,cAAc,EAAE,UAAU,GAAG,IAAI;IAI5D;;OAEG;IACI,kBAAkB,CAAC,cAAc,EAAE,UAAU,GAAG,UAAU;IAIjE;;OAEG;IACU,kBAAkB,CAC7B,cAAc,EAAE,eAAe,GAC9B,OAAO,CAAC,eAAe,CAAC;IAqI3B;;OAEG;IACU,kBAAkB,CAC7B,mBAAmB,EAAE,aAAa,GACjC,OAAO,CAAC,kBAAkB,CAAC;CAsD/B"}
+{"version":3,"file":"IModelCloneContext.d.ts","sourceRoot":"","sources":["../../src/IModelCloneContext.ts"],"names":[],"mappings":"AAQA,OAAO,EAAQ,UAAU,EAAU,MAAM,qBAAqB,CAAC;AAC/D,OAAO,EAKL,kBAAkB,EAClB,YAAY,EACZ,eAAe,EAKhB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAEL,OAAO,EACP,aAAa,EAEb,yBAAyB,EACzB,cAAc,EACf,MAAM,qBAAqB,CAAC;AAS7B;;GAEG;AACH,qBAAa,kBAAmB,SAAQ,yBAAyB;IAC/D,OAAO,CAAC,cAAc,CAA+B;IACrD,OAAO,CAAC,iBAAiB,CAA4B;IAErD,0HAA0H;IACpG,UAAU;IAIhC;;OAEG;IACmB,YAAY,CAChC,aAAa,EAAE,OAAO,EACtB,YAAY,CAAC,EAAE,cAAc,CAAC,mBAAmB,GAChD,OAAO,CAAC,YAAY,CAAC;IA2DxB,uGAAuG;IAChG,kBAAkB,CACvB,cAAc,EAAE,UAAU,EAC1B,cAAc,EAAE,UAAU,GACzB,IAAI;IAIP,mEAAmE;IAC5D,mBAAmB,CAAC,cAAc,EAAE,UAAU,GAAG,IAAI;IAI5D;;OAEG;IACI,kBAAkB,CAAC,cAAc,EAAE,UAAU,GAAG,UAAU;IAIjE;;OAEG;IACU,kBAAkB,CAC7B,cAAc,EAAE,eAAe,GAC9B,OAAO,CAAC,eAAe,CAAC;IAqI3B;;OAEG;IACU,kBAAkB,CAC7B,mBAAmB,EAAE,aAAa,GACjC,OAAO,CAAC,kBAAkB,CAAC;CAsD/B"}