Added a timeout on pQueue instance

scambier · Jan 17, 2023 · 0b78b97 · 0b78b97
1 parent dbd37d0
commit 0b78b97
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 9 deletions.
diff --git a/lib/src/globals.ts b/lib/src/globals.ts
@@ -11,8 +11,9 @@ if (backgroundProcesses == cpuCount) {
 }
 
 console.info(
-  `Text Extractor - Number of available workers: ${backgroundProcesses}`
+  `Text Extractor - Number of available workers: ${backgroundProcesses}`,
 )
 
-// export const processQueue = pLimit(backgroundProcesses)
-export const processQueue = new PQueue({ concurrency: backgroundProcesses })
+export const workerTimeout = 120_000
+
+export const processQueue = new PQueue({ concurrency: backgroundProcesses, timeout: workerTimeout + 100 })
diff --git a/lib/src/ocr-manager.ts b/lib/src/ocr-manager.ts
@@ -1,11 +1,9 @@
 import { Platform, TFile } from 'obsidian'
 import Tesseract, { createWorker } from 'tesseract.js'
 import { getCachePath, readCache, writeCache } from './cache'
-import { processQueue } from './globals'
+import { processQueue, workerTimeout } from './globals'
 import type { OcrOptions } from './types'
 
-const workerTimeout = 120_000
-
 class OCRWorker {
   static #pool: OCRWorker[] = []
   #running = false

diff --git a/lib/src/pdf-manager.ts b/lib/src/pdf-manager.ts
@@ -1,10 +1,8 @@
 import { Platform, TFile } from 'obsidian'
 import WebWorker from 'web-worker:./pdf-worker.ts'
-import { processQueue } from './globals'
+import { processQueue, workerTimeout } from './globals'
 import { getCachePath, readCache, writeCache } from './cache'
 
-const workerTimeout = 120_000
-
 class PDFWorker {
   static #pool: PDFWorker[] = []
   #running = false