Set Legacy Discover Table for Discover 2.0 (opensearch-project#8233)

Signed-off-by: Suchit Sahoo <[email protected]>
virajsanghvi · Sep 24, 2024 · 1b8b8a7 · 1b8b8a7
1 parent cff56e1
commit 1b8b8a7
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 29 deletions.
diff --git a/src/plugins/discover/public/application/components/chart/chart.tsx b/src/plugins/discover/public/application/components/chart/chart.tsx
@@ -118,7 +118,6 @@ export const DiscoverChart = ({
       <EuiFlexItem grow={true} style={{ justifyContent: 'flex-start' }}>
         {isTimeBased && timeChartHeader}
       </EuiFlexItem>
-      <EuiFlexItem grow={false}>{discoverOptions}</EuiFlexItem>
     </EuiFlexGroup>
   );
 

diff --git a/src/plugins/discover/public/application/components/data_grid/data_grid_table.tsx b/src/plugins/discover/public/application/components/data_grid/data_grid_table.tsx
@@ -6,6 +6,7 @@
 import { i18n } from '@osd/i18n';
 import React, { useState } from 'react';
 import { EuiPanel } from '@elastic/eui';
+import { QUERY_ENHANCEMENT_ENABLED_SETTING } from '../../../../common';
 import { IndexPattern, getServices } from '../../../opensearch_dashboards_services';
 import { DataGridFlyout } from './data_grid_table_flyout';
 import { DiscoverGridContextProvider } from './data_grid_table_context';
@@ -71,35 +72,37 @@ export const DataGridTable = ({
   }
 
   const newDiscoverEnabled = getNewDiscoverSetting(services.storage);
+  const isQueryEnhancementEnabled = services.uiSettings.get(QUERY_ENHANCEMENT_ENABLED_SETTING);
 
-  const panelContent = newDiscoverEnabled ? (
-    <DataGrid
-      columns={columns}
-      indexPattern={indexPattern}
-      sort={sort}
-      onSort={onSort}
-      rows={rows}
-      onSetColumns={onSetColumns}
-      isToolbarVisible={isToolbarVisible}
-      isContextView={isContextView}
-    />
-  ) : (
-    <DefaultDiscoverTable
-      columns={adjustedColumns}
-      indexPattern={indexPattern}
-      sort={sort}
-      onSort={onSort}
-      rows={rows}
-      hits={hits}
-      onAddColumn={onAddColumn}
-      onMoveColumn={onMoveColumn}
-      onRemoveColumn={onRemoveColumn}
-      onFilter={onFilter}
-      onClose={() => setInspectedHit(undefined)}
-      showPagination={showPagination}
-      scrollToTop={scrollToTop}
-    />
-  );
+  const panelContent =
+    isQueryEnhancementEnabled || !newDiscoverEnabled ? (
+      <DefaultDiscoverTable
+        columns={adjustedColumns}
+        indexPattern={indexPattern}
+        sort={sort}
+        onSort={onSort}
+        rows={rows}
+        hits={hits}
+        onAddColumn={onAddColumn}
+        onMoveColumn={onMoveColumn}
+        onRemoveColumn={onRemoveColumn}
+        onFilter={onFilter}
+        onClose={() => setInspectedHit(undefined)}
+        showPagination={showPagination}
+        scrollToTop={scrollToTop}
+      />
+    ) : (
+      <DataGrid
+        columns={columns}
+        indexPattern={indexPattern}
+        sort={sort}
+        onSort={onSort}
+        rows={rows}
+        onSetColumns={onSetColumns}
+        isToolbarVisible={isToolbarVisible}
+        isContextView={isContextView}
+      />
+    );
 
   const tablePanelProps = {
     paddingSize: 'none' as const,