fix brotli cache

This commit is contained in:
Rizky
2024-06-26 16:32:24 +07:00
parent 2b9c84270f
commit d781756ab9
8 changed files with 123 additions and 47 deletions
+43
View File
@@ -0,0 +1,43 @@
import brotliPromise from "brotli-wasm"; // Import the default export
import { g } from "./global";
import { dir } from "./dir";
const encoder = new TextEncoder();
const brotli = await brotliPromise;
export const loadCachedBr = (hash: string, content: string) => {
if (!g.cache.br[hash]) {
if (!g.cache.br_progress.pending[hash]) {
g.cache.br_progress.pending[hash] = content;
recurseCompressBr();
}
}
};
const recurseCompressBr = () => {
clearTimeout(g.cache.br_progress.timeout);
g.cache.br_progress.timeout = setTimeout(async () => {
if (g.cache.br_progress.running) {
return;
}
g.cache.br_progress.running = true;
const entries = Object.entries(g.cache.br_progress.pending);
if (entries.length > 0) {
const [hash, content] = entries.shift() as [string, string | Uint8Array];
const file = Bun.file(dir(`${g.datadir}/br-cache/${hash}`));
if (await file.exists()) {
g.cache.br[hash] = new Uint8Array(await file.arrayBuffer());
} else {
g.cache.br[hash] = brotli.compress(
typeof content === "string" ? encoder.encode(content) : content,
{ quality: 11 }
);
await Bun.write(file, g.cache.br[hash]);
}
delete g.cache.br_progress.pending[hash];
g.cache.br_progress.running = false;
recurseCompressBr();
}
}, 50);
};
+13 -13
View File
@@ -27,7 +27,7 @@ export const deploy = {
console.log(`Loading site: ${this.config.site_id} ${ts}`);
try {
g.deploy.gz = JSON.parse(
g.deploy.content = JSON.parse(
decoder.decode(
await gunzipAsync(
new Uint8Array(
@@ -37,45 +37,45 @@ export const deploy = {
)
);
if (g.deploy.gz) {
if (g.deploy.content) {
if (exists(dir("public"))) {
await removeAsync(dir("public"));
if (g.deploy.gz.public) {
if (g.deploy.content.public) {
await dirAsync(dir("public"));
for (const [k, v] of Object.entries(g.deploy.gz.public)) {
for (const [k, v] of Object.entries(g.deploy.content.public)) {
await writeAsync(dir(`public/${k}`), v);
}
}
}
for (const page of g.deploy.gz.layouts) {
for (const page of g.deploy.content.layouts) {
if (page.is_default_layout) {
g.deploy.layout = page.content_tree;
break;
}
}
if (!g.deploy.layout && g.deploy.gz.layouts.length > 0) {
g.deploy.layout = g.deploy.gz.layouts[0].content_tree;
if (!g.deploy.layout && g.deploy.content.layouts.length > 0) {
g.deploy.layout = g.deploy.content.layouts[0].content_tree;
}
g.deploy.router = createRouter();
g.deploy.pages = {};
for (const page of g.deploy.gz.pages) {
for (const page of g.deploy.content.pages) {
g.deploy.pages[page.id] = page;
g.deploy.router.insert(page.url, page);
}
g.deploy.comps = {};
for (const comp of g.deploy.gz.comps) {
for (const comp of g.deploy.content.comps) {
g.deploy.comps[comp.id] = comp.content_tree;
}
if (g.deploy.gz.code.server) {
if (g.deploy.content.code.server) {
setTimeout(async () => {
if (g.deploy.gz) {
if (g.deploy.content) {
delete require.cache[dir(`app/web/server/index.js`)];
await removeAsync(dir(`app/web/server`));
await dirAsync(dir(`app/web/server`));
for (const [k, v] of Object.entries(g.deploy.gz.code.server)) {
for (const [k, v] of Object.entries(g.deploy.content.code.server)) {
await writeAsync(dir(`app/web/server/${k}`), v);
}
@@ -139,7 +139,7 @@ export const deploy = {
config: { deploy: { ts: "" }, site_id: "" },
init: false,
raw: null,
gz: null,
content: null,
server: null,
};
}
+3 -2
View File
@@ -70,7 +70,8 @@ export const g = global as unknown as {
};
cache: {
br: Record<string, Uint8Array>;
br_timeout: Set<string>;
br_progress: { pending: Record<string, any>; running: boolean; timeout: any };
gz: Record<string, Uint8Array>;
};
createServer: (
arg: PrasiServer & { api: any; db: any }
@@ -85,7 +86,7 @@ export const g = global as unknown as {
string,
{ id: string; url: string; name: true; content_tree: any }
>;
gz: null | {
content: null | {
layouts: SinglePage[];
pages: SinglePage[];
site: any;