feat: add lazyWithRetry for chunk load failure recovery

This commit is contained in:
ashim-hq
2026-04-20 21:56:08 +08:00
parent 1ad916448a
commit d40d7b3847
2 changed files with 76 additions and 0 deletions
+34
View File
@@ -0,0 +1,34 @@
import { type ComponentType, lazy } from "react";
function isChunkError(error: unknown): boolean {
if (!(error instanceof Error)) return false;
const msg = error.message.toLowerCase();
return (
msg.includes("dynamically imported module") ||
msg.includes("loading chunk") ||
msg.includes("loading css chunk") ||
msg.includes("failed to fetch")
);
}
export async function retryDynamicImport<T>(
importFn: () => Promise<T>,
retries = 3,
delay = 1000,
): Promise<T> {
for (let attempt = 1; attempt <= retries; attempt++) {
try {
return await importFn();
} catch (error) {
if (!isChunkError(error) || attempt === retries) throw error;
await new Promise((resolve) => setTimeout(resolve, delay));
}
}
throw new Error("unreachable");
}
export function lazyWithRetry<T extends ComponentType<unknown>>(
importFn: () => Promise<{ default: T }>,
): React.LazyExoticComponent<T> {
return lazy(() => retryDynamicImport(importFn));
}
+42
View File
@@ -0,0 +1,42 @@
// @vitest-environment jsdom
import { describe, expect, it, vi } from "vitest";
import { retryDynamicImport } from "@/lib/lazy-with-retry";
describe("retryDynamicImport", () => {
it("resolves on first success", async () => {
const mod = { default: () => null };
const importFn = vi.fn().mockResolvedValue(mod);
const result = await retryDynamicImport(importFn);
expect(result).toBe(mod);
expect(importFn).toHaveBeenCalledTimes(1);
});
it("retries on failure and resolves on eventual success", async () => {
const mod = { default: () => null };
const importFn = vi
.fn()
.mockRejectedValueOnce(new TypeError("Failed to fetch dynamically imported module"))
.mockRejectedValueOnce(new TypeError("Failed to fetch dynamically imported module"))
.mockResolvedValue(mod);
const result = await retryDynamicImport(importFn, 3, 0);
expect(result).toBe(mod);
expect(importFn).toHaveBeenCalledTimes(3);
});
it("rejects after all retries exhausted", async () => {
const err = new TypeError("Failed to fetch dynamically imported module");
const importFn = vi.fn().mockRejectedValue(err);
await expect(retryDynamicImport(importFn, 3, 0)).rejects.toThrow(
"Failed to fetch dynamically imported module",
);
expect(importFn).toHaveBeenCalledTimes(3);
});
it("only retries chunk-related errors, not other errors", async () => {
const err = new Error("Some other error");
const importFn = vi.fn().mockRejectedValue(err);
await expect(retryDynamicImport(importFn, 3, 0)).rejects.toThrow("Some other error");
expect(importFn).toHaveBeenCalledTimes(1);
});
});