michael/astro
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

feat: cache getCollection() calls in production (#6503)

Browse source 
* feat: cache getCollection() in prod

* chore: changeset
This commit is contained in:
Ben Holmes 2023-03-10 13:50:34 -05:00 committed by GitHub
parent c44aa15534
commit f6eddffa04
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 34 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
.changeset/nasty-cougars-double.md Normal file
View file

@ -0,0 +1,5 @@
---
'astro': patch
---
Add caching to `getCollection()` queries for faster SSG production builds

49
packages/astro/src/content/internal.ts
View file

@ -38,6 +38,7 @@ export function createCollectionToGlobResultMap({
return collectionToGlobResultMap;
}
const cacheEntriesByCollection = new Map<string, any[]>();
export function createGetCollection({
collectionToEntryMap,
collectionToRenderEntryMap,
@ -47,27 +48,35 @@ export function createGetCollection({
}) {
return async function getCollection(collection: string, filter?: (entry: any) => unknown) {
const lazyImports = Object.values(collectionToEntryMap[collection] ?? {});
const entries = Promise.all(
lazyImports.map(async (lazyImport) => {
const entry = await lazyImport();
return {
id: entry.id,
slug: entry.slug,
body: entry.body,
collection: entry.collection,
data: entry.data,
async render() {
return render({
collection: entry.collection,
id: entry.id,
collectionToRenderEntryMap,
});
},
};
})
);
let entries: any[] = [];
// Cache `getCollection()` calls in production only
// prevents stale cache in development
if (import.meta.env.PROD && cacheEntriesByCollection.has(collection)) {
entries = cacheEntriesByCollection.get(collection)!;
} else {
entries = await Promise.all(
lazyImports.map(async (lazyImport) => {
const entry = await lazyImport();
return {
id: entry.id,
slug: entry.slug,
body: entry.body,
collection: entry.collection,
data: entry.data,
async render() {
return render({
collection: entry.collection,
id: entry.id,
collectionToRenderEntryMap,
});
},
};
})
);
cacheEntriesByCollection.set(collection, entries);
}
if (typeof filter === 'function') {
return (await entries).filter(filter);
return entries.filter(filter);
} else {
return entries;
}