@itwin/core-backend 5.1.0-dev.32 → 5.1.0-dev.36

package/lib/cjs/ChangesetECAdaptor.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ChangesetECAdaptor = exports.PartialECChangeUnifier = void 0;
+exports.ChangesetECAdaptor = exports.PartialECChangeUnifier = exports.ECChangeUnifierCache = void 0;
 /*---------------------------------------------------------------------------------------------
 * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
 * See LICENSE.md in the project root for license terms and full copyright notice.
@@ -9,6 +9,9 @@ exports.ChangesetECAdaptor = exports.PartialECChangeUnifier = void 0;
  * @module ECDb
  */
 const core_bentley_1 = require("@itwin/core-bentley");
+const core_common_1 = require("@itwin/core-common");
+const ECDb_1 = require("./ECDb");
+const Symbols_1 = require("./internal/Symbols");
 class ECDbMap {
     db;
     _cachedClassMaps = new Map();
@@ -17,14 +20,14 @@ class ECDbMap {
         this.db = db;
     }
     getAllDerivedClasses(classFullName) {
-        const sql = `
-      SELECT format('0x%x', ch.ClassId)
-      FROM   [ec_cache_ClassHierarchy] [ch]
-            JOIN [ec_Class] [cs] ON [cs].[Id] = [ch].[BaseClassId]
-            JOIN [ec_Schema] [sc] ON [sc].[Id] = [cs].[SchemaId]
-      WHERE  (([sc].[Alias] = :schemaNameOrAlias
-              OR [sc].[Name] = :schemaNameOrAlias)
-              AND ([cs].[Name] = :className))
+        const sql = `
+      SELECT format('0x%x', ch.ClassId)
+      FROM   [ec_cache_ClassHierarchy] [ch]
+            JOIN [ec_Class] [cs] ON [cs].[Id] = [ch].[BaseClassId]
+            JOIN [ec_Schema] [sc] ON [sc].[Id] = [cs].[SchemaId]
+      WHERE  (([sc].[Alias] = :schemaNameOrAlias
+              OR [sc].[Name] = :schemaNameOrAlias)
+              AND ([cs].[Name] = :className))
     `;
         return this.db.withPreparedSqliteStatement(sql, (stmt) => {
             const parts = classFullName.indexOf(".") !== -1 ? classFullName.split(".") : classFullName.split(":");
@@ -39,39 +42,39 @@ class ECDbMap {
     getTable(tableName) {
         if (this._cacheTables.has(tableName))
             return this._cacheTables.get(tableName);
-        const sql = `
-      SELECT
-        JSON_OBJECT (
-        'id', FORMAT ('0x%x', [t].[id]),
-        'name', [t].[Name],
-        'type', (
-          CASE
-            [t].[type]
-            WHEN 0 THEN 'Primary'
-            WHEN 1 THEN 'Joined'
-            WHEN 2 THEN 'Existing'
-            WHEN 3 THEN 'Overflow'
-            WHEN 4 THEN 'Virtual'
-          END
-        ),
-        'exclusiveRootClassId', FORMAT ('0x%x',
-          COALESCE (
-            [t].[ExclusiveRootClassId], (
-              SELECT [parent].[ExclusiveRootClassId]
-              FROM [ec_Table] [parent]
-              WHERE [parent].[Id] = [t].[ParentTableId] AND [parent].[Type] = 1))),
-        'isClassIdVirtual', (
-          SELECT
-            [c].[IsVirtual]
-          FROM
-            [ec_Column] [c]
-          WHERE
-            [c].[Name] = 'ECClassId' AND [c].[TableId] = [t].[Id]
-        )
-      )
-      FROM [ec_Table] [t]
-      WHERE
-        [t].[Name] = ?;
+        const sql = `
+      SELECT
+        JSON_OBJECT (
+        'id', FORMAT ('0x%x', [t].[id]),
+        'name', [t].[Name],
+        'type', (
+          CASE
+            [t].[type]
+            WHEN 0 THEN 'Primary'
+            WHEN 1 THEN 'Joined'
+            WHEN 2 THEN 'Existing'
+            WHEN 3 THEN 'Overflow'
+            WHEN 4 THEN 'Virtual'
+          END
+        ),
+        'exclusiveRootClassId', FORMAT ('0x%x',
+          COALESCE (
+            [t].[ExclusiveRootClassId], (
+              SELECT [parent].[ExclusiveRootClassId]
+              FROM [ec_Table] [parent]
+              WHERE [parent].[Id] = [t].[ParentTableId] AND [parent].[Type] = 1))),
+        'isClassIdVirtual', (
+          SELECT
+            [c].[IsVirtual]
+          FROM
+            [ec_Column] [c]
+          WHERE
+            [c].[Name] = 'ECClassId' AND [c].[TableId] = [t].[Id]
+        )
+      )
+      FROM [ec_Table] [t]
+      WHERE
+        [t].[Name] = ?;
     `;
         return this.db.withPreparedSqliteStatement(sql, (stmt) => {
             stmt.bindString(1, tableName);
@@ -92,186 +95,186 @@ class ECDbMap {
     getClassMap(classId) {
         if (this._cachedClassMaps.has(classId))
             return this._cachedClassMaps.get(classId);
-        const sql = `
-      SELECT
-      JSON_OBJECT(
-        'id', format('0x%x', cs.id),
-        'name', format('%s:%s', ss.Name, cs.Name),
-        'mapStrategy',
-        (
-          CASE cm.MapStrategy
-            WHEN 0 THEN 'NotMapped'
-            WHEN 1 THEN 'OwnTable'
-            WHEN 2 THEN 'TablePerHierarchy'
-            WHEN 3 THEN 'ExistingTable'
-            WHEN 10 THEN 'ForeignKeyInTargetTable'
-            WHEN 11 THEN 'ForeignKeyInSourceTable'
-          END
-        ),
-        'type',
-        (
-          CASE cs.Type
-            WHEN 0 THEN 'Entity'
-            WHEN 1 THEN 'Relationship'
-            WHEN 2 THEN 'Struct'
-            WHEN 3 THEN 'CustomAttribute'
-          END
-        ),
-        'modifier',
-        (
-          CASE cs.Modifier
-            WHEN 0 THEN 'None'
-            WHEN 1 THEN 'Abstract'
-            WHEN 2 THEN 'Sealed'
-          END
-        ),
-        'properties',
-        (
-          SELECT
-            JSON_GROUP_ARRAY(JSON(propJson))
-          FROM
-            (
-              SELECT
-                JSON_OBJECT(
-                  'id', format('0x%x', pt.id),
-                  'name', pt.Name,
-                  'kind',
-                  (
-                    CASE pt.Kind
-                      WHEN 0 THEN 'Primitive'
-                      WHEN 1 THEN 'Struct'
-                      WHEN 2 THEN 'PrimitiveArray'
-                      WHEN 3 THEN 'StructArray'
-                      WHEN 4 THEN 'Navigation'
-                    END
-                  ),
-                  'primitiveType',
-                  (
-                    CASE pt.PrimitiveType
-                      WHEN 0x101 THEN 'Binary'
-                      WHEN 0x201 THEN 'Boolean'
-                      WHEN 0x301 THEN 'DateTime'
-                      WHEN 0x401 THEN 'Double'
-                      WHEN 0x501 THEN 'Integer'
-                      WHEN 0x601 THEN 'Long'
-                      WHEN 0x701 THEN 'Point2d'
-                      WHEN 0x801 THEN 'Point3d'
-                      WHEN 0x901 THEN 'String'
-                      WHEN 0xa01 THEN 'IGeometry'
-                    END
-                  ),
-                  'extendedTypeName', ExtendedTypeName,
-                  'navigationRelationship',
-                  (
-                    SELECT
-                      JSON_OBJECT(
-                        'classId', format('0x%x', nc.Id),
-                        'classFullName', format('%s:%s', ns.Name, nc.Name)
-                      )
-                    FROM ec_Class nc
-                      JOIN ec_Schema ns ON ns.Id = nc.SchemaId
-                    WHERE
-                      nc.Id = pt.NavigationRelationshipClassId
-                  ),
-                  'structClass',
-                  (
-                    SELECT
-                      JSON_OBJECT(
-                        'classId', format('0x%x', nc.Id),
-                        'classFullName', format('%s:%s', ns.Name, nc.Name)
-                      )
-                    FROM ec_Class nc
-                      JOIN ec_Schema ns ON ns.Id = nc.SchemaId
-                    WHERE
-                      nc.Id = pt.StructClassId
-                  ),
-                  'dateTimeInfo', (
-                      SELECT
-                      JSON_OBJECT (
-                        'dateTimeKind', (
-                          CASE
-                            WHEN [ca].[Instance] LIKE '%<DateTimeKind>Utc</DateTimeKind>%' COLLATE [NoCase] THEN 'Utc'
-                            WHEN [ca].[Instance] LIKE '%<DateTimeKind>Local</DateTimeKind>%' COLLATE [NoCase] THEN 'Local'
-                            ELSE 'Unspecified'
-                          END
-                        ),
-                        'dateTimeComponent', (
-                          CASE
-                            WHEN [ca].[Instance] LIKE '%<DateTimeComponent>DateTime</DateTimeComponent>%' COLLATE [NoCase] THEN 'DateTime'
-                            WHEN [ca].[Instance] LIKE '%<DateTimeComponent>Date</DateTimeComponent>%' COLLATE [NoCase] THEN 'Date'
-                            WHEN [ca].[Instance] LIKE '%<DateTimeComponent>TimeOfDay</DateTimeComponent>%' COLLATE [NoCase] THEN 'TimeOfDay'
-                            ELSE 'DateTime'
-                          END
-                        )
-                      )
-                    FROM
-                      [ec_CustomAttribute] [ca]
-                      JOIN [ec_Class] [cl] ON [cl].[Id] = [ca].[ClassId]
-                      JOIN [ec_Schema] [sc] ON [sc].[Id] = [cl].[SchemaId]
-                    WHERE
-                      [ca].[ContainerType] = 992
-                      AND [cl].[Name] = 'DateTimeInfo'
-                      AND [sc].[Name] = 'CoreCustomAttributes'
-                      AND [ca].[ContainerId] = [pt].[Id]
-                  ),
-                  'columns',
-                  (
-                    SELECT
-                      JSON_GROUP_ARRAY(JSON(columnJson))
-                    FROM
-                      (
-                        SELECT
-                          JSON_OBJECT(
-                            'table', tb.Name,
-                            'column', cc.Name,
-                            'type',
-                            (
-                              CASE cc.Type
-                                WHEN 0 THEN 'Any'
-                                WHEN 1 THEN 'Boolean'
-                                WHEN 2 THEN 'Blob'
-                                WHEN 3 THEN 'Timestamp'
-                                WHEN 4 THEN 'Real'
-                                WHEN 5 THEN 'Integer'
-                                WHEN 6 THEN 'Text'
-                              END
-                            ),
-                            'columnKind',
-                            (
-                              CASE cc.ColumnKind
-                                WHEN 0 THEN 'Default'
-                                WHEN 1 THEN 'Id'
-                                WHEN 2 THEN 'ClassId'
-                                WHEN 4 THEN 'SharedData'
-                              END
-                            ),
-                            'accessString', pp0.AccessString,
-                            'isVirtual', cc.IsVirtual OR tb.Type = 4
-                          ) columnJson
-                        FROM [ec_PropertyMap] [pm0]
-                          JOIN [ec_Column] [cc] ON [cc].[Id] = [pm0].[ColumnId]
-                          JOIN [ec_Table] [tb] ON [tb].[Id] = [cc].[TableId]
-                          JOIN [ec_PropertyPath] [pp0] ON [pp0].[Id] = [pm0].[PropertyPathId]
-                        WHERE
-                          [pp0].[RootPropertyId] = pt.Id AND pm0.ClassId = cs.Id
-                      )
-                  )
-                ) propJson
-              FROM [ec_PropertyMap] [pm]
-                JOIN [ec_PropertyPath] [pp] ON [pp].[Id] = [pm].[PropertyPathId]
-                JOIN [ec_Property] [pt] ON [pt].[Id] = [pp].[RootPropertyId]
-              WHERE
-                pm.ClassId = cs.Id
-              GROUP BY
-                pt.Id
-            )
-        )
-      ) classDef
-    FROM [ec_Class] [cs]
-      JOIN [ec_ClassMap] [cm] ON [cm].[ClassId] = [cs].[Id]
-      JOIN [ec_Schema] [ss] ON [ss].[Id] = [cs].[SchemaId]
-    WHERE
-      [cs].[Id] = ?
+        const sql = `
+      SELECT
+      JSON_OBJECT(
+        'id', format('0x%x', cs.id),
+        'name', format('%s:%s', ss.Name, cs.Name),
+        'mapStrategy',
+        (
+          CASE cm.MapStrategy
+            WHEN 0 THEN 'NotMapped'
+            WHEN 1 THEN 'OwnTable'
+            WHEN 2 THEN 'TablePerHierarchy'
+            WHEN 3 THEN 'ExistingTable'
+            WHEN 10 THEN 'ForeignKeyInTargetTable'
+            WHEN 11 THEN 'ForeignKeyInSourceTable'
+          END
+        ),
+        'type',
+        (
+          CASE cs.Type
+            WHEN 0 THEN 'Entity'
+            WHEN 1 THEN 'Relationship'
+            WHEN 2 THEN 'Struct'
+            WHEN 3 THEN 'CustomAttribute'
+          END
+        ),
+        'modifier',
+        (
+          CASE cs.Modifier
+            WHEN 0 THEN 'None'
+            WHEN 1 THEN 'Abstract'
+            WHEN 2 THEN 'Sealed'
+          END
+        ),
+        'properties',
+        (
+          SELECT
+            JSON_GROUP_ARRAY(JSON(propJson))
+          FROM
+            (
+              SELECT
+                JSON_OBJECT(
+                  'id', format('0x%x', pt.id),
+                  'name', pt.Name,
+                  'kind',
+                  (
+                    CASE pt.Kind
+                      WHEN 0 THEN 'Primitive'
+                      WHEN 1 THEN 'Struct'
+                      WHEN 2 THEN 'PrimitiveArray'
+                      WHEN 3 THEN 'StructArray'
+                      WHEN 4 THEN 'Navigation'
+                    END
+                  ),
+                  'primitiveType',
+                  (
+                    CASE pt.PrimitiveType
+                      WHEN 0x101 THEN 'Binary'
+                      WHEN 0x201 THEN 'Boolean'
+                      WHEN 0x301 THEN 'DateTime'
+                      WHEN 0x401 THEN 'Double'
+                      WHEN 0x501 THEN 'Integer'
+                      WHEN 0x601 THEN 'Long'
+                      WHEN 0x701 THEN 'Point2d'
+                      WHEN 0x801 THEN 'Point3d'
+                      WHEN 0x901 THEN 'String'
+                      WHEN 0xa01 THEN 'IGeometry'
+                    END
+                  ),
+                  'extendedTypeName', ExtendedTypeName,
+                  'navigationRelationship',
+                  (
+                    SELECT
+                      JSON_OBJECT(
+                        'classId', format('0x%x', nc.Id),
+                        'classFullName', format('%s:%s', ns.Name, nc.Name)
+                      )
+                    FROM ec_Class nc
+                      JOIN ec_Schema ns ON ns.Id = nc.SchemaId
+                    WHERE
+                      nc.Id = pt.NavigationRelationshipClassId
+                  ),
+                  'structClass',
+                  (
+                    SELECT
+                      JSON_OBJECT(
+                        'classId', format('0x%x', nc.Id),
+                        'classFullName', format('%s:%s', ns.Name, nc.Name)
+                      )
+                    FROM ec_Class nc
+                      JOIN ec_Schema ns ON ns.Id = nc.SchemaId
+                    WHERE
+                      nc.Id = pt.StructClassId
+                  ),
+                  'dateTimeInfo', (
+                      SELECT
+                      JSON_OBJECT (
+                        'dateTimeKind', (
+                          CASE
+                            WHEN [ca].[Instance] LIKE '%<DateTimeKind>Utc</DateTimeKind>%' COLLATE [NoCase] THEN 'Utc'
+                            WHEN [ca].[Instance] LIKE '%<DateTimeKind>Local</DateTimeKind>%' COLLATE [NoCase] THEN 'Local'
+                            ELSE 'Unspecified'
+                          END
+                        ),
+                        'dateTimeComponent', (
+                          CASE
+                            WHEN [ca].[Instance] LIKE '%<DateTimeComponent>DateTime</DateTimeComponent>%' COLLATE [NoCase] THEN 'DateTime'
+                            WHEN [ca].[Instance] LIKE '%<DateTimeComponent>Date</DateTimeComponent>%' COLLATE [NoCase] THEN 'Date'
+                            WHEN [ca].[Instance] LIKE '%<DateTimeComponent>TimeOfDay</DateTimeComponent>%' COLLATE [NoCase] THEN 'TimeOfDay'
+                            ELSE 'DateTime'
+                          END
+                        )
+                      )
+                    FROM
+                      [ec_CustomAttribute] [ca]
+                      JOIN [ec_Class] [cl] ON [cl].[Id] = [ca].[ClassId]
+                      JOIN [ec_Schema] [sc] ON [sc].[Id] = [cl].[SchemaId]
+                    WHERE
+                      [ca].[ContainerType] = 992
+                      AND [cl].[Name] = 'DateTimeInfo'
+                      AND [sc].[Name] = 'CoreCustomAttributes'
+                      AND [ca].[ContainerId] = [pt].[Id]
+                  ),
+                  'columns',
+                  (
+                    SELECT
+                      JSON_GROUP_ARRAY(JSON(columnJson))
+                    FROM
+                      (
+                        SELECT
+                          JSON_OBJECT(
+                            'table', tb.Name,
+                            'column', cc.Name,
+                            'type',
+                            (
+                              CASE cc.Type
+                                WHEN 0 THEN 'Any'
+                                WHEN 1 THEN 'Boolean'
+                                WHEN 2 THEN 'Blob'
+                                WHEN 3 THEN 'Timestamp'
+                                WHEN 4 THEN 'Real'
+                                WHEN 5 THEN 'Integer'
+                                WHEN 6 THEN 'Text'
+                              END
+                            ),
+                            'columnKind',
+                            (
+                              CASE cc.ColumnKind
+                                WHEN 0 THEN 'Default'
+                                WHEN 1 THEN 'Id'
+                                WHEN 2 THEN 'ClassId'
+                                WHEN 4 THEN 'SharedData'
+                              END
+                            ),
+                            'accessString', pp0.AccessString,
+                            'isVirtual', cc.IsVirtual OR tb.Type = 4
+                          ) columnJson
+                        FROM [ec_PropertyMap] [pm0]
+                          JOIN [ec_Column] [cc] ON [cc].[Id] = [pm0].[ColumnId]
+                          JOIN [ec_Table] [tb] ON [tb].[Id] = [cc].[TableId]
+                          JOIN [ec_PropertyPath] [pp0] ON [pp0].[Id] = [pm0].[PropertyPathId]
+                        WHERE
+                          [pp0].[RootPropertyId] = pt.Id AND pm0.ClassId = cs.Id
+                      )
+                  )
+                ) propJson
+              FROM [ec_PropertyMap] [pm]
+                JOIN [ec_PropertyPath] [pp] ON [pp].[Id] = [pm].[PropertyPathId]
+                JOIN [ec_Property] [pt] ON [pt].[Id] = [pp].[RootPropertyId]
+              WHERE
+                pm.ClassId = cs.Id
+              GROUP BY
+                pt.Id
+            )
+        )
+      ) classDef
+    FROM [ec_Class] [cs]
+      JOIN [ec_ClassMap] [cm] ON [cm].[ClassId] = [cs].[Id]
+      JOIN [ec_Schema] [ss] ON [ss].[Id] = [cs].[SchemaId]
+    WHERE
+      [cs].[Id] = ?
     `;
         return this.db.withPreparedSqliteStatement(sql, (stmt) => {
             stmt.bindId(1, classId);
@@ -321,6 +324,186 @@ var DateTime;
     }
     DateTime.fromJulianDay = fromJulianDay;
 })(DateTime || (DateTime = {}));
+var ECChangeUnifierCache;
+(function (ECChangeUnifierCache) {
+    function createInMemory() {
+        return new InMemoryInstanceCache();
+    }
+    ECChangeUnifierCache.createInMemory = createInMemory;
+    function createSqliteBacked(db, bufferedReadInstanceSizeInBytes = 1024 * 1024 * 10) {
+        return new SqliteBackedInstanceCache(db, bufferedReadInstanceSizeInBytes);
+    }
+    ECChangeUnifierCache.createSqliteBacked = createSqliteBacked;
+})(ECChangeUnifierCache || (exports.ECChangeUnifierCache = ECChangeUnifierCache = {}));
+/**
+ * In-memory cache for storing changed EC instances.
+ */
+class InMemoryInstanceCache {
+    _cache = new Map();
+    /**
+     * Retrieves the changed EC instance associated with the specified key.
+     * @param key - The key used to retrieve the instance.
+     * @returns The changed EC instance, or undefined if not found.
+     */
+    get(key) {
+        return this._cache.get(key);
+    }
+    /**
+     * Sets the changed EC instance associated with the specified key.
+     * @param key - The key used to store the instance.
+     * @param value - The changed EC instance to be stored.
+     */
+    set(key, value) {
+        const meta = value.$meta;
+        // Remove undefined keys
+        if (meta) {
+            Object.keys(meta).forEach((k) => meta[k] === undefined && delete meta[k]);
+        }
+        this._cache.set(key, value);
+    }
+    /**
+     * Returns an iterator over all the changed EC instances in the cache.
+     * @returns An iterator over all the changed EC instances.
+     */
+    *all() {
+        for (const key of Array.from(this._cache.keys()).sort()) {
+            const instance = this._cache.get(key);
+            if (instance) {
+                yield instance;
+            }
+        }
+    }
+    /**
+     * Returns the number of changed EC instances in the cache.
+     * @returns The number of changed EC instances.
+     */
+    count() {
+        return this._cache.size;
+    }
+    /**
+     * Disposes the cache.
+     */
+    [Symbol.dispose]() {
+        // Implementation details
+    }
+}
+/**
+ * Represents a cache for unifying EC changes in a SQLite-backed instance cache.
+ */
+class SqliteBackedInstanceCache {
+    _db;
+    bufferedReadInstanceSizeInBytes;
+    _cacheTable = `[temp].[${core_bentley_1.Guid.createValue()}]`;
+    static defaultBufferSize = 1024 * 1024 * 10; // 10MB
+    /**
+     * Creates an instance of SqliteBackedInstanceCache.
+     * @param _db The underlying database connection.
+     * @param bufferedReadInstanceSizeInBytes The size of read instance buffer defaults to 10Mb.
+     * @throws Error if bufferedReadInstanceSizeInBytes is less than or equal to 0.
+     */
+    constructor(_db, bufferedReadInstanceSizeInBytes = SqliteBackedInstanceCache.defaultBufferSize) {
+        this._db = _db;
+        this.bufferedReadInstanceSizeInBytes = bufferedReadInstanceSizeInBytes;
+        if (bufferedReadInstanceSizeInBytes <= 0)
+            throw new Error("bufferedReadInstanceCount must be greater than 0");
+        this.createTempTable();
+    }
+    /**
+     * Creates a temporary table in the database for caching instances.
+     * @throws Error if unable to create the temporary table.
+     */
+    createTempTable() {
+        this._db.withSqliteStatement(`CREATE TABLE ${this._cacheTable} ([key] text primary key, [value] text)`, (stmt) => {
+            if (core_bentley_1.DbResult.BE_SQLITE_DONE !== stmt.step())
+                throw new Error("unable to create temp table");
+        });
+    }
+    /**
+     * Drops the temporary table from the database.
+     * @throws Error if unable to drop the temporary table.
+     */
+    dropTempTable() {
+        this._db.saveChanges();
+        if (this._db instanceof ECDb_1.ECDb)
+            this._db.clearStatementCache();
+        else {
+            this._db.clearCaches();
+            this._db[Symbols_1._nativeDb].clearECDbCache();
+        }
+        this._db.withSqliteStatement(`DROP TABLE IF EXISTS ${this._cacheTable}`, (stmt) => {
+            if (core_bentley_1.DbResult.BE_SQLITE_DONE !== stmt.step())
+                throw new Error("unable to drop temp table");
+        });
+    }
+    /**
+     * Retrieves the changed EC instance from the cache based on the specified key.
+     * @param key The key of the instance.
+     * @returns The changed EC instance if found, otherwise undefined.
+     */
+    get(key) {
+        return this._db.withPreparedSqliteStatement(`SELECT [value] FROM ${this._cacheTable} WHERE [key]=?`, (stmt) => {
+            stmt.bindString(1, key);
+            if (stmt.step() === core_bentley_1.DbResult.BE_SQLITE_ROW) {
+                const out = JSON.parse(stmt.getValueString(0), core_common_1.Base64EncodedString.reviver);
+                return out;
+            }
+            return undefined;
+        });
+    }
+    /**
+     * Sets the changed EC instance in the cache with the specified key.
+     * @param key The key of the instance.
+     * @param value The changed EC instance to be set.
+     */
+    set(key, value) {
+        const shallowCopy = Object.assign({}, value);
+        this._db.withPreparedSqliteStatement(`INSERT INTO ${this._cacheTable} ([key], [value]) VALUES (?, ?) ON CONFLICT ([key]) DO UPDATE SET [value] = [excluded].[value]`, (stmt) => {
+            stmt.bindString(1, key);
+            stmt.bindString(2, JSON.stringify(shallowCopy, core_common_1.Base64EncodedString.replacer));
+            stmt.step();
+        });
+    }
+    /**
+     * Returns an iterator for all the changed EC instances in the cache.
+     * @returns An iterator for all the changed EC instances.
+     */
+    *all() {
+        const sql = `
+      SELECT JSON_GROUP_ARRAY (JSON([value]))
+      FROM   (SELECT
+                    [value],
+                    SUM (LENGTH ([value])) OVER (ORDER BY [key] ROWS UNBOUNDED PRECEDING) / ${this.bufferedReadInstanceSizeInBytes} AS [bucket]
+              FROM   ${this._cacheTable})
+      GROUP  BY [bucket]`;
+        const stmt = this._db.prepareSqliteStatement(sql);
+        while (stmt.step() === core_bentley_1.DbResult.BE_SQLITE_ROW) {
+            const instanceBucket = JSON.parse(stmt.getValueString(0), core_common_1.Base64EncodedString.reviver);
+            for (const value of instanceBucket) {
+                yield value;
+            }
+        }
+        stmt[Symbol.dispose]();
+    }
+    /**
+     * Returns the number of instances in the cache.
+     * @returns The number of instances in the cache.
+     */
+    count() {
+        return this._db.withPreparedSqliteStatement(`SELECT COUNT(*) FROM ${this._cacheTable}`, (stmt) => {
+            if (stmt.step() === core_bentley_1.DbResult.BE_SQLITE_ROW)
+                return stmt.getValue(0).getInteger();
+            return 0;
+        });
+    }
+    /**
+     * Disposes the cache by dropping the temporary table.
+     */
+    [Symbol.dispose]() {
+        if (this._db.isOpen) {
+            this.dropTempTable();
+        }
+    }
+}
 /**
  * Combine partial changed instance into single instance.
  * Partial changes is per table and a single instance can
@@ -328,38 +511,49 @@ var DateTime;
  * @beta
  */
 class PartialECChangeUnifier {
-    _cache = new Map();
+    _db;
+    _cache;
     _readonly = false;
+    constructor(_db, _cache = new InMemoryInstanceCache()) {
+        this._db = _db;
+        this._cache = _cache;
+    }
+    /**
+     * Dispose the instance.
+     */
+    [Symbol.dispose]() {
+        this._cache[Symbol.dispose]();
+    }
     /**
      * Get root class id for a given class
      * @param classId given class id
      * @param db use to find root class
      * @returns return root class id
      */
-    static getRootClassId(classId, db) {
-        const sql = `
-      WITH
-      [base_class]([classId], [baseClassId], [Level]) AS(
-        SELECT [ch].[ClassId], [ch].[BaseClassId], 0
-        FROM   [ec_ClassHasBaseClasses] [ch] WHERE  [ch].[ClassId] = ?
-        UNION ALL
-        SELECT [ch].[ClassId], [ch].[BaseClassId], [Level] + 1
-        FROM   [ec_ClassHasBaseClasses] [ch], [base_class] [bc] WHERE  [bc].[BaseClassId] = [ch].[ClassId]
-
-      )
-      SELECT FORMAT('0x%x', [bc].[BaseClassId]) rootClass
-      FROM   [base_class] [bc]
-      WHERE  [bc].[ClassId] <> [bc].[BaseClassId]
-              AND [bc].[BaseClassId] NOT IN (SELECT [ca].[ContainerId]
-            FROM   [ec_CustomAttribute] [ca]
-            WHERE  [ca].[ContainerType] = 30
-                      AND [ca].[ClassId] IN (SELECT [cc].[Id]
-                    FROM   [ec_Class] [cc]
-                          JOIN [ec_Schema] [ss] ON [ss].[Id] = [cc].[SchemaId]
-                    WHERE  [cc].[Name] = 'IsMixIn'
-                            AND [ss].[Name] = 'CoreCustomAttributes'))
+    getRootClassId(classId) {
+        const sql = `
+      WITH
+      [base_class]([classId], [baseClassId], [Level]) AS(
+        SELECT [ch].[ClassId], [ch].[BaseClassId], 0
+        FROM   [ec_ClassHasBaseClasses] [ch] WHERE  [ch].[ClassId] = ?
+        UNION ALL
+        SELECT [ch].[ClassId], [ch].[BaseClassId], [Level] + 1
+        FROM   [ec_ClassHasBaseClasses] [ch], [base_class] [bc] WHERE  [bc].[BaseClassId] = [ch].[ClassId]
+
+      )
+      SELECT FORMAT('0x%x', [bc].[BaseClassId]) rootClass
+      FROM   [base_class] [bc]
+      WHERE  [bc].[ClassId] <> [bc].[BaseClassId]
+              AND [bc].[BaseClassId] NOT IN (SELECT [ca].[ContainerId]
+            FROM   [ec_CustomAttribute] [ca]
+            WHERE  [ca].[ContainerType] = 30
+                      AND [ca].[ClassId] IN (SELECT [cc].[Id]
+                    FROM   [ec_Class] [cc]
+                          JOIN [ec_Schema] [ss] ON [ss].[Id] = [cc].[SchemaId]
+                    WHERE  [cc].[Name] = 'IsMixIn'
+                            AND [ss].[Name] = 'CoreCustomAttributes'))
       ORDER BY [Level] DESC`;
-        return db.withSqliteStatement(sql, (stmt) => {
+        return this._db.withSqliteStatement(sql, (stmt) => {
             stmt.bindId(1, classId);
             if (stmt.step() === core_bentley_1.DbResult.BE_SQLITE_ROW && !stmt.isValueNull(0)) {
                 return stmt.getValueString(0);
@@ -367,15 +561,30 @@ class PartialECChangeUnifier {
             return classId;
         });
     }
+    /**
+     * Checks if the given `rhsClassId` is an instance of the `lhsClassId`.
+     * @param rhsClassId The ID of the right-hand side class.
+     * @param lhsClassId The ID of the left-hand side class.
+     * @returns `true` if `rhsClassId` is an instance of `lhsClassId`, `false` otherwise.
+     */
+    instanceOf(rhsClassId, lhsClassId) {
+        // eslint-disable-next-line @typescript-eslint/no-deprecated
+        return this._db.withPreparedStatement("SELECT ec_instanceof(?,?)", (stmt) => {
+            stmt.bindId(1, rhsClassId);
+            stmt.bindId(2, lhsClassId);
+            stmt.step();
+            return stmt.getValue(0).getInteger() === 1;
+        });
+    }
     /**
      * Combine partial instance with instance with same key if already exists.
      * @param rhs partial instance
      */
-    combine(rhs, db) {
+    combine(rhs) {
         if (!rhs.$meta) {
             throw new Error("PartialECChange being combine must have '$meta' property");
         }
-        const key = PartialECChangeUnifier.buildKey(rhs, db);
+        const key = this.buildKey(rhs);
         const lhs = this._cache.get(key);
         if (lhs) {
             const { $meta: _, ...restOfRhs } = rhs;
@@ -384,37 +593,39 @@ class PartialECChangeUnifier {
                 lhs.$meta.tables = [...rhs.$meta?.tables, ...lhs.$meta?.tables];
                 lhs.$meta.changeIndexes = [...rhs.$meta?.changeIndexes, ...lhs.$meta?.changeIndexes];
                 // we preserve child class name & id when merging instance.
-                if (rhs.$meta.fallbackClassId && lhs.$meta.fallbackClassId && db && rhs.$meta.fallbackClassId !== lhs.$meta.fallbackClassId) {
+                if (rhs.$meta.fallbackClassId && lhs.$meta.fallbackClassId && rhs.$meta.fallbackClassId !== lhs.$meta.fallbackClassId) {
                     const lhsClassId = lhs.$meta.fallbackClassId;
                     const rhsClassId = rhs.$meta.fallbackClassId;
-                    // eslint-disable-next-line @typescript-eslint/no-deprecated
-                    const isRhsIsSubClassOfLhs = db.withPreparedStatement("SELECT ec_instanceof(?,?)", (stmt) => {
-                        stmt.bindId(1, rhsClassId);
-                        stmt.bindId(2, lhsClassId);
-                        stmt.step();
-                        return stmt.getValue(0).getInteger() === 1;
-                    });
+                    const isRhsIsSubClassOfLhs = this.instanceOf(rhsClassId, lhsClassId);
                     if (isRhsIsSubClassOfLhs) {
                         lhs.$meta.fallbackClassId = rhs.$meta.fallbackClassId;
                         lhs.$meta.classFullName = rhs.$meta.classFullName;
                     }
                 }
             }
+            this._cache.set(key, lhs);
         }
         else {
             this._cache.set(key, rhs);
         }
     }
+    /**
+     * Returns the number of instances in the cache.
+     * @returns The number of instances in the cache.
+     */
+    getInstanceCount() {
+        return this._cache.count();
+    }
     /**
      * Build key from EC change.
      * @param change EC change
      * @returns key created from EC change.
      */
-    static buildKey(change, db) {
+    buildKey(change) {
         let classId = change.ECClassId;
         if (typeof classId === "undefined") {
-            if (db && change.$meta?.fallbackClassId) {
-                classId = this.getRootClassId(change.$meta.fallbackClassId, db);
+            if (change.$meta?.fallbackClassId) {
+                classId = this.getRootClassId(change.$meta.fallbackClassId);
             }
             if (typeof classId === "undefined") {
                 throw new Error(`unable to resolve ECClassId to root class id.`);
@@ -437,32 +648,23 @@ class PartialECChangeUnifier {
             throw new Error("this instance is marked as readonly.");
         }
         if (adaptor.op === "Updated" && adaptor.inserted && adaptor.deleted) {
-            this.combine(adaptor.inserted, adaptor.reader.db);
-            this.combine(adaptor.deleted, adaptor.reader.db);
+            this.combine(adaptor.inserted);
+            this.combine(adaptor.deleted);
         }
         else if (adaptor.op === "Inserted" && adaptor.inserted) {
-            this.combine(adaptor.inserted, adaptor.reader.db);
+            this.combine(adaptor.inserted);
         }
         else if (adaptor.op === "Deleted" && adaptor.deleted) {
-            this.combine(adaptor.deleted, adaptor.reader.db);
-        }
-    }
-    /**
-     * Delete $meta from all the instances.
-     */
-    stripMetaData() {
-        for (const inst of this._cache.values()) {
-            if ("$meta" in inst) {
-                delete inst.$meta;
-            }
+            this.combine(adaptor.deleted);
         }
-        this._readonly = true;
     }
     /**
      * Returns complete EC change instances.
      * @beta
      */
-    get instances() { return this._cache.values(); }
+    get instances() {
+        return this._cache.all();
+    }
 }
 exports.PartialECChangeUnifier = PartialECChangeUnifier;
 /**