[Discover 2.0] Loading fix for databases

changelogs/fragments/7567.yml

@@ -0,0 +1,2 @@
+fix:
+- Fixes databases not being displayed upon success ([#7567](https://github.com/opensearch-project/OpenSearch-Dashboards/pull/7567))

src/plugins/data/public/ui/dataset_navigator/dataset_navigator.tsx

@@ -286,6 +286,29 @@
     [selectedDataSetState?.dataSourceRef, tablesLoadStatus, startLoadingTables]
   );
 
+  useEffect(() => {
+    const status = databasesLoadStatus.toLowerCase();
+    if (
+      selectedDataSetState?.dataSourceRef &&
+      selectedDataSetState.dataSourceRef.type === SIMPLE_DATA_SOURCE_TYPES.EXTERNAL
+    ) {
+      const dataSourceCache = CatalogCacheManager.getOrCreateDataSource(
+        selectedDataSetState.dataSourceRef.name,
+        selectedDataSetState.dataSourceRef.id
+      );
+      if (status === DirectQueryLoadingStatus.SUCCESS) {
+        setCachedDatabases(
+          dataSourceCache.databases.map((db) => ({ id: db.name, title: db.name }))
+        );
+      } else if (
+        status === DirectQueryLoadingStatus.CANCELED ||
+        status === DirectQueryLoadingStatus.FAILED
+      ) {
+        notifications.toasts.addWarning('Error loading databases');
+      }
+    }
+  }, [databasesLoadStatus, selectedDataSetState?.dataSourceRef, notifications.toasts]);
+
   useEffect(() => {
     if (
       selectedDataSetState?.dataSourceRef &&