Austin/firecrawl
1
0
Fork 0
mirror of https://github.com/mendableai/firecrawl.git synced 2024-11-16 03:32:22 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

feat(batchScrape): handle timeout

Browse Source
This commit is contained in:
Móricz Gergő 2024-11-12 12:42:39 +01:00
parent f6db9f1428
commit 91f52287db
1 changed files with 24 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
apps/api/src/services/queue-worker.ts
View File

@ -276,10 +276,18 @@ async function processJob(job: Job & { id: string }, token: string) {
}); });
const start = Date.now(); const start = Date.now();
const pipeline = await startWebScraperPipeline({ const pipeline = await Promise.race([
job, startWebScraperPipeline({
token, job,
}); token,
}),
...(job.data.scrapeOptions.timeout !== undefined ? [
(async () => {
await sleep(job.data.scrapeOptions.timeout);
throw new Error("timeout")
})(),
] : [])
]);
if (!pipeline.success) { if (!pipeline.success) {
// TODO: let's Not do this // TODO: let's Not do this
@ -486,13 +494,19 @@ async function processJob(job: Job & { id: string }, token: string) {
logger.info(`🐂 Job done ${job.id}`); logger.info(`🐂 Job done ${job.id}`);
return data; return data;
} catch (error) { } catch (error) {
logger.error(`🐂 Job errored ${job.id} - ${error}`); const isEarlyTimeout = error instanceof Error && error.message === "timeout";
Sentry.captureException(error, { if (!isEarlyTimeout) {
data: { logger.error(`🐂 Job errored ${job.id} - ${error}`);
job: job.id,
}, Sentry.captureException(error, {
}); data: {
job: job.id,
},
});
} else {
logger.error(`🐂 Job timed out ${job.id}`);
}
if (error instanceof CustomError) { if (error instanceof CustomError) {
// Here we handle the error, then save the failed job // Here we handle the error, then save the failed job