fix: firebase limit on document size causing cache failures

This commit is contained in:
yanlong.wang 2024-04-25 12:24:05 +08:00
parent 9dd5af0cb5
commit f1016649ac
No known key found for this signature in database
GPG Key ID: C0A623C0BADF9F37
3 changed files with 61 additions and 27 deletions

View File

@ -13,7 +13,7 @@ import normalizeUrl from "@esm2cjs/normalize-url";
import { AltTextService } from '../services/alt-text';
import TurndownService from 'turndown';
import { parseString as parseSetCookieString } from 'set-cookie-parser';
import { CookieParam } from 'puppeteer';
import type { CookieParam } from 'puppeteer';
import { Crawled } from '../db/crawled';
import { tidyMarkdown } from '../utils/markdown';
import { cleanAttribute } from '../utils/misc';
@ -408,28 +408,44 @@ ${this.content}
const cache = (await Crawled.fromFirestoreQuery(Crawled.COLLECTION.where('urlPathDigest', '==', digest).orderBy('createdAt', 'desc').limit(1)))?.[0];
if (cache) {
const age = Date.now() - cache.createdAt.valueOf();
const stale = cache.createdAt.valueOf() > (Date.now() - this.cacheValidMs);
this.logger.info(`${stale ? 'Only stale ' : ''}Cache exists for ${urlToCrawl}, normalized digest: ${digest}, ${age}ms old`, {
url: urlToCrawl, digest, age, stale
});
const r = cache.snapshot;
return {
isFresh: !stale,
...cache,
snapshot: {
...r,
screenshot: undefined,
screenshotUrl: cache.screenshotAvailable ?
await this.firebaseObjectStorage.signDownloadUrl(`screenshots/${cache._id}`, Date.now() + this.urlValidMs) : undefined,
} as PageSnapshot & { screenshotUrl?: string; }
};
if (!cache) {
return undefined;
}
return undefined;
const age = Date.now() - cache.createdAt.valueOf();
const stale = cache.createdAt.valueOf() < (Date.now() - this.cacheValidMs);
this.logger.info(`${stale ? 'Stale cache exists' : 'Cache hit'} for ${urlToCrawl}, normalized digest: ${digest}, ${age}ms old`, {
url: urlToCrawl, digest, age, stale
});
let snapshot: PageSnapshot | undefined;
let screenshotUrl: string | undefined;
const preparations = [
this.firebaseObjectStorage.downloadFile(`snapshots/${cache._id}`).then((r) => {
snapshot = JSON.parse(r.toString('utf-8'));
}),
cache.screenshotAvailable ?
this.firebaseObjectStorage.signDownloadUrl(`screenshots/${cache._id}`, Date.now() + this.urlValidMs).then((r) => {
screenshotUrl = r;
}) :
Promise.resolve(undefined)
];
try {
await Promise.all(preparations);
} catch (_err) {
// Swallow cache errors.
return undefined;
}
return {
isFresh: !stale,
...cache,
snapshot: {
...snapshot,
screenshot: undefined,
screenshotUrl,
} as PageSnapshot & { screenshotUrl?: string; }
};
}
async setToCache(urlToCrawl: URL, snapshot: PageSnapshot) {
@ -444,10 +460,24 @@ ${this.content}
createdAt: nowDate,
expireAt: new Date(nowDate.valueOf() + this.cacheRetentionMs),
urlPathDigest: digest,
snapshot: {
...snapshot,
screenshot: null
},
});
const savingOfSnapshot = this.firebaseObjectStorage.saveFile(`snapshots/${cache._id}`,
Buffer.from(
JSON.stringify({
...snapshot,
screenshot: undefined
}),
'utf-8'
),
{
metadata: {
contentType: 'application/json',
}
}
).then((r) => {
cache.snapshotAvailable = true;
return r;
});
if (snapshot.screenshot) {
@ -458,6 +488,7 @@ ${this.content}
});
cache.screenshotAvailable = true;
}
await savingOfSnapshot;
const r = await Crawled.save(cache.degradeForFireStore()).catch((err) => {
this.logger.error(`Failed to save cache for ${urlToCrawl}`, { err: marshalErrorLike(err) });

View File

@ -22,11 +22,14 @@ export class Crawled extends FirestoreRecord {
urlPathDigest!: string;
@Prop()
snapshot!: PageSnapshot & { screenshot: never; };
snapshot?: PageSnapshot & { screenshot: never; };
@Prop()
screenshotAvailable?: boolean;
@Prop()
snapshotAvailable?: boolean;
@Prop()
createdAt!: Date;

@ -1 +1 @@
Subproject commit 577131db50d5c86ffb3d085a593eaed8950eabcd
Subproject commit 64157bc57ef9ce2cec69f37b5f55fccb71742b6f