perf(algolia): optimize chart indexing code a bit

owid · Apr 10, 2024 · 5aa3cd9 · 5aa3cd9
1 parent 6fba963
commit 5aa3cd9
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/baker/algolia/indexChartsToAlgolia.ts b/baker/algolia/indexChartsToAlgolia.ts
@@ -23,12 +23,14 @@ const computeScore = (record: Omit<ChartRecord, "score">): number => {
     return numRelatedArticles * 500 + views_7d
 }
 
-const processAvailableEntities = (availableEntities: string[] | null) => {
-    if (!availableEntities) return []
-
-    const countriesWithVariantNames = countries
+const countriesWithVariantNames = new Set(
+    countries
         .filter((country) => country.variantNames?.length || country.shortName)
         .map((country) => country.name)
+)
+
+const processAvailableEntities = (availableEntities: string[] | null) => {
+    if (!availableEntities) return []
 
     // Algolia is a bit weird with synonyms:
     // If we have a synonym "USA" -> "United States", and we search for "USA",
@@ -41,7 +43,7 @@ const processAvailableEntities = (availableEntities: string[] | null) => {
         availableEntities,
         [
             (entityName) =>
-                countriesWithVariantNames.includes(
+                countriesWithVariantNames.has(
                     removeTrailingParenthetical(entityName)
                 ),
             (entityName) => entityName,