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 index 41922312c..3841a3641 100644 --- 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], }; }