fix: group node neighborhood (#2565)

hypertrace · Jan 2, 2024 · e9cbabb · e9cbabb
1 parent 014a1ff
commit e9cbabb
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/projects/observability/src/shared/components/topology/utils/topology-neighborhood-finder.ts b/projects/observability/src/shared/components/topology/utils/topology-neighborhood-finder.ts
@@ -1,10 +1,15 @@
 import { uniq } from 'lodash-es';
 import { TopologyEdge, TopologyNeighborhood, TopologyNode } from '../topology';
+import { TopologyGroupNodeUtil } from './topology-group-node.util';
 
 export class TopologyNeighborhoodFinder {
   public neighborhoodForNode(node: TopologyNode): TopologyNeighborhood {
     return {
-      nodes: uniq([node, ...node.edges.flatMap(edge => [edge.fromNode, edge.toNode])]),
+      nodes: uniq([
+        node,
+        ...node.edges.flatMap(edge => [edge.fromNode, edge.toNode]),
+        ...(TopologyGroupNodeUtil.isTopologyGroupNode(node) && node.expanded ? node.children : []),
+      ]),
       edges: [...node.edges],
     };
   }