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

feat: register worker immediately

Browse Source
This commit is contained in:
Rohit Rajan
2025-09-10 10:21:29 +05:30
parent 7834f1730a
commit b31daa6c1b
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
server/src/routes/storage.ts
View File

@@ -17,7 +17,7 @@ import { capture } from "../utils/analytics";
import { encrypt, decrypt } from '../utils/auth'; import { encrypt, decrypt } from '../utils/auth';
import { WorkflowFile } from 'maxun-core'; import { WorkflowFile } from 'maxun-core';
import { cancelScheduledWorkflow, scheduleWorkflow } from '../schedule-worker'; import { cancelScheduledWorkflow, scheduleWorkflow } from '../schedule-worker';
import { pgBoss } from '../pgboss-worker'; import { pgBoss, registerWorkerForQueue, registerAbortWorkerForQueue } from '../pgboss-worker';
chromium.use(stealthPlugin()); chromium.use(stealthPlugin());
export const router = Router(); export const router = Router();
@@ -573,6 +573,7 @@ router.put('/runs/:id', requireSignIn, async (req: AuthenticatedRequest, res) =>
try { try {
const userQueueName = `execute-run-user-${req.user.id}`; const userQueueName = `execute-run-user-${req.user.id}`;
await pgBoss.createQueue(userQueueName); await pgBoss.createQueue(userQueueName);
await registerWorkerForQueue(userQueueName);
const jobId = await pgBoss.send(userQueueName, { const jobId = await pgBoss.send(userQueueName, {
userId: req.user.id, userId: req.user.id,
@@ -690,6 +691,7 @@ router.post('/runs/run/:id', requireSignIn, async (req: AuthenticatedRequest, re
// Queue the execution job // Queue the execution job
await pgBoss.createQueue(userQueueName); await pgBoss.createQueue(userQueueName);
await registerWorkerForQueue(userQueueName);
const jobId = await pgBoss.send(userQueueName, { const jobId = await pgBoss.send(userQueueName, {
userId: req.user.id, userId: req.user.id,
@@ -962,6 +964,7 @@ router.post('/runs/abort/:id', requireSignIn, async (req: AuthenticatedRequest,
const userQueueName = `abort-run-user-${req.user.id}`; const userQueueName = `abort-run-user-${req.user.id}`;
await pgBoss.createQueue(userQueueName); await pgBoss.createQueue(userQueueName);
await registerAbortWorkerForQueue(userQueueName);
const jobId = await pgBoss.send(userQueueName, { const jobId = await pgBoss.send(userQueueName, {
userId: req.user.id, userId: req.user.id,
@@ -1029,6 +1032,7 @@ async function processQueuedRuns() {
const userQueueName = `execute-run-user-${userId}`; const userQueueName = `execute-run-user-${userId}`;
await pgBoss.createQueue(userQueueName); await pgBoss.createQueue(userQueueName);
await registerWorkerForQueue(userQueueName);
const jobId = await pgBoss.send(userQueueName, { const jobId = await pgBoss.send(userQueueName, {
userId: userId, userId: userId,