perf: optimize trie construction (#448)

nolanlawson · Sep 7, 2024 · 9954aef · 9954aef
1 parent c0dad1d
commit 9954aef
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/database/customEmojiIndex.js b/src/database/customEmojiIndex.js
@@ -16,9 +16,17 @@ export function customEmojiIndex (customEmojis) {
   //
   // search()
   //
-  const emojiToTokens = emoji => (
-    [...new Set((emoji.shortcodes || []).map(shortcode => extractTokens(shortcode)).flat())]
-  )
+  const emojiToTokens = emoji => {
+    const set = new Set()
+    if (emoji.shortcodes) {
+      for (const shortcode of emoji.shortcodes) {
+        for (const token of extractTokens(shortcode)) {
+          set.add(token)
+        }
+      }
+    }
+    return set
+  }
   const searchTrie = trie(customEmojis, emojiToTokens)
   const searchByExactMatch = _ => searchTrie(_, true)
   const searchByPrefix = _ => searchTrie(_, false)