@itwin/core-backend 4.0.0-dev.1 → 4.0.0-dev.100

package/lib/cjs/ServerBasedLocks.js

@@ -1,169 +1,169 @@
-"use strict";
-/*---------------------------------------------------------------------------------------------
-* Copyright (c) Bentley Systems, Incorporated. All rights reserved.
-* See LICENSE.md in the project root for license terms and full copyright notice.
-*--------------------------------------------------------------------------------------------*/
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.ServerBasedLocks = void 0;
-/** @packageDocumentation
- * @module iModels
- */
-const core_bentley_1 = require("@itwin/core-bentley");
-const core_common_1 = require("@itwin/core-common");
-const BackendHubAccess_1 = require("./BackendHubAccess");
-const IModelHost_1 = require("./IModelHost");
-const SQLiteDb_1 = require("./SQLiteDb");
-/** @internal */
-class ServerBasedLocks {
-    constructor(iModel) {
-        this.lockDb = new SQLiteDb_1.SQLiteDb();
-        this.briefcase = iModel;
-        const dbName = `${iModel.nativeDb.getTempFileBaseName()}-locks`;
-        try {
-            this.lockDb.openDb(dbName, core_bentley_1.OpenMode.ReadWrite);
-        }
-        catch (_e) {
-            this.lockDb.createDb(dbName);
-            this.lockDb.executeSQL("CREATE TABLE locks(id INTEGER PRIMARY KEY NOT NULL,state INTEGER NOT NULL,origin INTEGER)");
-            this.lockDb.saveChanges();
-        }
-    }
-    get isServerBased() { return true; }
-    close() {
-        if (this.lockDb.isOpen)
-            this.lockDb.closeDb();
-    }
-    getOwners(id) {
-        return this.briefcase.withPreparedSqliteStatement("SELECT ModelId,ParentId FROM bis_Element WHERE id=?", (stmt) => {
-            stmt.bindId(1, id);
-            const rc = stmt.step();
-            if (core_bentley_1.DbResult.BE_SQLITE_ROW !== rc)
-                throw new core_common_1.IModelError(rc, `element ${id} not found`);
-            return { modelId: stmt.getValueId(0), parentId: stmt.getValueId(1) };
-        });
-    }
-    getLockState(id) {
-        return (id === undefined || !core_bentley_1.Id64.isValid(id)) ? undefined : this.lockDb.withPreparedSqliteStatement("SELECT state FROM locks WHERE id=?", (stmt) => {
-            stmt.bindId(1, id);
-            return (core_bentley_1.DbResult.BE_SQLITE_ROW === stmt.step()) ? stmt.getValueInteger(0) : undefined;
-        });
-    }
-    /** Clear the cache of locally held locks.
-     * Note: does *not* release locks from server.
-     */
-    clearAllLocks() {
-        this.lockDb.executeSQL("DELETE FROM locks");
-        this.lockDb.saveChanges();
-    }
-    /** only for tests */
-    getLockCount(state) {
-        return this.lockDb.withSqliteStatement("SELECT count(*) FROM locks WHERE state=?", (stmt) => {
-            stmt.bindInteger(1, state);
-            stmt.step();
-            return stmt.getValueInteger(0);
-        });
-    }
-    async releaseAllLocks() {
-        await IModelHost_1.IModelHost.hubAccess.releaseAllLocks(this.briefcase); // throws if unsuccessful
-        this.clearAllLocks();
-    }
-    insertLock(id, state, origin) {
-        this.lockDb.withPreparedSqliteStatement("INSERT INTO locks(id,state,origin) VALUES (?,?,?) ON CONFLICT(id) DO UPDATE SET state=excluded.state,origin=excluded.origin", (stmt) => {
-            stmt.bindId(1, id);
-            stmt.bindInteger(2, state);
-            stmt.bindInteger(3, origin);
-            const rc = stmt.step();
-            if (core_bentley_1.DbResult.BE_SQLITE_DONE !== rc)
-                throw new core_common_1.IModelError(rc, "can't insert lock into database");
-        });
-        return true;
-    }
-    ownerHoldsExclusiveLock(id) {
-        if (id === undefined || id === core_common_1.IModel.rootSubjectId)
-            return false; // has no owners
-        const { modelId, parentId } = this.getOwners(id);
-        if (this.getLockState(modelId) === BackendHubAccess_1.LockState.Exclusive || this.getLockState(parentId) === BackendHubAccess_1.LockState.Exclusive)
-            return true;
-        // see if this model is exclusively locked by one of its owners. If so, save that fact on modelId so future tests won't have to descend.
-        if (this.ownerHoldsExclusiveLock(modelId))
-            return this.insertLock(modelId, BackendHubAccess_1.LockState.Exclusive, 2 /* Discovered */);
-        // see if the parent is exclusively locked by one of its owners. If so, save that fact on parentId so future tests won't have to descend.
-        return this.ownerHoldsExclusiveLock(parentId) ? this.insertLock(parentId, BackendHubAccess_1.LockState.Exclusive, 2 /* Discovered */) : false; // eslint-disable-line @typescript-eslint/no-non-null-assertion
-    }
-    /** Determine whether an the exclusive lock is already held by an element (or one of its owners) */
-    holdsExclusiveLock(id) {
-        // see if we hold the exclusive lock. or if one of the element's owners is exclusively locked (recursively)
-        return this.getLockState(id) === BackendHubAccess_1.LockState.Exclusive || this.ownerHoldsExclusiveLock(id);
-    }
-    holdsSharedLock(id) {
-        const state = this.getLockState(id);
-        // see if we hold shared or exclusive lock, or if an owner has exclusive lock. If so we implicitly have shared lock, but owner holding shared lock doesn't help.
-        return (state === BackendHubAccess_1.LockState.Shared || state === BackendHubAccess_1.LockState.Exclusive) || this.ownerHoldsExclusiveLock(id);
-    }
-    /** if the shared lock on the element supplied is not already held, add it to the set of shared locks required. Then, check owners. */
-    addSharedLock(id, locks) {
-        // if the id is not valid, or of the lock is already in the set, or if we already hold a shared lock, we're done
-        // Note: if we hold a shared lock, it is guaranteed that we also hold all required shared locks on owners.
-        if (id === undefined || !core_bentley_1.Id64.isValid(id) || locks.has(id) || this.holdsSharedLock(id))
-            return;
-        locks.add(id); // add to set of needed shared locks
-        this.addOwnerSharedLocks(id, locks); // check parent models and groups
-    }
-    /** add owners (recursively) of an element to a list of required shared locks, if not already held. */
-    addOwnerSharedLocks(id, locks) {
-        const el = this.getOwners(id);
-        this.addSharedLock(el.parentId, locks); // if this element is in a group
-        this.addSharedLock(el.modelId, locks); // check its model
-    }
-    /** attempt to acquire all necessary locks for a set of elements */
-    async acquireAllLocks(locks) {
-        if (locks.size === 0) // no locks are required.
-            return;
-        const sharedLocks = new Set();
-        for (const lock of locks)
-            this.addOwnerSharedLocks(lock[0], sharedLocks);
-        for (const shared of sharedLocks) {
-            if (!locks.has(shared)) // we may already be asking for exclusive lock
-                locks.set(shared, BackendHubAccess_1.LockState.Shared);
-        }
-        await IModelHost_1.IModelHost.hubAccess.acquireLocks(this.briefcase, locks); // throws if unsuccessful
-        for (const lock of locks)
-            this.insertLock(lock[0], lock[1], 0 /* Acquired */);
-        this.lockDb.saveChanges();
-    }
-    async acquireLocks(arg) {
-        const locks = new Map();
-        if (arg.shared) {
-            for (const id of core_bentley_1.Id64.iterable(arg.shared)) {
-                if (!this.holdsSharedLock(id))
-                    locks.set(id, BackendHubAccess_1.LockState.Shared);
-            }
-        }
-        if (arg.exclusive) {
-            for (const id of core_bentley_1.Id64.iterable(arg.exclusive)) {
-                if (!this.holdsExclusiveLock(id))
-                    locks.set(id, BackendHubAccess_1.LockState.Exclusive);
-            }
-        }
-        return this.acquireAllLocks(locks);
-    }
-    /** When an element is newly created in a session, we hold the lock on it implicitly. Save that fact. */
-    elementWasCreated(id) {
-        this.insertLock(id, BackendHubAccess_1.LockState.Exclusive, 1 /* NewElement */);
-        this.lockDb.saveChanges();
-    }
-    /** locks are not necessary during change propagation. */
-    get _locksAreRequired() { return !this.briefcase.txns.isIndirectChanges; }
-    /** throw if locks are currently required and the exclusive lock is not held on the supplied element */
-    checkExclusiveLock(id, type, operation) {
-        if (this._locksAreRequired && !this.holdsExclusiveLock(id))
-            throw new core_common_1.IModelError(core_bentley_1.IModelStatus.LockNotHeld, `exclusive lock not held on ${type} for ${operation} (id=${id})`);
-    }
-    /** throw if locks are currently required and a shared lock is not held on the supplied element */
-    checkSharedLock(id, type, operation) {
-        if (this._locksAreRequired && !this.holdsSharedLock(id))
-            throw new core_common_1.IModelError(core_bentley_1.IModelStatus.LockNotHeld, `shared lock not held on ${type} for ${operation} (id=${id})`);
-    }
-}
-exports.ServerBasedLocks = ServerBasedLocks;
+"use strict";
+/*---------------------------------------------------------------------------------------------
+* Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+* See LICENSE.md in the project root for license terms and full copyright notice.
+*--------------------------------------------------------------------------------------------*/
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ServerBasedLocks = void 0;
+/** @packageDocumentation
+ * @module iModels
+ */
+const core_bentley_1 = require("@itwin/core-bentley");
+const core_common_1 = require("@itwin/core-common");
+const BackendHubAccess_1 = require("./BackendHubAccess");
+const IModelHost_1 = require("./IModelHost");
+const SQLiteDb_1 = require("./SQLiteDb");
+/** @internal */
+class ServerBasedLocks {
+    get isServerBased() { return true; }
+    constructor(iModel) {
+        this.lockDb = new SQLiteDb_1.SQLiteDb();
+        this.briefcase = iModel;
+        const dbName = `${iModel.nativeDb.getTempFileBaseName()}-locks`;
+        try {
+            this.lockDb.openDb(dbName, core_bentley_1.OpenMode.ReadWrite);
+        }
+        catch (_e) {
+            this.lockDb.createDb(dbName);
+            this.lockDb.executeSQL("CREATE TABLE locks(id INTEGER PRIMARY KEY NOT NULL,state INTEGER NOT NULL,origin INTEGER)");
+            this.lockDb.saveChanges();
+        }
+    }
+    close() {
+        if (this.lockDb.isOpen)
+            this.lockDb.closeDb();
+    }
+    getOwners(id) {
+        return this.briefcase.withPreparedSqliteStatement("SELECT ModelId,ParentId FROM bis_Element WHERE id=?", (stmt) => {
+            stmt.bindId(1, id);
+            const rc = stmt.step();
+            if (core_bentley_1.DbResult.BE_SQLITE_ROW !== rc)
+                throw new core_common_1.IModelError(rc, `element ${id} not found`);
+            return { modelId: stmt.getValueId(0), parentId: stmt.getValueId(1) };
+        });
+    }
+    getLockState(id) {
+        return (id === undefined || !core_bentley_1.Id64.isValid(id)) ? undefined : this.lockDb.withPreparedSqliteStatement("SELECT state FROM locks WHERE id=?", (stmt) => {
+            stmt.bindId(1, id);
+            return (core_bentley_1.DbResult.BE_SQLITE_ROW === stmt.step()) ? stmt.getValueInteger(0) : undefined;
+        });
+    }
+    /** Clear the cache of locally held locks.
+     * Note: does *not* release locks from server.
+     */
+    clearAllLocks() {
+        this.lockDb.executeSQL("DELETE FROM locks");
+        this.lockDb.saveChanges();
+    }
+    /** only for tests */
+    getLockCount(state) {
+        return this.lockDb.withSqliteStatement("SELECT count(*) FROM locks WHERE state=?", (stmt) => {
+            stmt.bindInteger(1, state);
+            stmt.step();
+            return stmt.getValueInteger(0);
+        });
+    }
+    async releaseAllLocks() {
+        await IModelHost_1.IModelHost.hubAccess.releaseAllLocks(this.briefcase); // throws if unsuccessful
+        this.clearAllLocks();
+    }
+    insertLock(id, state, origin) {
+        this.lockDb.withPreparedSqliteStatement("INSERT INTO locks(id,state,origin) VALUES (?,?,?) ON CONFLICT(id) DO UPDATE SET state=excluded.state,origin=excluded.origin", (stmt) => {
+            stmt.bindId(1, id);
+            stmt.bindInteger(2, state);
+            stmt.bindInteger(3, origin);
+            const rc = stmt.step();
+            if (core_bentley_1.DbResult.BE_SQLITE_DONE !== rc)
+                throw new core_common_1.IModelError(rc, "can't insert lock into database");
+        });
+        return true;
+    }
+    ownerHoldsExclusiveLock(id) {
+        if (id === undefined || id === core_common_1.IModel.rootSubjectId)
+            return false; // has no owners
+        const { modelId, parentId } = this.getOwners(id);
+        if (this.getLockState(modelId) === BackendHubAccess_1.LockState.Exclusive || this.getLockState(parentId) === BackendHubAccess_1.LockState.Exclusive)
+            return true;
+        // see if this model is exclusively locked by one of its owners. If so, save that fact on modelId so future tests won't have to descend.
+        if (this.ownerHoldsExclusiveLock(modelId))
+            return this.insertLock(modelId, BackendHubAccess_1.LockState.Exclusive, 2 /* LockOrigin.Discovered */);
+        // see if the parent is exclusively locked by one of its owners. If so, save that fact on parentId so future tests won't have to descend.
+        return this.ownerHoldsExclusiveLock(parentId) ? this.insertLock(parentId, BackendHubAccess_1.LockState.Exclusive, 2 /* LockOrigin.Discovered */) : false; // eslint-disable-line @typescript-eslint/no-non-null-assertion
+    }
+    /** Determine whether an the exclusive lock is already held by an element (or one of its owners) */
+    holdsExclusiveLock(id) {
+        // see if we hold the exclusive lock. or if one of the element's owners is exclusively locked (recursively)
+        return this.getLockState(id) === BackendHubAccess_1.LockState.Exclusive || this.ownerHoldsExclusiveLock(id);
+    }
+    holdsSharedLock(id) {
+        const state = this.getLockState(id);
+        // see if we hold shared or exclusive lock, or if an owner has exclusive lock. If so we implicitly have shared lock, but owner holding shared lock doesn't help.
+        return (state === BackendHubAccess_1.LockState.Shared || state === BackendHubAccess_1.LockState.Exclusive) || this.ownerHoldsExclusiveLock(id);
+    }
+    /** if the shared lock on the element supplied is not already held, add it to the set of shared locks required. Then, check owners. */
+    addSharedLock(id, locks) {
+        // if the id is not valid, or of the lock is already in the set, or if we already hold a shared lock, we're done
+        // Note: if we hold a shared lock, it is guaranteed that we also hold all required shared locks on owners.
+        if (id === undefined || !core_bentley_1.Id64.isValid(id) || locks.has(id) || this.holdsSharedLock(id))
+            return;
+        locks.add(id); // add to set of needed shared locks
+        this.addOwnerSharedLocks(id, locks); // check parent models and groups
+    }
+    /** add owners (recursively) of an element to a list of required shared locks, if not already held. */
+    addOwnerSharedLocks(id, locks) {
+        const el = this.getOwners(id);
+        this.addSharedLock(el.parentId, locks); // if this element is in a group
+        this.addSharedLock(el.modelId, locks); // check its model
+    }
+    /** attempt to acquire all necessary locks for a set of elements */
+    async acquireAllLocks(locks) {
+        if (locks.size === 0) // no locks are required.
+            return;
+        const sharedLocks = new Set();
+        for (const lock of locks)
+            this.addOwnerSharedLocks(lock[0], sharedLocks);
+        for (const shared of sharedLocks) {
+            if (!locks.has(shared)) // we may already be asking for exclusive lock
+                locks.set(shared, BackendHubAccess_1.LockState.Shared);
+        }
+        await IModelHost_1.IModelHost.hubAccess.acquireLocks(this.briefcase, locks); // throws if unsuccessful
+        for (const lock of locks)
+            this.insertLock(lock[0], lock[1], 0 /* LockOrigin.Acquired */);
+        this.lockDb.saveChanges();
+    }
+    async acquireLocks(arg) {
+        const locks = new Map();
+        if (arg.shared) {
+            for (const id of core_bentley_1.Id64.iterable(arg.shared)) {
+                if (!this.holdsSharedLock(id))
+                    locks.set(id, BackendHubAccess_1.LockState.Shared);
+            }
+        }
+        if (arg.exclusive) {
+            for (const id of core_bentley_1.Id64.iterable(arg.exclusive)) {
+                if (!this.holdsExclusiveLock(id))
+                    locks.set(id, BackendHubAccess_1.LockState.Exclusive);
+            }
+        }
+        return this.acquireAllLocks(locks);
+    }
+    /** When an element is newly created in a session, we hold the lock on it implicitly. Save that fact. */
+    elementWasCreated(id) {
+        this.insertLock(id, BackendHubAccess_1.LockState.Exclusive, 1 /* LockOrigin.NewElement */);
+        this.lockDb.saveChanges();
+    }
+    /** locks are not necessary during change propagation. */
+    get _locksAreRequired() { return !this.briefcase.txns.isIndirectChanges; }
+    /** throw if locks are currently required and the exclusive lock is not held on the supplied element */
+    checkExclusiveLock(id, type, operation) {
+        if (this._locksAreRequired && !this.holdsExclusiveLock(id))
+            throw new core_common_1.IModelError(core_bentley_1.IModelStatus.LockNotHeld, `exclusive lock not held on ${type} for ${operation} (id=${id})`);
+    }
+    /** throw if locks are currently required and a shared lock is not held on the supplied element */
+    checkSharedLock(id, type, operation) {
+        if (this._locksAreRequired && !this.holdsSharedLock(id))
+            throw new core_common_1.IModelError(core_bentley_1.IModelStatus.LockNotHeld, `shared lock not held on ${type} for ${operation} (id=${id})`);
+    }
+}
+exports.ServerBasedLocks = ServerBasedLocks;
 //# sourceMappingURL=ServerBasedLocks.js.map

package/lib/cjs/ServerBasedLocks.js.map

@@ -1 +1 @@
-{"version":3,"file":"ServerBasedLocks.js","sourceRoot":"","sources":["../../src/ServerBasedLocks.ts"],"names":[],"mappings":";AAAA;;;+FAG+F;;;AAE/F;;GAEG;AAEH,sDAAkG;AAClG,oDAAyD;AACzD,yDAAwD;AAExD,6CAA0C;AAC1C,yCAAsC;AAoBtC,gBAAgB;AAChB,MAAa,gBAAgB;IAK3B,YAAmB,MAAmB;QAHnB,WAAM,GAAG,IAAI,mBAAQ,EAAE,CAAC;QAIzC,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC;QACxB,MAAM,MAAM,GAAG,GAAG,MAAM,CAAC,QAAQ,CAAC,mBAAmB,EAAE,QAAQ,CAAC;QAChE,IAAI;YACF,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,uBAAQ,CAAC,SAAS,CAAC,CAAC;SAChD;QAAC,OAAO,EAAE,EAAE;YACX,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;YAC7B,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,2FAA2F,CAAC,CAAC;YACpH,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;SAC3B;IACH,CAAC;IAdD,IAAW,aAAa,KAAK,OAAO,IAAI,CAAC,CAAC,CAAC;IAgBpC,KAAK;QACV,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM;YACpB,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;IAC1B,CAAC;IAEO,SAAS,CAAC,EAAc;QAC9B,OAAO,IAAI,CAAC,SAAS,CAAC,2BAA2B,CAAC,qDAAqD,EAAE,CAAC,IAAI,EAAE,EAAE;YAChH,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACnB,MAAM,EAAE,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;YACvB,IAAI,uBAAQ,CAAC,aAAa,KAAK,EAAE;gBAC/B,MAAM,IAAI,yBAAW,CAAC,EAAE,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;YAEvD,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;QACvE,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,YAAY,CAAC,EAAe;QAClC,OAAO,CAAC,EAAE,KAAK,SAAS,IAAI,CAAC,mBAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,2BAA2B,CAAC,oCAAoC,EAAE,CAAC,IAAI,EAAE,EAAE;YAClJ,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACnB,OAAO,CAAC,uBAAQ,CAAC,aAAa,KAAK,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACxF,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,aAAa;QACnB,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC;QAC5C,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;IAC5B,CAAC;IAED,qBAAqB;IACd,YAAY,CAAC,KAAgB;QAClC,OAAO,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,0CAA0C,EAAE,CAAC,IAAI,EAAE,EAAE;YAC1F,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;YAC3B,IAAI,CAAC,IAAI,EAAE,CAAC;YACZ,OAAO,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,eAAe;QAC1B,MAAM,uBAAU,CAAC,SAAS,CAAC,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,yBAAyB;QACrF,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC;IAEO,UAAU,CAAC,EAAc,EAAE,KAAgB,EAAE,MAAkB;QACrE,IAAI,CAAC,MAAM,CAAC,2BAA2B,CAAC,6HAA6H,EAAE,CAAC,IAAI,EAAE,EAAE;YAC9K,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACnB,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;YAC3B,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;YAC5B,MAAM,EAAE,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;YACvB,IAAI,uBAAQ,CAAC,cAAc,KAAK,EAAE;gBAChC,MAAM,IAAI,yBAAW,CAAC,EAAE,EAAE,iCAAiC,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,uBAAuB,CAAC,EAA0B;QACxD,IAAI,EAAE,KAAK,SAAS,IAAI,EAAE,KAAK,oBAAM,CAAC,aAAa;YACjD,OAAO,KAAK,CAAC,CAAC,gBAAgB;QAEhC,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QACjD,IAAI,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,4BAAS,CAAC,SAAS,IAAI,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,KAAK,4BAAS,CAAC,SAAS;YAC3G,OAAO,IAAI,CAAC;QAEd,wIAAwI;QACxI,IAAI,IAAI,CAAC,uBAAuB,CAAC,OAAO,CAAC;YACvC,OAAO,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,4BAAS,CAAC,SAAS,qBAAwB,CAAC;QAE9E,yIAAyI;QACzI,OAAO,IAAI,CAAC,uBAAuB,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAS,EAAE,4BAAS,CAAC,SAAS,qBAAwB,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,+DAA+D;IACjM,CAAC;IAED,mGAAmG;IAC5F,kBAAkB,CAAC,EAAc;QACtC,2GAA2G;QAC3G,OAAO,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,KAAK,4BAAS,CAAC,SAAS,IAAI,IAAI,CAAC,uBAAuB,CAAC,EAAE,CAAC,CAAC;IAC3F,CAAC;IAEM,eAAe,CAAC,EAAc;QACnC,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;QACpC,gKAAgK;QAChK,OAAO,CAAC,KAAK,KAAK,4BAAS,CAAC,MAAM,IAAI,KAAK,KAAK,4BAAS,CAAC,SAAS,CAAC,IAAI,IAAI,CAAC,uBAAuB,CAAC,EAAE,CAAC,CAAC;IAC3G,CAAC;IAED,sIAAsI;IAC9H,aAAa,CAAC,EAA0B,EAAE,KAAsB;QACtE,gHAAgH;QAChH,0GAA0G;QAC1G,IAAI,EAAE,KAAK,SAAS,IAAI,CAAC,mBAAI,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;YACpF,OAAO;QAET,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,oCAAoC;QACnD,IAAI,CAAC,mBAAmB,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,CAAC,iCAAiC;IACxE,CAAC;IAED,sGAAsG;IAC9F,mBAAmB,CAAC,EAAc,EAAE,KAAsB;QAChE,MAAM,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QAC9B,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC,CAAC,gCAAgC;QACxE,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,kBAAkB;IAC3D,CAAC;IAED,mEAAmE;IAC3D,KAAK,CAAC,eAAe,CAAC,KAAc;QAC1C,IAAI,KAAK,CAAC,IAAI,KAAK,CAAC,EAAE,yBAAyB;YAC7C,OAAO;QAET,MAAM,WAAW,GAAG,IAAI,GAAG,EAAc,CAAC;QAC1C,KAAK,MAAM,IAAI,IAAI,KAAK;YACtB,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC;QAEjD,KAAK,MAAM,MAAM,IAAI,WAAW,EAAE;YAChC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,8CAA8C;gBACpE,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,4BAAS,CAAC,MAAM,CAAC,CAAC;SACvC;QAED,MAAM,uBAAU,CAAC,SAAS,CAAC,YAAY,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC,CAAC,yBAAyB;QACzF,KAAK,MAAM,IAAI,IAAI,KAAK;YACtB,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,mBAAsB,CAAC;QACzD,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;IAC5B,CAAC;IAEM,KAAK,CAAC,YAAY,CAAC,GAA8C;QACtE,MAAM,KAAK,GAAG,IAAI,GAAG,EAAyB,CAAC;QAC/C,IAAI,GAAG,CAAC,MAAM,EAAE;YACd,KAAK,MAAM,EAAE,IAAI,mBAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE;gBAC1C,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;oBAC3B,KAAK,CAAC,GAAG,CAAC,EAAE,EAAE,4BAAS,CAAC,MAAM,CAAC,CAAC;aACnC;SACF;QACD,IAAI,GAAG,CAAC,SAAS,EAAE;YACjB,KAAK,MAAM,EAAE,IAAI,mBAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE;gBAC7C,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,EAAE,CAAC;oBAC9B,KAAK,CAAC,GAAG,CAAC,EAAE,EAAE,4BAAS,CAAC,SAAS,CAAC,CAAC;aACtC;SACF;QACD,OAAO,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IACrC,CAAC;IAED,wGAAwG;IACjG,iBAAiB,CAAC,EAAc;QACrC,IAAI,CAAC,UAAU,CAAC,EAAE,EAAE,4BAAS,CAAC,SAAS,qBAAwB,CAAC;QAChE,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;IAC5B,CAAC;IAED,yDAAyD;IACzD,IAAY,iBAAiB,KAAK,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC;IAElF,uGAAuG;IAChG,kBAAkB,CAAC,EAAc,EAAE,IAAY,EAAE,SAAiB;QACvE,IAAI,IAAI,CAAC,iBAAiB,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,EAAE,CAAC;YACxD,MAAM,IAAI,yBAAW,CAAC,2BAAY,CAAC,WAAW,EAAE,8BAA8B,IAAI,QAAQ,SAAS,QAAQ,EAAE,GAAG,CAAC,CAAC;IACtH,CAAC;IAED,kGAAkG;IAC3F,eAAe,CAAC,EAAc,EAAE,IAAY,EAAE,SAAiB;QACpE,IAAI,IAAI,CAAC,iBAAiB,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;YACrD,MAAM,IAAI,yBAAW,CAAC,2BAAY,CAAC,WAAW,EAAE,2BAA2B,IAAI,QAAQ,SAAS,QAAQ,EAAE,GAAG,CAAC,CAAC;IACnH,CAAC;CAEF;AAlLD,4CAkLC","sourcesContent":["/*---------------------------------------------------------------------------------------------\r\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\r\n* See LICENSE.md in the project root for license terms and full copyright notice.\r\n*--------------------------------------------------------------------------------------------*/\r\n\r\n/** @packageDocumentation\r\n * @module iModels\r\n */\r\n\r\nimport { DbResult, Id64, Id64Arg, Id64String, IModelStatus, OpenMode } from \"@itwin/core-bentley\";\r\nimport { IModel, IModelError } from \"@itwin/core-common\";\r\nimport { LockMap, LockState } from \"./BackendHubAccess\";\r\nimport { BriefcaseDb, LockControl } from \"./IModelDb\";\r\nimport { IModelHost } from \"./IModelHost\";\r\nimport { SQLiteDb } from \"./SQLiteDb\";\r\n\r\n/**\r\n * Both the Model and Parent of an element are considered \"owners\" of their member elements. That means:\r\n * 1) they must hold at least a shared lock before an exclusive lock can be acquired for their members\r\n * 2) if they hold an exclusive lock, then all of their members are exclusively locked implicitly.\r\n * @internal\r\n */\r\nexport interface ElementOwners {\r\n  readonly modelId: Id64String;\r\n  readonly parentId: Id64String | undefined;\r\n}\r\n\r\n// eslint-disable-next-line no-restricted-syntax\r\nconst enum LockOrigin {\r\n  Acquired = 0,\r\n  NewElement = 1,\r\n  Discovered = 2,\r\n}\r\n\r\n/** @internal */\r\nexport class ServerBasedLocks implements LockControl {\r\n  public get isServerBased() { return true; }\r\n  protected readonly lockDb = new SQLiteDb();\r\n  protected readonly briefcase: BriefcaseDb;\r\n\r\n  public constructor(iModel: BriefcaseDb) {\r\n    this.briefcase = iModel;\r\n    const dbName = `${iModel.nativeDb.getTempFileBaseName()}-locks`;\r\n    try {\r\n      this.lockDb.openDb(dbName, OpenMode.ReadWrite);\r\n    } catch (_e) {\r\n      this.lockDb.createDb(dbName);\r\n      this.lockDb.executeSQL(\"CREATE TABLE locks(id INTEGER PRIMARY KEY NOT NULL,state INTEGER NOT NULL,origin INTEGER)\");\r\n      this.lockDb.saveChanges();\r\n    }\r\n  }\r\n\r\n  public close() {\r\n    if (this.lockDb.isOpen)\r\n      this.lockDb.closeDb();\r\n  }\r\n\r\n  private getOwners(id: Id64String): ElementOwners {\r\n    return this.briefcase.withPreparedSqliteStatement(\"SELECT ModelId,ParentId FROM bis_Element WHERE id=?\", (stmt) => {\r\n      stmt.bindId(1, id);\r\n      const rc = stmt.step();\r\n      if (DbResult.BE_SQLITE_ROW !== rc)\r\n        throw new IModelError(rc, `element ${id} not found`);\r\n\r\n      return { modelId: stmt.getValueId(0), parentId: stmt.getValueId(1) };\r\n    });\r\n  }\r\n\r\n  private getLockState(id?: Id64String): LockState | undefined {\r\n    return (id === undefined || !Id64.isValid(id)) ? undefined : this.lockDb.withPreparedSqliteStatement(\"SELECT state FROM locks WHERE id=?\", (stmt) => {\r\n      stmt.bindId(1, id);\r\n      return (DbResult.BE_SQLITE_ROW === stmt.step()) ? stmt.getValueInteger(0) : undefined;\r\n    });\r\n  }\r\n\r\n  /** Clear the cache of locally held locks.\r\n   * Note: does *not* release locks from server.\r\n   */\r\n  private clearAllLocks() {\r\n    this.lockDb.executeSQL(\"DELETE FROM locks\");\r\n    this.lockDb.saveChanges();\r\n  }\r\n\r\n  /** only for tests */\r\n  public getLockCount(state: LockState): number {\r\n    return this.lockDb.withSqliteStatement(\"SELECT count(*) FROM locks WHERE state=?\", (stmt) => {\r\n      stmt.bindInteger(1, state);\r\n      stmt.step();\r\n      return stmt.getValueInteger(0);\r\n    });\r\n  }\r\n\r\n  public async releaseAllLocks(): Promise<void> {\r\n    await IModelHost.hubAccess.releaseAllLocks(this.briefcase); // throws if unsuccessful\r\n    this.clearAllLocks();\r\n  }\r\n\r\n  private insertLock(id: Id64String, state: LockState, origin: LockOrigin): true {\r\n    this.lockDb.withPreparedSqliteStatement(\"INSERT INTO locks(id,state,origin) VALUES (?,?,?) ON CONFLICT(id) DO UPDATE SET state=excluded.state,origin=excluded.origin\", (stmt) => {\r\n      stmt.bindId(1, id);\r\n      stmt.bindInteger(2, state);\r\n      stmt.bindInteger(3, origin);\r\n      const rc = stmt.step();\r\n      if (DbResult.BE_SQLITE_DONE !== rc)\r\n        throw new IModelError(rc, \"can't insert lock into database\");\r\n    });\r\n    return true;\r\n  }\r\n\r\n  private ownerHoldsExclusiveLock(id: Id64String | undefined): boolean {\r\n    if (id === undefined || id === IModel.rootSubjectId)\r\n      return false; // has no owners\r\n\r\n    const { modelId, parentId } = this.getOwners(id);\r\n    if (this.getLockState(modelId) === LockState.Exclusive || this.getLockState(parentId) === LockState.Exclusive)\r\n      return true;\r\n\r\n    // see if this model is exclusively locked by one of its owners. If so, save that fact on modelId so future tests won't have to descend.\r\n    if (this.ownerHoldsExclusiveLock(modelId))\r\n      return this.insertLock(modelId, LockState.Exclusive, LockOrigin.Discovered);\r\n\r\n    // see if the parent is exclusively locked by one of its owners. If so, save that fact on parentId so future tests won't have to descend.\r\n    return this.ownerHoldsExclusiveLock(parentId) ? this.insertLock(parentId!, LockState.Exclusive, LockOrigin.Discovered) : false; // eslint-disable-line @typescript-eslint/no-non-null-assertion\r\n  }\r\n\r\n  /** Determine whether an the exclusive lock is already held by an element (or one of its owners) */\r\n  public holdsExclusiveLock(id: Id64String): boolean {\r\n    // see if we hold the exclusive lock. or if one of the element's owners is exclusively locked (recursively)\r\n    return this.getLockState(id) === LockState.Exclusive || this.ownerHoldsExclusiveLock(id);\r\n  }\r\n\r\n  public holdsSharedLock(id: Id64String): boolean {\r\n    const state = this.getLockState(id);\r\n    // see if we hold shared or exclusive lock, or if an owner has exclusive lock. If so we implicitly have shared lock, but owner holding shared lock doesn't help.\r\n    return (state === LockState.Shared || state === LockState.Exclusive) || this.ownerHoldsExclusiveLock(id);\r\n  }\r\n\r\n  /** if the shared lock on the element supplied is not already held, add it to the set of shared locks required. Then, check owners. */\r\n  private addSharedLock(id: Id64String | undefined, locks: Set<Id64String>) {\r\n    // if the id is not valid, or of the lock is already in the set, or if we already hold a shared lock, we're done\r\n    // Note: if we hold a shared lock, it is guaranteed that we also hold all required shared locks on owners.\r\n    if (id === undefined || !Id64.isValid(id) || locks.has(id) || this.holdsSharedLock(id))\r\n      return;\r\n\r\n    locks.add(id); // add to set of needed shared locks\r\n    this.addOwnerSharedLocks(id, locks); // check parent models and groups\r\n  }\r\n\r\n  /** add owners (recursively) of an element to a list of required shared locks, if not already held. */\r\n  private addOwnerSharedLocks(id: Id64String, locks: Set<Id64String>) {\r\n    const el = this.getOwners(id);\r\n    this.addSharedLock(el.parentId, locks); // if this element is in a group\r\n    this.addSharedLock(el.modelId, locks); // check its model\r\n  }\r\n\r\n  /** attempt to acquire all necessary locks for a set of elements */\r\n  private async acquireAllLocks(locks: LockMap) {\r\n    if (locks.size === 0) // no locks are required.\r\n      return;\r\n\r\n    const sharedLocks = new Set<Id64String>();\r\n    for (const lock of locks)\r\n      this.addOwnerSharedLocks(lock[0], sharedLocks);\r\n\r\n    for (const shared of sharedLocks) {\r\n      if (!locks.has(shared)) // we may already be asking for exclusive lock\r\n        locks.set(shared, LockState.Shared);\r\n    }\r\n\r\n    await IModelHost.hubAccess.acquireLocks(this.briefcase, locks); // throws if unsuccessful\r\n    for (const lock of locks)\r\n      this.insertLock(lock[0], lock[1], LockOrigin.Acquired);\r\n    this.lockDb.saveChanges();\r\n  }\r\n\r\n  public async acquireLocks(arg: { shared?: Id64Arg, exclusive?: Id64Arg }): Promise<void> {\r\n    const locks = new Map<Id64String, LockState>();\r\n    if (arg.shared) {\r\n      for (const id of Id64.iterable(arg.shared)) {\r\n        if (!this.holdsSharedLock(id))\r\n          locks.set(id, LockState.Shared);\r\n      }\r\n    }\r\n    if (arg.exclusive) {\r\n      for (const id of Id64.iterable(arg.exclusive)) {\r\n        if (!this.holdsExclusiveLock(id))\r\n          locks.set(id, LockState.Exclusive);\r\n      }\r\n    }\r\n    return this.acquireAllLocks(locks);\r\n  }\r\n\r\n  /** When an element is newly created in a session, we hold the lock on it implicitly. Save that fact. */\r\n  public elementWasCreated(id: Id64String) {\r\n    this.insertLock(id, LockState.Exclusive, LockOrigin.NewElement);\r\n    this.lockDb.saveChanges();\r\n  }\r\n\r\n  /** locks are not necessary during change propagation. */\r\n  private get _locksAreRequired() { return !this.briefcase.txns.isIndirectChanges; }\r\n\r\n  /** throw if locks are currently required and the exclusive lock is not held on the supplied element */\r\n  public checkExclusiveLock(id: Id64String, type: string, operation: string) {\r\n    if (this._locksAreRequired && !this.holdsExclusiveLock(id))\r\n      throw new IModelError(IModelStatus.LockNotHeld, `exclusive lock not held on ${type} for ${operation} (id=${id})`);\r\n  }\r\n\r\n  /** throw if locks are currently required and a shared lock is not held on the supplied element */\r\n  public checkSharedLock(id: Id64String, type: string, operation: string) {\r\n    if (this._locksAreRequired && !this.holdsSharedLock(id))\r\n      throw new IModelError(IModelStatus.LockNotHeld, `shared lock not held on ${type} for ${operation} (id=${id})`);\r\n  }\r\n\r\n}\r\n\r\n"]}
+{"version":3,"file":"ServerBasedLocks.js","sourceRoot":"","sources":["../../src/ServerBasedLocks.ts"],"names":[],"mappings":";AAAA;;;+FAG+F;;;AAE/F;;GAEG;AAEH,sDAAkG;AAClG,oDAAyD;AACzD,yDAAwD;AAExD,6CAA0C;AAC1C,yCAAsC;AAoBtC,gBAAgB;AAChB,MAAa,gBAAgB;IAC3B,IAAW,aAAa,KAAK,OAAO,IAAI,CAAC,CAAC,CAAC;IAI3C,YAAmB,MAAmB;QAHnB,WAAM,GAAG,IAAI,mBAAQ,EAAE,CAAC;QAIzC,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC;QACxB,MAAM,MAAM,GAAG,GAAG,MAAM,CAAC,QAAQ,CAAC,mBAAmB,EAAE,QAAQ,CAAC;QAChE,IAAI;YACF,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,uBAAQ,CAAC,SAAS,CAAC,CAAC;SAChD;QAAC,OAAO,EAAE,EAAE;YACX,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;YAC7B,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,2FAA2F,CAAC,CAAC;YACpH,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;SAC3B;IACH,CAAC;IAEM,KAAK;QACV,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM;YACpB,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;IAC1B,CAAC;IAEO,SAAS,CAAC,EAAc;QAC9B,OAAO,IAAI,CAAC,SAAS,CAAC,2BAA2B,CAAC,qDAAqD,EAAE,CAAC,IAAI,EAAE,EAAE;YAChH,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACnB,MAAM,EAAE,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;YACvB,IAAI,uBAAQ,CAAC,aAAa,KAAK,EAAE;gBAC/B,MAAM,IAAI,yBAAW,CAAC,EAAE,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;YAEvD,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;QACvE,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,YAAY,CAAC,EAAe;QAClC,OAAO,CAAC,EAAE,KAAK,SAAS,IAAI,CAAC,mBAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,2BAA2B,CAAC,oCAAoC,EAAE,CAAC,IAAI,EAAE,EAAE;YAClJ,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACnB,OAAO,CAAC,uBAAQ,CAAC,aAAa,KAAK,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACxF,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,aAAa;QACnB,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC;QAC5C,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;IAC5B,CAAC;IAED,qBAAqB;IACd,YAAY,CAAC,KAAgB;QAClC,OAAO,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,0CAA0C,EAAE,CAAC,IAAI,EAAE,EAAE;YAC1F,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;YAC3B,IAAI,CAAC,IAAI,EAAE,CAAC;YACZ,OAAO,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,eAAe;QAC1B,MAAM,uBAAU,CAAC,SAAS,CAAC,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,yBAAyB;QACrF,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC;IAEO,UAAU,CAAC,EAAc,EAAE,KAAgB,EAAE,MAAkB;QACrE,IAAI,CAAC,MAAM,CAAC,2BAA2B,CAAC,6HAA6H,EAAE,CAAC,IAAI,EAAE,EAAE;YAC9K,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACnB,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;YAC3B,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;YAC5B,MAAM,EAAE,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;YACvB,IAAI,uBAAQ,CAAC,cAAc,KAAK,EAAE;gBAChC,MAAM,IAAI,yBAAW,CAAC,EAAE,EAAE,iCAAiC,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,uBAAuB,CAAC,EAA0B;QACxD,IAAI,EAAE,KAAK,SAAS,IAAI,EAAE,KAAK,oBAAM,CAAC,aAAa;YACjD,OAAO,KAAK,CAAC,CAAC,gBAAgB;QAEhC,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QACjD,IAAI,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,4BAAS,CAAC,SAAS,IAAI,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,KAAK,4BAAS,CAAC,SAAS;YAC3G,OAAO,IAAI,CAAC;QAEd,wIAAwI;QACxI,IAAI,IAAI,CAAC,uBAAuB,CAAC,OAAO,CAAC;YACvC,OAAO,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,4BAAS,CAAC,SAAS,gCAAwB,CAAC;QAE9E,yIAAyI;QACzI,OAAO,IAAI,CAAC,uBAAuB,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAS,EAAE,4BAAS,CAAC,SAAS,gCAAwB,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,+DAA+D;IACjM,CAAC;IAED,mGAAmG;IAC5F,kBAAkB,CAAC,EAAc;QACtC,2GAA2G;QAC3G,OAAO,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,KAAK,4BAAS,CAAC,SAAS,IAAI,IAAI,CAAC,uBAAuB,CAAC,EAAE,CAAC,CAAC;IAC3F,CAAC;IAEM,eAAe,CAAC,EAAc;QACnC,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;QACpC,gKAAgK;QAChK,OAAO,CAAC,KAAK,KAAK,4BAAS,CAAC,MAAM,IAAI,KAAK,KAAK,4BAAS,CAAC,SAAS,CAAC,IAAI,IAAI,CAAC,uBAAuB,CAAC,EAAE,CAAC,CAAC;IAC3G,CAAC;IAED,sIAAsI;IAC9H,aAAa,CAAC,EAA0B,EAAE,KAAsB;QACtE,gHAAgH;QAChH,0GAA0G;QAC1G,IAAI,EAAE,KAAK,SAAS,IAAI,CAAC,mBAAI,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;YACpF,OAAO;QAET,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,oCAAoC;QACnD,IAAI,CAAC,mBAAmB,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,CAAC,iCAAiC;IACxE,CAAC;IAED,sGAAsG;IAC9F,mBAAmB,CAAC,EAAc,EAAE,KAAsB;QAChE,MAAM,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QAC9B,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC,CAAC,gCAAgC;QACxE,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,kBAAkB;IAC3D,CAAC;IAED,mEAAmE;IAC3D,KAAK,CAAC,eAAe,CAAC,KAAc;QAC1C,IAAI,KAAK,CAAC,IAAI,KAAK,CAAC,EAAE,yBAAyB;YAC7C,OAAO;QAET,MAAM,WAAW,GAAG,IAAI,GAAG,EAAc,CAAC;QAC1C,KAAK,MAAM,IAAI,IAAI,KAAK;YACtB,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC;QAEjD,KAAK,MAAM,MAAM,IAAI,WAAW,EAAE;YAChC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,8CAA8C;gBACpE,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,4BAAS,CAAC,MAAM,CAAC,CAAC;SACvC;QAED,MAAM,uBAAU,CAAC,SAAS,CAAC,YAAY,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC,CAAC,yBAAyB;QACzF,KAAK,MAAM,IAAI,IAAI,KAAK;YACtB,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,8BAAsB,CAAC;QACzD,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;IAC5B,CAAC;IAEM,KAAK,CAAC,YAAY,CAAC,GAA8C;QACtE,MAAM,KAAK,GAAG,IAAI,GAAG,EAAyB,CAAC;QAC/C,IAAI,GAAG,CAAC,MAAM,EAAE;YACd,KAAK,MAAM,EAAE,IAAI,mBAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE;gBAC1C,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;oBAC3B,KAAK,CAAC,GAAG,CAAC,EAAE,EAAE,4BAAS,CAAC,MAAM,CAAC,CAAC;aACnC;SACF;QACD,IAAI,GAAG,CAAC,SAAS,EAAE;YACjB,KAAK,MAAM,EAAE,IAAI,mBAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE;gBAC7C,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,EAAE,CAAC;oBAC9B,KAAK,CAAC,GAAG,CAAC,EAAE,EAAE,4BAAS,CAAC,SAAS,CAAC,CAAC;aACtC;SACF;QACD,OAAO,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IACrC,CAAC;IAED,wGAAwG;IACjG,iBAAiB,CAAC,EAAc;QACrC,IAAI,CAAC,UAAU,CAAC,EAAE,EAAE,4BAAS,CAAC,SAAS,gCAAwB,CAAC;QAChE,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;IAC5B,CAAC;IAED,yDAAyD;IACzD,IAAY,iBAAiB,KAAK,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC;IAElF,uGAAuG;IAChG,kBAAkB,CAAC,EAAc,EAAE,IAAY,EAAE,SAAiB;QACvE,IAAI,IAAI,CAAC,iBAAiB,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,EAAE,CAAC;YACxD,MAAM,IAAI,yBAAW,CAAC,2BAAY,CAAC,WAAW,EAAE,8BAA8B,IAAI,QAAQ,SAAS,QAAQ,EAAE,GAAG,CAAC,CAAC;IACtH,CAAC;IAED,kGAAkG;IAC3F,eAAe,CAAC,EAAc,EAAE,IAAY,EAAE,SAAiB;QACpE,IAAI,IAAI,CAAC,iBAAiB,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;YACrD,MAAM,IAAI,yBAAW,CAAC,2BAAY,CAAC,WAAW,EAAE,2BAA2B,IAAI,QAAQ,SAAS,QAAQ,EAAE,GAAG,CAAC,CAAC;IACnH,CAAC;CAEF;AAlLD,4CAkLC","sourcesContent":["/*---------------------------------------------------------------------------------------------\r\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\r\n* See LICENSE.md in the project root for license terms and full copyright notice.\r\n*--------------------------------------------------------------------------------------------*/\r\n\r\n/** @packageDocumentation\r\n * @module iModels\r\n */\r\n\r\nimport { DbResult, Id64, Id64Arg, Id64String, IModelStatus, OpenMode } from \"@itwin/core-bentley\";\r\nimport { IModel, IModelError } from \"@itwin/core-common\";\r\nimport { LockMap, LockState } from \"./BackendHubAccess\";\r\nimport { BriefcaseDb, LockControl } from \"./IModelDb\";\r\nimport { IModelHost } from \"./IModelHost\";\r\nimport { SQLiteDb } from \"./SQLiteDb\";\r\n\r\n/**\r\n * Both the Model and Parent of an element are considered \"owners\" of their member elements. That means:\r\n * 1) they must hold at least a shared lock before an exclusive lock can be acquired for their members\r\n * 2) if they hold an exclusive lock, then all of their members are exclusively locked implicitly.\r\n * @internal\r\n */\r\nexport interface ElementOwners {\r\n  readonly modelId: Id64String;\r\n  readonly parentId: Id64String | undefined;\r\n}\r\n\r\n// eslint-disable-next-line no-restricted-syntax\r\nconst enum LockOrigin {\r\n  Acquired = 0,\r\n  NewElement = 1,\r\n  Discovered = 2,\r\n}\r\n\r\n/** @internal */\r\nexport class ServerBasedLocks implements LockControl {\r\n  public get isServerBased() { return true; }\r\n  protected readonly lockDb = new SQLiteDb();\r\n  protected readonly briefcase: BriefcaseDb;\r\n\r\n  public constructor(iModel: BriefcaseDb) {\r\n    this.briefcase = iModel;\r\n    const dbName = `${iModel.nativeDb.getTempFileBaseName()}-locks`;\r\n    try {\r\n      this.lockDb.openDb(dbName, OpenMode.ReadWrite);\r\n    } catch (_e) {\r\n      this.lockDb.createDb(dbName);\r\n      this.lockDb.executeSQL(\"CREATE TABLE locks(id INTEGER PRIMARY KEY NOT NULL,state INTEGER NOT NULL,origin INTEGER)\");\r\n      this.lockDb.saveChanges();\r\n    }\r\n  }\r\n\r\n  public close() {\r\n    if (this.lockDb.isOpen)\r\n      this.lockDb.closeDb();\r\n  }\r\n\r\n  private getOwners(id: Id64String): ElementOwners {\r\n    return this.briefcase.withPreparedSqliteStatement(\"SELECT ModelId,ParentId FROM bis_Element WHERE id=?\", (stmt) => {\r\n      stmt.bindId(1, id);\r\n      const rc = stmt.step();\r\n      if (DbResult.BE_SQLITE_ROW !== rc)\r\n        throw new IModelError(rc, `element ${id} not found`);\r\n\r\n      return { modelId: stmt.getValueId(0), parentId: stmt.getValueId(1) };\r\n    });\r\n  }\r\n\r\n  private getLockState(id?: Id64String): LockState | undefined {\r\n    return (id === undefined || !Id64.isValid(id)) ? undefined : this.lockDb.withPreparedSqliteStatement(\"SELECT state FROM locks WHERE id=?\", (stmt) => {\r\n      stmt.bindId(1, id);\r\n      return (DbResult.BE_SQLITE_ROW === stmt.step()) ? stmt.getValueInteger(0) : undefined;\r\n    });\r\n  }\r\n\r\n  /** Clear the cache of locally held locks.\r\n   * Note: does *not* release locks from server.\r\n   */\r\n  private clearAllLocks() {\r\n    this.lockDb.executeSQL(\"DELETE FROM locks\");\r\n    this.lockDb.saveChanges();\r\n  }\r\n\r\n  /** only for tests */\r\n  public getLockCount(state: LockState): number {\r\n    return this.lockDb.withSqliteStatement(\"SELECT count(*) FROM locks WHERE state=?\", (stmt) => {\r\n      stmt.bindInteger(1, state);\r\n      stmt.step();\r\n      return stmt.getValueInteger(0);\r\n    });\r\n  }\r\n\r\n  public async releaseAllLocks(): Promise<void> {\r\n    await IModelHost.hubAccess.releaseAllLocks(this.briefcase); // throws if unsuccessful\r\n    this.clearAllLocks();\r\n  }\r\n\r\n  private insertLock(id: Id64String, state: LockState, origin: LockOrigin): true {\r\n    this.lockDb.withPreparedSqliteStatement(\"INSERT INTO locks(id,state,origin) VALUES (?,?,?) ON CONFLICT(id) DO UPDATE SET state=excluded.state,origin=excluded.origin\", (stmt) => {\r\n      stmt.bindId(1, id);\r\n      stmt.bindInteger(2, state);\r\n      stmt.bindInteger(3, origin);\r\n      const rc = stmt.step();\r\n      if (DbResult.BE_SQLITE_DONE !== rc)\r\n        throw new IModelError(rc, \"can't insert lock into database\");\r\n    });\r\n    return true;\r\n  }\r\n\r\n  private ownerHoldsExclusiveLock(id: Id64String | undefined): boolean {\r\n    if (id === undefined || id === IModel.rootSubjectId)\r\n      return false; // has no owners\r\n\r\n    const { modelId, parentId } = this.getOwners(id);\r\n    if (this.getLockState(modelId) === LockState.Exclusive || this.getLockState(parentId) === LockState.Exclusive)\r\n      return true;\r\n\r\n    // see if this model is exclusively locked by one of its owners. If so, save that fact on modelId so future tests won't have to descend.\r\n    if (this.ownerHoldsExclusiveLock(modelId))\r\n      return this.insertLock(modelId, LockState.Exclusive, LockOrigin.Discovered);\r\n\r\n    // see if the parent is exclusively locked by one of its owners. If so, save that fact on parentId so future tests won't have to descend.\r\n    return this.ownerHoldsExclusiveLock(parentId) ? this.insertLock(parentId!, LockState.Exclusive, LockOrigin.Discovered) : false; // eslint-disable-line @typescript-eslint/no-non-null-assertion\r\n  }\r\n\r\n  /** Determine whether an the exclusive lock is already held by an element (or one of its owners) */\r\n  public holdsExclusiveLock(id: Id64String): boolean {\r\n    // see if we hold the exclusive lock. or if one of the element's owners is exclusively locked (recursively)\r\n    return this.getLockState(id) === LockState.Exclusive || this.ownerHoldsExclusiveLock(id);\r\n  }\r\n\r\n  public holdsSharedLock(id: Id64String): boolean {\r\n    const state = this.getLockState(id);\r\n    // see if we hold shared or exclusive lock, or if an owner has exclusive lock. If so we implicitly have shared lock, but owner holding shared lock doesn't help.\r\n    return (state === LockState.Shared || state === LockState.Exclusive) || this.ownerHoldsExclusiveLock(id);\r\n  }\r\n\r\n  /** if the shared lock on the element supplied is not already held, add it to the set of shared locks required. Then, check owners. */\r\n  private addSharedLock(id: Id64String | undefined, locks: Set<Id64String>) {\r\n    // if the id is not valid, or of the lock is already in the set, or if we already hold a shared lock, we're done\r\n    // Note: if we hold a shared lock, it is guaranteed that we also hold all required shared locks on owners.\r\n    if (id === undefined || !Id64.isValid(id) || locks.has(id) || this.holdsSharedLock(id))\r\n      return;\r\n\r\n    locks.add(id); // add to set of needed shared locks\r\n    this.addOwnerSharedLocks(id, locks); // check parent models and groups\r\n  }\r\n\r\n  /** add owners (recursively) of an element to a list of required shared locks, if not already held. */\r\n  private addOwnerSharedLocks(id: Id64String, locks: Set<Id64String>) {\r\n    const el = this.getOwners(id);\r\n    this.addSharedLock(el.parentId, locks); // if this element is in a group\r\n    this.addSharedLock(el.modelId, locks); // check its model\r\n  }\r\n\r\n  /** attempt to acquire all necessary locks for a set of elements */\r\n  private async acquireAllLocks(locks: LockMap) {\r\n    if (locks.size === 0) // no locks are required.\r\n      return;\r\n\r\n    const sharedLocks = new Set<Id64String>();\r\n    for (const lock of locks)\r\n      this.addOwnerSharedLocks(lock[0], sharedLocks);\r\n\r\n    for (const shared of sharedLocks) {\r\n      if (!locks.has(shared)) // we may already be asking for exclusive lock\r\n        locks.set(shared, LockState.Shared);\r\n    }\r\n\r\n    await IModelHost.hubAccess.acquireLocks(this.briefcase, locks); // throws if unsuccessful\r\n    for (const lock of locks)\r\n      this.insertLock(lock[0], lock[1], LockOrigin.Acquired);\r\n    this.lockDb.saveChanges();\r\n  }\r\n\r\n  public async acquireLocks(arg: { shared?: Id64Arg, exclusive?: Id64Arg }): Promise<void> {\r\n    const locks = new Map<Id64String, LockState>();\r\n    if (arg.shared) {\r\n      for (const id of Id64.iterable(arg.shared)) {\r\n        if (!this.holdsSharedLock(id))\r\n          locks.set(id, LockState.Shared);\r\n      }\r\n    }\r\n    if (arg.exclusive) {\r\n      for (const id of Id64.iterable(arg.exclusive)) {\r\n        if (!this.holdsExclusiveLock(id))\r\n          locks.set(id, LockState.Exclusive);\r\n      }\r\n    }\r\n    return this.acquireAllLocks(locks);\r\n  }\r\n\r\n  /** When an element is newly created in a session, we hold the lock on it implicitly. Save that fact. */\r\n  public elementWasCreated(id: Id64String) {\r\n    this.insertLock(id, LockState.Exclusive, LockOrigin.NewElement);\r\n    this.lockDb.saveChanges();\r\n  }\r\n\r\n  /** locks are not necessary during change propagation. */\r\n  private get _locksAreRequired() { return !this.briefcase.txns.isIndirectChanges; }\r\n\r\n  /** throw if locks are currently required and the exclusive lock is not held on the supplied element */\r\n  public checkExclusiveLock(id: Id64String, type: string, operation: string) {\r\n    if (this._locksAreRequired && !this.holdsExclusiveLock(id))\r\n      throw new IModelError(IModelStatus.LockNotHeld, `exclusive lock not held on ${type} for ${operation} (id=${id})`);\r\n  }\r\n\r\n  /** throw if locks are currently required and a shared lock is not held on the supplied element */\r\n  public checkSharedLock(id: Id64String, type: string, operation: string) {\r\n    if (this._locksAreRequired && !this.holdsSharedLock(id))\r\n      throw new IModelError(IModelStatus.LockNotHeld, `shared lock not held on ${type} for ${operation} (id=${id})`);\r\n  }\r\n\r\n}\r\n\r\n"]}