Address PR comments

elastic · Jul 2, 2024 · 11dbb52 · 11dbb52
1 parent 14c43b4
commit 11dbb52
Showing 1 changed file with 6 additions and 13 deletions.
diff --git a/x-pack/plugins/lens/common/expressions/map_to_columns/map_to_columns_fn_textbased.ts b/x-pack/plugins/lens/common/expressions/map_to_columns/map_to_columns_fn_textbased.ts
@@ -28,18 +28,11 @@ export const mapToOriginalColumnsTextBased: MapToColumnsExpressionFunction['fn']
 
       return mappedRow;
     }),
-    columns: data.columns
-      .filter((column) => idMap[column.id])
-      .map((column) => {
-        const originalColumns = idMap[column.id];
-        return (
-          originalColumns &&
-          originalColumns.map((originalColumn) => ({
-            ...column,
-            id: originalColumn.id,
-          }))
-        );
-      })
-      .flat(),
+    columns: data.columns.flatMap((column) => {
+      if (!(column.id in idMap)) {
+        return [];
+      }
+      return idMap[column.id].map((originalColumn) => ({ ...column, id: originalColumn.id }));
+    }),
   };
 };