vodorod/parcer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add support for stop and interpret workflow

Browse Source
This commit is contained in:
Rohit
2025-03-11 21:55:27 +05:30
parent 4dd745c95c
commit 7b940ab965
1 changed files with 21 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
server/src/pgboss-worker.ts
View File

@@ -16,8 +16,17 @@ interface InitializeBrowserData {
userId: string; userId: string;
} }
interface InterpretWorkflow {
userId: string;
}
interface StopInterpretWorkflow {
userId: string;
}
interface DestroyBrowserData { interface DestroyBrowserData {
browserId: string; browserId: string;
userId: string;
} }
const pgBoss = new PgBoss({connectionString: pgBossConnectionString, schema: 'public'}); const pgBoss = new PgBoss({connectionString: pgBossConnectionString, schema: 'public'});
@@ -65,10 +74,10 @@ async function startWorkers() {
await pgBoss.work('destroy-browser', async (job: Job<DestroyBrowserData> | Job<DestroyBrowserData>[]) => { await pgBoss.work('destroy-browser', async (job: Job<DestroyBrowserData> | Job<DestroyBrowserData>[]) => {
try { try {
const data = extractJobData(job); const data = extractJobData(job);
const browserId = data.browserId; const { browserId, userId } = data;
logger.log('info', `Starting browser destruction job for browser: ${browserId}`); logger.log('info', `Starting browser destruction job for browser: ${browserId}`);
const success = await destroyRemoteBrowser(browserId); const success = await destroyRemoteBrowser(browserId, userId);
logger.log('info', `Browser destruction job completed with result: ${success}`); logger.log('info', `Browser destruction job completed with result: ${success}`);
return { success }; return { success };
} catch (error: unknown) { } catch (error: unknown) {
@@ -79,10 +88,13 @@ async function startWorkers() {
}); });
// Worker for interpreting workflow // Worker for interpreting workflow
await pgBoss.work('interpret-workflow', async () => { await pgBoss.work('interpret-workflow', async (job: Job<InterpretWorkflow> | Job<InterpretWorkflow>[]) => {
try { try {
const data = extractJobData(job);
const userId = data.userId;
logger.log('info', 'Starting workflow interpretation job'); logger.log('info', 'Starting workflow interpretation job');
await interpretWholeWorkflow(); await interpretWholeWorkflow(userId);
logger.log('info', 'Workflow interpretation job completed'); logger.log('info', 'Workflow interpretation job completed');
return { success: true }; return { success: true };
} catch (error: unknown) { } catch (error: unknown) {
@@ -93,10 +105,13 @@ async function startWorkers() {
}); });
// Worker for stopping workflow interpretation // Worker for stopping workflow interpretation
await pgBoss.work('stop-interpretation', async () => { await pgBoss.work('stop-interpretation', async (job: Job<StopInterpretWorkflow> | Job<StopInterpretWorkflow>[]) => {
try { try {
const data = extractJobData(job);
const userId = data.userId;
logger.log('info', 'Starting stop interpretation job'); logger.log('info', 'Starting stop interpretation job');
await stopRunningInterpretation(); await stopRunningInterpretation(userId);
logger.log('info', 'Stop interpretation job completed'); logger.log('info', 'Stop interpretation job completed');
return { success: true }; return { success: true };
} catch (error: unknown) { } catch (error: unknown) {