fix: ensure lock stays on

server/src/domain/library/library.service.ts

@@ -74,11 +74,9 @@ export class LibraryService extends EventEmitter {
     const { watch, scan } = config.library;
     this.watchLibraries = false;
     if (watch.enabled) {
-      this.databaseRepository.withTryLock(DatabaseLock.LibraryWatch, async () => {
       // This ensures that library watching only occurs in one microservice
       // TODO: we could make the lock be per-library instead of global
-        this.watchLibraries = true;
+      this.watchLibraries = await this.databaseRepository.tryLock(DatabaseLock.LibraryWatch);
-      });
     }
     this.jobRepository.addCronJob(
       'libraryScan',

server/src/domain/repositories/database.repository.ts

@@ -47,7 +47,7 @@ export interface IDatabaseRepository {
   shouldReindex(name: VectorIndex): Promise<boolean>;
   runMigrations(options?: { transaction?: 'all' | 'none' | 'each' }): Promise<void>;
   withLock<R>(lock: DatabaseLock, callback: () => Promise<R>): Promise<R>;
-  withTryLock<R>(lock: DatabaseLock, callback: () => Promise<R>): Promise<R>;
+  tryLock(lock: DatabaseLock): Promise<boolean>;
   isBusy(lock: DatabaseLock): boolean;
   wait(lock: DatabaseLock): Promise<void>;
 }

server/src/infra/repositories/database.repository.ts

@@ -210,23 +210,9 @@ export class DatabaseRepository implements IDatabaseRepository {
     return res as R;
   }
 
-  async withTryLock<R>(lock: DatabaseLock, callback: () => Promise<R>): Promise<R> {
+  async tryLock(lock: DatabaseLock): Promise<boolean> {
-    let res;
     const queryRunner = this.dataSource.createQueryRunner();
-    try {
+    return await this.acquireTryLock(lock, queryRunner);
-      const lockAcquired = await this.tryLock(lock, queryRunner);
-      if (lockAcquired) {
-        res = await callback();
-      }
-    } finally {
-      try {
-        await this.releaseLock(lock, queryRunner);
-      } finally {
-        await queryRunner.release();
-      }
-    }
-
-    return res as R;
   }
 
   isBusy(lock: DatabaseLock): boolean {
@@ -241,7 +227,7 @@ export class DatabaseRepository implements IDatabaseRepository {
     return queryRunner.query('SELECT pg_advisory_lock($1)', [lock]);
   }
 
-  private async tryLock(lock: DatabaseLock, queryRunner: QueryRunner): Promise<boolean> {
+  private async acquireTryLock(lock: DatabaseLock, queryRunner: QueryRunner): Promise<boolean> {
     return queryRunner.query('SELECT pg_try_advisory_lock($1)', [lock]);
   }
 

server/test/repositories/database.repository.mock.ts

@@ -13,6 +13,7 @@ export const newDatabaseRepositoryMock = (): jest.Mocked<IDatabaseRepository> =>
     shouldReindex: jest.fn(),
     runMigrations: jest.fn(),
     withLock: jest.fn().mockImplementation((_, function_: <R>() => Promise<R>) => function_()),
+    tryLock: jest.fn(),
     isBusy: jest.fn(),
     wait: jest.fn(),
   };