perf(core): cache rule index to reduce query index (#4892)

lghuahua · zyyv · web-flow · commit 48ea5e30f7c0 · 2025-08-31T06:50:39.000Z
Co-authored-by: Chris &lt;hizyyv@gmail.com&gt;
diff --git a/packages-engine/core/src/config.ts b/packages-engine/core/src/config.ts
@@ -1,4 +1,4 @@
-import type { ContentOptions, FilterPattern, Preset, PresetFactory, PresetFactoryAwaitable, PresetOrFactoryAwaitable, ResolvedConfig, Rule, Shortcut, ToArray, UserConfig, UserConfigDefaults, UserShortcuts } from './types'
+import type { ContentOptions, DynamicRule, FilterPattern, Preset, PresetFactory, PresetFactoryAwaitable, PresetOrFactoryAwaitable, ResolvedConfig, Rule, Shortcut, ToArray, UserConfig, UserConfigDefaults, UserShortcuts } from './types'
 import { DEFAULT_LAYERS } from './constants'
 import { extractorSplit } from './extractors'
 import { clone, isStaticRule, mergeDeep, normalizeVariant, toArray, uniq, uniqueBy } from './utils'
@@ -155,22 +155,25 @@ export async function resolveConfig<Theme extends object = object>(
   extractors.sort((a, b) => (a.order || 0) - (b.order || 0))
 
   const rules = getMerged('rules')
+  const rulesSize = rules.length
   const rulesStaticMap: ResolvedConfig<Theme>['rulesStaticMap'] = {}
+  const rulesDynamic: ResolvedConfig<Theme>['rulesDynamic'] = []
 
-  const rulesSize = rules.length
+  for (const [index, rule] of rules.entries()) {
+    const meta = rule[2] ?? (rule[2] = {})
+    meta.__index = index
 
-  const rulesDynamic = rules
-    .filter((rule) => {
-      if (!isStaticRule(rule))
-        return true
+    if (isStaticRule(rule)) {
       // Put static rules into the map for faster lookup
-      const prefixes = toArray(rule[2]?.prefix || '')
+      const prefixes = toArray(meta.prefix ?? '')
       prefixes.forEach((prefix) => {
         rulesStaticMap[prefix + rule[0]] = rule
       })
-      return false
-    })
-    .reverse() as ResolvedConfig<Theme>['rulesDynamic']
+    }
+    else {
+      (rulesDynamic as DynamicRule<Theme>[]).unshift(rule)
+    }
+  }
 
   const autocomplete = {
     templates: uniq(sources.flatMap(p => toArray(p.autocomplete?.templates))),
diff --git a/packages-engine/core/src/generator.ts b/packages-engine/core/src/generator.ts
@@ -672,12 +672,11 @@ class UnoGeneratorInternal<Theme extends object = object> {
         context.rules!.push(rule)
       }
       context.generator.activatedRules.add(rule)
-      const index = context.generator.config.rules.indexOf(rule)
       const meta = rule[2]
 
       return entries.map((css): ParsedUtil | RawUtil => {
         if (isString(css))
-          return [index, css, meta]
+          return [meta!.__index!, css, meta]
 
         // Extract variants from special symbols
         let variants = context.variantHandlers
@@ -729,7 +728,7 @@ class UnoGeneratorInternal<Theme extends object = object> {
           }
         }
 
-        return [index, raw, css as CSSEntries, entryMeta, variants]
+        return [meta!.__index!, raw, css as CSSEntries, entryMeta, variants]
       })
     }
   }
diff --git a/packages-engine/core/src/types.ts b/packages-engine/core/src/types.ts
@@ -228,6 +228,13 @@ export interface RuleMeta {
    */
   __hash?: string
 
+  /**
+   * Internal index of the rulelist
+   * @internal
+   * @private
+   */
+  __index?: number
+
   /**
    * Custom metadata
    */
diff --git a/packages-engine/core/test/__snapshots__/extended-info.test.ts.snap b/packages-engine/core/test/__snapshots__/extended-info.test.ts.snap
@@ -11,6 +11,7 @@ Map {
         "name:bar1;",
         undefined,
         {
+          "__index": 0,
           "layer": "a",
           "sort": undefined,
         },
@@ -24,6 +25,7 @@ Map {
                 "name": "bar1",
               },
               {
+                "__index": 0,
                 "layer": "a",
               },
             ],
@@ -61,6 +63,7 @@ Map {
         "name:bar2;",
         undefined,
         {
+          "__index": 1,
           "layer": "b",
           "sort": undefined,
         },
@@ -74,6 +77,7 @@ Map {
                 "name": "bar2",
               },
               {
+                "__index": 1,
                 "layer": "b",
               },
             ],
@@ -111,6 +115,7 @@ Map {
         "name:1;",
         undefined,
         {
+          "__index": 2,
           "layer": "c",
           "sort": undefined,
         },
@@ -122,6 +127,7 @@ Map {
               /\\^c\\(\\\\d\\+\\)\\$/,
               [Function],
               {
+                "__index": 2,
                 "layer": "c",
               },
             ],
diff --git a/packages-integrations/svelte-scoped/src/_preprocess/transformApply/getUtils.test.ts b/packages-integrations/svelte-scoped/src/_preprocess/transformApply/getUtils.test.ts
@@ -34,6 +34,7 @@ describe('getUtils', async () => {
           "margin-bottom:0.25rem;margin-right:0.25rem;",
           undefined,
           {
+            "__index": 90,
             "layer": undefined,
             "sort": undefined,
           },
@@ -46,6 +47,7 @@ describe('getUtils', async () => {
           "margin-right:0.5rem;",
           undefined,
           {
+            "__index": 90,
             "layer": undefined,
             "sort": 24,
           },
@@ -69,6 +71,7 @@ describe('getUtils', async () => {
           "margin-right:0.25rem;font-weight:500;",
           undefined,
           {
+            "__index": 90,
             "layer": undefined,
             "sort": 24,
           },

Original file line number	Diff line number	Diff line change
`@@ -672,12 +672,11 @@ class UnoGeneratorInternal<Theme extends object = object> {`
`672`	`672`	`context.rules!.push(rule)`
`673`	`673`	`}`
`674`	`674`	`context.generator.activatedRules.add(rule)`
`675`		`- const index = context.generator.config.rules.indexOf(rule)`
`676`	`675`	`const meta = rule[2]`
`677`	`676`
`678`	`677`	`return entries.map((css): ParsedUtil \| RawUtil => {`
`679`	`678`	`if (isString(css))`
`680`		`- return [index, css, meta]`
	`679`	`+ return [meta!.__index!, css, meta]`
`681`	`680`
`682`	`681`	`// Extract variants from special symbols`
`683`	`682`	`let variants = context.variantHandlers`
`@@ -729,7 +728,7 @@ class UnoGeneratorInternal<Theme extends object = object> {`
`729`	`728`	`}`
`730`	`729`	`}`
`731`	`730`
`732`		`- return [index, raw, css as CSSEntries, entryMeta, variants]`
	`731`	`+ return [meta!.__index!, raw, css as CSSEntries, entryMeta, variants]`
`733`	`732`	`})`
`734`	`733`	`}`
`735`	`734`	`}`