@itwin/imodel-transformer 2.0.0-dev.4 → 2.0.0-dev.40

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/ElementCascadingDeleter.d.ts

@@ -1,13 +1,13 @@
 /** @packageDocumentation
  * @module iModels
  */
-import { ElementTreeDeleter, ElementTreeWalkerScope, IModelDb } from "@itwin/core-backend";
+import { EditTxn, ElementTreeDeleter, ElementTreeWalkerScope } from "@itwin/core-backend";
 import { Id64String } from "@itwin/core-bentley";
 /** Deletes an element tree and code scope references starting with the specified top element. The top element is also deleted. Uses ElementCascadeDeleter.
- * @param iModel The iModel
+ * @param editTxn The active EditTxn for the iModel
  * @param topElement The parent of the sub-tree
  */
-export declare function deleteElementTreeCascade(iModel: IModelDb, topElement: Id64String): void;
+export declare function deleteElementTreeCascade(editTxn: EditTxn, topElement: Id64String): void;
 /** Deletes an entire element tree, including sub-models, child elements and code scope references.
  * Items are deleted in bottom-up order. Definitions and Subjects are deleted after normal elements.
  * Call deleteNormalElements on each tree. Then call deleteSpecialElements.

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

@@ -1 +1 @@
-{"version":3,"file":"ElementCascadingDeleter.d.ts","sourceRoot":"","sources":["../../src/ElementCascadingDeleter.ts"],"names":[],"mappings":"AAIA;;GAEG;AACH,OAAO,EACL,kBAAkB,EAClB,sBAAsB,EACtB,QAAQ,EACT,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAY,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAE3D;;;GAGG;AACH,wBAAgB,wBAAwB,CACtC,MAAM,EAAE,QAAQ,EAChB,UAAU,EAAE,UAAU,GACrB,IAAI,CAIN;AAED;;;GAGG;AACH,qBAAa,uBAAwB,SAAQ,kBAAkB;IAC7D,SAAS,CAAC,qBAAqB,CAC7B,UAAU,EAAE,UAAU,EACtB,MAAM,EAAE,sBAAsB;IAKhC,qCAAqC;cAClB,kBAAkB,CACnC,OAAO,EAAE,UAAU,EACnB,KAAK,EAAE,sBAAsB,GAC5B,IAAI;IAMP,oCAAoC;IACpC,OAAO,CAAC,kBAAkB;CAsB3B"}
+{"version":3,"file":"ElementCascadingDeleter.d.ts","sourceRoot":"","sources":["../../src/ElementCascadingDeleter.ts"],"names":[],"mappings":"AAIA;;GAEG;AACH,OAAO,EACL,OAAO,EACP,kBAAkB,EAClB,sBAAsB,EACvB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAGjD;;;GAGG;AACH,wBAAgB,wBAAwB,CACtC,OAAO,EAAE,OAAO,EAChB,UAAU,EAAE,UAAU,GACrB,IAAI,CAIN;AAED;;;GAGG;AACH,qBAAa,uBAAwB,SAAQ,kBAAkB;IAC7D,SAAS,CAAC,qBAAqB,CAC7B,UAAU,EAAE,UAAU,EACtB,MAAM,EAAE,sBAAsB;IAKhC,qCAAqC;cAClB,kBAAkB,CACnC,OAAO,EAAE,UAAU,EACnB,KAAK,EAAE,sBAAsB,GAC5B,IAAI;IAMP,oCAAoC;IACpC,OAAO,CAAC,kBAAkB;CAyB3B"}

package/lib/cjs/ElementCascadingDeleter.js

@@ -10,13 +10,13 @@ exports.deleteElementTreeCascade = deleteElementTreeCascade;
  * @module iModels
  */
 const core_backend_1 = require("@itwin/core-backend");
-const core_bentley_1 = require("@itwin/core-bentley");
+const core_common_1 = require("@itwin/core-common");
 /** Deletes an element tree and code scope references starting with the specified top element. The top element is also deleted. Uses ElementCascadeDeleter.
- * @param iModel The iModel
+ * @param editTxn The active EditTxn for the iModel
  * @param topElement The parent of the sub-tree
  */
-function deleteElementTreeCascade(iModel, topElement) {
-    const del = new ElementCascadingDeleter(iModel);
+function deleteElementTreeCascade(editTxn, topElement) {
+    const del = new ElementCascadingDeleter(editTxn);
     del.deleteNormalElements(topElement);
     del.deleteSpecialElements();
 }
@@ -38,19 +38,19 @@ class ElementCascadingDeleter extends core_backend_1.ElementTreeDeleter {
     /** Process code scope references */
     _processCodeScopes(element, scope) {
         const newScope = new core_backend_1.ElementTreeWalkerScope(scope, element);
-        // eslint-disable-next-line @itwin/no-internal, @typescript-eslint/no-deprecated
-        this._iModel.withPreparedStatement(`
+        const query = `
       SELECT ECInstanceId
       FROM bis.Element
-      WHERE CodeScope.id=?
+      WHERE CodeScope.id=:scopeId
         AND Parent.id IS NULL
-    `, (stmt) => {
-            stmt.bindId(1, element);
-            while (stmt.step() === core_bentley_1.DbResult.BE_SQLITE_ROW) {
-                const elementId = stmt.getValue(0).getId();
+    `;
+        const params = new core_common_1.QueryBinder().bindId("scopeId", element);
+        this.txn.iModel.withQueryReader(query, (reader) => {
+            for (const row of reader) {
+                const elementId = row[0];
                 this.processElementTree(elementId, newScope);
             }
-        });
+        }, params);
     }
 }
 exports.ElementCascadingDeleter = ElementCascadingDeleter;

package/lib/cjs/ElementCascadingDeleter.js.map

@@ -1 +1 @@
-{"version":3,"file":"ElementCascadingDeleter.js","sourceRoot":"","sources":["../../src/ElementCascadingDeleter.ts"],"names":[],"mappings":";;;AAkBA,4DAOC;AAzBD;;;gGAGgG;AAChG;;GAEG;AACH,sDAI6B;AAC7B,sDAA2D;AAE3D;;;GAGG;AACH,SAAgB,wBAAwB,CACtC,MAAgB,EAChB,UAAsB;IAEtB,MAAM,GAAG,GAAG,IAAI,uBAAuB,CAAC,MAAM,CAAC,CAAC;IAChD,GAAG,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC;IACrC,GAAG,CAAC,qBAAqB,EAAE,CAAC;AAC9B,CAAC;AAED;;;GAGG;AACH,MAAa,uBAAwB,SAAQ,iCAAkB;IACnD,qBAAqB,CAC7B,UAAsB,EACtB,MAA8B;QAE9B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,qCAAqC;IAClB,kBAAkB,CACnC,OAAmB,EACnB,KAA6B;QAE7B,IAAI,IAAI,CAAC,qBAAqB,CAAC,OAAO,EAAE,KAAK,CAAC,EAAE,CAAC;YAC/C,IAAI,CAAC,kBAAkB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAC1C,CAAC;QACD,KAAK,CAAC,kBAAkB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;IAC3C,CAAC;IACD,oCAAoC;IAC5B,kBAAkB,CACxB,OAAmB,EACnB,KAA6B;QAE7B,MAAM,QAAQ,GAAG,IAAI,qCAAsB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAC5D,gFAAgF;QAChF,IAAI,CAAC,OAAO,CAAC,qBAAqB,CAChC;;;;;KAKD,EACC,CAAC,IAAI,EAAE,EAAE;YACP,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;YACxB,OAAO,IAAI,CAAC,IAAI,EAAE,KAAK,uBAAQ,CAAC,aAAa,EAAE,CAAC;gBAC9C,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC;gBAC3C,IAAI,CAAC,kBAAkB,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YAC/C,CAAC;QACH,CAAC,CACF,CAAC;IACJ,CAAC;CACF;AAzCD,0DAyCC","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/** @packageDocumentation\n * @module iModels\n */\nimport {\n  ElementTreeDeleter,\n  ElementTreeWalkerScope,\n  IModelDb,\n} from \"@itwin/core-backend\";\nimport { DbResult, Id64String } from \"@itwin/core-bentley\";\n\n/** Deletes an element tree and code scope references starting with the specified top element. The top element is also deleted. Uses ElementCascadeDeleter.\n * @param iModel The iModel\n * @param topElement The parent of the sub-tree\n */\nexport function deleteElementTreeCascade(\n  iModel: IModelDb,\n  topElement: Id64String\n): void {\n  const del = new ElementCascadingDeleter(iModel);\n  del.deleteNormalElements(topElement);\n  del.deleteSpecialElements();\n}\n\n/** Deletes an entire element tree, including sub-models, child elements and code scope references.\n * Items are deleted in bottom-up order. Definitions and Subjects are deleted after normal elements.\n * Call deleteNormalElements on each tree. Then call deleteSpecialElements.\n */\nexport class ElementCascadingDeleter extends ElementTreeDeleter {\n  protected shouldVisitCodeScopes(\n    _elementId: Id64String,\n    _scope: ElementTreeWalkerScope\n  ) {\n    return true;\n  }\n\n  /** The main tree-walking function */\n  protected override processElementTree(\n    element: Id64String,\n    scope: ElementTreeWalkerScope\n  ): void {\n    if (this.shouldVisitCodeScopes(element, scope)) {\n      this._processCodeScopes(element, scope);\n    }\n    super.processElementTree(element, scope);\n  }\n  /** Process code scope references */\n  private _processCodeScopes(\n    element: Id64String,\n    scope: ElementTreeWalkerScope\n  ) {\n    const newScope = new ElementTreeWalkerScope(scope, element);\n    // eslint-disable-next-line @itwin/no-internal, @typescript-eslint/no-deprecated\n    this._iModel.withPreparedStatement(\n      `\n      SELECT ECInstanceId\n      FROM bis.Element\n      WHERE CodeScope.id=?\n        AND Parent.id IS NULL\n    `,\n      (stmt) => {\n        stmt.bindId(1, element);\n        while (stmt.step() === DbResult.BE_SQLITE_ROW) {\n          const elementId = stmt.getValue(0).getId();\n          this.processElementTree(elementId, newScope);\n        }\n      }\n    );\n  }\n}\n"]}
+{"version":3,"file":"ElementCascadingDeleter.js","sourceRoot":"","sources":["../../src/ElementCascadingDeleter.ts"],"names":[],"mappings":";;;AAmBA,4DAOC;AA1BD;;;gGAGgG;AAChG;;GAEG;AACH,sDAI6B;AAE7B,oDAAiD;AAEjD;;;GAGG;AACH,SAAgB,wBAAwB,CACtC,OAAgB,EAChB,UAAsB;IAEtB,MAAM,GAAG,GAAG,IAAI,uBAAuB,CAAC,OAAO,CAAC,CAAC;IACjD,GAAG,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC;IACrC,GAAG,CAAC,qBAAqB,EAAE,CAAC;AAC9B,CAAC;AAED;;;GAGG;AACH,MAAa,uBAAwB,SAAQ,iCAAkB;IACnD,qBAAqB,CAC7B,UAAsB,EACtB,MAA8B;QAE9B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,qCAAqC;IAClB,kBAAkB,CACnC,OAAmB,EACnB,KAA6B;QAE7B,IAAI,IAAI,CAAC,qBAAqB,CAAC,OAAO,EAAE,KAAK,CAAC,EAAE,CAAC;YAC/C,IAAI,CAAC,kBAAkB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAC1C,CAAC;QACD,KAAK,CAAC,kBAAkB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;IAC3C,CAAC;IACD,oCAAoC;IAC5B,kBAAkB,CACxB,OAAmB,EACnB,KAA6B;QAE7B,MAAM,QAAQ,GAAG,IAAI,qCAAsB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAC5D,MAAM,KAAK,GAAG;;;;;KAKb,CAAC;QAEF,MAAM,MAAM,GAAG,IAAI,yBAAW,EAAE,CAAC,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAE5D,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,eAAe,CAC7B,KAAK,EACL,CAAC,MAAM,EAAE,EAAE;YACT,KAAK,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;gBACzB,MAAM,SAAS,GAAG,GAAG,CAAC,CAAC,CAAe,CAAC;gBACvC,IAAI,CAAC,kBAAkB,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YAC/C,CAAC;QACH,CAAC,EACD,MAAM,CACP,CAAC;IACJ,CAAC;CACF;AA5CD,0DA4CC","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/** @packageDocumentation\n * @module iModels\n */\nimport {\n  EditTxn,\n  ElementTreeDeleter,\n  ElementTreeWalkerScope,\n} from \"@itwin/core-backend\";\nimport { Id64String } from \"@itwin/core-bentley\";\nimport { QueryBinder } from \"@itwin/core-common\";\n\n/** Deletes an element tree and code scope references starting with the specified top element. The top element is also deleted. Uses ElementCascadeDeleter.\n * @param editTxn The active EditTxn for the iModel\n * @param topElement The parent of the sub-tree\n */\nexport function deleteElementTreeCascade(\n  editTxn: EditTxn,\n  topElement: Id64String\n): void {\n  const del = new ElementCascadingDeleter(editTxn);\n  del.deleteNormalElements(topElement);\n  del.deleteSpecialElements();\n}\n\n/** Deletes an entire element tree, including sub-models, child elements and code scope references.\n * Items are deleted in bottom-up order. Definitions and Subjects are deleted after normal elements.\n * Call deleteNormalElements on each tree. Then call deleteSpecialElements.\n */\nexport class ElementCascadingDeleter extends ElementTreeDeleter {\n  protected shouldVisitCodeScopes(\n    _elementId: Id64String,\n    _scope: ElementTreeWalkerScope\n  ) {\n    return true;\n  }\n\n  /** The main tree-walking function */\n  protected override processElementTree(\n    element: Id64String,\n    scope: ElementTreeWalkerScope\n  ): void {\n    if (this.shouldVisitCodeScopes(element, scope)) {\n      this._processCodeScopes(element, scope);\n    }\n    super.processElementTree(element, scope);\n  }\n  /** Process code scope references */\n  private _processCodeScopes(\n    element: Id64String,\n    scope: ElementTreeWalkerScope\n  ) {\n    const newScope = new ElementTreeWalkerScope(scope, element);\n    const query = `\n      SELECT ECInstanceId\n      FROM bis.Element\n      WHERE CodeScope.id=:scopeId\n        AND Parent.id IS NULL\n    `;\n\n    const params = new QueryBinder().bindId(\"scopeId\", element);\n\n    this.txn.iModel.withQueryReader(\n      query,\n      (reader) => {\n        for (const row of reader) {\n          const elementId = row[0] as Id64String;\n          this.processElementTree(elementId, newScope);\n        }\n      },\n      params\n    );\n  }\n}\n"]}

package/lib/cjs/EntityUnifier.d.ts

@@ -4,18 +4,14 @@
  * for entity-generic operations in the transformer
  */
 import { EntityReference } from "@itwin/core-common";
-import { ConcreteEntity, ConcreteEntityProps, IModelDb } from "@itwin/core-backend";
+import { ConcreteEntity, IModelDb } from "@itwin/core-backend";
 /** @internal */
 export declare namespace EntityUnifier {
-    export function getReadableType(entity: ConcreteEntity): "element" | "element aspect" | "relationship" | "unknown entity type";
-    type EntityUpdater = (entityProps: ConcreteEntityProps) => void;
-    /** needs to return a widened type otherwise typescript complains when result is used with a narrow type */
-    export function updaterFor(db: IModelDb, entity: ConcreteEntity): EntityUpdater;
-    export function exists(db: IModelDb, arg: {
+    function getReadableType(entity: ConcreteEntity): "element" | "element aspect" | "relationship" | "unknown entity type";
+    function exists(db: IModelDb, arg: {
         entity: ConcreteEntity;
     } | {
         entityReference: EntityReference;
-    }): boolean;
-    export {};
+    }): Promise<boolean>;
 }
 //# sourceMappingURL=EntityUnifier.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"EntityUnifier.d.ts","sourceRoot":"","sources":["../../src/EntityUnifier.ts"],"names":[],"mappings":"AAIA;;;;GAIG;AAGH,OAAO,EAEL,eAAe,EAEhB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,cAAc,EACd,mBAAmB,EAKnB,QAAQ,EAET,MAAM,qBAAqB,CAAC;AAG7B,gBAAgB;AAChB,yBAAiB,aAAa,CAAC;IAC7B,MAAM,UAAU,eAAe,CAAC,MAAM,EAAE,cAAc,yEAKrD;IAED,KAAK,aAAa,GAAG,CAAC,WAAW,EAAE,mBAAmB,KAAK,IAAI,CAAC;IAEhE,2GAA2G;IAC3G,MAAM,UAAU,UAAU,CAAC,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,cAAc,iBAc9D;IAED,MAAM,UAAU,MAAM,CACpB,EAAE,EAAE,QAAQ,EACZ,GAAG,EAAE;QAAE,MAAM,EAAE,cAAc,CAAA;KAAE,GAAG;QAAE,eAAe,EAAE,eAAe,CAAA;KAAE,WAyBvE;;CACF"}
+{"version":3,"file":"EntityUnifier.d.ts","sourceRoot":"","sources":["../../src/EntityUnifier.ts"],"names":[],"mappings":"AAIA;;;;GAIG;AAEH,OAAO,EAEL,eAAe,EAEhB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,cAAc,EAKd,QAAQ,EAET,MAAM,qBAAqB,CAAC;AAG7B,gBAAgB;AAChB,yBAAiB,aAAa,CAAC;IAC7B,SAAgB,eAAe,CAAC,MAAM,EAAE,cAAc,yEAKrD;IAED,SAAsB,MAAM,CAC1B,EAAE,EAAE,QAAQ,EACZ,GAAG,EAAE;QAAE,MAAM,EAAE,cAAc,CAAA;KAAE,GAAG;QAAE,eAAe,EAAE,eAAe,CAAA;KAAE,oBAoBvE;CACF"}

package/lib/cjs/EntityUnifier.js

@@ -10,7 +10,6 @@
  */
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.EntityUnifier = void 0;
-const assert = require("assert");
 const core_common_1 = require("@itwin/core-common");
 const core_backend_1 = require("@itwin/core-backend");
 const core_bentley_1 = require("@itwin/core-bentley");
@@ -28,19 +27,7 @@ var EntityUnifier;
             return "unknown entity type";
     }
     EntityUnifier.getReadableType = getReadableType;
-    /** needs to return a widened type otherwise typescript complains when result is used with a narrow type */
-    function updaterFor(db, entity) {
-        if (entity instanceof core_backend_1.Element)
-            return db.elements.updateElement.bind(db.elements);
-        else if (entity instanceof core_backend_1.Relationship)
-            return db.relationships.updateInstance.bind(db.relationships);
-        else if (entity instanceof core_backend_1.ElementAspect)
-            return db.elements.updateAspect.bind(db.elements);
-        else
-            assert(false, `unreachable; entity was '${entity.constructor.name}' not an Element, Relationship, or ElementAspect`);
-    }
-    EntityUnifier.updaterFor = updaterFor;
-    function exists(db, arg) {
+    async function exists(db, arg) {
         const [type, id] = "entityReference" in arg
             ? core_backend_1.EntityReferences.split(arg.entityReference)
             : [undefined, arg.entity.id];
@@ -50,17 +37,12 @@ var EntityUnifier;
             : `[${arg.entity.schemaName}].[${arg.entity.className}]`;
         if (id === undefined || core_bentley_1.Id64.isInvalid(id))
             return false;
-        // eslint-disable-next-line @itwin/no-internal, @typescript-eslint/no-deprecated
-        return db.withPreparedStatement(`SELECT 1 FROM ${classFullName} WHERE ECInstanceId=?`, (stmt) => {
-            stmt.bindId(1, id);
-            const matchesResult = stmt.step();
-            if (matchesResult === core_bentley_1.DbResult.BE_SQLITE_ROW)
-                return true;
-            if (matchesResult === core_bentley_1.DbResult.BE_SQLITE_DONE)
-                return false;
-            else
-                throw new core_common_1.IModelError(matchesResult, "query failed");
+        const query = `SELECT 1 FROM ${classFullName} WHERE ECInstanceId=:id`;
+        const params = new core_common_1.QueryBinder().bindId("id", id);
+        const reader = db.createQueryReader(query, params, {
+            usePrimaryConn: true,
         });
+        return reader.step();
     }
     EntityUnifier.exists = exists;
 })(EntityUnifier || (exports.EntityUnifier = EntityUnifier = {}));