[Backport 2.x] [Bug] Traces/Services remove toast message on empty data

public/components/trace_analytics/requests/services_request_handler.ts

@@ -25,11 +25,11 @@
 
 export const handleServicesRequest = async (
   http: HttpSetup,
   DSL: any,
   setItems: any,
   mode: TraceAnalyticsMode,
   dataSourceMDSId?: string,
   setServiceMap?: any,
   serviceNameFilter?: string
 ) => {
   return handleDslRequest(
@@ -47,10 +47,16 @@
         dataSourceMDSId,
         setServiceMap
       );
+
+      if (!serviceObject || Object.keys(serviceObject).length === 0) {
+        setItems([]);
+        return [];
+      }
+
       return Promise.all(
         response.aggregations.service.buckets
           .filter((bucket: any) => serviceObject[bucket.key])
           .map((bucket: any) => {
             const connectedServices = [
               ...serviceObject[bucket.key].targetServices,
               ...serviceObject[bucket.key].destServices,
@@ -82,10 +88,10 @@
 
 export const handleServiceMapRequest = async (
   http: HttpSetup,
   DSL: DSLService | any,
   mode: TraceAnalyticsMode,
   dataSourceMDSId?: string,
   setItems?: any,
   includeMetrics = true
 ) => {
   let minutesInDateRange: number;
@@ -103,38 +109,42 @@
     getServiceNodesQuery(mode),
     mode,
     dataSourceMDSId
-  )
-    .then((response) => {
-      response.aggregations.service_name.buckets.map(
-        (bucket: any) =>
-          (map[bucket.key] = {
-            serviceName: bucket.key,
-            id: id++,
-            targetResources: bucket.target_resource.buckets.map((res: any) => res.key),
-            targetServices: [],
-            destServices: [],
-          })
-      );
-      return true;
-    })
-    .catch((error) => {
-      console.error('Error retrieving service nodes:', error);
-      coreRefs.core?.notifications.toasts.addError(error, {
-        title: 'Failed to retrieve service nodes',
-        toastLifeTimeMs: 10000,
-      });
-      return false;
+  ).catch((error) => {
+    console.error('Error retrieving service nodes:', error);
+    coreRefs.core?.notifications.toasts.addError(error, {
+      title: 'Failed to retrieve service nodes',
+      toastLifeTimeMs: 10000,
     });
+    return null;
+  });
 
-  // Early return if service node not found
-  if (!serviceNodesResponse) {
+  if (
+    !serviceNodesResponse ||
+    !serviceNodesResponse.aggregations ||
+    !serviceNodesResponse.aggregations.service_name ||
+    !serviceNodesResponse.aggregations.service_name.buckets ||
+    serviceNodesResponse.aggregations.service_name.buckets.length === 0
+  ) {
+    if (setItems) {
+      setItems(map);
+    }
     return map;
   }
 
-  const targets = {};
+  serviceNodesResponse.aggregations.service_name.buckets.forEach((bucket: any) => {
+    map[bucket.key] = {
+      serviceName: bucket.key,
+      id: id++,
+      targetResources: bucket.target_resource.buckets.map((res: any) => res.key),
+      targetServices: [],
+      destServices: [],
+    };
+  });
+
+  const targets: Record<string, string> = {};
   await handleDslRequest(http, null, getServiceEdgesQuery('target', mode), mode, dataSourceMDSId)
     .then((response) =>
       response.aggregations.service_name.buckets.map((bucket: any) => {
         bucket.resource.buckets.map((resource: any) => {
           resource.domain.buckets.map((domain: any) => {
             targets[resource.key + ':' + domain.key] = bucket.key;

public/components/trace_analytics/requests/traces_request_handler.ts

@@ -153,6 +153,21 @@ export const handleTracesRequest = async (
         percentileRangesResult.status === 'fulfilled' ? percentileRangesResult.value : {};
       const response = responseResult.value;
 
+      if ((response.statusCode && response.statusCode >= 400) || response.error) {
+        return Promise.reject(response);
+      }
+
+      if (
+        !response ||
+        !response.aggregations ||
+        !response.aggregations.traces ||
+        !response.aggregations.traces.buckets ||
+        response.aggregations.traces.buckets.length === 0
+      ) {
+        setItems([]);
+        return [];
+      }
+
       return response.aggregations.traces.buckets.map((bucket: any) => {
         if (mode === 'data_prepper' || mode === 'custom_data_prepper') {
           return {