@itwin/imodel-transformer 2.0.0-dev.1 → 2.0.0-dev.11

package/CHANGELOG.md

@@ -1,9 +1,17 @@
 # Change Log - @itwin/imodel-transformer
 
-This log was last generated on Tue, 29 Apr 2025 14:01:13 GMT and should not be manually modified.
+This log was last generated on Mon, 30 Jun 2025 18:36:23 GMT and should not be manually modified.
 
 <!-- Start content -->
 
+## 1.2.0
+
+Mon, 30 Jun 2025 18:36:23 GMT
+
+### Minor changes
+
+- allow option to transform spatial elements of linearly located imodels ([commit](https://github.com/iTwin/imodel-transformer/commit/7971863b75b604e9d5b090d69938d17ef40a2d44))
+
 ## 1.1.3
 
 Tue, 29 Apr 2025 14:01:13 GMT

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

@@ -1 +1 @@
-{"version":3,"file":"BranchProvenanceInitializer.d.ts","sourceRoot":"","sources":["../../src/BranchProvenanceInitializer.ts"],"names":[],"mappings":"AAIA,OAAO,EAIL,QAAQ,EAGT,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAY,UAAU,EAAoB,MAAM,qBAAqB,CAAC;AAS7E;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,8DAA8D;IAC9D,MAAM,EAAE,QAAQ,CAAC;IACjB,6CAA6C;IAC7C,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,2CAA2C;IAC3C,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B;;;OAGG;IACH,MAAM,EAAE,QAAQ,CAAC;IACjB;;;;;;;OAOG;IACH,uBAAuB,CAAC,EAAE,IAAI,GAAG,KAAK,GAAG,kBAAkB,CAAC;CAC7D;AAED;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC,oBAAoB,EAAE,UAAU,CAAC;IACjC,sBAAsB,EAAE,UAAU,CAAC;IACnC,sBAAsB,EAAE,UAAU,CAAC;CACpC;AAED;;GAEG;AACH,wBAAsB,0BAA0B,CAC9C,IAAI,EAAE,kBAAkB,GACvB,OAAO,CAAC,oBAAoB,CAAC,CAsJ/B"}
+{"version":3,"file":"BranchProvenanceInitializer.d.ts","sourceRoot":"","sources":["../../src/BranchProvenanceInitializer.ts"],"names":[],"mappings":"AAIA,OAAO,EAIL,QAAQ,EAGT,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAY,UAAU,EAAoB,MAAM,qBAAqB,CAAC;AAS7E;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,8DAA8D;IAC9D,MAAM,EAAE,QAAQ,CAAC;IACjB,6CAA6C;IAC7C,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,2CAA2C;IAC3C,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B;;;OAGG;IACH,MAAM,EAAE,QAAQ,CAAC;IACjB;;;;;;;OAOG;IACH,uBAAuB,CAAC,EAAE,IAAI,GAAG,KAAK,GAAG,kBAAkB,CAAC;CAC7D;AAED;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC,oBAAoB,EAAE,UAAU,CAAC;IACjC,sBAAsB,EAAE,UAAU,CAAC;IACnC,sBAAsB,EAAE,UAAU,CAAC;CACpC;AAED;;GAEG;AACH,wBAAsB,0BAA0B,CAC9C,IAAI,EAAE,kBAAkB,GACvB,OAAO,CAAC,oBAAoB,CAAC,CAiJ/B"}

package/lib/cjs/BranchProvenanceInitializer.js

@@ -18,7 +18,6 @@ async function initializeBranchProvenance(args) {
     if (args.createFedGuidsForMaster) {
         // FIXME<LOW>: Consider enforcing that the master and branch dbs passed as part of ProvenanceInitArgs to this function
         // are identical. https://github.com/iTwin/imodel-transformer/issues/138
-        /* eslint-disable deprecation/deprecation */
         args.master.withSqliteStatement(`
         UPDATE bis_Element
         SET FederationGuid=randomblob(16)
@@ -51,7 +50,6 @@ async function initializeBranchProvenance(args) {
             // eslint-disable-next-line @itwin/no-internal
             assert(res === core_bentley_1.DbResult.BE_SQLITE_ERROR, args.branch.getLastError());
         });
-        /* eslint-enable deprecation/deprecation */
         args.branch.performCheckpoint();
         const reopenBranch = makeDbReopener(args.branch);
         // close dbs because element cache could be invalid
@@ -76,20 +74,20 @@ async function initializeBranchProvenance(args) {
         model: core_backend_1.IModelDb.rootSubjectId,
         code: core_common_1.Code.createEmpty(),
         repository: new core_backend_1.ExternalSourceIsInRepository(masterRepoLinkId),
-        /* eslint-disable @typescript-eslint/no-var-requires */
+        /* eslint-disable @typescript-eslint/no-require-imports */
         connectorName: require("../../package.json").name,
         connectorVersion: require("../../package.json").version,
-        /* eslint-enable @typescript-eslint/no-var-requires */
+        /* eslint-enable @typescript-eslint/no-require-imports */
     });
-    const fedGuidLessElemsSql = `
+    const fedGuidlessElemsSql = `
     SELECT ECInstanceId AS id
     FROM Bis.Element
     WHERE FederationGuid IS NULL
       AND ECInstanceId NOT IN (0x1, 0xe, 0x10) /* ignore special elems */
   `;
-    const elemReader = args.branch.createQueryReader(fedGuidLessElemsSql, undefined, { usePrimaryConn: true });
-    while (await elemReader.step()) {
-        const id = elemReader.current.toRow().id;
+    const elemReader = args.branch.createQueryReader(fedGuidlessElemsSql, undefined, { usePrimaryConn: true });
+    for await (const row of elemReader) {
+        const id = row.id;
         const aspectProps = IModelTransformer_1.IModelTransformer.initElementProvenanceOptions(id, id, {
             isReverseSynchronization: false,
             targetScopeElementId: masterExternalSourceId,
@@ -98,7 +96,7 @@ async function initializeBranchProvenance(args) {
         });
         args.branch.elements.insertAspect(aspectProps);
     }
-    const fedGuidLessRelsSql = `
+    const fedGuidlessRelsSql = `
     SELECT erte.ECInstanceId as id
     FROM Bis.ElementRefersToElements erte
     JOIN bis.Element se
@@ -107,10 +105,10 @@ async function initializeBranchProvenance(args) {
       ON te.ECInstanceId=erte.TargetECInstanceId
       WHERE se.FederationGuid IS NULL
       OR te.FederationGuid IS NULL`;
-    const relReader = args.branch.createQueryReader(fedGuidLessRelsSql, undefined, { usePrimaryConn: true });
-    while (await relReader.step()) {
-        const id = relReader.current.toRow().id;
-        const aspectProps = IModelTransformer_1.IModelTransformer.initRelationshipProvenanceOptions(id, id, {
+    const relReader = args.branch.createQueryReader(fedGuidlessRelsSql, undefined, { usePrimaryConn: true });
+    for await (const row of relReader) {
+        const id = row.id;
+        const aspectProps = await IModelTransformer_1.IModelTransformer.initRelationshipProvenanceOptions(id, id, {
             isReverseSynchronization: false,
             targetScopeElementId: masterExternalSourceId,
             sourceDb: args.master,

package/lib/cjs/BranchProvenanceInitializer.js.map

@@ -1 +1 @@
-{"version":3,"file":"BranchProvenanceInitializer.js","sourceRoot":"","sources":["../../src/BranchProvenanceInitializer.ts"],"names":[],"mappings":";;AA2DA,gEAwJC;AAnND;;;gGAGgG;AAChG,sDAO6B;AAC7B,sDAA6E;AAC7E,oDAI4B;AAC5B,iCAAiC;AACjC,2DAAwD;AACxD,6BAAoC;AAoCpC;;GAEG;AACI,KAAK,UAAU,0BAA0B,CAC9C,IAAwB;IAExB,IAAI,IAAI,CAAC,uBAAuB,EAAE,CAAC;QACjC,sHAAsH;QACtH,wEAAwE;QACxE,4CAA4C;QAC5C,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAC7B;;;;OAIC;QACD,8CAA8C;QAC9C,CAAC,CAAC,EAAE,EAAE,CACJ,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,KAAK,uBAAQ,CAAC,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC,CAC3E,CAAC;QACF,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;QACxC,MAAM,YAAY,GAAG,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACjD,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,eAAe;QACpC,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAC7B,oBAAoB,IAAA,mBAAa,EAAC,GAAG,UAAU,EAAE,CAAC,qBAAqB;QACvE,8CAA8C;QAC9C,CAAC,CAAC,EAAE,EAAE,CACJ,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,KAAK,uBAAQ,CAAC,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC,CAC3E,CAAC;QACF,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAC7B;;;;;;QAME;QAEF,8CAA8C;QAC9C,CAAC,CAAC,EAAE,EAAE,CACJ,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,KAAK,uBAAQ,CAAC,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC,CAC3E,CAAC;QACF,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,iFAAiF;QAC5G,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;QAC1B,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,wBAAwB,EAAE,CAAC,CAAC,EAAE,EAAE;YAC9D,MAAM,GAAG,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;YACrB,IAAI,GAAG,KAAK,uBAAQ,CAAC,cAAc;gBACjC,qBAAM,CAAC,QAAQ,CACb,4BAA4B,EAC5B,8CAA8C,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,EAAE,CAC3E,CAAC;YACJ,6CAA6C;YAC7C,8CAA8C;YAC9C,MAAM,CAAC,GAAG,KAAK,uBAAQ,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;QACH,2CAA2C;QAC3C,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC;QAEhC,MAAM,YAAY,GAAG,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACjD,mDAAmD;QACnD,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QACpB,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YAC7C,YAAY,EAAE;YACd,YAAY,EAAE;SACf,CAAC,CAAC;IACL,CAAC;IAED,wHAAwH;IACxH,MAAM,gBAAgB,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC;QAC1D,aAAa,EAAE,6BAAc,CAAC,aAAa;QAC3C,IAAI,EAAE,6BAAc,CAAC,UAAU,CAC7B,IAAI,CAAC,MAAM,EACX,uBAAQ,CAAC,iBAAiB,EAC1B,oBAAoB,CACrB;QACD,KAAK,EAAE,uBAAQ,CAAC,iBAAiB;QACjC,GAAG,EAAE,IAAI,CAAC,SAAS;QACnB,MAAM,EAAE,QAAQ;QAChB,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;QACpC,WAAW,EAAE,IAAI,CAAC,iBAAiB;KACb,CAAC,CAAC;IAE1B,MAAM,sBAAsB,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC;QAChE,aAAa,EAAE,6BAAc,CAAC,aAAa;QAC3C,KAAK,EAAE,uBAAQ,CAAC,aAAa;QAC7B,IAAI,EAAE,kBAAI,CAAC,WAAW,EAAE;QACxB,UAAU,EAAE,IAAI,2CAA4B,CAAC,gBAAgB,CAAC;QAC9D,uDAAuD;QACvD,aAAa,EAAE,OAAO,CAAC,oBAAoB,CAAC,CAAC,IAAI;QACjD,gBAAgB,EAAE,OAAO,CAAC,oBAAoB,CAAC,CAAC,OAAO;QACvD,sDAAsD;KAChC,CAAC,CAAC;IAE1B,MAAM,mBAAmB,GAAG;;;;;GAK3B,CAAC;IACF,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAC9C,mBAAmB,EACnB,SAAS,EACT,EAAE,cAAc,EAAE,IAAI,EAAE,CACzB,CAAC;IACF,OAAO,MAAM,UAAU,CAAC,IAAI,EAAE,EAAE,CAAC;QAC/B,MAAM,EAAE,GAAW,UAAU,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,EAAE,CAAC;QACjD,MAAM,WAAW,GAAG,qCAAiB,CAAC,4BAA4B,CAAC,EAAE,EAAE,EAAE,EAAE;YACzE,wBAAwB,EAAE,KAAK;YAC/B,oBAAoB,EAAE,sBAAsB;YAC5C,QAAQ,EAAE,IAAI,CAAC,MAAM;YACrB,QAAQ,EAAE,IAAI,CAAC,MAAM;SACtB,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC;IACjD,CAAC;IAED,MAAM,kBAAkB,GAAG;;;;;;;;mCAQM,CAAC;IAClC,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAC7C,kBAAkB,EAClB,SAAS,EACT,EAAE,cAAc,EAAE,IAAI,EAAE,CACzB,CAAC;IACF,OAAO,MAAM,SAAS,CAAC,IAAI,EAAE,EAAE,CAAC;QAC9B,MAAM,EAAE,GAAW,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,EAAE,CAAC;QAChD,MAAM,WAAW,GAAG,qCAAiB,CAAC,iCAAiC,CACrE,EAAE,EACF,EAAE,EACF;YACE,wBAAwB,EAAE,KAAK;YAC/B,oBAAoB,EAAE,sBAAsB;YAC5C,QAAQ,EAAE,IAAI,CAAC,MAAM;YACrB,QAAQ,EAAE,IAAI,CAAC,MAAM;YACrB,oCAAoC,EAAE,KAAK;SAC5C,CACF,CAAC;QACF,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC;IACjD,CAAC;IAED,IAAI,IAAI,CAAC,uBAAuB,KAAK,IAAI,EAAE,CAAC;QAC1C,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QACpB,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC;IAED,OAAO;QACL,oBAAoB,EAAE,sBAAsB;QAC5C,sBAAsB;QACtB,sBAAsB,EAAE,gBAAgB;KACzC,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CAAC,EAAY;IAClC,MAAM,YAAY,GAAG,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC,uBAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,uBAAQ,CAAC,SAAS,CAAC;IAC5E,MAAM,MAAM,GAAG,EAAE,CAAC,QAAQ,CAAC;IAC3B,IAAI,QAA2D,CAAC;IAChE,IAAI,EAAE,YAAY,0BAAW;QAC3B,QAAQ,GAAG,KAAK,EAAE,IAAI,GAAG,YAAY,EAAE,EAAE,CACvC,0BAAW,CAAC,IAAI,CAAC;YACf,QAAQ,EAAE,MAAM;YAChB,QAAQ,EAAE,IAAI,KAAK,uBAAQ,CAAC,QAAQ;SACrC,CAAC,CAAC;SACF,IAAI,EAAE,YAAY,2BAAY;QACjC,QAAQ,GAAG,CAAC,IAAI,GAAG,YAAY,EAAE,EAAE,CAAC,2BAAY,CAAC,QAAQ,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;;QACrE,MAAM,CAAC,KAAK,EAAE,YAAY,EAAE,CAAC,WAAW,CAAC,IAAI,iBAAiB,CAAC,CAAC;IACrE,OAAO,QAAQ,CAAC;AAClB,CAAC","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 *--------------------------------------------------------------------------------------------*/\nimport {\n  BriefcaseDb,\n  ExternalSource,\n  ExternalSourceIsInRepository,\n  IModelDb,\n  RepositoryLink,\n  StandaloneDb,\n} from \"@itwin/core-backend\";\nimport { DbResult, Id64String, Logger, OpenMode } from \"@itwin/core-bentley\";\nimport {\n  Code,\n  ExternalSourceProps,\n  RepositoryLinkProps,\n} from \"@itwin/core-common\";\nimport * as assert from \"assert\";\nimport { IModelTransformer } from \"./IModelTransformer\";\nimport { pathToFileURL } from \"url\";\n/**\n * @alpha\n */\nexport interface ProvenanceInitArgs {\n  /** the master iModel which is the source of the provenance */\n  master: IModelDb;\n  /** the canonical url of the master iModel */\n  masterUrl?: string;\n  /** the description of the master iModel */\n  masterDescription?: string;\n  /**\n   * @param {IModelDb} branchDb - the branch iModel which is the container of the provenance\n   *                              Must be opened Read/Write\n   */\n  branch: IModelDb;\n  /**\n   * insert Federation Guids in all lacking elements in the master database, which will prevent\n   * needing to insert External Source Aspects for provenance tracking\n   * @note requires a read/write master\n   * @note closes both the master and branch iModels to reset caches, so you must reopen them.\n   *       If you pass `\"keep-reopened-db\"`, this object's `master` and `branch` properties will\n   *       be set to new, open databases.\n   */\n  createFedGuidsForMaster?: true | false | \"keep-reopened-db\";\n}\n\n/**\n * @alpha\n */\nexport interface ProvenanceInitResult {\n  targetScopeElementId: Id64String;\n  masterExternalSourceId: Id64String;\n  masterRepositoryLinkId: Id64String;\n}\n\n/**\n * @alpha\n */\nexport async function initializeBranchProvenance(\n  args: ProvenanceInitArgs\n): Promise<ProvenanceInitResult> {\n  if (args.createFedGuidsForMaster) {\n    // FIXME<LOW>: Consider enforcing that the master and branch dbs passed as part of ProvenanceInitArgs to this function\n    // are identical. https://github.com/iTwin/imodel-transformer/issues/138\n    /* eslint-disable deprecation/deprecation */\n    args.master.withSqliteStatement(\n      `\n        UPDATE bis_Element\n        SET FederationGuid=randomblob(16)\n        WHERE FederationGuid IS NULL\n      `,\n      // eslint-disable-next-line @itwin/no-internal\n      (s) =>\n        assert(s.step() === DbResult.BE_SQLITE_DONE, args.branch.getLastError())\n    );\n    const masterPath = args.master.pathName;\n    const reopenMaster = makeDbReopener(args.master);\n    args.master.close(); // prevent busy\n    args.branch.withSqliteStatement(\n      `ATTACH DATABASE '${pathToFileURL(`${masterPath}`)}?mode=ro' AS master`,\n      // eslint-disable-next-line @itwin/no-internal\n      (s) =>\n        assert(s.step() === DbResult.BE_SQLITE_DONE, args.branch.getLastError())\n    );\n    args.branch.withSqliteStatement(\n      `\n      UPDATE main.bis_Element\n      SET FederationGuid = (\n        SELECT m.FederationGuid\n        FROM master.bis_Element m\n        WHERE m.Id=main.bis_Element.Id\n      )`,\n\n      // eslint-disable-next-line @itwin/no-internal\n      (s) =>\n        assert(s.step() === DbResult.BE_SQLITE_DONE, args.branch.getLastError())\n    );\n    args.branch.clearCaches(); // statements write lock attached db (clearing statement cache does not fix this)\n    args.branch.saveChanges();\n    args.branch.withSqliteStatement(\"DETACH DATABASE master\", (s) => {\n      const res = s.step();\n      if (res !== DbResult.BE_SQLITE_DONE)\n        Logger.logTrace(\n          \"initializeBranchProvenance\",\n          `Error detaching db (we will close anyway): ${args.branch.getLastError()}`\n        );\n      // this is the case until native side changes\n      // eslint-disable-next-line @itwin/no-internal\n      assert(res === DbResult.BE_SQLITE_ERROR, args.branch.getLastError());\n    });\n    /* eslint-enable deprecation/deprecation */\n    args.branch.performCheckpoint();\n\n    const reopenBranch = makeDbReopener(args.branch);\n    // close dbs because element cache could be invalid\n    args.branch.close();\n    [args.master, args.branch] = await Promise.all([\n      reopenMaster(),\n      reopenBranch(),\n    ]);\n  }\n\n  // create an external source and owning repository link to use as our *Target Scope Element* for future synchronizations\n  const masterRepoLinkId = args.branch.elements.insertElement({\n    classFullName: RepositoryLink.classFullName,\n    code: RepositoryLink.createCode(\n      args.branch,\n      IModelDb.repositoryModelId,\n      \"example-code-value\"\n    ),\n    model: IModelDb.repositoryModelId,\n    url: args.masterUrl,\n    format: \"iModel\",\n    repositoryGuid: args.master.iModelId,\n    description: args.masterDescription,\n  } as RepositoryLinkProps);\n\n  const masterExternalSourceId = args.branch.elements.insertElement({\n    classFullName: ExternalSource.classFullName,\n    model: IModelDb.rootSubjectId,\n    code: Code.createEmpty(),\n    repository: new ExternalSourceIsInRepository(masterRepoLinkId),\n    /* eslint-disable @typescript-eslint/no-var-requires */\n    connectorName: require(\"../../package.json\").name,\n    connectorVersion: require(\"../../package.json\").version,\n    /* eslint-enable @typescript-eslint/no-var-requires */\n  } as ExternalSourceProps);\n\n  const fedGuidLessElemsSql = `\n    SELECT ECInstanceId AS id\n    FROM Bis.Element\n    WHERE FederationGuid IS NULL\n      AND ECInstanceId NOT IN (0x1, 0xe, 0x10) /* ignore special elems */\n  `;\n  const elemReader = args.branch.createQueryReader(\n    fedGuidLessElemsSql,\n    undefined,\n    { usePrimaryConn: true }\n  );\n  while (await elemReader.step()) {\n    const id: string = elemReader.current.toRow().id;\n    const aspectProps = IModelTransformer.initElementProvenanceOptions(id, id, {\n      isReverseSynchronization: false,\n      targetScopeElementId: masterExternalSourceId,\n      sourceDb: args.master,\n      targetDb: args.branch,\n    });\n    args.branch.elements.insertAspect(aspectProps);\n  }\n\n  const fedGuidLessRelsSql = `\n    SELECT erte.ECInstanceId as id\n    FROM Bis.ElementRefersToElements erte\n    JOIN bis.Element se\n      ON se.ECInstanceId=erte.SourceECInstanceId\n    JOIN bis.Element te\n      ON te.ECInstanceId=erte.TargetECInstanceId\n      WHERE se.FederationGuid IS NULL\n      OR te.FederationGuid IS NULL`;\n  const relReader = args.branch.createQueryReader(\n    fedGuidLessRelsSql,\n    undefined,\n    { usePrimaryConn: true }\n  );\n  while (await relReader.step()) {\n    const id: string = relReader.current.toRow().id;\n    const aspectProps = IModelTransformer.initRelationshipProvenanceOptions(\n      id,\n      id,\n      {\n        isReverseSynchronization: false,\n        targetScopeElementId: masterExternalSourceId,\n        sourceDb: args.master,\n        targetDb: args.branch,\n        forceOldRelationshipProvenanceMethod: false,\n      }\n    );\n    args.branch.elements.insertAspect(aspectProps);\n  }\n\n  if (args.createFedGuidsForMaster === true) {\n    args.master.close();\n    args.branch.close();\n  }\n\n  return {\n    targetScopeElementId: masterExternalSourceId,\n    masterExternalSourceId,\n    masterRepositoryLinkId: masterRepoLinkId,\n  };\n}\n\nfunction makeDbReopener(db: IModelDb) {\n  const originalMode = db.isReadonly ? OpenMode.Readonly : OpenMode.ReadWrite;\n  const dbPath = db.pathName;\n  let reopenDb: (mode?: OpenMode) => IModelDb | Promise<IModelDb>;\n  if (db instanceof BriefcaseDb)\n    reopenDb = async (mode = originalMode) =>\n      BriefcaseDb.open({\n        fileName: dbPath,\n        readonly: mode === OpenMode.Readonly,\n      });\n  else if (db instanceof StandaloneDb)\n    reopenDb = (mode = originalMode) => StandaloneDb.openFile(dbPath, mode);\n  else assert(false, `db type '${db.constructor.name}' not supported`);\n  return reopenDb;\n}\n"]}
+{"version":3,"file":"BranchProvenanceInitializer.js","sourceRoot":"","sources":["../../src/BranchProvenanceInitializer.ts"],"names":[],"mappings":";;AA2DA,gEAmJC;AA9MD;;;gGAGgG;AAChG,sDAO6B;AAC7B,sDAA6E;AAC7E,oDAI4B;AAC5B,iCAAiC;AACjC,2DAAwD;AACxD,6BAAoC;AAoCpC;;GAEG;AACI,KAAK,UAAU,0BAA0B,CAC9C,IAAwB;IAExB,IAAI,IAAI,CAAC,uBAAuB,EAAE,CAAC;QACjC,sHAAsH;QACtH,wEAAwE;QACxE,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAC7B;;;;OAIC;QACD,8CAA8C;QAC9C,CAAC,CAAC,EAAE,EAAE,CACJ,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,KAAK,uBAAQ,CAAC,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC,CAC3E,CAAC;QACF,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;QACxC,MAAM,YAAY,GAAG,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACjD,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,eAAe;QACpC,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAC7B,oBAAoB,IAAA,mBAAa,EAAC,GAAG,UAAU,EAAE,CAAC,qBAAqB;QACvE,8CAA8C;QAC9C,CAAC,CAAC,EAAE,EAAE,CACJ,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,KAAK,uBAAQ,CAAC,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC,CAC3E,CAAC;QACF,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAC7B;;;;;;QAME;QAEF,8CAA8C;QAC9C,CAAC,CAAC,EAAE,EAAE,CACJ,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,KAAK,uBAAQ,CAAC,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC,CAC3E,CAAC;QACF,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,iFAAiF;QAC5G,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;QAC1B,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,wBAAwB,EAAE,CAAC,CAAC,EAAE,EAAE;YAC9D,MAAM,GAAG,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;YACrB,IAAI,GAAG,KAAK,uBAAQ,CAAC,cAAc;gBACjC,qBAAM,CAAC,QAAQ,CACb,4BAA4B,EAC5B,8CAA8C,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,EAAE,CAC3E,CAAC;YACJ,6CAA6C;YAC7C,8CAA8C;YAC9C,MAAM,CAAC,GAAG,KAAK,uBAAQ,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC;QAEhC,MAAM,YAAY,GAAG,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACjD,mDAAmD;QACnD,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QACpB,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YAC7C,YAAY,EAAE;YACd,YAAY,EAAE;SACf,CAAC,CAAC;IACL,CAAC;IAED,wHAAwH;IACxH,MAAM,gBAAgB,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC;QAC1D,aAAa,EAAE,6BAAc,CAAC,aAAa;QAC3C,IAAI,EAAE,6BAAc,CAAC,UAAU,CAC7B,IAAI,CAAC,MAAM,EACX,uBAAQ,CAAC,iBAAiB,EAC1B,oBAAoB,CACrB;QACD,KAAK,EAAE,uBAAQ,CAAC,iBAAiB;QACjC,GAAG,EAAE,IAAI,CAAC,SAAS;QACnB,MAAM,EAAE,QAAQ;QAChB,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;QACpC,WAAW,EAAE,IAAI,CAAC,iBAAiB;KACb,CAAC,CAAC;IAE1B,MAAM,sBAAsB,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC;QAChE,aAAa,EAAE,6BAAc,CAAC,aAAa;QAC3C,KAAK,EAAE,uBAAQ,CAAC,aAAa;QAC7B,IAAI,EAAE,kBAAI,CAAC,WAAW,EAAE;QACxB,UAAU,EAAE,IAAI,2CAA4B,CAAC,gBAAgB,CAAC;QAC9D,0DAA0D;QAC1D,aAAa,EAAE,OAAO,CAAC,oBAAoB,CAAC,CAAC,IAAI;QACjD,gBAAgB,EAAE,OAAO,CAAC,oBAAoB,CAAC,CAAC,OAAO;QACvD,yDAAyD;KACnC,CAAC,CAAC;IAE1B,MAAM,mBAAmB,GAAG;;;;;GAK3B,CAAC;IACF,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAC9C,mBAAmB,EACnB,SAAS,EACT,EAAE,cAAc,EAAE,IAAI,EAAE,CACzB,CAAC;IACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,UAAU,EAAE,CAAC;QACnC,MAAM,EAAE,GAAW,GAAG,CAAC,EAAE,CAAC;QAC1B,MAAM,WAAW,GAAG,qCAAiB,CAAC,4BAA4B,CAAC,EAAE,EAAE,EAAE,EAAE;YACzE,wBAAwB,EAAE,KAAK;YAC/B,oBAAoB,EAAE,sBAAsB;YAC5C,QAAQ,EAAE,IAAI,CAAC,MAAM;YACrB,QAAQ,EAAE,IAAI,CAAC,MAAM;SACtB,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC;IACjD,CAAC;IAED,MAAM,kBAAkB,GAAG;;;;;;;;mCAQM,CAAC;IAClC,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAC7C,kBAAkB,EAClB,SAAS,EACT,EAAE,cAAc,EAAE,IAAI,EAAE,CACzB,CAAC;IACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,SAAS,EAAE,CAAC;QAClC,MAAM,EAAE,GAAW,GAAG,CAAC,EAAE,CAAC;QAC1B,MAAM,WAAW,GACf,MAAM,qCAAiB,CAAC,iCAAiC,CAAC,EAAE,EAAE,EAAE,EAAE;YAChE,wBAAwB,EAAE,KAAK;YAC/B,oBAAoB,EAAE,sBAAsB;YAC5C,QAAQ,EAAE,IAAI,CAAC,MAAM;YACrB,QAAQ,EAAE,IAAI,CAAC,MAAM;YACrB,oCAAoC,EAAE,KAAK;SAC5C,CAAC,CAAC;QACL,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC;IACjD,CAAC;IAED,IAAI,IAAI,CAAC,uBAAuB,KAAK,IAAI,EAAE,CAAC;QAC1C,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QACpB,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC;IAED,OAAO;QACL,oBAAoB,EAAE,sBAAsB;QAC5C,sBAAsB;QACtB,sBAAsB,EAAE,gBAAgB;KACzC,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CAAC,EAAY;IAClC,MAAM,YAAY,GAAG,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC,uBAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,uBAAQ,CAAC,SAAS,CAAC;IAC5E,MAAM,MAAM,GAAG,EAAE,CAAC,QAAQ,CAAC;IAC3B,IAAI,QAA2D,CAAC;IAChE,IAAI,EAAE,YAAY,0BAAW;QAC3B,QAAQ,GAAG,KAAK,EAAE,IAAI,GAAG,YAAY,EAAE,EAAE,CACvC,0BAAW,CAAC,IAAI,CAAC;YACf,QAAQ,EAAE,MAAM;YAChB,QAAQ,EAAE,IAAI,KAAK,uBAAQ,CAAC,QAAQ;SACrC,CAAC,CAAC;SACF,IAAI,EAAE,YAAY,2BAAY;QACjC,QAAQ,GAAG,CAAC,IAAI,GAAG,YAAY,EAAE,EAAE,CAAC,2BAAY,CAAC,QAAQ,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;;QACrE,MAAM,CAAC,KAAK,EAAE,YAAY,EAAE,CAAC,WAAW,CAAC,IAAI,iBAAiB,CAAC,CAAC;IACrE,OAAO,QAAQ,CAAC;AAClB,CAAC","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 *--------------------------------------------------------------------------------------------*/\nimport {\n  BriefcaseDb,\n  ExternalSource,\n  ExternalSourceIsInRepository,\n  IModelDb,\n  RepositoryLink,\n  StandaloneDb,\n} from \"@itwin/core-backend\";\nimport { DbResult, Id64String, Logger, OpenMode } from \"@itwin/core-bentley\";\nimport {\n  Code,\n  ExternalSourceProps,\n  RepositoryLinkProps,\n} from \"@itwin/core-common\";\nimport * as assert from \"assert\";\nimport { IModelTransformer } from \"./IModelTransformer\";\nimport { pathToFileURL } from \"url\";\n/**\n * @alpha\n */\nexport interface ProvenanceInitArgs {\n  /** the master iModel which is the source of the provenance */\n  master: IModelDb;\n  /** the canonical url of the master iModel */\n  masterUrl?: string;\n  /** the description of the master iModel */\n  masterDescription?: string;\n  /**\n   * @param {IModelDb} branchDb - the branch iModel which is the container of the provenance\n   *                              Must be opened Read/Write\n   */\n  branch: IModelDb;\n  /**\n   * insert Federation Guids in all lacking elements in the master database, which will prevent\n   * needing to insert External Source Aspects for provenance tracking\n   * @note requires a read/write master\n   * @note closes both the master and branch iModels to reset caches, so you must reopen them.\n   *       If you pass `\"keep-reopened-db\"`, this object's `master` and `branch` properties will\n   *       be set to new, open databases.\n   */\n  createFedGuidsForMaster?: true | false | \"keep-reopened-db\";\n}\n\n/**\n * @alpha\n */\nexport interface ProvenanceInitResult {\n  targetScopeElementId: Id64String;\n  masterExternalSourceId: Id64String;\n  masterRepositoryLinkId: Id64String;\n}\n\n/**\n * @alpha\n */\nexport async function initializeBranchProvenance(\n  args: ProvenanceInitArgs\n): Promise<ProvenanceInitResult> {\n  if (args.createFedGuidsForMaster) {\n    // FIXME<LOW>: Consider enforcing that the master and branch dbs passed as part of ProvenanceInitArgs to this function\n    // are identical. https://github.com/iTwin/imodel-transformer/issues/138\n    args.master.withSqliteStatement(\n      `\n        UPDATE bis_Element\n        SET FederationGuid=randomblob(16)\n        WHERE FederationGuid IS NULL\n      `,\n      // eslint-disable-next-line @itwin/no-internal\n      (s) =>\n        assert(s.step() === DbResult.BE_SQLITE_DONE, args.branch.getLastError())\n    );\n    const masterPath = args.master.pathName;\n    const reopenMaster = makeDbReopener(args.master);\n    args.master.close(); // prevent busy\n    args.branch.withSqliteStatement(\n      `ATTACH DATABASE '${pathToFileURL(`${masterPath}`)}?mode=ro' AS master`,\n      // eslint-disable-next-line @itwin/no-internal\n      (s) =>\n        assert(s.step() === DbResult.BE_SQLITE_DONE, args.branch.getLastError())\n    );\n    args.branch.withSqliteStatement(\n      `\n      UPDATE main.bis_Element\n      SET FederationGuid = (\n        SELECT m.FederationGuid\n        FROM master.bis_Element m\n        WHERE m.Id=main.bis_Element.Id\n      )`,\n\n      // eslint-disable-next-line @itwin/no-internal\n      (s) =>\n        assert(s.step() === DbResult.BE_SQLITE_DONE, args.branch.getLastError())\n    );\n    args.branch.clearCaches(); // statements write lock attached db (clearing statement cache does not fix this)\n    args.branch.saveChanges();\n    args.branch.withSqliteStatement(\"DETACH DATABASE master\", (s) => {\n      const res = s.step();\n      if (res !== DbResult.BE_SQLITE_DONE)\n        Logger.logTrace(\n          \"initializeBranchProvenance\",\n          `Error detaching db (we will close anyway): ${args.branch.getLastError()}`\n        );\n      // this is the case until native side changes\n      // eslint-disable-next-line @itwin/no-internal\n      assert(res === DbResult.BE_SQLITE_ERROR, args.branch.getLastError());\n    });\n    args.branch.performCheckpoint();\n\n    const reopenBranch = makeDbReopener(args.branch);\n    // close dbs because element cache could be invalid\n    args.branch.close();\n    [args.master, args.branch] = await Promise.all([\n      reopenMaster(),\n      reopenBranch(),\n    ]);\n  }\n\n  // create an external source and owning repository link to use as our *Target Scope Element* for future synchronizations\n  const masterRepoLinkId = args.branch.elements.insertElement({\n    classFullName: RepositoryLink.classFullName,\n    code: RepositoryLink.createCode(\n      args.branch,\n      IModelDb.repositoryModelId,\n      \"example-code-value\"\n    ),\n    model: IModelDb.repositoryModelId,\n    url: args.masterUrl,\n    format: \"iModel\",\n    repositoryGuid: args.master.iModelId,\n    description: args.masterDescription,\n  } as RepositoryLinkProps);\n\n  const masterExternalSourceId = args.branch.elements.insertElement({\n    classFullName: ExternalSource.classFullName,\n    model: IModelDb.rootSubjectId,\n    code: Code.createEmpty(),\n    repository: new ExternalSourceIsInRepository(masterRepoLinkId),\n    /* eslint-disable @typescript-eslint/no-require-imports */\n    connectorName: require(\"../../package.json\").name,\n    connectorVersion: require(\"../../package.json\").version,\n    /* eslint-enable @typescript-eslint/no-require-imports */\n  } as ExternalSourceProps);\n\n  const fedGuidlessElemsSql = `\n    SELECT ECInstanceId AS id\n    FROM Bis.Element\n    WHERE FederationGuid IS NULL\n      AND ECInstanceId NOT IN (0x1, 0xe, 0x10) /* ignore special elems */\n  `;\n  const elemReader = args.branch.createQueryReader(\n    fedGuidlessElemsSql,\n    undefined,\n    { usePrimaryConn: true }\n  );\n  for await (const row of elemReader) {\n    const id: string = row.id;\n    const aspectProps = IModelTransformer.initElementProvenanceOptions(id, id, {\n      isReverseSynchronization: false,\n      targetScopeElementId: masterExternalSourceId,\n      sourceDb: args.master,\n      targetDb: args.branch,\n    });\n    args.branch.elements.insertAspect(aspectProps);\n  }\n\n  const fedGuidlessRelsSql = `\n    SELECT erte.ECInstanceId as id\n    FROM Bis.ElementRefersToElements erte\n    JOIN bis.Element se\n      ON se.ECInstanceId=erte.SourceECInstanceId\n    JOIN bis.Element te\n      ON te.ECInstanceId=erte.TargetECInstanceId\n      WHERE se.FederationGuid IS NULL\n      OR te.FederationGuid IS NULL`;\n  const relReader = args.branch.createQueryReader(\n    fedGuidlessRelsSql,\n    undefined,\n    { usePrimaryConn: true }\n  );\n  for await (const row of relReader) {\n    const id: string = row.id;\n    const aspectProps =\n      await IModelTransformer.initRelationshipProvenanceOptions(id, id, {\n        isReverseSynchronization: false,\n        targetScopeElementId: masterExternalSourceId,\n        sourceDb: args.master,\n        targetDb: args.branch,\n        forceOldRelationshipProvenanceMethod: false,\n      });\n    args.branch.elements.insertAspect(aspectProps);\n  }\n\n  if (args.createFedGuidsForMaster === true) {\n    args.master.close();\n    args.branch.close();\n  }\n\n  return {\n    targetScopeElementId: masterExternalSourceId,\n    masterExternalSourceId,\n    masterRepositoryLinkId: masterRepoLinkId,\n  };\n}\n\nfunction makeDbReopener(db: IModelDb) {\n  const originalMode = db.isReadonly ? OpenMode.Readonly : OpenMode.ReadWrite;\n  const dbPath = db.pathName;\n  let reopenDb: (mode?: OpenMode) => IModelDb | Promise<IModelDb>;\n  if (db instanceof BriefcaseDb)\n    reopenDb = async (mode = originalMode) =>\n      BriefcaseDb.open({\n        fileName: dbPath,\n        readonly: mode === OpenMode.Readonly,\n      });\n  else if (db instanceof StandaloneDb)\n    reopenDb = (mode = originalMode) => StandaloneDb.openFile(dbPath, mode);\n  else assert(false, `db type '${db.constructor.name}' not supported`);\n  return reopenDb;\n}\n"]}

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

@@ -1 +1 @@
-{"version":3,"file":"DetachedExportElementAspectsStrategy.d.ts","sourceRoot":"","sources":["../../src/DetachedExportElementAspectsStrategy.ts"],"names":[],"mappings":"AAWA,OAAO,EAAE,4BAA4B,EAAE,MAAM,gCAAgC,CAAC;AAQ9E;;;;;;;GAOG;AACH,qBAAa,oCAAqC,SAAQ,4BAA4B;IAC9D,uBAAuB,IAAI,OAAO,CAAC,IAAI,CAAC;YA+ChD,iBAAiB;YAahB,YAAY;IA8DL,8BAA8B,CAClD,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,IAAI,CAAC;CAGjB"}
+{"version":3,"file":"DetachedExportElementAspectsStrategy.d.ts","sourceRoot":"","sources":["../../src/DetachedExportElementAspectsStrategy.ts"],"names":[],"mappings":"AAWA,OAAO,EAAE,4BAA4B,EAAE,MAAM,gCAAgC,CAAC;AAQ9E;;;;;;;GAOG;AACH,qBAAa,oCAAqC,SAAQ,4BAA4B;IAC9D,uBAAuB,IAAI,OAAO,CAAC,IAAI,CAAC;YAsDhD,iBAAiB;YAahB,YAAY;IA+DL,8BAA8B,CAClD,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,IAAI,CAAC;CAGjB"}

package/lib/cjs/DetachedExportElementAspectsStrategy.js

@@ -25,7 +25,7 @@ class DetachedExportElementAspectsStrategy extends ExportElementAspectsStrategy_
             const doExport = this.aspectChanges === undefined || isInsertChange || isUpdateChange;
             if (doExport) {
                 const isKnownUpdate = this.aspectChanges ? isUpdateChange : undefined;
-                this.handler.onExportElementUniqueAspect(uniqueAspect, isKnownUpdate);
+                await this.handler.onExportElementUniqueAspect(uniqueAspect, isKnownUpdate);
                 await this.handler.trackProgress();
             }
         });
@@ -37,7 +37,7 @@ class DetachedExportElementAspectsStrategy extends ExportElementAspectsStrategy_
             }
             // element id changed so all element's aspects are in the array and can be exported
             if (batchedElementMultiAspects[0].element.id !== multiAspect.element.id) {
-                this.handler.onExportElementMultiAspects(batchedElementMultiAspects);
+                await this.handler.onExportElementMultiAspects(batchedElementMultiAspects);
                 await this.handler.trackProgress();
                 batchedElementMultiAspects = [];
             }
@@ -45,7 +45,7 @@ class DetachedExportElementAspectsStrategy extends ExportElementAspectsStrategy_
         });
         if (batchedElementMultiAspects.length > 0) {
             // aspects that are left in the array have not been exported
-            this.handler.onExportElementMultiAspects(batchedElementMultiAspects);
+            await this.handler.onExportElementMultiAspects(batchedElementMultiAspects);
             await this.handler.trackProgress();
         }
     }
@@ -65,7 +65,7 @@ class DetachedExportElementAspectsStrategy extends ExportElementAspectsStrategy_
       JOIN ECDbMeta.ECClassDef c ON c.ECInstanceId = r.SourceECInstanceId
       WHERE r.TargetECInstanceId = ec_classId(:baseClassName)
     `;
-        const aspectClassesQueryReader = this.sourceDb.createQueryReader(optimizesAspectClassesSql, new core_common_1.QueryBinder().bindString("baseClassName", baseElementAspectClassFullName));
+        const aspectClassesQueryReader = this.sourceDb.createQueryReader(optimizesAspectClassesSql, new core_common_1.QueryBinder().bindString("baseClassName", baseElementAspectClassFullName), { usePrimaryConn: true });
         const aspectClassesAsyncQueryReader = (0, ECSqlReaderAsyncIterableIteratorAdapter_1.ensureECSqlReaderIsAsyncIterableIterator)(aspectClassesQueryReader);
         for await (const rowProxy of aspectClassesAsyncQueryReader) {
             const row = rowProxy.toRow();
@@ -79,7 +79,7 @@ class DetachedExportElementAspectsStrategy extends ExportElementAspectsStrategy_
             if (this.excludedElementAspectClassFullNames.has(classFullName))
                 continue;
             const getAspectPropsSql = `SELECT * FROM [${schemaName}]:[${className}] WHERE ECClassId = :classId ORDER BY Element.Id`;
-            const aspectQueryReader = this.sourceDb.createQueryReader(getAspectPropsSql, new core_common_1.QueryBinder().bindId("classId", classId), { rowFormat: core_common_1.QueryRowFormat.UseJsPropertyNames });
+            const aspectQueryReader = this.sourceDb.createQueryReader(getAspectPropsSql, new core_common_1.QueryBinder().bindId("classId", classId), { rowFormat: core_common_1.QueryRowFormat.UseJsPropertyNames, usePrimaryConn: true });
             const aspectAsyncQueryReader = (0, ECSqlReaderAsyncIterableIteratorAdapter_1.ensureECSqlReaderIsAsyncIterableIterator)(aspectQueryReader);
             let firstDone = false;
             for await (const rowProxy of aspectAsyncQueryReader) {

package/lib/cjs/DetachedExportElementAspectsStrategy.js.map

@@ -1 +1 @@
-{"version":3,"file":"DetachedExportElementAspectsStrategy.js","sourceRoot":"","sources":["../../src/DetachedExportElementAspectsStrategy.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG,sDAI6B;AAE7B,iFAA8E;AAC9E,uGAAqG;AACrG,oDAI4B;AAE5B;;;;;;;GAOG;AACH,MAAa,oCAAqC,SAAQ,2DAA4B;IACpE,KAAK,CAAC,uBAAuB;QAC3C,MAAM,IAAI,CAAC,iBAAiB,CAC1B,kCAAmB,CAAC,aAAa,EACjC,KAAK,EAAE,YAAY,EAAE,EAAE;YACrB,MAAM,cAAc,GAClB,IAAI,CAAC,aAAa,EAAE,SAAS,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC,IAAI,KAAK,CAAC;YAC9D,MAAM,cAAc,GAClB,IAAI,CAAC,aAAa,EAAE,SAAS,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC,IAAI,KAAK,CAAC;YAC9D,MAAM,QAAQ,GACZ,IAAI,CAAC,aAAa,KAAK,SAAS,IAAI,cAAc,IAAI,cAAc,CAAC;YACvE,IAAI,QAAQ,EAAE,CAAC;gBACb,MAAM,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,SAAS,CAAC;gBACtE,IAAI,CAAC,OAAO,CAAC,2BAA2B,CAAC,YAAY,EAAE,aAAa,CAAC,CAAC;gBACtE,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;YACrC,CAAC;QACH,CAAC,CACF,CAAC;QAEF,IAAI,0BAA0B,GAAyB,EAAE,CAAC;QAC1D,MAAM,IAAI,CAAC,iBAAiB,CAC1B,iCAAkB,CAAC,aAAa,EAChC,KAAK,EAAE,WAAW,EAAE,EAAE;YACpB,IAAI,0BAA0B,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC5C,0BAA0B,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;gBAC7C,OAAO;YACT,CAAC;YAED,mFAAmF;YACnF,IACE,0BAA0B,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,WAAW,CAAC,OAAO,CAAC,EAAE,EACnE,CAAC;gBACD,IAAI,CAAC,OAAO,CAAC,2BAA2B,CAAC,0BAA0B,CAAC,CAAC;gBACrE,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;gBACnC,0BAA0B,GAAG,EAAE,CAAC;YAClC,CAAC;YAED,0BAA0B,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC/C,CAAC,CACF,CAAC;QAEF,IAAI,0BAA0B,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1C,4DAA4D;YAC5D,IAAI,CAAC,OAAO,CAAC,2BAA2B,CAAC,0BAA0B,CAAC,CAAC;YACrE,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;QACrC,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,iBAAiB,CAC7B,eAAuB,EACvB,YAA0C;QAE1C,IAAI,KAAK,EAAE,MAAM,MAAM,IAAI,IAAI,CAAC,YAAY,CAAI,eAAe,CAAC,EAAE,CAAC;YACjE,IAAI,CAAC,IAAI,CAAC,yBAAyB,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC5C,SAAS;YACX,CAAC;YAED,MAAM,YAAY,CAAC,MAAM,CAAC,CAAC;QAC7B,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,YAAY,CACzB,8BAAsC;QAEtC,MAAM,oBAAoB,GAAG,IAAI,GAAG,EAGjC,CAAC;QAEJ,MAAM,yBAAyB,GAAG;;;;;KAKjC,CAAC;QACF,MAAM,wBAAwB,GAAG,IAAI,CAAC,QAAQ,CAAC,iBAAiB,CAC9D,yBAAyB,EACzB,IAAI,yBAAW,EAAE,CAAC,UAAU,CAC1B,eAAe,EACf,8BAA8B,CAC/B,CACF,CAAC;QACF,MAAM,6BAA6B,GACjC,IAAA,kFAAwC,EAAC,wBAAwB,CAAC,CAAC;QACrE,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,6BAA6B,EAAE,CAAC;YAC3D,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;QAED,KAAK,MAAM,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,IAAI,oBAAoB,EAAE,CAAC;YACxE,MAAM,aAAa,GAAG,GAAG,UAAU,IAAI,SAAS,EAAE,CAAC;YACnD,IAAI,IAAI,CAAC,mCAAmC,CAAC,GAAG,CAAC,aAAa,CAAC;gBAAE,SAAS;YAE1E,MAAM,iBAAiB,GAAG,kBAAkB,UAAU,MAAM,SAAS,kDAAkD,CAAC;YACxH,MAAM,iBAAiB,GAAG,IAAI,CAAC,QAAQ,CAAC,iBAAiB,CACvD,iBAAiB,EACjB,IAAI,yBAAW,EAAE,CAAC,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,EAC5C,EAAE,SAAS,EAAE,4BAAc,CAAC,kBAAkB,EAAE,CACjD,CAAC;YACF,MAAM,sBAAsB,GAC1B,IAAA,kFAAwC,EAAC,iBAAiB,CAAC,CAAC;YAC9D,IAAI,SAAS,GAAG,KAAK,CAAC;YACtB,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,sBAAsB,EAAE,CAAC;gBACpD,MAAM,GAAG,GAAG,QAAQ,CAAC,KAAK,EAAE,CAAC;gBAC7B,MAAM,WAAW,GAAuB;oBACtC,GAAG,GAAG;oBACN,aAAa;oBACb,SAAS,EAAE,SAAS;iBACrB,CAAC,CAAC,0CAA0C;gBAC7C,IAAI,CAAC,SAAS,EAAE,CAAC;oBACf,SAAS,GAAG,IAAI,CAAC;gBACnB,CAAC;gBACD,OAAQ,WAAmB,CAAC,SAAS,CAAC,CAAC,wEAAwE;gBAC/G,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAI,WAAW,CAAC,CAAC;gBAEnE,MAAM,YAAY,CAAC;YACrB,CAAC;QACH,CAAC;IACH,CAAC;IAEe,KAAK,CAAC,8BAA8B,CAClD,UAAkB;QAElB,kHAAkH;IACpH,CAAC;CACF;AAhID,oFAgIC","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} from \"@itwin/core-backend\";\nimport { Id64String } from \"@itwin/core-bentley\";\nimport { ExportElementAspectsStrategy } from \"./ExportElementAspectsStrategy\";\nimport { ensureECSqlReaderIsAsyncIterableIterator } from \"./ECSqlReaderAsyncIterableIteratorAdapter\";\nimport {\n  ElementAspectProps,\n  QueryBinder,\n  QueryRowFormat,\n} from \"@itwin/core-common\";\n\n/**\n * Detached ElementAspect export strategy for [[IModelExporter]].\n * This strategy exports all ElementAspects separately from the Elements that own them.\n *\n * @note Since aspects are exported separately from elements that own them, this strategy will export aspects of filtered out elements by default and\n * this needs to be handled by ElementAspectHandler\n * @internal\n */\nexport class DetachedExportElementAspectsStrategy extends ExportElementAspectsStrategy {\n  public override async exportAllElementAspects(): Promise<void> {\n    await this.exportAspectsLoop<ElementUniqueAspect>(\n      ElementUniqueAspect.classFullName,\n      async (uniqueAspect) => {\n        const isInsertChange =\n          this.aspectChanges?.insertIds.has(uniqueAspect.id) ?? false;\n        const isUpdateChange =\n          this.aspectChanges?.updateIds.has(uniqueAspect.id) ?? false;\n        const doExport =\n          this.aspectChanges === undefined || isInsertChange || isUpdateChange;\n        if (doExport) {\n          const isKnownUpdate = this.aspectChanges ? isUpdateChange : undefined;\n          this.handler.onExportElementUniqueAspect(uniqueAspect, isKnownUpdate);\n          await this.handler.trackProgress();\n        }\n      }\n    );\n\n    let batchedElementMultiAspects: ElementMultiAspect[] = [];\n    await this.exportAspectsLoop<ElementMultiAspect>(\n      ElementMultiAspect.classFullName,\n      async (multiAspect) => {\n        if (batchedElementMultiAspects.length === 0) {\n          batchedElementMultiAspects.push(multiAspect);\n          return;\n        }\n\n        // element id changed so all element's aspects are in the array and can be exported\n        if (\n          batchedElementMultiAspects[0].element.id !== multiAspect.element.id\n        ) {\n          this.handler.onExportElementMultiAspects(batchedElementMultiAspects);\n          await this.handler.trackProgress();\n          batchedElementMultiAspects = [];\n        }\n\n        batchedElementMultiAspects.push(multiAspect);\n      }\n    );\n\n    if (batchedElementMultiAspects.length > 0) {\n      // aspects that are left in the array have not been exported\n      this.handler.onExportElementMultiAspects(batchedElementMultiAspects);\n      await this.handler.trackProgress();\n    }\n  }\n\n  private async exportAspectsLoop<T extends ElementAspect>(\n    baseAspectClass: string,\n    exportAspect: (aspect: T) => Promise<void>\n  ) {\n    for await (const aspect of this.queryAspects<T>(baseAspectClass)) {\n      if (!this.shouldExportElementAspect(aspect)) {\n        continue;\n      }\n\n      await exportAspect(aspect);\n    }\n  }\n\n  private async *queryAspects<T extends ElementAspect>(\n    baseElementAspectClassFullName: string\n  ) {\n    const aspectClassNameIdMap = new Map<\n      Id64String,\n      { schemaName: string; className: string }\n    >();\n\n    const optimizesAspectClassesSql = `\n      SELECT c.ECInstanceId as classId, (ec_className(c.ECInstanceId, 's')) as schemaName, (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    `;\n    const aspectClassesQueryReader = this.sourceDb.createQueryReader(\n      optimizesAspectClassesSql,\n      new QueryBinder().bindString(\n        \"baseClassName\",\n        baseElementAspectClassFullName\n      )\n    );\n    const aspectClassesAsyncQueryReader =\n      ensureECSqlReaderIsAsyncIterableIterator(aspectClassesQueryReader);\n    for await (const rowProxy of aspectClassesAsyncQueryReader) {\n      const row = rowProxy.toRow();\n      aspectClassNameIdMap.set(row.classId, {\n        schemaName: row.schemaName,\n        className: row.className,\n      });\n    }\n\n    for (const [classId, { schemaName, className }] of aspectClassNameIdMap) {\n      const classFullName = `${schemaName}:${className}`;\n      if (this.excludedElementAspectClassFullNames.has(classFullName)) continue;\n\n      const getAspectPropsSql = `SELECT * FROM [${schemaName}]:[${className}] WHERE ECClassId = :classId ORDER BY Element.Id`;\n      const aspectQueryReader = this.sourceDb.createQueryReader(\n        getAspectPropsSql,\n        new QueryBinder().bindId(\"classId\", classId),\n        { rowFormat: QueryRowFormat.UseJsPropertyNames }\n      );\n      const aspectAsyncQueryReader =\n        ensureECSqlReaderIsAsyncIterableIterator(aspectQueryReader);\n      let firstDone = false;\n      for await (const rowProxy of aspectAsyncQueryReader) {\n        const row = rowProxy.toRow();\n        const aspectProps: ElementAspectProps = {\n          ...row,\n          classFullName,\n          className: undefined,\n        }; // add in property required by EntityProps\n        if (!firstDone) {\n          firstDone = true;\n        }\n        delete (aspectProps as any).className; // clear property from SELECT * that we don't want in the final instance\n        const aspectEntity = this.sourceDb.constructEntity<T>(aspectProps);\n\n        yield aspectEntity;\n      }\n    }\n  }\n\n  public override async exportElementAspectsForElement(\n    _elementId: string\n  ): Promise<void> {\n    // All aspects are exported separately from their elements and don't need to be exported when element is exported.\n  }\n}\n"]}
+{"version":3,"file":"DetachedExportElementAspectsStrategy.js","sourceRoot":"","sources":["../../src/DetachedExportElementAspectsStrategy.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG,sDAI6B;AAE7B,iFAA8E;AAC9E,uGAAqG;AACrG,oDAI4B;AAE5B;;;;;;;GAOG;AACH,MAAa,oCAAqC,SAAQ,2DAA4B;IACpE,KAAK,CAAC,uBAAuB;QAC3C,MAAM,IAAI,CAAC,iBAAiB,CAC1B,kCAAmB,CAAC,aAAa,EACjC,KAAK,EAAE,YAAY,EAAE,EAAE;YACrB,MAAM,cAAc,GAClB,IAAI,CAAC,aAAa,EAAE,SAAS,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC,IAAI,KAAK,CAAC;YAC9D,MAAM,cAAc,GAClB,IAAI,CAAC,aAAa,EAAE,SAAS,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC,IAAI,KAAK,CAAC;YAC9D,MAAM,QAAQ,GACZ,IAAI,CAAC,aAAa,KAAK,SAAS,IAAI,cAAc,IAAI,cAAc,CAAC;YACvE,IAAI,QAAQ,EAAE,CAAC;gBACb,MAAM,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,SAAS,CAAC;gBACtE,MAAM,IAAI,CAAC,OAAO,CAAC,2BAA2B,CAC5C,YAAY,EACZ,aAAa,CACd,CAAC;gBACF,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;YACrC,CAAC;QACH,CAAC,CACF,CAAC;QAEF,IAAI,0BAA0B,GAAyB,EAAE,CAAC;QAC1D,MAAM,IAAI,CAAC,iBAAiB,CAC1B,iCAAkB,CAAC,aAAa,EAChC,KAAK,EAAE,WAAW,EAAE,EAAE;YACpB,IAAI,0BAA0B,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC5C,0BAA0B,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;gBAC7C,OAAO;YACT,CAAC;YAED,mFAAmF;YACnF,IACE,0BAA0B,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,WAAW,CAAC,OAAO,CAAC,EAAE,EACnE,CAAC;gBACD,MAAM,IAAI,CAAC,OAAO,CAAC,2BAA2B,CAC5C,0BAA0B,CAC3B,CAAC;gBACF,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;gBACnC,0BAA0B,GAAG,EAAE,CAAC;YAClC,CAAC;YAED,0BAA0B,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC/C,CAAC,CACF,CAAC;QAEF,IAAI,0BAA0B,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1C,4DAA4D;YAC5D,MAAM,IAAI,CAAC,OAAO,CAAC,2BAA2B,CAC5C,0BAA0B,CAC3B,CAAC;YACF,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;QACrC,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,iBAAiB,CAC7B,eAAuB,EACvB,YAA0C;QAE1C,IAAI,KAAK,EAAE,MAAM,MAAM,IAAI,IAAI,CAAC,YAAY,CAAI,eAAe,CAAC,EAAE,CAAC;YACjE,IAAI,CAAC,IAAI,CAAC,yBAAyB,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC5C,SAAS;YACX,CAAC;YAED,MAAM,YAAY,CAAC,MAAM,CAAC,CAAC;QAC7B,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,YAAY,CACzB,8BAAsC;QAEtC,MAAM,oBAAoB,GAAG,IAAI,GAAG,EAGjC,CAAC;QAEJ,MAAM,yBAAyB,GAAG;;;;;KAKjC,CAAC;QACF,MAAM,wBAAwB,GAAG,IAAI,CAAC,QAAQ,CAAC,iBAAiB,CAC9D,yBAAyB,EACzB,IAAI,yBAAW,EAAE,CAAC,UAAU,CAC1B,eAAe,EACf,8BAA8B,CAC/B,EACD,EAAE,cAAc,EAAE,IAAI,EAAE,CACzB,CAAC;QACF,MAAM,6BAA6B,GACjC,IAAA,kFAAwC,EAAC,wBAAwB,CAAC,CAAC;QACrE,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,6BAA6B,EAAE,CAAC;YAC3D,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;QAED,KAAK,MAAM,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,IAAI,oBAAoB,EAAE,CAAC;YACxE,MAAM,aAAa,GAAG,GAAG,UAAU,IAAI,SAAS,EAAE,CAAC;YACnD,IAAI,IAAI,CAAC,mCAAmC,CAAC,GAAG,CAAC,aAAa,CAAC;gBAAE,SAAS;YAE1E,MAAM,iBAAiB,GAAG,kBAAkB,UAAU,MAAM,SAAS,kDAAkD,CAAC;YACxH,MAAM,iBAAiB,GAAG,IAAI,CAAC,QAAQ,CAAC,iBAAiB,CACvD,iBAAiB,EACjB,IAAI,yBAAW,EAAE,CAAC,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,EAC5C,EAAE,SAAS,EAAE,4BAAc,CAAC,kBAAkB,EAAE,cAAc,EAAE,IAAI,EAAE,CACvE,CAAC;YACF,MAAM,sBAAsB,GAC1B,IAAA,kFAAwC,EAAC,iBAAiB,CAAC,CAAC;YAC9D,IAAI,SAAS,GAAG,KAAK,CAAC;YACtB,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,sBAAsB,EAAE,CAAC;gBACpD,MAAM,GAAG,GAAG,QAAQ,CAAC,KAAK,EAAE,CAAC;gBAC7B,MAAM,WAAW,GAAuB;oBACtC,GAAG,GAAG;oBACN,aAAa;oBACb,SAAS,EAAE,SAAS;iBACrB,CAAC,CAAC,0CAA0C;gBAC7C,IAAI,CAAC,SAAS,EAAE,CAAC;oBACf,SAAS,GAAG,IAAI,CAAC;gBACnB,CAAC;gBACD,OAAQ,WAAmB,CAAC,SAAS,CAAC,CAAC,wEAAwE;gBAC/G,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAI,WAAW,CAAC,CAAC;gBAEnE,MAAM,YAAY,CAAC;YACrB,CAAC;QACH,CAAC;IACH,CAAC;IAEe,KAAK,CAAC,8BAA8B,CAClD,UAAkB;QAElB,kHAAkH;IACpH,CAAC;CACF;AAxID,oFAwIC","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} from \"@itwin/core-backend\";\nimport { Id64String } from \"@itwin/core-bentley\";\nimport { ExportElementAspectsStrategy } from \"./ExportElementAspectsStrategy\";\nimport { ensureECSqlReaderIsAsyncIterableIterator } from \"./ECSqlReaderAsyncIterableIteratorAdapter\";\nimport {\n  ElementAspectProps,\n  QueryBinder,\n  QueryRowFormat,\n} from \"@itwin/core-common\";\n\n/**\n * Detached ElementAspect export strategy for [[IModelExporter]].\n * This strategy exports all ElementAspects separately from the Elements that own them.\n *\n * @note Since aspects are exported separately from elements that own them, this strategy will export aspects of filtered out elements by default and\n * this needs to be handled by ElementAspectHandler\n * @internal\n */\nexport class DetachedExportElementAspectsStrategy extends ExportElementAspectsStrategy {\n  public override async exportAllElementAspects(): Promise<void> {\n    await this.exportAspectsLoop<ElementUniqueAspect>(\n      ElementUniqueAspect.classFullName,\n      async (uniqueAspect) => {\n        const isInsertChange =\n          this.aspectChanges?.insertIds.has(uniqueAspect.id) ?? false;\n        const isUpdateChange =\n          this.aspectChanges?.updateIds.has(uniqueAspect.id) ?? false;\n        const doExport =\n          this.aspectChanges === undefined || isInsertChange || isUpdateChange;\n        if (doExport) {\n          const isKnownUpdate = this.aspectChanges ? isUpdateChange : undefined;\n          await this.handler.onExportElementUniqueAspect(\n            uniqueAspect,\n            isKnownUpdate\n          );\n          await this.handler.trackProgress();\n        }\n      }\n    );\n\n    let batchedElementMultiAspects: ElementMultiAspect[] = [];\n    await this.exportAspectsLoop<ElementMultiAspect>(\n      ElementMultiAspect.classFullName,\n      async (multiAspect) => {\n        if (batchedElementMultiAspects.length === 0) {\n          batchedElementMultiAspects.push(multiAspect);\n          return;\n        }\n\n        // element id changed so all element's aspects are in the array and can be exported\n        if (\n          batchedElementMultiAspects[0].element.id !== multiAspect.element.id\n        ) {\n          await this.handler.onExportElementMultiAspects(\n            batchedElementMultiAspects\n          );\n          await this.handler.trackProgress();\n          batchedElementMultiAspects = [];\n        }\n\n        batchedElementMultiAspects.push(multiAspect);\n      }\n    );\n\n    if (batchedElementMultiAspects.length > 0) {\n      // aspects that are left in the array have not been exported\n      await this.handler.onExportElementMultiAspects(\n        batchedElementMultiAspects\n      );\n      await this.handler.trackProgress();\n    }\n  }\n\n  private async exportAspectsLoop<T extends ElementAspect>(\n    baseAspectClass: string,\n    exportAspect: (aspect: T) => Promise<void>\n  ) {\n    for await (const aspect of this.queryAspects<T>(baseAspectClass)) {\n      if (!this.shouldExportElementAspect(aspect)) {\n        continue;\n      }\n\n      await exportAspect(aspect);\n    }\n  }\n\n  private async *queryAspects<T extends ElementAspect>(\n    baseElementAspectClassFullName: string\n  ) {\n    const aspectClassNameIdMap = new Map<\n      Id64String,\n      { schemaName: string; className: string }\n    >();\n\n    const optimizesAspectClassesSql = `\n      SELECT c.ECInstanceId as classId, (ec_className(c.ECInstanceId, 's')) as schemaName, (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    `;\n    const aspectClassesQueryReader = this.sourceDb.createQueryReader(\n      optimizesAspectClassesSql,\n      new QueryBinder().bindString(\n        \"baseClassName\",\n        baseElementAspectClassFullName\n      ),\n      { usePrimaryConn: true }\n    );\n    const aspectClassesAsyncQueryReader =\n      ensureECSqlReaderIsAsyncIterableIterator(aspectClassesQueryReader);\n    for await (const rowProxy of aspectClassesAsyncQueryReader) {\n      const row = rowProxy.toRow();\n      aspectClassNameIdMap.set(row.classId, {\n        schemaName: row.schemaName,\n        className: row.className,\n      });\n    }\n\n    for (const [classId, { schemaName, className }] of aspectClassNameIdMap) {\n      const classFullName = `${schemaName}:${className}`;\n      if (this.excludedElementAspectClassFullNames.has(classFullName)) continue;\n\n      const getAspectPropsSql = `SELECT * FROM [${schemaName}]:[${className}] WHERE ECClassId = :classId ORDER BY Element.Id`;\n      const aspectQueryReader = this.sourceDb.createQueryReader(\n        getAspectPropsSql,\n        new QueryBinder().bindId(\"classId\", classId),\n        { rowFormat: QueryRowFormat.UseJsPropertyNames, usePrimaryConn: true }\n      );\n      const aspectAsyncQueryReader =\n        ensureECSqlReaderIsAsyncIterableIterator(aspectQueryReader);\n      let firstDone = false;\n      for await (const rowProxy of aspectAsyncQueryReader) {\n        const row = rowProxy.toRow();\n        const aspectProps: ElementAspectProps = {\n          ...row,\n          classFullName,\n          className: undefined,\n        }; // add in property required by EntityProps\n        if (!firstDone) {\n          firstDone = true;\n        }\n        delete (aspectProps as any).className; // clear property from SELECT * that we don't want in the final instance\n        const aspectEntity = this.sourceDb.constructEntity<T>(aspectProps);\n\n        yield aspectEntity;\n      }\n    }\n  }\n\n  public override async exportElementAspectsForElement(\n    _elementId: string\n  ): Promise<void> {\n    // All aspects are exported separately from their elements and don't need to be exported when element is exported.\n  }\n}\n"]}

package/lib/cjs/ECReferenceTypesCache.d.ts

@@ -22,6 +22,9 @@ export declare class ECReferenceTypesCache {
     private _propQualifierToRefType;
     private _relClassNameEndToRefTypes;
     private _initedSchemas;
+    private _rootBisClassCache;
+    private _relationshipInfoCache;
+    private _constraintClassCache;
     private static bisRootClassToRefType;
     private getRootBisClass;
     private getAbstractConstraintClass;

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

@@ -1 +1 @@
-{"version":3,"file":"ECReferenceTypesCache.d.ts","sourceRoot":"","sources":["../../src/ECReferenceTypesCache.ts"],"names":[],"mappings":"AAIA;;GAEG;AAGH,OAAO,EACL,mBAAmB,EAEnB,WAAW,EACZ,MAAM,oBAAoB,CAAC;AAY5B,OAAO,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AAE/C;;GAEG;AACH,qBAAa,mBAAoB,SAAQ,KAAK;;CAI7C;AAED;;;;;;;GAOG;AACH,qBAAa,qBAAqB;IAChC,uGAAuG;IACvG,OAAO,CAAC,uBAAuB,CAG3B;IACJ,OAAO,CAAC,0BAA0B,CAG9B;IACJ,OAAO,CAAC,cAAc,CAAgC;IAEtD,OAAO,CAAC,MAAM,CAAC,qBAAqB,CAalC;YAEY,eAAe;YA2Bf,0BAA0B;IAaxC,8CAA8C;IACjC,sBAAsB,CAAC,MAAM,EAAE,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC;YAkCtD,kBAAkB;YAalB,UAAU;YAqCV,mBAAmB;IA4D1B,iBAAiB,CACtB,UAAU,EAAE,MAAM,EAClB,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,MAAM,GACf,SAAS,GAAG,mBAAmB;IAS3B,sBAAsB,CAC3B,UAAU,EAAE,MAAM,EAClB,SAAS,EAAE,MAAM,GAChB,SAAS,GAAG,WAAW;IAQnB,KAAK;CAIb"}
+{"version":3,"file":"ECReferenceTypesCache.d.ts","sourceRoot":"","sources":["../../src/ECReferenceTypesCache.ts"],"names":[],"mappings":"AAIA;;GAEG;AAGH,OAAO,EAAE,mBAAmB,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAWtE,OAAO,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AAG/C;;GAEG;AACH,qBAAa,mBAAoB,SAAQ,KAAK;;CAI7C;AAED;;;;;;;GAOG;AACH,qBAAa,qBAAqB;IAChC,uGAAuG;IACvG,OAAO,CAAC,uBAAuB,CAG3B;IACJ,OAAO,CAAC,0BAA0B,CAG9B;IACJ,OAAO,CAAC,cAAc,CAAgC;IAGtD,OAAO,CAAC,kBAAkB,CAA8B;IACxD,OAAO,CAAC,sBAAsB,CAA8C;IAC5E,OAAO,CAAC,qBAAqB,CAA8B;IAE3D,OAAO,CAAC,MAAM,CAAC,qBAAqB,CAalC;YAEY,eAAe;YAmCf,0BAA0B;IAqBxC,8CAA8C;IACjC,sBAAsB,CAAC,MAAM,EAAE,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC;YAkDtD,kBAAkB;YAalB,UAAU;YA4FV,mBAAmB;IA4D1B,iBAAiB,CACtB,UAAU,EAAE,MAAM,EAClB,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,MAAM,GACf,SAAS,GAAG,mBAAmB;IAS3B,sBAAsB,CAC3B,UAAU,EAAE,MAAM,EAClB,SAAS,EAAE,MAAM,GAChB,SAAS,GAAG,WAAW;IAQnB,KAAK;CAQb"}

package/lib/cjs/ECReferenceTypesCache.js

@@ -12,6 +12,7 @@ const core_bentley_1 = require("@itwin/core-bentley");
 const core_common_1 = require("@itwin/core-common");
 const ecschema_metadata_1 = require("@itwin/ecschema-metadata");
 const assert = require("assert");
+const TransformerLoggerCategory_1 = require("./TransformerLoggerCategory");
 /** The context for transforming a *source* Element to a *target* Element and remapping internal identifiers to the target iModel.
  * @internal
  */
@@ -34,6 +35,10 @@ class ECReferenceTypesCache {
     _propQualifierToRefType = new core_bentley_1.TupleKeyedMap();
     _relClassNameEndToRefTypes = new core_bentley_1.TupleKeyedMap();
     _initedSchemas = new Map();
+    // Performance optimization caches
+    _rootBisClassCache = new Map();
+    _relationshipInfoCache = new Map();
+    _constraintClassCache = new Map();
     static bisRootClassToRefType = {
         /* eslint-disable quote-props, @typescript-eslint/naming-convention */
         Element: core_common_1.ConcreteEntityTypes.Element,
@@ -46,6 +51,11 @@ class ECReferenceTypesCache {
         /* eslint-enable quote-props, @typescript-eslint/naming-convention */
     };
     async getRootBisClass(ecclass) {
+        const cacheKey = ecclass.fullName;
+        const cached = this._rootBisClassCache.get(cacheKey);
+        if (cached) {
+            return cached;
+        }
         let bisRootForConstraint = ecclass;
         await ecclass.traverseBaseClasses((baseClass) => {
             // The depth first traversal will descend all the way to the root class before making any lateral traversal
@@ -64,43 +74,58 @@ class ECReferenceTypesCache {
             assert(bisRootForConstraint.appliesTo !== undefined, "The referenced AppliesToEntityClass could not be found, how did it pass schema validation?");
             bisRootForConstraint = await this.getRootBisClass(await bisRootForConstraint.appliesTo);
         }
+        this._rootBisClassCache.set(cacheKey, bisRootForConstraint);
         return bisRootForConstraint;
     }
     async getAbstractConstraintClass(constraint) {
+        const cacheKey = `${constraint.fullName}_${constraint.constraintClasses?.[0]?.fullName || "abstract"}`;
+        const cached = this._constraintClassCache.get(cacheKey);
+        if (cached) {
+            return cached;
+        }
         // constraint classes must share a base so we can get the root from any of them, just use the first
         const ecclass = await (constraint.constraintClasses?.[0] ||
             constraint.abstractConstraint);
         assert(ecclass !== undefined, "At least one constraint class or an abstract constraint must have been defined, the constraint is not valid");
+        this._constraintClassCache.set(cacheKey, ecclass);
         return ecclass;
     }
     /** initialize from an imodel with metadata */
     async initAllSchemasInIModel(imodel) {
-        const schemaLoader = new ecschema_metadata_1.SchemaLoader((name) => imodel.getSchemaProps(name));
-        // Issue for `createQueryReader` reported: https://github.com/iTwin/itwinjs-core/issues/7984
-        // eslint-disable-next-line @itwin/no-internal, deprecation/deprecation
-        await imodel.withPreparedStatement(`
+        let schemaCount = 0;
+        const initStartTime = performance.now();
+        const query = `
       WITH RECURSIVE refs(SchemaId) AS (
         SELECT ECInstanceId FROM ECDbMeta.ECSchemaDef WHERE Name='BisCore'
-        UNION ALL
+        UNION
         SELECT sr.SourceECInstanceId
         FROM ECDbMeta.SchemaHasSchemaReferences sr
         JOIN refs ON sr.TargetECInstanceId = refs.SchemaId
       )
-      SELECT s.Name
+      SELECT s.Name as name
       FROM refs
       JOIN ECDbMeta.ECSchemaDef s ON refs.SchemaId=s.ECInstanceId
       -- ensure schema dependency order
-      ORDER BY ECInstanceId
-    `, async (stmt) => {
-            let status;
-            while ((status = stmt.step()) === core_bentley_1.DbResult.BE_SQLITE_ROW) {
-                const schemaName = stmt.getValue(0).getString();
-                const schema = schemaLoader.getSchema(schemaName);
-                await this.considerInitSchema(schema);
+      ORDER BY s.ECInstanceId
+    `;
+        for await (const row of imodel.createQueryReader(query, undefined, {
+            usePrimaryConn: true,
+        })) {
+            const schemaName = row.name;
+            const startTime = performance.now();
+            core_bentley_1.Logger.logTrace(TransformerLoggerCategory_1.TransformerLoggerCategory.ECReferenceTypesCache, `Loading schema: ${schemaName}`);
+            const schemaItemKey = new ecschema_metadata_1.SchemaKey(schemaName);
+            const schema = await imodel.schemaContext.getSchema(schemaItemKey);
+            if (!schema) {
+                throw new Error(`Failed to load schema: ${schemaName}`);
             }
-            if (status !== core_bentley_1.DbResult.BE_SQLITE_DONE)
-                throw new core_common_1.IModelError(status, "unexpected query failure");
-        });
+            await this.considerInitSchema(schema);
+            const endTime = performance.now();
+            core_bentley_1.Logger.logTrace(TransformerLoggerCategory_1.TransformerLoggerCategory.ECReferenceTypesCache, `Completed schema: ${schemaName} in ${(endTime - startTime).toFixed(2)}ms`);
+            schemaCount++;
+        }
+        const initEndTime = performance.now();
+        core_bentley_1.Logger.logTrace(TransformerLoggerCategory_1.TransformerLoggerCategory.ECReferenceTypesCache, `Completed ${schemaCount} schemas in ${(initEndTime - initStartTime).toFixed(2)}ms`);
     }
     async considerInitSchema(schema) {
         if (this._initedSchemas.has(schema.name)) {
@@ -114,33 +139,67 @@ class ECReferenceTypesCache {
         return this.initSchema(schema);
     }
     async initSchema(schema) {
-        for (const ecclass of schema.getItems()) {
-            if (!ecschema_metadata_1.ECClass.isECClass(ecclass))
+        core_bentley_1.Logger.logInfo(TransformerLoggerCategory_1.TransformerLoggerCategory.ECReferenceTypesCache, `Init Schema: ${schema.name}`);
+        const schemaNameLower = schema.name.toLowerCase();
+        // Pre-collect all items to reduce iterator overhead
+        const allItems = Array.from(schema.getItems());
+        const ecClasses = [];
+        const relationshipClasses = [];
+        // Single pass through items with type checking
+        for (const item of allItems) {
+            // eslint-disable-next-line @itwin/no-internal
+            if (!ecschema_metadata_1.ECClass.isECClass(item))
                 continue;
-            const properties = await ecclass.getProperties();
-            if (properties === undefined)
-                continue;
-            for (const prop of properties) {
-                if (!prop.isNavigation())
-                    continue;
-                const relClass = await prop.relationshipClass;
-                const relInfo = await this.relInfoFromRelClass(relClass);
-                if (relInfo === undefined)
-                    continue;
-                const navPropRefType = prop.direction === ecschema_metadata_1.StrengthDirection.Forward
-                    ? relInfo.target
-                    : relInfo.source;
-                this._propQualifierToRefType.set([
-                    schema.name.toLowerCase(),
-                    ecclass.name.toLowerCase(),
-                    prop.name.toLowerCase(),
-                ], navPropRefType);
+            ecClasses.push(item);
+            if (item instanceof ecschema_metadata_1.RelationshipClass) {
+                relationshipClasses.push(item);
             }
-            if (ecclass instanceof ecschema_metadata_1.RelationshipClass) {
-                const relInfo = await this.relInfoFromRelClass(ecclass);
-                if (relInfo)
-                    this._relClassNameEndToRefTypes.set([schema.name.toLowerCase(), ecclass.name.toLowerCase()], relInfo);
+        }
+        // Process relationship classes in parallel and populate global cache
+        const relInfoPromises = relationshipClasses.map(async (relClass) => {
+            const relInfo = await this.relInfoFromRelClass(relClass);
+            this._relationshipInfoCache.set(relClass.fullName, relInfo);
+            if (relInfo) {
+                this._relClassNameEndToRefTypes.set([schemaNameLower, relClass.name.toLowerCase()], relInfo);
             }
+            return relInfo;
+        });
+        // Wait for all relationship info to be cached
+        await Promise.all(relInfoPromises);
+        // Process navigation properties with optimized batching
+        const propertyBatchSize = 25;
+        for (let i = 0; i < ecClasses.length; i += propertyBatchSize) {
+            const classBatch = ecClasses.slice(i, i + propertyBatchSize);
+            const classPromises = classBatch.map(async (ecclass) => {
+                const properties = await ecclass.getProperties();
+                if (!properties)
+                    return;
+                const classNameLower = ecclass.name.toLowerCase();
+                // Efficiently filter navigation properties
+                const navProps = Array.from(properties).filter((prop) => prop.isNavigation());
+                if (navProps.length === 0)
+                    return;
+                const navPropPromises = navProps.map(async (prop) => {
+                    const relClass = await prop.relationshipClass;
+                    // Use cached relation info
+                    let relInfo = this._relationshipInfoCache.get(relClass.fullName);
+                    if (relInfo === undefined &&
+                        !this._relationshipInfoCache.has(relClass.fullName)) {
+                        relInfo = await this.relInfoFromRelClass(relClass);
+                        this._relationshipInfoCache.set(relClass.fullName, relInfo);
+                    }
+                    if (relInfo === undefined)
+                        return;
+                    const navPropRefType = prop.direction === ecschema_metadata_1.StrengthDirection.Forward
+                        ? // eslint-disable-next-line @itwin/no-internal
+                            relInfo.target
+                        : // eslint-disable-next-line @itwin/no-internal
+                            relInfo.source;
+                    this._propQualifierToRefType.set([schemaNameLower, classNameLower, prop.name.toLowerCase()], navPropRefType);
+                });
+                await Promise.all(navPropPromises);
+            });
+            await Promise.all(classPromises);
         }
         this._initedSchemas.set(schema.name, schema.schemaKey);
     }
@@ -199,6 +258,10 @@ class ECReferenceTypesCache {
     clear() {
         this._initedSchemas.clear();
         this._propQualifierToRefType.clear();
+        this._relClassNameEndToRefTypes.clear();
+        this._rootBisClassCache.clear();
+        this._relationshipInfoCache.clear();
+        this._constraintClassCache.clear();
     }
 }
 exports.ECReferenceTypesCache = ECReferenceTypesCache;

package/lib/cjs/ECReferenceTypesCache.js.map

@@ -1 +1 @@
-{"version":3,"file":"ECReferenceTypesCache.js","sourceRoot":"","sources":["../../src/ECReferenceTypesCache.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;AAChG;;GAEG;;;AAEH,sDAA8D;AAC9D,oDAI4B;AAC5B,gEASkC;AAClC,iCAAiC;AAGjC;;GAEG;AACH,MAAa,mBAAoB,SAAQ,KAAK;IAC5C;QACE,KAAK,CAAC,iDAAiD,CAAC,CAAC;IAC3D,CAAC;CACF;AAJD,kDAIC;AAED;;;;;;;GAOG;AACH,MAAa,qBAAqB;IAChC,uGAAuG;IAC/F,uBAAuB,GAAG,IAAI,4BAAa,EAGhD,CAAC;IACI,0BAA0B,GAAG,IAAI,4BAAa,EAGnD,CAAC;IACI,cAAc,GAAG,IAAI,GAAG,EAAqB,CAAC;IAE9C,MAAM,CAAC,qBAAqB,GAGhC;QACF,sEAAsE;QACtE,OAAO,EAAE,iCAAmB,CAAC,OAAO;QACpC,KAAK,EAAE,iCAAmB,CAAC,KAAK;QAChC,aAAa,EAAE,iCAAmB,CAAC,aAAa;QAChD,uBAAuB,EAAE,iCAAmB,CAAC,YAAY;QACzD,oBAAoB,EAAE,iCAAmB,CAAC,YAAY;QACtD,8EAA8E;QAC9E,qCAAqC;QACrC,qEAAqE;KACtE,CAAC;IAEM,KAAK,CAAC,eAAe,CAAC,OAAgB;QAC5C,IAAI,oBAAoB,GAAY,OAAO,CAAC;QAC5C,MAAM,OAAO,CAAC,mBAAmB,CAAC,CAAC,SAAS,EAAE,EAAE;YAC9C,2GAA2G;YAC3G,+GAA+G;YAC/G,iEAAiE;YACjE,MAAM,WAAW,GAAG,oBAAoB,KAAK,OAAO,CAAC;YACrD,MAAM,yBAAyB,GAC7B,SAAS,CAAC,IAAI,KAAK,oBAAoB,CAAC,SAAS,EAAE,IAAI,CAAC;YAC1D,MAAM,uBAAuB,GAAG,WAAW,IAAI,CAAC,yBAAyB,CAAC;YAC1E,IAAI,CAAC,uBAAuB;gBAAE,OAAO,IAAI,CAAC,CAAC,uBAAuB;YAClE,oBAAoB,GAAG,SAAS,CAAC;YACjC,OAAO,KAAK,CAAC;QACf,CAAC,CAAC,CAAC;QACH,gFAAgF;QAChF,IAAI,oBAAoB,YAAY,yBAAK,EAAE,CAAC;YAC1C,MAAM,CACJ,oBAAoB,CAAC,SAAS,KAAK,SAAS,EAC5C,4FAA4F,CAC7F,CAAC;YACF,oBAAoB,GAAG,MAAM,IAAI,CAAC,eAAe,CAC/C,MAAM,oBAAoB,CAAC,SAAS,CACrC,CAAC;QACJ,CAAC;QACD,OAAO,oBAAoB,CAAC;IAC9B,CAAC;IAEO,KAAK,CAAC,0BAA0B,CACtC,UAAkC;QAElC,mGAAmG;QACnG,MAAM,OAAO,GAAG,MAAM,CAAC,UAAU,CAAC,iBAAiB,EAAE,CAAC,CAAC,CAAC;YACtD,UAAU,CAAC,kBAAkB,CAAC,CAAC;QACjC,MAAM,CACJ,OAAO,KAAK,SAAS,EACrB,6GAA6G,CAC9G,CAAC;QACF,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,8CAA8C;IACvC,KAAK,CAAC,sBAAsB,CAAC,MAAgB;QAClD,MAAM,YAAY,GAAG,IAAI,gCAAY,CAAC,CAAC,IAAY,EAAE,EAAE,CACrD,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,CAC5B,CAAC;QACF,4FAA4F;QAC5F,uEAAuE;QACvE,MAAM,MAAM,CAAC,qBAAqB,CAChC;;;;;;;;;;;;;KAaD,EACC,KAAK,EAAE,IAAI,EAAE,EAAE;YACb,IAAI,MAAgB,CAAC;YACrB,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC,KAAK,uBAAQ,CAAC,aAAa,EAAE,CAAC;gBACzD,MAAM,UAAU,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;gBAChD,MAAM,MAAM,GAAG,YAAY,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;gBAClD,MAAM,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;YACxC,CAAC;YACD,IAAI,MAAM,KAAK,uBAAQ,CAAC,cAAc;gBACpC,MAAM,IAAI,yBAAW,CAAC,MAAM,EAAE,0BAA0B,CAAC,CAAC;QAC9D,CAAC,CACF,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAAC,MAAc;QAC7C,IAAI,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YACzC,MAAM,eAAe,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC7D,MAAM,CAAC,eAAe,KAAK,SAAS,CAAC,CAAC;YACtC,MAAM,4BAA4B,GAChC,MAAM,CAAC,SAAS,CAAC,gBAAgB,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAC1D,IAAI,4BAA4B,EAAE,CAAC;gBACjC,OAAO;YACT,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;IACjC,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,MAAc;QACrC,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,QAAQ,EAAE,EAAE,CAAC;YACxC,IAAI,CAAC,2BAAO,CAAC,SAAS,CAAC,OAAO,CAAC;gBAAE,SAAS;YAC1C,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,aAAa,EAAE,CAAC;YACjD,IAAI,UAAU,KAAK,SAAS;gBAAE,SAAS;YACvC,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;gBAC9B,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE;oBAAE,SAAS;gBACnC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC;gBAC9C,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC;gBACzD,IAAI,OAAO,KAAK,SAAS;oBAAE,SAAS;gBACpC,MAAM,cAAc,GAClB,IAAI,CAAC,SAAS,KAAK,qCAAiB,CAAC,OAAO;oBAC1C,CAAC,CAAC,OAAO,CAAC,MAAM;oBAChB,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC;gBACrB,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAC9B;oBACE,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE;oBACzB,OAAO,CAAC,IAAI,CAAC,WAAW,EAAE;oBAC1B,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE;iBACxB,EACD,cAAc,CACf,CAAC;YACJ,CAAC;YAED,IAAI,OAAO,YAAY,qCAAiB,EAAE,CAAC;gBACzC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;gBACxD,IAAI,OAAO;oBACT,IAAI,CAAC,0BAA0B,CAAC,GAAG,CACjC,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,OAAO,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,EACvD,OAAO,CACR,CAAC;YACN,CAAC;QACH,CAAC;QAED,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC;IACzD,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAC/B,OAA0B;QAE1B,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,iBAAiB,KAAK,SAAS,CAAC,CAAC;QACvD,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,iBAAiB,KAAK,SAAS,CAAC,CAAC;QACvD,MAAM,CACJ,CAAC,WAAW,EAAE,kBAAkB,CAAC,EACjC,CAAC,WAAW,EAAE,kBAAkB,CAAC,EAClC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACpB,IAAI,CAAC,0BAA0B,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAClD,KAAK,EAAE,eAAe,EAAE,EAAE,CAAC;gBACzB,eAAe;gBACf,MAAM,IAAI,CAAC,eAAe,CAAC,eAAe,CAAC;aAC5C,CACF;YACD,IAAI,CAAC,0BAA0B,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAClD,KAAK,EAAE,eAAe,EAAE,EAAE,CAAC;gBACzB,eAAe;gBACf,MAAM,IAAI,CAAC,eAAe,CAAC,eAAe,CAAC;aAC5C,CACF;SACF,CAAC,CAAC;QAEH,IACE,kBAAkB,CAAC,IAAI,KAAK,UAAU;YACtC,kBAAkB,CAAC,IAAI,KAAK,UAAU;YAEtC,OAAO,SAAS,CAAC;QACnB,MAAM,UAAU,GACd,qBAAqB,CAAC,qBAAqB,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;QACvE,MAAM,UAAU,GACd,qBAAqB,CAAC,qBAAqB,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;QACvE,IACE,CAAC,CAAC,UAAU;YACV,kBAAkB,CAAC,gBAAgB,EAAE,GAAG,CAAC,mBAAmB,CAAC,CAAC;YAChE,CAAC,CAAC,UAAU;gBACV,kBAAkB,CAAC,gBAAgB,EAAE,GAAG,CAAC,mBAAmB,CAAC,CAAC,EAChE,CAAC;YACD,qFAAqF;YACrF,wGAAwG;YACxG,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,aAAa,GAAG,CAAC,IAAa,EAAE,GAAY,EAAE,EAAE,CACpD;YACE,0BAA0B,IAAI,CAAC,QAAQ,oCAAoC;YAC3E,yCAAyC,GAAG,CAAC,QAAQ,GAAG;YACxD,gBAAgB;SACjB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACf,MAAM,CACJ,UAAU,KAAK,SAAS,EACxB,aAAa,CAAC,kBAAkB,EAAE,WAAW,CAAC,CAC/C,CAAC;QACF,MAAM,CACJ,UAAU,KAAK,SAAS,EACxB,aAAa,CAAC,kBAAkB,EAAE,WAAW,CAAC,CAC/C,CAAC;QACF,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC;IACpD,CAAC;IAEM,iBAAiB,CACtB,UAAkB,EAClB,SAAiB,EACjB,QAAgB;QAEhB,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,UAAU,CAAC;YAAE,MAAM,IAAI,mBAAmB,EAAE,CAAC;QAC1E,OAAO,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC;YACtC,UAAU,CAAC,WAAW,EAAE;YACxB,SAAS,CAAC,WAAW,EAAE;YACvB,QAAQ,CAAC,WAAW,EAAE;SACvB,CAAC,CAAC;IACL,CAAC;IAEM,sBAAsB,CAC3B,UAAkB,EAClB,SAAiB;QAEjB,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,UAAU,CAAC;YAAE,MAAM,IAAI,mBAAmB,EAAE,CAAC;QAC1E,OAAO,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC;YACzC,UAAU,CAAC,WAAW,EAAE;YACxB,SAAS,CAAC,WAAW,EAAE;SACxB,CAAC,CAAC;IACL,CAAC;IAEM,KAAK;QACV,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAC;QAC5B,IAAI,CAAC,uBAAuB,CAAC,KAAK,EAAE,CAAC;IACvC,CAAC;;AA/OH,sDAgPC","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 */\n\nimport { DbResult, TupleKeyedMap } from \"@itwin/core-bentley\";\nimport {\n  ConcreteEntityTypes,\n  IModelError,\n  RelTypeInfo,\n} from \"@itwin/core-common\";\nimport {\n  ECClass,\n  Mixin,\n  RelationshipClass,\n  RelationshipConstraint,\n  Schema,\n  SchemaKey,\n  SchemaLoader,\n  StrengthDirection,\n} from \"@itwin/ecschema-metadata\";\nimport * as assert from \"assert\";\nimport { IModelDb } from \"@itwin/core-backend\";\n\n/** The context for transforming a *source* Element to a *target* Element and remapping internal identifiers to the target iModel.\n * @internal\n */\nexport class SchemaNotInCacheErr extends Error {\n  public constructor() {\n    super(\"Schema was not in cache, initialize that schema\");\n  }\n}\n\n/**\n * A cache of the entity types referenced by navprops in ecchemas, as well as the source and target entity types of\n * The transformer needs the referenced type to determine how to resolve references.\n *\n * Using multiple of these usually performs redundant computation, for static schemas at least. A possible future optimization\n * would be to seed the computation from a global cache of non-dynamic schemas, but dynamic schemas can collide willy-nilly\n * @internal\n */\nexport class ECReferenceTypesCache {\n  /** nesting based tuple map keyed by qualified property path tuple [schemaName, className, propName] */\n  private _propQualifierToRefType = new TupleKeyedMap<\n    [string, string, string],\n    ConcreteEntityTypes\n  >();\n  private _relClassNameEndToRefTypes = new TupleKeyedMap<\n    [string, string],\n    RelTypeInfo\n  >();\n  private _initedSchemas = new Map<string, SchemaKey>();\n\n  private static bisRootClassToRefType: Record<\n    string,\n    ConcreteEntityTypes | undefined\n  > = {\n    /* eslint-disable quote-props, @typescript-eslint/naming-convention */\n    Element: ConcreteEntityTypes.Element,\n    Model: ConcreteEntityTypes.Model,\n    ElementAspect: ConcreteEntityTypes.ElementAspect,\n    ElementRefersToElements: ConcreteEntityTypes.Relationship,\n    ElementDrivesElement: ConcreteEntityTypes.Relationship,\n    // code spec is technically a potential root class but it is ignored currently\n    // see [ConcreteEntityTypes]($common)\n    /* eslint-enable quote-props, @typescript-eslint/naming-convention */\n  };\n\n  private async getRootBisClass(ecclass: ECClass) {\n    let bisRootForConstraint: ECClass = ecclass;\n    await ecclass.traverseBaseClasses((baseClass) => {\n      // The depth first traversal will descend all the way to the root class before making any lateral traversal\n      // of mixin hierarchies, (or if the constraint is a mixin, it will traverse to the root of the mixin hierarchy)\n      // Once we see that we've moved laterally, we can terminate early\n      const isFirstTest = bisRootForConstraint === ecclass;\n      const traversalSwitchedRootPath =\n        baseClass.name !== bisRootForConstraint.baseClass?.name;\n      const stillTraversingRootPath = isFirstTest || !traversalSwitchedRootPath;\n      if (!stillTraversingRootPath) return true; // stop traversal early\n      bisRootForConstraint = baseClass;\n      return false;\n    });\n    // if the root class of the constraint was a mixin, use its AppliesToEntityClass\n    if (bisRootForConstraint instanceof Mixin) {\n      assert(\n        bisRootForConstraint.appliesTo !== undefined,\n        \"The referenced AppliesToEntityClass could not be found, how did it pass schema validation?\"\n      );\n      bisRootForConstraint = await this.getRootBisClass(\n        await bisRootForConstraint.appliesTo\n      );\n    }\n    return bisRootForConstraint;\n  }\n\n  private async getAbstractConstraintClass(\n    constraint: RelationshipConstraint\n  ): Promise<ECClass> {\n    // constraint classes must share a base so we can get the root from any of them, just use the first\n    const ecclass = await (constraint.constraintClasses?.[0] ||\n      constraint.abstractConstraint);\n    assert(\n      ecclass !== undefined,\n      \"At least one constraint class or an abstract constraint must have been defined, the constraint is not valid\"\n    );\n    return ecclass;\n  }\n\n  /** initialize from an imodel with metadata */\n  public async initAllSchemasInIModel(imodel: IModelDb): Promise<void> {\n    const schemaLoader = new SchemaLoader((name: string) =>\n      imodel.getSchemaProps(name)\n    );\n    // Issue for `createQueryReader` reported: https://github.com/iTwin/itwinjs-core/issues/7984\n    // eslint-disable-next-line @itwin/no-internal, deprecation/deprecation\n    await imodel.withPreparedStatement(\n      `\n      WITH RECURSIVE refs(SchemaId) AS (\n        SELECT ECInstanceId FROM ECDbMeta.ECSchemaDef WHERE Name='BisCore'\n        UNION ALL\n        SELECT sr.SourceECInstanceId\n        FROM ECDbMeta.SchemaHasSchemaReferences sr\n        JOIN refs ON sr.TargetECInstanceId = refs.SchemaId\n      )\n      SELECT s.Name\n      FROM refs\n      JOIN ECDbMeta.ECSchemaDef s ON refs.SchemaId=s.ECInstanceId\n      -- ensure schema dependency order\n      ORDER BY ECInstanceId\n    `,\n      async (stmt) => {\n        let status: DbResult;\n        while ((status = stmt.step()) === DbResult.BE_SQLITE_ROW) {\n          const schemaName = stmt.getValue(0).getString();\n          const schema = schemaLoader.getSchema(schemaName);\n          await this.considerInitSchema(schema);\n        }\n        if (status !== DbResult.BE_SQLITE_DONE)\n          throw new IModelError(status, \"unexpected query failure\");\n      }\n    );\n  }\n\n  private async considerInitSchema(schema: Schema): Promise<void> {\n    if (this._initedSchemas.has(schema.name)) {\n      const cachedSchemaKey = this._initedSchemas.get(schema.name);\n      assert(cachedSchemaKey !== undefined);\n      const incomingSchemaIsEqualOrOlder =\n        schema.schemaKey.compareByVersion(cachedSchemaKey) <= 0;\n      if (incomingSchemaIsEqualOrOlder) {\n        return;\n      }\n    }\n    return this.initSchema(schema);\n  }\n\n  private async initSchema(schema: Schema): Promise<void> {\n    for (const ecclass of schema.getItems()) {\n      if (!ECClass.isECClass(ecclass)) continue;\n      const properties = await ecclass.getProperties();\n      if (properties === undefined) continue;\n      for (const prop of properties) {\n        if (!prop.isNavigation()) continue;\n        const relClass = await prop.relationshipClass;\n        const relInfo = await this.relInfoFromRelClass(relClass);\n        if (relInfo === undefined) continue;\n        const navPropRefType =\n          prop.direction === StrengthDirection.Forward\n            ? relInfo.target\n            : relInfo.source;\n        this._propQualifierToRefType.set(\n          [\n            schema.name.toLowerCase(),\n            ecclass.name.toLowerCase(),\n            prop.name.toLowerCase(),\n          ],\n          navPropRefType\n        );\n      }\n\n      if (ecclass instanceof RelationshipClass) {\n        const relInfo = await this.relInfoFromRelClass(ecclass);\n        if (relInfo)\n          this._relClassNameEndToRefTypes.set(\n            [schema.name.toLowerCase(), ecclass.name.toLowerCase()],\n            relInfo\n          );\n      }\n    }\n\n    this._initedSchemas.set(schema.name, schema.schemaKey);\n  }\n\n  private async relInfoFromRelClass(\n    ecclass: RelationshipClass\n  ): Promise<RelTypeInfo | undefined> {\n    assert(ecclass.source.constraintClasses !== undefined);\n    assert(ecclass.target.constraintClasses !== undefined);\n    const [\n      [sourceClass, sourceRootBisClass],\n      [targetClass, targetRootBisClass],\n    ] = await Promise.all([\n      this.getAbstractConstraintClass(ecclass.source).then(\n        async (constraintClass) => [\n          constraintClass,\n          await this.getRootBisClass(constraintClass),\n        ]\n      ),\n      this.getAbstractConstraintClass(ecclass.target).then(\n        async (constraintClass) => [\n          constraintClass,\n          await this.getRootBisClass(constraintClass),\n        ]\n      ),\n    ]);\n\n    if (\n      sourceRootBisClass.name === \"CodeSpec\" ||\n      targetRootBisClass.name === \"CodeSpec\"\n    )\n      return undefined;\n    const sourceType =\n      ECReferenceTypesCache.bisRootClassToRefType[sourceRootBisClass.name];\n    const targetType =\n      ECReferenceTypesCache.bisRootClassToRefType[targetRootBisClass.name];\n    if (\n      (!sourceType &&\n        sourceRootBisClass.customAttributes?.has(\"ECDbMap.QueryView\")) ||\n      (!targetType &&\n        targetRootBisClass.customAttributes?.has(\"ECDbMap.QueryView\"))\n    ) {\n      // ECView elements are not \"real\" data and transformer does not need to process them.\n      // Relationships that point to ECView elements can only be used in ECViews so the mapping is not needed.\n      return undefined;\n    }\n\n    const makeAssertMsg = (root: ECClass, cls: ECClass) =>\n      [\n        `An unknown root class '${root.fullName}' was encountered while populating`,\n        `the nav prop reference type cache for ${cls.fullName}.`,\n        \"This is a bug.\",\n      ].join(\"\\n\");\n    assert(\n      sourceType !== undefined,\n      makeAssertMsg(sourceRootBisClass, sourceClass)\n    );\n    assert(\n      targetType !== undefined,\n      makeAssertMsg(targetRootBisClass, targetClass)\n    );\n    return { source: sourceType, target: targetType };\n  }\n\n  public getNavPropRefType(\n    schemaName: string,\n    className: string,\n    propName: string\n  ): undefined | ConcreteEntityTypes {\n    if (!this._initedSchemas.has(schemaName)) throw new SchemaNotInCacheErr();\n    return this._propQualifierToRefType.get([\n      schemaName.toLowerCase(),\n      className.toLowerCase(),\n      propName.toLowerCase(),\n    ]);\n  }\n\n  public getRelationshipEndType(\n    schemaName: string,\n    className: string\n  ): undefined | RelTypeInfo {\n    if (!this._initedSchemas.has(schemaName)) throw new SchemaNotInCacheErr();\n    return this._relClassNameEndToRefTypes.get([\n      schemaName.toLowerCase(),\n      className.toLowerCase(),\n    ]);\n  }\n\n  public clear() {\n    this._initedSchemas.clear();\n    this._propQualifierToRefType.clear();\n  }\n}\n"]}
+{"version":3,"file":"ECReferenceTypesCache.js","sourceRoot":"","sources":["../../src/ECReferenceTypesCache.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;AAChG;;GAEG;;;AAEH,sDAA4D;AAC5D,oDAAsE;AACtE,gEAQkC;AAClC,iCAAiC;AAEjC,2EAAwE;AAExE;;GAEG;AACH,MAAa,mBAAoB,SAAQ,KAAK;IAC5C;QACE,KAAK,CAAC,iDAAiD,CAAC,CAAC;IAC3D,CAAC;CACF;AAJD,kDAIC;AAED;;;;;;;GAOG;AACH,MAAa,qBAAqB;IAChC,uGAAuG;IAC/F,uBAAuB,GAAG,IAAI,4BAAa,EAGhD,CAAC;IACI,0BAA0B,GAAG,IAAI,4BAAa,EAGnD,CAAC;IACI,cAAc,GAAG,IAAI,GAAG,EAAqB,CAAC;IAEtD,kCAAkC;IAC1B,kBAAkB,GAAG,IAAI,GAAG,EAAmB,CAAC;IAChD,sBAAsB,GAAG,IAAI,GAAG,EAAmC,CAAC;IACpE,qBAAqB,GAAG,IAAI,GAAG,EAAmB,CAAC;IAEnD,MAAM,CAAC,qBAAqB,GAGhC;QACF,sEAAsE;QACtE,OAAO,EAAE,iCAAmB,CAAC,OAAO;QACpC,KAAK,EAAE,iCAAmB,CAAC,KAAK;QAChC,aAAa,EAAE,iCAAmB,CAAC,aAAa;QAChD,uBAAuB,EAAE,iCAAmB,CAAC,YAAY;QACzD,oBAAoB,EAAE,iCAAmB,CAAC,YAAY;QACtD,8EAA8E;QAC9E,qCAAqC;QACrC,qEAAqE;KACtE,CAAC;IAEM,KAAK,CAAC,eAAe,CAAC,OAAgB;QAC5C,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QAClC,MAAM,MAAM,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACrD,IAAI,MAAM,EAAE,CAAC;YACX,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,IAAI,oBAAoB,GAAY,OAAO,CAAC;QAC5C,MAAM,OAAO,CAAC,mBAAmB,CAAC,CAAC,SAAS,EAAE,EAAE;YAC9C,2GAA2G;YAC3G,+GAA+G;YAC/G,iEAAiE;YACjE,MAAM,WAAW,GAAG,oBAAoB,KAAK,OAAO,CAAC;YACrD,MAAM,yBAAyB,GAC7B,SAAS,CAAC,IAAI,KAAK,oBAAoB,CAAC,SAAS,EAAE,IAAI,CAAC;YAC1D,MAAM,uBAAuB,GAAG,WAAW,IAAI,CAAC,yBAAyB,CAAC;YAC1E,IAAI,CAAC,uBAAuB;gBAAE,OAAO,IAAI,CAAC,CAAC,uBAAuB;YAClE,oBAAoB,GAAG,SAAS,CAAC;YACjC,OAAO,KAAK,CAAC;QACf,CAAC,CAAC,CAAC;QACH,gFAAgF;QAChF,IAAI,oBAAoB,YAAY,yBAAK,EAAE,CAAC;YAC1C,MAAM,CACJ,oBAAoB,CAAC,SAAS,KAAK,SAAS,EAC5C,4FAA4F,CAC7F,CAAC;YACF,oBAAoB,GAAG,MAAM,IAAI,CAAC,eAAe,CAC/C,MAAM,oBAAoB,CAAC,SAAS,CACrC,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,QAAQ,EAAE,oBAAoB,CAAC,CAAC;QAC5D,OAAO,oBAAoB,CAAC;IAC9B,CAAC;IAEO,KAAK,CAAC,0BAA0B,CACtC,UAAkC;QAElC,MAAM,QAAQ,GAAG,GAAG,UAAU,CAAC,QAAQ,IAAI,UAAU,CAAC,iBAAiB,EAAE,CAAC,CAAC,CAAC,EAAE,QAAQ,IAAI,UAAU,EAAE,CAAC;QACvG,MAAM,MAAM,GAAG,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACxD,IAAI,MAAM,EAAE,CAAC;YACX,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,mGAAmG;QACnG,MAAM,OAAO,GAAG,MAAM,CAAC,UAAU,CAAC,iBAAiB,EAAE,CAAC,CAAC,CAAC;YACtD,UAAU,CAAC,kBAAkB,CAAC,CAAC;QACjC,MAAM,CACJ,OAAO,KAAK,SAAS,EACrB,6GAA6G,CAC9G,CAAC;QAEF,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAClD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,8CAA8C;IACvC,KAAK,CAAC,sBAAsB,CAAC,MAAgB;QAClD,IAAI,WAAW,GAAG,CAAC,CAAC;QAEpB,MAAM,aAAa,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QAExC,MAAM,KAAK,GAAG;;;;;;;;;;;;;KAab,CAAC;QAEF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,CAAC,iBAAiB,CAAC,KAAK,EAAE,SAAS,EAAE;YACjE,cAAc,EAAE,IAAI;SACrB,CAAC,EAAE,CAAC;YACH,MAAM,UAAU,GAAG,GAAG,CAAC,IAAI,CAAC;YAC5B,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;YACpC,qBAAM,CAAC,QAAQ,CACb,qDAAyB,CAAC,qBAAqB,EAC/C,mBAAmB,UAAU,EAAE,CAChC,CAAC;YACF,MAAM,aAAa,GAAG,IAAI,6BAAS,CAAC,UAAU,CAAC,CAAC;YAChD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;YACnE,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,IAAI,KAAK,CAAC,0BAA0B,UAAU,EAAE,CAAC,CAAC;YAC1D,CAAC;YACD,MAAM,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;YACtC,MAAM,OAAO,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;YAClC,qBAAM,CAAC,QAAQ,CACb,qDAAyB,CAAC,qBAAqB,EAC/C,qBAAqB,UAAU,OAAO,CAAC,OAAO,GAAG,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAC3E,CAAC;YACF,WAAW,EAAE,CAAC;QAChB,CAAC;QAED,MAAM,WAAW,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACtC,qBAAM,CAAC,QAAQ,CACb,qDAAyB,CAAC,qBAAqB,EAC/C,aAAa,WAAW,eAAe,CAAC,WAAW,GAAG,aAAa,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CACpF,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAAC,MAAc;QAC7C,IAAI,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YACzC,MAAM,eAAe,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC7D,MAAM,CAAC,eAAe,KAAK,SAAS,CAAC,CAAC;YACtC,MAAM,4BAA4B,GAChC,MAAM,CAAC,SAAS,CAAC,gBAAgB,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAC1D,IAAI,4BAA4B,EAAE,CAAC;gBACjC,OAAO;YACT,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;IACjC,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,MAAc;QACrC,qBAAM,CAAC,OAAO,CACZ,qDAAyB,CAAC,qBAAqB,EAC/C,gBAAgB,MAAM,CAAC,IAAI,EAAE,CAC9B,CAAC;QACF,MAAM,eAAe,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;QAElD,oDAAoD;QACpD,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC/C,MAAM,SAAS,GAAc,EAAE,CAAC;QAChC,MAAM,mBAAmB,GAAwB,EAAE,CAAC;QAEpD,+CAA+C;QAC/C,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;YAC5B,8CAA8C;YAC9C,IAAI,CAAC,2BAAO,CAAC,SAAS,CAAC,IAAI,CAAC;gBAAE,SAAS;YACvC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACrB,IAAI,IAAI,YAAY,qCAAiB,EAAE,CAAC;gBACtC,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjC,CAAC;QACH,CAAC;QAED,qEAAqE;QACrE,MAAM,eAAe,GAAG,mBAAmB,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE;YACjE,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC;YACzD,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YAC5D,IAAI,OAAO,EAAE,CAAC;gBACZ,IAAI,CAAC,0BAA0B,CAAC,GAAG,CACjC,CAAC,eAAe,EAAE,QAAQ,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,EAC9C,OAAO,CACR,CAAC;YACJ,CAAC;YACD,OAAO,OAAO,CAAC;QACjB,CAAC,CAAC,CAAC;QAEH,8CAA8C;QAC9C,MAAM,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;QAEnC,wDAAwD;QACxD,MAAM,iBAAiB,GAAG,EAAE,CAAC;QAC7B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC,IAAI,iBAAiB,EAAE,CAAC;YAC7D,MAAM,UAAU,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,iBAAiB,CAAC,CAAC;YAE7D,MAAM,aAAa,GAAG,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE;gBACrD,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,aAAa,EAAE,CAAC;gBACjD,IAAI,CAAC,UAAU;oBAAE,OAAO;gBAExB,MAAM,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;gBAElD,2CAA2C;gBAC3C,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CACtD,IAAI,CAAC,YAAY,EAAE,CACpB,CAAC;gBACF,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;oBAAE,OAAO;gBAElC,MAAM,eAAe,GAAG,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;oBAClD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC;oBAE9C,2BAA2B;oBAC3B,IAAI,OAAO,GAAG,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;oBACjE,IACE,OAAO,KAAK,SAAS;wBACrB,CAAC,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,EACnD,CAAC;wBACD,OAAO,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC;wBACnD,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;oBAC9D,CAAC;oBAED,IAAI,OAAO,KAAK,SAAS;wBAAE,OAAO;oBAElC,MAAM,cAAc,GAClB,IAAI,CAAC,SAAS,KAAK,qCAAiB,CAAC,OAAO;wBAC1C,CAAC,CAAC,8CAA8C;4BAC9C,OAAO,CAAC,MAAM;wBAChB,CAAC,CAAC,8CAA8C;4BAC9C,OAAO,CAAC,MAAM,CAAC;oBAErB,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAC9B,CAAC,eAAe,EAAE,cAAc,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,EAC1D,cAAc,CACf,CAAC;gBACJ,CAAC,CAAC,CAAC;gBAEH,MAAM,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;YACrC,CAAC,CAAC,CAAC;YAEH,MAAM,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QACnC,CAAC;QAED,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC;IACzD,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAC/B,OAA0B;QAE1B,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,iBAAiB,KAAK,SAAS,CAAC,CAAC;QACvD,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,iBAAiB,KAAK,SAAS,CAAC,CAAC;QACvD,MAAM,CACJ,CAAC,WAAW,EAAE,kBAAkB,CAAC,EACjC,CAAC,WAAW,EAAE,kBAAkB,CAAC,EAClC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACpB,IAAI,CAAC,0BAA0B,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAClD,KAAK,EAAE,eAAe,EAAE,EAAE,CAAC;gBACzB,eAAe;gBACf,MAAM,IAAI,CAAC,eAAe,CAAC,eAAe,CAAC;aAC5C,CACF;YACD,IAAI,CAAC,0BAA0B,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAClD,KAAK,EAAE,eAAe,EAAE,EAAE,CAAC;gBACzB,eAAe;gBACf,MAAM,IAAI,CAAC,eAAe,CAAC,eAAe,CAAC;aAC5C,CACF;SACF,CAAC,CAAC;QAEH,IACE,kBAAkB,CAAC,IAAI,KAAK,UAAU;YACtC,kBAAkB,CAAC,IAAI,KAAK,UAAU;YAEtC,OAAO,SAAS,CAAC;QACnB,MAAM,UAAU,GACd,qBAAqB,CAAC,qBAAqB,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;QACvE,MAAM,UAAU,GACd,qBAAqB,CAAC,qBAAqB,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;QACvE,IACE,CAAC,CAAC,UAAU;YACV,kBAAkB,CAAC,gBAAgB,EAAE,GAAG,CAAC,mBAAmB,CAAC,CAAC;YAChE,CAAC,CAAC,UAAU;gBACV,kBAAkB,CAAC,gBAAgB,EAAE,GAAG,CAAC,mBAAmB,CAAC,CAAC,EAChE,CAAC;YACD,qFAAqF;YACrF,wGAAwG;YACxG,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,aAAa,GAAG,CAAC,IAAa,EAAE,GAAY,EAAE,EAAE,CACpD;YACE,0BAA0B,IAAI,CAAC,QAAQ,oCAAoC;YAC3E,yCAAyC,GAAG,CAAC,QAAQ,GAAG;YACxD,gBAAgB;SACjB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACf,MAAM,CACJ,UAAU,KAAK,SAAS,EACxB,aAAa,CAAC,kBAAkB,EAAE,WAAW,CAAC,CAC/C,CAAC;QACF,MAAM,CACJ,UAAU,KAAK,SAAS,EACxB,aAAa,CAAC,kBAAkB,EAAE,WAAW,CAAC,CAC/C,CAAC;QACF,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC;IACpD,CAAC;IAEM,iBAAiB,CACtB,UAAkB,EAClB,SAAiB,EACjB,QAAgB;QAEhB,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,UAAU,CAAC;YAAE,MAAM,IAAI,mBAAmB,EAAE,CAAC;QAC1E,OAAO,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC;YACtC,UAAU,CAAC,WAAW,EAAE;YACxB,SAAS,CAAC,WAAW,EAAE;YACvB,QAAQ,CAAC,WAAW,EAAE;SACvB,CAAC,CAAC;IACL,CAAC;IAEM,sBAAsB,CAC3B,UAAkB,EAClB,SAAiB;QAEjB,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,UAAU,CAAC;YAAE,MAAM,IAAI,mBAAmB,EAAE,CAAC;QAC1E,OAAO,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC;YACzC,UAAU,CAAC,WAAW,EAAE;YACxB,SAAS,CAAC,WAAW,EAAE;SACxB,CAAC,CAAC;IACL,CAAC;IAEM,KAAK;QACV,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAC;QAC5B,IAAI,CAAC,uBAAuB,CAAC,KAAK,EAAE,CAAC;QACrC,IAAI,CAAC,0BAA0B,CAAC,KAAK,EAAE,CAAC;QACxC,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,CAAC;QAChC,IAAI,CAAC,sBAAsB,CAAC,KAAK,EAAE,CAAC;QACpC,IAAI,CAAC,qBAAqB,CAAC,KAAK,EAAE,CAAC;IACrC,CAAC;;AA/UH,sDAgVC","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 */\n\nimport { Logger, TupleKeyedMap } from \"@itwin/core-bentley\";\nimport { ConcreteEntityTypes, RelTypeInfo } from \"@itwin/core-common\";\nimport {\n  ECClass,\n  Mixin,\n  RelationshipClass,\n  RelationshipConstraint,\n  Schema,\n  SchemaKey,\n  StrengthDirection,\n} from \"@itwin/ecschema-metadata\";\nimport * as assert from \"assert\";\nimport { IModelDb } from \"@itwin/core-backend\";\nimport { TransformerLoggerCategory } from \"./TransformerLoggerCategory\";\n\n/** The context for transforming a *source* Element to a *target* Element and remapping internal identifiers to the target iModel.\n * @internal\n */\nexport class SchemaNotInCacheErr extends Error {\n  public constructor() {\n    super(\"Schema was not in cache, initialize that schema\");\n  }\n}\n\n/**\n * A cache of the entity types referenced by navprops in ecchemas, as well as the source and target entity types of\n * The transformer needs the referenced type to determine how to resolve references.\n *\n * Using multiple of these usually performs redundant computation, for static schemas at least. A possible future optimization\n * would be to seed the computation from a global cache of non-dynamic schemas, but dynamic schemas can collide willy-nilly\n * @internal\n */\nexport class ECReferenceTypesCache {\n  /** nesting based tuple map keyed by qualified property path tuple [schemaName, className, propName] */\n  private _propQualifierToRefType = new TupleKeyedMap<\n    [string, string, string],\n    ConcreteEntityTypes\n  >();\n  private _relClassNameEndToRefTypes = new TupleKeyedMap<\n    [string, string],\n    RelTypeInfo\n  >();\n  private _initedSchemas = new Map<string, SchemaKey>();\n\n  // Performance optimization caches\n  private _rootBisClassCache = new Map<string, ECClass>();\n  private _relationshipInfoCache = new Map<string, RelTypeInfo | undefined>();\n  private _constraintClassCache = new Map<string, ECClass>();\n\n  private static bisRootClassToRefType: Record<\n    string,\n    ConcreteEntityTypes | undefined\n  > = {\n    /* eslint-disable quote-props, @typescript-eslint/naming-convention */\n    Element: ConcreteEntityTypes.Element,\n    Model: ConcreteEntityTypes.Model,\n    ElementAspect: ConcreteEntityTypes.ElementAspect,\n    ElementRefersToElements: ConcreteEntityTypes.Relationship,\n    ElementDrivesElement: ConcreteEntityTypes.Relationship,\n    // code spec is technically a potential root class but it is ignored currently\n    // see [ConcreteEntityTypes]($common)\n    /* eslint-enable quote-props, @typescript-eslint/naming-convention */\n  };\n\n  private async getRootBisClass(ecclass: ECClass) {\n    const cacheKey = ecclass.fullName;\n    const cached = this._rootBisClassCache.get(cacheKey);\n    if (cached) {\n      return cached;\n    }\n\n    let bisRootForConstraint: ECClass = ecclass;\n    await ecclass.traverseBaseClasses((baseClass) => {\n      // The depth first traversal will descend all the way to the root class before making any lateral traversal\n      // of mixin hierarchies, (or if the constraint is a mixin, it will traverse to the root of the mixin hierarchy)\n      // Once we see that we've moved laterally, we can terminate early\n      const isFirstTest = bisRootForConstraint === ecclass;\n      const traversalSwitchedRootPath =\n        baseClass.name !== bisRootForConstraint.baseClass?.name;\n      const stillTraversingRootPath = isFirstTest || !traversalSwitchedRootPath;\n      if (!stillTraversingRootPath) return true; // stop traversal early\n      bisRootForConstraint = baseClass;\n      return false;\n    });\n    // if the root class of the constraint was a mixin, use its AppliesToEntityClass\n    if (bisRootForConstraint instanceof Mixin) {\n      assert(\n        bisRootForConstraint.appliesTo !== undefined,\n        \"The referenced AppliesToEntityClass could not be found, how did it pass schema validation?\"\n      );\n      bisRootForConstraint = await this.getRootBisClass(\n        await bisRootForConstraint.appliesTo\n      );\n    }\n\n    this._rootBisClassCache.set(cacheKey, bisRootForConstraint);\n    return bisRootForConstraint;\n  }\n\n  private async getAbstractConstraintClass(\n    constraint: RelationshipConstraint\n  ): Promise<ECClass> {\n    const cacheKey = `${constraint.fullName}_${constraint.constraintClasses?.[0]?.fullName || \"abstract\"}`;\n    const cached = this._constraintClassCache.get(cacheKey);\n    if (cached) {\n      return cached;\n    }\n\n    // constraint classes must share a base so we can get the root from any of them, just use the first\n    const ecclass = await (constraint.constraintClasses?.[0] ||\n      constraint.abstractConstraint);\n    assert(\n      ecclass !== undefined,\n      \"At least one constraint class or an abstract constraint must have been defined, the constraint is not valid\"\n    );\n\n    this._constraintClassCache.set(cacheKey, ecclass);\n    return ecclass;\n  }\n\n  /** initialize from an imodel with metadata */\n  public async initAllSchemasInIModel(imodel: IModelDb): Promise<void> {\n    let schemaCount = 0;\n\n    const initStartTime = performance.now();\n\n    const query = `\n      WITH RECURSIVE refs(SchemaId) AS (\n        SELECT ECInstanceId FROM ECDbMeta.ECSchemaDef WHERE Name='BisCore'\n        UNION\n        SELECT sr.SourceECInstanceId\n        FROM ECDbMeta.SchemaHasSchemaReferences sr\n        JOIN refs ON sr.TargetECInstanceId = refs.SchemaId\n      )\n      SELECT s.Name as name\n      FROM refs\n      JOIN ECDbMeta.ECSchemaDef s ON refs.SchemaId=s.ECInstanceId\n      -- ensure schema dependency order\n      ORDER BY s.ECInstanceId\n    `;\n\n    for await (const row of imodel.createQueryReader(query, undefined, {\n      usePrimaryConn: true,\n    })) {\n      const schemaName = row.name;\n      const startTime = performance.now();\n      Logger.logTrace(\n        TransformerLoggerCategory.ECReferenceTypesCache,\n        `Loading schema: ${schemaName}`\n      );\n      const schemaItemKey = new SchemaKey(schemaName);\n      const schema = await imodel.schemaContext.getSchema(schemaItemKey);\n      if (!schema) {\n        throw new Error(`Failed to load schema: ${schemaName}`);\n      }\n      await this.considerInitSchema(schema);\n      const endTime = performance.now();\n      Logger.logTrace(\n        TransformerLoggerCategory.ECReferenceTypesCache,\n        `Completed schema: ${schemaName} in ${(endTime - startTime).toFixed(2)}ms`\n      );\n      schemaCount++;\n    }\n\n    const initEndTime = performance.now();\n    Logger.logTrace(\n      TransformerLoggerCategory.ECReferenceTypesCache,\n      `Completed ${schemaCount} schemas in ${(initEndTime - initStartTime).toFixed(2)}ms`\n    );\n  }\n\n  private async considerInitSchema(schema: Schema): Promise<void> {\n    if (this._initedSchemas.has(schema.name)) {\n      const cachedSchemaKey = this._initedSchemas.get(schema.name);\n      assert(cachedSchemaKey !== undefined);\n      const incomingSchemaIsEqualOrOlder =\n        schema.schemaKey.compareByVersion(cachedSchemaKey) <= 0;\n      if (incomingSchemaIsEqualOrOlder) {\n        return;\n      }\n    }\n    return this.initSchema(schema);\n  }\n\n  private async initSchema(schema: Schema): Promise<void> {\n    Logger.logInfo(\n      TransformerLoggerCategory.ECReferenceTypesCache,\n      `Init Schema: ${schema.name}`\n    );\n    const schemaNameLower = schema.name.toLowerCase();\n\n    // Pre-collect all items to reduce iterator overhead\n    const allItems = Array.from(schema.getItems());\n    const ecClasses: ECClass[] = [];\n    const relationshipClasses: RelationshipClass[] = [];\n\n    // Single pass through items with type checking\n    for (const item of allItems) {\n      // eslint-disable-next-line @itwin/no-internal\n      if (!ECClass.isECClass(item)) continue;\n      ecClasses.push(item);\n      if (item instanceof RelationshipClass) {\n        relationshipClasses.push(item);\n      }\n    }\n\n    // Process relationship classes in parallel and populate global cache\n    const relInfoPromises = relationshipClasses.map(async (relClass) => {\n      const relInfo = await this.relInfoFromRelClass(relClass);\n      this._relationshipInfoCache.set(relClass.fullName, relInfo);\n      if (relInfo) {\n        this._relClassNameEndToRefTypes.set(\n          [schemaNameLower, relClass.name.toLowerCase()],\n          relInfo\n        );\n      }\n      return relInfo;\n    });\n\n    // Wait for all relationship info to be cached\n    await Promise.all(relInfoPromises);\n\n    // Process navigation properties with optimized batching\n    const propertyBatchSize = 25;\n    for (let i = 0; i < ecClasses.length; i += propertyBatchSize) {\n      const classBatch = ecClasses.slice(i, i + propertyBatchSize);\n\n      const classPromises = classBatch.map(async (ecclass) => {\n        const properties = await ecclass.getProperties();\n        if (!properties) return;\n\n        const classNameLower = ecclass.name.toLowerCase();\n\n        // Efficiently filter navigation properties\n        const navProps = Array.from(properties).filter((prop) =>\n          prop.isNavigation()\n        );\n        if (navProps.length === 0) return;\n\n        const navPropPromises = navProps.map(async (prop) => {\n          const relClass = await prop.relationshipClass;\n\n          // Use cached relation info\n          let relInfo = this._relationshipInfoCache.get(relClass.fullName);\n          if (\n            relInfo === undefined &&\n            !this._relationshipInfoCache.has(relClass.fullName)\n          ) {\n            relInfo = await this.relInfoFromRelClass(relClass);\n            this._relationshipInfoCache.set(relClass.fullName, relInfo);\n          }\n\n          if (relInfo === undefined) return;\n\n          const navPropRefType =\n            prop.direction === StrengthDirection.Forward\n              ? // eslint-disable-next-line @itwin/no-internal\n                relInfo.target\n              : // eslint-disable-next-line @itwin/no-internal\n                relInfo.source;\n\n          this._propQualifierToRefType.set(\n            [schemaNameLower, classNameLower, prop.name.toLowerCase()],\n            navPropRefType\n          );\n        });\n\n        await Promise.all(navPropPromises);\n      });\n\n      await Promise.all(classPromises);\n    }\n\n    this._initedSchemas.set(schema.name, schema.schemaKey);\n  }\n\n  private async relInfoFromRelClass(\n    ecclass: RelationshipClass\n  ): Promise<RelTypeInfo | undefined> {\n    assert(ecclass.source.constraintClasses !== undefined);\n    assert(ecclass.target.constraintClasses !== undefined);\n    const [\n      [sourceClass, sourceRootBisClass],\n      [targetClass, targetRootBisClass],\n    ] = await Promise.all([\n      this.getAbstractConstraintClass(ecclass.source).then(\n        async (constraintClass) => [\n          constraintClass,\n          await this.getRootBisClass(constraintClass),\n        ]\n      ),\n      this.getAbstractConstraintClass(ecclass.target).then(\n        async (constraintClass) => [\n          constraintClass,\n          await this.getRootBisClass(constraintClass),\n        ]\n      ),\n    ]);\n\n    if (\n      sourceRootBisClass.name === \"CodeSpec\" ||\n      targetRootBisClass.name === \"CodeSpec\"\n    )\n      return undefined;\n    const sourceType =\n      ECReferenceTypesCache.bisRootClassToRefType[sourceRootBisClass.name];\n    const targetType =\n      ECReferenceTypesCache.bisRootClassToRefType[targetRootBisClass.name];\n    if (\n      (!sourceType &&\n        sourceRootBisClass.customAttributes?.has(\"ECDbMap.QueryView\")) ||\n      (!targetType &&\n        targetRootBisClass.customAttributes?.has(\"ECDbMap.QueryView\"))\n    ) {\n      // ECView elements are not \"real\" data and transformer does not need to process them.\n      // Relationships that point to ECView elements can only be used in ECViews so the mapping is not needed.\n      return undefined;\n    }\n\n    const makeAssertMsg = (root: ECClass, cls: ECClass) =>\n      [\n        `An unknown root class '${root.fullName}' was encountered while populating`,\n        `the nav prop reference type cache for ${cls.fullName}.`,\n        \"This is a bug.\",\n      ].join(\"\\n\");\n    assert(\n      sourceType !== undefined,\n      makeAssertMsg(sourceRootBisClass, sourceClass)\n    );\n    assert(\n      targetType !== undefined,\n      makeAssertMsg(targetRootBisClass, targetClass)\n    );\n    return { source: sourceType, target: targetType };\n  }\n\n  public getNavPropRefType(\n    schemaName: string,\n    className: string,\n    propName: string\n  ): undefined | ConcreteEntityTypes {\n    if (!this._initedSchemas.has(schemaName)) throw new SchemaNotInCacheErr();\n    return this._propQualifierToRefType.get([\n      schemaName.toLowerCase(),\n      className.toLowerCase(),\n      propName.toLowerCase(),\n    ]);\n  }\n\n  public getRelationshipEndType(\n    schemaName: string,\n    className: string\n  ): undefined | RelTypeInfo {\n    if (!this._initedSchemas.has(schemaName)) throw new SchemaNotInCacheErr();\n    return this._relClassNameEndToRefTypes.get([\n      schemaName.toLowerCase(),\n      className.toLowerCase(),\n    ]);\n  }\n\n  public clear() {\n    this._initedSchemas.clear();\n    this._propQualifierToRefType.clear();\n    this._relClassNameEndToRefTypes.clear();\n    this._rootBisClassCache.clear();\n    this._relationshipInfoCache.clear();\n    this._constraintClassCache.clear();\n  }\n}\n"]}

package/lib/cjs/ElementCascadingDeleter.js

@@ -38,7 +38,7 @@ 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, deprecation/deprecation
+        // eslint-disable-next-line @itwin/no-internal, @typescript-eslint/no-deprecated
         this._iModel.withPreparedStatement(`
       SELECT ECInstanceId
       FROM bis.Element

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,uEAAuE;QACvE,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, deprecation/deprecation\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":";;;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"]}