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

package/lib/cjs/IModelExporter.js

@@ -48,7 +48,7 @@ class IModelExportHandler {
      */
     onExportModel(_model, _isUpdate) { }
     /** Called when a model should be deleted. */
-    onDeleteModel(_modelId) { }
+    async onDeleteModel(_modelId) { }
     /** If `true` is returned, then the element will be exported.
      * @note This method can optionally be overridden to exclude an individual Element (and its children and ElementAspects) from the export. The base implementation always returns `true`.
      */
@@ -64,7 +64,7 @@ class IModelExportHandler {
      * @param isUpdate If defined, then `true` indicates an UPDATE operation while `false` indicates an INSERT operation. If not defined, then INSERT vs. UPDATE is not known.
      * @note This should be overridden to actually do the export.
      */
-    onExportElement(_element, _isUpdate) { }
+    async onExportElement(_element, _isUpdate) { }
     /**
      * Do any asynchronous actions before exporting an element
      * @note Do not implement this handler manually, it is internal, it will be removed.
@@ -85,11 +85,11 @@ class IModelExportHandler {
      * @param isUpdate If defined, then `true` indicates an UPDATE operation while `false` indicates an INSERT operation. If not defined, then INSERT vs. UPDATE is not known.
      * @note This should be overridden to actually do the export.
      */
-    onExportElementUniqueAspect(_aspect, _isUpdate) { }
+    async onExportElementUniqueAspect(_aspect, _isUpdate) { }
     /** Called when ElementMultiAspects should be exported.
      * @note This should be overridden to actually do the export.
      */
-    onExportElementMultiAspects(_aspects) { }
+    async onExportElementMultiAspects(_aspects) { }
     /** If `true` is returned, then the relationship will be exported.
      * @note This method can optionally be overridden to exclude an individual CodeSpec from the export. The base implementation always returns `true`.
      */
@@ -101,7 +101,7 @@ class IModelExportHandler {
      * @param isUpdate If defined, then `true` indicates an UPDATE operation while `false` indicates an INSERT operation. If not defined, then INSERT vs. UPDATE is not known.
      * @note This should be overridden to actually do the export.
      */
-    onExportRelationship(_relationship, _isUpdate) { }
+    async onExportRelationship(_relationship, _isUpdate) { }
     /** Called when a relationship should be deleted. */
     onDeleteRelationship(_relInstanceId) { }
     /** If `true` is returned, then the schema will be exported.
@@ -198,8 +198,8 @@ class IModelExporter {
     constructor(sourceDb, elementAspectsStrategy = ExportElementAspectsWithElementsStrategy_1.ExportElementAspectsWithElementsStrategy) {
         this.sourceDb = sourceDb;
         this._exportElementAspectsStrategy = new elementAspectsStrategy(this.sourceDb, {
-            onExportElementMultiAspects: (aspects) => this.handler.onExportElementMultiAspects(aspects),
-            onExportElementUniqueAspect: (aspect, isUpdate) => this.handler.onExportElementUniqueAspect(aspect, isUpdate),
+            onExportElementMultiAspects: async (aspects) => this.handler.onExportElementMultiAspects(aspects),
+            onExportElementUniqueAspect: async (aspect, isUpdate) => this.handler.onExportElementUniqueAspect(aspect, isUpdate),
             shouldExportElementAspect: (aspect) => this.handler.shouldExportElementAspect(aspect),
             trackProgress: async () => this.trackProgress(),
         });
@@ -297,7 +297,7 @@ class IModelExporter {
         if (this.visitElements) {
             // must delete models first since they have a constraint on the submodeling element which may also be deleted
             for (const modelId of this._sourceDbChanges.model.deleteIds) {
-                this.handler.onDeleteModel(modelId);
+                await this.handler.onDeleteModel(modelId);
             }
             for (const elementId of this._sourceDbChanges.element.deleteIds) {
                 // We don't know how the handler wants to handle deletions, and we don't have enough information
@@ -335,7 +335,6 @@ class IModelExporter {
      * @note This must be called separately from [[exportAll]] or [[exportChanges]].
      */
     async exportSchemas() {
-        /* eslint-disable @typescript-eslint/indent */
         const sql = `
       SELECT s.Name, s.VersionMajor, s.VersionWrite, s.VersionMinor
       FROM ECDbMeta.ECSchemaDef s
@@ -346,27 +345,27 @@ class IModelExporter {
       `}
       ORDER BY ECInstanceId
     `;
-        /* eslint-enable @typescript-eslint/indent */
-        const schemaNamesToExport = [];
-        // eslint-disable-next-line @itwin/no-internal, deprecation/deprecation
-        this.sourceDb.withPreparedStatement(sql, (statement) => {
-            while (core_bentley_1.DbResult.BE_SQLITE_ROW === statement.step()) {
-                const schemaName = statement.getValue(0).getString();
-                const versionMajor = statement.getValue(1).getInteger();
-                const versionWrite = statement.getValue(2).getInteger();
-                const versionMinor = statement.getValue(3).getInteger();
-                const schemaKey = new ecschema_metadata_1.SchemaKey(schemaName, new ecschema_metadata_1.ECVersion(versionMajor, versionWrite, versionMinor));
-                if (this.handler.shouldExportSchema(schemaKey)) {
-                    schemaNamesToExport.push(schemaName);
-                }
+        const schemaKeysToExport = [];
+        for await (const row of this.sourceDb.createQueryReader(sql, undefined, {
+            usePrimaryConn: true,
+        })) {
+            const schemaName = row[0];
+            const versionMajor = row[1];
+            const versionWrite = row[2];
+            const versionMinor = row[3];
+            const schemaKey = new ecschema_metadata_1.SchemaKey(schemaName, new ecschema_metadata_1.ECVersion(versionMajor, versionWrite, versionMinor));
+            if (this.handler.shouldExportSchema(schemaKey)) {
+                schemaKeysToExport.push(schemaKey);
             }
-        });
-        if (schemaNamesToExport.length === 0)
+        }
+        if (schemaKeysToExport.length === 0)
             return;
-        const schemaLoader = new ecschema_metadata_1.SchemaLoader((name) => this.sourceDb.getSchemaProps(name));
-        await Promise.all(schemaNamesToExport.map(async (schemaName) => {
-            const schema = schemaLoader.getSchema(schemaName);
-            core_bentley_1.Logger.logTrace(loggerCategory, `exportSchema(${schemaName})`);
+        await Promise.all(schemaKeysToExport.map(async (schemaKey) => {
+            const schema = await this.sourceDb.schemaContext.getSchema(schemaKey);
+            if (!schema) {
+                throw new Error(`Failed to load schema: ${schemaKey.name}`);
+            }
+            core_bentley_1.Logger.logTrace(loggerCategory, `exportSchema(${schemaKey.name})`);
             return this.handler.onExportSchema(schema);
         }));
     }
@@ -380,15 +379,11 @@ class IModelExporter {
     async exportCodeSpecs() {
         core_bentley_1.Logger.logTrace(loggerCategory, "exportCodeSpecs()");
         const sql = "SELECT Name FROM BisCore:CodeSpec ORDER BY ECInstanceId";
-        // eslint-disable-next-line @itwin/no-internal, deprecation/deprecation
-        await this.sourceDb.withPreparedStatement(sql, 
-        // eslint-disable-next-line @itwin/no-internal, deprecation/deprecation
-        async (statement) => {
-            while (core_bentley_1.DbResult.BE_SQLITE_ROW === statement.step()) {
-                const codeSpecName = statement.getValue(0).getString();
-                await this.exportCodeSpecByName(codeSpecName);
-            }
-        });
+        for await (const row of this.sourceDb.createQueryReader(sql, undefined, {
+            usePrimaryConn: true,
+        })) {
+            await this.exportCodeSpecByName(row[0]);
+        }
     }
     /** Export a single CodeSpec from the source iModel.
      * @note This method is called from [[exportChanges]] and [[exportAll]], so it only needs to be called directly when exporting a subset of an iModel.
@@ -432,18 +427,39 @@ class IModelExporter {
      */
     async exportFonts() {
         core_bentley_1.Logger.logTrace(loggerCategory, "exportFonts()");
-        for (const font of this.sourceDb.fontMap.fonts.values()) {
-            await this.exportFontByNumber(font.id);
+        for (const font of this.sourceDb.fonts.queryMappedFamilies()) {
+            await this.exportFontByFontProps(font);
         }
     }
     /** Export a single font from the source iModel.
-     * @note This method is called from [[exportChanges]] and [[exportAll]], so it only needs to be called directly when exporting a subset of an iModel.
+     * @note multiple fonts can have the same font name, if a font with a specific type is needed use exportFontByFontFamilyDescriptor.
+     * If not this will only export the first font with this type in the db.
      */
     async exportFontByName(fontName) {
         core_bentley_1.Logger.logTrace(loggerCategory, `exportFontByName(${fontName})`);
-        const font = this.sourceDb.fontMap.getFont(fontName);
-        if (undefined !== font) {
-            await this.exportFontByNumber(font.id);
+        const fontId = this.sourceDb.fonts.findId({
+            name: fontName,
+        });
+        if (undefined !== fontId) {
+            await this.exportFontByNumber(fontId);
+        }
+    }
+    /** Export a single font from the source iModel.
+     * @note This method is called from [[exportChanges]] and [[exportAll]], so it only needs to be called directly when exporting a subset of an iModel.
+     */
+    async exportFontByFontProps(fontProps) {
+        core_bentley_1.Logger.logTrace(loggerCategory, `exportFontByFamily(${fontProps.name}, ${fontProps.type})`);
+        this.handler.onExportFont(fontProps, true);
+        return this.trackProgress();
+    }
+    /** Export a single font from the source iModel.
+     * @note This method is called from [[exportChanges]] and [[exportAll]], so it only needs to be called directly when exporting a subset of an iModel.
+     */
+    async exportFontByFontFamilyDescriptor(fontFamily) {
+        core_bentley_1.Logger.logTrace(loggerCategory, `exportFontByFamilyDescriptor(${fontFamily.name}, ${fontFamily.type})`);
+        const fontId = await this.sourceDb.fonts.acquireId(fontFamily);
+        if (undefined !== fontId) {
+            await this.exportFontByFontProps({ ...fontFamily, id: fontId });
         }
     }
     /** Export a single font from the source iModel.
@@ -455,12 +471,10 @@ class IModelExporter {
          * It is very rare and even problematic for the font table to reach a large size, so it is not a bottleneck in transforming changes.
          * See https://github.com/iTwin/imodel-transformer/pull/135 for removed code.
          */
-        const isUpdate = true;
         core_bentley_1.Logger.logTrace(loggerCategory, `exportFontById(${fontNumber})`);
-        const font = this.sourceDb.fontMap.getFont(fontNumber);
+        const font = this.sourceDb.fonts.findDescriptor(fontNumber);
         if (undefined !== font) {
-            this.handler.onExportFont(font, isUpdate);
-            return this.trackProgress();
+            await this.exportFontByFontFamilyDescriptor(font);
         }
     }
     /** Export the model container, contents, and sub-models from the source iModel.
@@ -535,19 +549,16 @@ class IModelExporter {
         else {
             sql = `SELECT ECInstanceId FROM ${elementClassFullName} WHERE Parent.Id IS NULL AND Model.Id=:modelId ORDER BY ECInstanceId`;
         }
-        // eslint-disable-next-line @itwin/no-internal, deprecation/deprecation
-        await this.sourceDb.withPreparedStatement(sql, 
-        // eslint-disable-next-line @itwin/no-internal, deprecation/deprecation
-        async (statement) => {
-            statement.bindId("modelId", modelId);
-            if (skipRootSubject) {
-                statement.bindId("rootSubjectId", core_common_1.IModel.rootSubjectId);
-            }
-            while (core_bentley_1.DbResult.BE_SQLITE_ROW === statement.step()) {
-                await this.exportElement(statement.getValue(0).getId());
-                await this._yieldManager.allowYield();
-            }
-        });
+        const params = new core_common_1.QueryBinder().bindId("modelId", modelId);
+        if (skipRootSubject) {
+            params.bindId("rootSubjectId", core_common_1.IModel.rootSubjectId);
+        }
+        for await (const row of this.sourceDb.createQueryReader(sql, params, {
+            usePrimaryConn: true,
+        })) {
+            await this.exportElement(row.id);
+            await this._yieldManager.allowYield();
+        }
     }
     /** Export the sub-models directly below the specified model.
      * @note This method is called from [[exportChanges]] and [[exportAll]], so it only needs to be called directly when exporting a subset of an iModel.
@@ -557,22 +568,19 @@ class IModelExporter {
         const definitionModelIds = [];
         const otherModelIds = [];
         const sql = `SELECT ECInstanceId FROM ${core_backend_1.Model.classFullName} WHERE ParentModel.Id=:parentModelId ORDER BY ECInstanceId`;
-        // eslint-disable-next-line @itwin/no-internal, deprecation/deprecation
-        this.sourceDb.withPreparedStatement(sql, 
-        // eslint-disable-next-line @itwin/no-internal, deprecation/deprecation
-        (statement) => {
-            statement.bindId("parentModelId", parentModelId);
-            while (core_bentley_1.DbResult.BE_SQLITE_ROW === statement.step()) {
-                const modelId = statement.getValue(0).getId();
-                const model = this.sourceDb.models.getModel(modelId);
-                if (model instanceof core_backend_1.DefinitionModel) {
-                    definitionModelIds.push(modelId);
-                }
-                else {
-                    otherModelIds.push(modelId);
-                }
+        const params = new core_common_1.QueryBinder().bindId("parentModelId", parentModelId);
+        for await (const row of this.sourceDb.createQueryReader(sql, params, {
+            usePrimaryConn: true,
+        })) {
+            const modelId = row.id;
+            const model = this.sourceDb.models.getModel(modelId);
+            if (model instanceof core_backend_1.DefinitionModel) {
+                definitionModelIds.push(modelId);
             }
-        });
+            else {
+                otherModelIds.push(modelId);
+            }
+        }
         // export DefinitionModels before other types of Models
         for (const definitionModelId of definitionModelIds) {
             await this.exportModel(definitionModelId);
@@ -639,7 +647,7 @@ class IModelExporter {
         // the order and `await`ing of calls beyond here is depended upon by the IModelTransformer for a current bug workaround
         if (this.shouldExportElement(element)) {
             await this.handler.preExportElement(element);
-            this.handler.onExportElement(element, isUpdate);
+            await this.handler.onExportElement(element, isUpdate);
             await this.trackProgress();
             await this._exportElementAspectsStrategy.exportElementAspectsForElement(elementId);
             return this.exportChildElements(elementId);
@@ -678,23 +686,18 @@ class IModelExporter {
             return;
         }
         core_bentley_1.Logger.logTrace(loggerCategory, `exportRelationships(${baseRelClassFullName})`);
-        const sql = `SELECT r.ECInstanceId, r.ECClassId FROM ${baseRelClassFullName} r
+        const sql = `SELECT r.ECInstanceId, ec_className(r.ECClassId, 's.c') as className FROM ${baseRelClassFullName} r
                   JOIN bis.Element s ON s.ECInstanceId = r.SourceECInstanceId
                   JOIN bis.Element t ON t.ECInstanceId = r.TargetECInstanceId
                   WHERE s.ECInstanceId IS NOT NULL AND t.ECInstanceId IS NOT NULL`;
-        // eslint-disable-next-line @itwin/no-internal, deprecation/deprecation
-        await this.sourceDb.withPreparedStatement(sql, 
-        // eslint-disable-next-line @itwin/no-internal, deprecation/deprecation
-        async (statement) => {
-            while (core_bentley_1.DbResult.BE_SQLITE_ROW === statement.step()) {
-                const relationshipId = statement.getValue(0).getId();
-                const relationshipClass = statement
-                    .getValue(1)
-                    .getClassNameForClassId();
-                await this.exportRelationship(relationshipClass, relationshipId); // must call exportRelationship using the actual classFullName, not baseRelClassFullName
-                await this._yieldManager.allowYield();
-            }
-        });
+        for await (const row of this.sourceDb.createQueryReader(sql, undefined, {
+            usePrimaryConn: true,
+        })) {
+            const relationshipId = row.id;
+            const relationshipClass = row.className;
+            await this.exportRelationship(relationshipClass, relationshipId); // must call exportRelationship using the actual classFullName, not baseRelClassFullName
+            await this._yieldManager.allowYield();
+        }
     }
     /** Export a relationship from the source iModel. */
     async exportRelationship(relClassFullName, relInstanceId) {
@@ -726,7 +729,7 @@ class IModelExporter {
         }
         // relationship has passed standard exclusion rules, now give handler a chance to accept/reject export
         if (this.handler.shouldExportRelationship(relationship)) {
-            this.handler.onExportRelationship(relationship, isUpdate);
+            await this.handler.onExportRelationship(relationship, isUpdate);
             await this.trackProgress();
         }
     }
@@ -797,7 +800,7 @@ class ChangedInstanceIds {
         this._relationshipSubclassIds = new Set();
         this._relationshipSubclassIdsToSkip = new Set();
         const addECClassIdsToSet = async (setToModify, baseClass) => {
-            for await (const row of this._db.createQueryReader(`SELECT ECInstanceId FROM ECDbMeta.ECClassDef where ECInstanceId IS (${baseClass})`)) {
+            for await (const row of this._db.createQueryReader(`SELECT ECInstanceId FROM ECDbMeta.ECClassDef where ECInstanceId IS (${baseClass})`, undefined, { usePrimaryConn: true })) {
                 setToModify.add(row.ECInstanceId);
             }
         };
@@ -960,7 +963,9 @@ class ChangedInstanceIds {
         SELECT parentId FROM hierarchy where parentId is not null
     `;
         const parentModelIds = new Set();
-        for await (const row of this._db.createQueryReader(ecQuery, params)) {
+        for await (const row of this._db.createQueryReader(ecQuery, params, {
+            usePrimaryConn: true,
+        })) {
             // Transformer handles update as insert when element does not exist in target.
             // Which means that in scenario where child and parent model are filtered out from target,
             // and child element is inserted trough custom change, its parent model will be marked as updated.
@@ -978,7 +983,9 @@ class ChangedInstanceIds {
         WHERE InVirtualSet(:elementIds, TargetECInstanceId)
         OR InVirtualSet(:elementIds, SourceECInstanceId)`;
         const queryBinder = new core_common_1.QueryBinder().bindIdSet("elementIds", elementIds);
-        const queryReader = this._db.createQueryReader(ecQuery, queryBinder);
+        const queryReader = this._db.createQueryReader(ecQuery, queryBinder, {
+            usePrimaryConn: true,
+        });
         for await (const row of queryReader) {
             this.handleChange(this.relationship, "Inserted", row.ECInstanceId);
         }
@@ -990,7 +997,9 @@ class ChangedInstanceIds {
         ]) {
             const ecQuery = `Select ECInstanceId from ${aspectClassName} where InVirtualSet(:elementIds, Element.Id)`;
             const queryBinder = new core_common_1.QueryBinder().bindIdSet("elementIds", elementIds);
-            const queryReader = this._db.createQueryReader(ecQuery, queryBinder);
+            const queryReader = this._db.createQueryReader(ecQuery, queryBinder, {
+                usePrimaryConn: true,
+            });
             for await (const row of queryReader) {
                 this.addCustomAspectChange("Inserted", row.toArray()[0]);
             }
@@ -1066,7 +1075,7 @@ class ChangedInstanceIds {
                 disableSchemaCheck: true,
             });
             const csAdaptor = new core_backend_1.ChangesetECAdaptor(csReader);
-            const ecChangeUnifier = new core_backend_1.PartialECChangeUnifier();
+            const ecChangeUnifier = new core_backend_1.PartialECChangeUnifier(opts.iModel);
             while (csAdaptor.step()) {
                 ecChangeUnifier.appendFrom(csAdaptor);
             }