swap for lru-cache with 10KB

vercel · Dec 3, 2024 · 5f598cb · 5f598cb
1 parent 1be89a5
commit 5f598cb
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/packages/next/src/build/output/log.ts b/packages/next/src/build/output/log.ts
@@ -1,4 +1,5 @@
 import { bold, green, magenta, red, yellow, white } from '../../lib/picocolors'
+import { LRUCache } from '../../server/lib/lru-cache'
 
 export const prefixes = {
   wait: white(bold('○')),
@@ -76,15 +77,12 @@ export function trace(...message: any[]) {
   prefixedLog('trace', ...message)
 }
 
-const warnOnceMessages = new Set()
-export function warnOnce(...message: any[]) {
-  if (!warnOnceMessages.has(message[0])) {
-    if (warnOnceMessages.size > 1000) {
-      // delete the oldest message (FIFO) to prevent memory leak
-      warnOnceMessages.delete(warnOnceMessages.values().next().value)
-    }
-    warnOnceMessages.add(message.join(' '))
+const warnOnceCache = new LRUCache<string>(10_000, (message) => message.length)
 
+export function warnOnce(...message: any[]) {
+  const key = message.join(' ')
+  if (!warnOnceCache.has(key)) {
+    warnOnceCache.set(key)
     warn(...message)
   }
 }