AI/firecrawl
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/mendableai/firecrawl synced 2025-08-12 14:28:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(sitemap): propagate crawlid

Browse Source
This commit is contained in:
Móricz Gergő 2025-01-23 07:19:00 +01:00
parent a922aac805
commit aa2c369060
2 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
apps/api/src/scraper/WebScraper/crawler.ts
View File

@ -555,6 +555,7 @@ export class WebCrawler {
sitemapCount = await getLinksFromSitemap(
{ sitemapUrl, urlsHandler, mode: "fire-engine" },
this.logger,
this.jobId,
);
} catch (error) {
this.logger.debug(`Failed to fetch sitemap from ${sitemapUrl}`, {
@ -595,6 +596,7 @@ export class WebCrawler {
mode: "fire-engine",
},
this.logger,
this.jobId,
);
} catch (error) {
this.logger.debug(
@ -618,6 +620,7 @@ export class WebCrawler {
sitemapCount += await getLinksFromSitemap(
{ sitemapUrl: baseUrlSitemap, urlsHandler, mode: "fire-engine" },
this.logger,
this.jobId,
);
} catch (error) {
this.logger.debug(`Failed to fetch sitemap from ${baseUrlSitemap}`, {
@ -631,6 +634,7 @@ export class WebCrawler {
sitemapCount += await getLinksFromSitemap(
{ sitemapUrl: baseUrlSitemap, urlsHandler, mode: "fire-engine" },
this.logger,
this.jobId,
);
}
}

8
apps/api/src/scraper/WebScraper/sitemap.ts
View File

@ -18,13 +18,14 @@ export async function getLinksFromSitemap(
mode?: "axios" | "fire-engine";
},
logger: Logger,
crawlId: string,
): Promise<number> {
try {
let content: string = "";
try {
if (mode === "fire-engine" && useFireEngine) {
const fetchResponse = await scrapeURL(
"sitemap",
"sitemap;" + crawlId,
sitemapUrl,
scrapeOptions.parse({ formats: ["rawHtml"] }),
{ forceEngine: "fetch" },
@ -79,7 +80,7 @@ export async function getLinksFromSitemap(
}
} else {
const fetchResponse = await scrapeURL(
"sitemap",
"sitemap;" + crawlId,
sitemapUrl,
scrapeOptions.parse({ formats: ["rawHtml"] }),
{ forceEngine: "fetch" },
@ -125,7 +126,7 @@ export async function getLinksFromSitemap(
.map((sitemap) => sitemap.loc[0].trim());
const sitemapPromises: Promise<number>[] = sitemapUrls.map((sitemapUrl) =>
getLinksFromSitemap({ sitemapUrl, urlsHandler, mode }, logger),
getLinksFromSitemap({ sitemapUrl, urlsHandler, mode }, logger, crawlId),
);
const results = await Promise.all(sitemapPromises);
@ -147,6 +148,7 @@ export async function getLinksFromSitemap(
getLinksFromSitemap(
{ sitemapUrl: sitemapUrl, urlsHandler, mode },
logger,
crawlId,
),
);
count += (await Promise.all(sitemapPromises)).reduce(