Interaction and UI updates in Sankey Chart (#2997)

This commit is contained in:
Sudheer Salavadi 2025-02-03 08:40:15 -05:00 committed by GitHub
parent ea8729dd93
commit 2a52de073d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 180 additions and 215 deletions

View file

@ -1,21 +1,23 @@
// START GEN
import React from 'react';
import { echarts, defaultOptions } from './init';
import { SankeyChart } from 'echarts/charts';
import { sankeyTooltip, getEventPriority, getNodeName } from './sankeyUtils'
import { sankeyTooltip, getEventPriority, getNodeName } from './sankeyUtils';
import { boxShadow } from 'html2canvas/dist/types/css/property-descriptors/box-shadow';
echarts.use([SankeyChart]);
interface SankeyNode {
name: string | null; // e.g. "/en/deployment/", or null
eventType?: string; // e.g. "LOCATION" (not strictly needed by ECharts)
name: string | null;
eventType?: string;
depth?: number;
id?: string | number;
}
interface SankeyLink {
source: number; // index of source node
target: number; // index of target node
value: number; // percentage
source: number;
target: number;
value: number;
sessionsCount: number;
eventType?: string; // optional
eventType?: string;
}
interface Data {
@ -29,35 +31,6 @@ interface Props {
onChartClick?: (filters: any[]) => void;
}
// Not working properly
function findHighestContributors(nodeIndex: number, links: SankeyLink[]) {
const contributors: SankeyLink[] = [];
let currentNode = nodeIndex;
while (true) {
let maxContribution = -Infinity;
let primaryLink: SankeyLink | null = null;
for (const link of links) {
if (link.target === currentNode) {
if (link.value > maxContribution) {
maxContribution = link.value;
primaryLink = link;
}
}
}
if (primaryLink) {
contributors.push(primaryLink);
currentNode = primaryLink.source;
} else {
break;
}
}
return contributors;
}
const EChartsSankey: React.FC<Props> = (props) => {
const { data, height = 240, onChartClick } = props;
const chartRef = React.useRef<HTMLDivElement>(null);
@ -67,27 +40,49 @@ const EChartsSankey: React.FC<Props> = (props) => {
const chart = echarts.init(chartRef.current);
const nodeValues = new Array(data.nodes.length).fill(0);
const echartNodes = data.nodes
.map((n, i) => ({
name: getNodeName(n.eventType || 'Other', n.name),
const echartNodes = data.nodes.map((n) => {
let computedName = getNodeName(n.eventType || 'Minor Paths', n.name);
if (computedName === 'Other') {
computedName = 'Minor Paths';
}
const itemColor =
computedName === 'Minor Paths'
? '#222F99'
: n.eventType === 'DROP'
? '#B5B7C8'
: '#394eff';
return {
name: computedName,
depth: n.depth,
type: n.eventType,
id: n.id,
}))
draggable: false,
itemStyle: { color: itemColor },
};
})
.sort((a, b) => {
if (a.depth === b.depth) {
return getEventPriority(a.type || '') - getEventPriority(b.type || '')
return getEventPriority(a.type || '') - getEventPriority(b.type || '');
} else {
return a.depth - b.depth;
return (a.depth as number) - (b.depth as number);
}
});
const echartLinks = data.links.map((l, i) => ({
const echartLinks = data.links.map((l) => ({
source: echartNodes.findIndex((n) => n.id === l.source),
target: echartNodes.findIndex((n) => n.id === l.target),
value: l.sessionsCount,
percentage: l.value,
lineStyle: { opacity: 0.1 },
}));
nodeValues.forEach((v, i) => {
const outgoingValues = echartLinks
.filter((l) => l.source === i)
@ -96,13 +91,18 @@ const EChartsSankey: React.FC<Props> = (props) => {
.filter((l) => l.target === i)
.reduce((p, c) => p + c.value, 0);
nodeValues[i] = Math.max(outgoingValues, incomingValues);
})
});
const option = {
...defaultOptions,
tooltip: {
trigger: 'item',
},
toolbox: {
feature: {
saveAsImage: { show: false }
}
},
series: [
{
layoutIterations: 0,
@ -110,26 +110,28 @@ const EChartsSankey: React.FC<Props> = (props) => {
data: echartNodes,
links: echartLinks,
emphasis: {
focus: 'adjacency',
blurScope: 'global',
focus: 'none',
lineStyle: {
opacity: 0.5,
},
},
label: {
formatter: '{b} - {c}'
formatter: '{b} - {c}',
},
tooltip: {
formatter: sankeyTooltip(echartNodes, nodeValues)
formatter: sankeyTooltip(echartNodes, nodeValues),
},
nodeAlign: 'right',
nodeWidth: 10,
nodeWidth: 40,
nodeGap: 8,
lineStyle: {
color: 'source',
curveness: 0.5,
opacity: 0.3,
curveness: 0.2,
opacity: 0.1,
},
itemStyle: {
color: '#394eff',
borderRadius: 4,
borderRadius: 2,
},
},
],
@ -137,146 +139,112 @@ const EChartsSankey: React.FC<Props> = (props) => {
chart.setOption(option);
const seriesIndex = 0;
function highlightNode(nodeIdx: number) {
chart.dispatchAction({
type: 'highlight',
seriesIndex,
dataType: 'node',
dataIndex: nodeIdx,
});
}
function highlightLink(linkIdx: number) {
chart.dispatchAction({
type: 'highlight',
seriesIndex,
dataType: 'edge',
dataIndex: linkIdx,
});
}
function resetHighlight() {
chart.dispatchAction({
type: 'downplay',
seriesIndex,
function getUpstreamNodes(nodeIdx: number, visited = new Set<number>()) {
if (visited.has(nodeIdx)) return;
visited.add(nodeIdx);
echartLinks.forEach((link) => {
if (link.target === nodeIdx && !visited.has(link.source)) {
getUpstreamNodes(link.source, visited);
}
});
return visited;
}
chart.on('click', function (params) {
if (!onChartClick) return;
const unsupported = ['other', 'drop']
function getDownstreamNodes(nodeIdx: number, visited = new Set<number>()) {
if (visited.has(nodeIdx)) return;
visited.add(nodeIdx);
echartLinks.forEach((link) => {
if (link.source === nodeIdx && !visited.has(link.target)) {
getDownstreamNodes(link.target, visited);
}
});
return visited;
}
function getConnectedChain(nodeIdx: number): Set<number> {
const upstream = getUpstreamNodes(nodeIdx) || new Set<number>();
const downstream = getDownstreamNodes(nodeIdx) || new Set<number>();
return new Set<number>([...upstream, ...downstream]);
}
const originalNodes = [...echartNodes];
const originalLinks = [...echartLinks];
chart.on('mouseover', function (params: any) {
if (params.dataType === 'node') {
const node: any = params.data;
const filters = []
if (node.type) {
const type = node.type.toLowerCase();
if (unsupported.includes(type)) {
return
}
filters.push({
operator: 'is',
type: type,
value: [node.name],
isEvent: true,
});
}
onChartClick?.(filters);
} else if (params.dataType === 'edge') {
const linkIndex = params.dataIndex;
const link = data.links[linkIndex];
const firstNode = data.nodes.find(n => n.id === link.source)
const lastNode = data.nodes.find(n => n.id === link.target)
const hoveredIndex = params.dataIndex;
const connectedChain = getConnectedChain(hoveredIndex);
const firstNodeType = firstNode?.eventType?.toLowerCase() ?? 'location';
const lastNodeType = lastNode?.eventType?.toLowerCase() ?? 'location';
if (unsupported.includes(firstNodeType) || unsupported.includes(lastNodeType)) {
return
}
const filters = [];
if (firstNode) {
filters.push({
operator: 'is',
type: firstNodeType,
value: [firstNode.name],
isEvent: true
});
}
const updatedNodes = echartNodes.map((node, idx) => {
const baseOpacity = connectedChain.has(idx) ? 1 : 0.35;
if (lastNode) {
filters.push({
operator: 'is',
type: lastNodeType,
value: [lastNode.name],
isEvent: true
});
}
const extraStyle = idx === hoveredIndex
? {borderColor: '#000000', borderWidth:1, borderType: 'dotted' }
: {};
return {
...node,
itemStyle: {
...node.itemStyle,
opacity: baseOpacity,
...extraStyle,
},
};
});
onChartClick?.(filters);
const updatedLinks = echartLinks.map((link) => ({
...link,
lineStyle: {
...link.lineStyle,
opacity: (connectedChain.has(link.source) && connectedChain.has(link.target))
? 0.5
: 0.1,
},
}));
chart.setOption({
series: [
{
data: updatedNodes,
links: updatedLinks,
},
],
});
}
});
chart.on('mouseout', function (params: any) {
if (params.dataType === 'node') {
// Restore original styles on mouseout.
chart.setOption({
series: [
{
data: originalNodes,
links: originalLinks,
},
],
});
}
});
chart.on('click', function (params: any) {
if (!onChartClick) return;
if (params.dataType === 'node') {
const nodeIndex = params.dataIndex;
const node = data.nodes[nodeIndex];
onChartClick([{ node }]);
} else if (params.dataType === 'edge') {
const linkIndex = params.dataIndex;
const link = data.links[linkIndex];
onChartClick([{ link }]);
}
});
// chart.on('mouseover', function (params) {
// if (params.seriesIndex !== seriesIndex) return; // ignore if not sankey
// resetHighlight(); // dim everything first
//
// if (params.dataType === 'node') {
// const hoveredNodeIndex = params.dataIndex;
// // find outgoing links
// const outgoingLinks: number[] = [];
// data.links.forEach((link, linkIdx) => {
// if (link.source === hoveredNodeIndex) {
// outgoingLinks.push(linkIdx);
// }
// });
//
// // find incoming highest contributors
// const highestContribLinks = findHighestContributors(hoveredNodeIndex, data.links);
//
// // highlight outgoing links
// outgoingLinks.forEach((linkIdx) => highlightLink(linkIdx));
// // highlight the "highest path" of incoming links
// highestContribLinks.forEach((lk) => {
// // We need to find which link index in data.links => lk
// const linkIndex = data.links.indexOf(lk);
// if (linkIndex >= 0) {
// highlightLink(linkIndex);
// }
// });
//
// // highlight the node itself
// highlightNode(hoveredNodeIndex);
//
// // highlight the nodes that are "source/target" of the highlighted links
// const highlightNodeSet = new Set<number>();
// outgoingLinks.forEach((lIdx) => {
// highlightNodeSet.add(data.links[lIdx].target);
// highlightNodeSet.add(data.links[lIdx].source);
// });
// highestContribLinks.forEach((lk) => {
// highlightNodeSet.add(lk.source);
// highlightNodeSet.add(lk.target);
// });
// // also add the hovered node
// highlightNodeSet.add(hoveredNodeIndex);
//
// // highlight those nodes
// highlightNodeSet.forEach((nIdx) => highlightNode(nIdx));
//
// } else if (params.dataType === 'edge') {
// const hoveredLinkIndex = params.dataIndex;
// // highlight just that edge
// highlightLink(hoveredLinkIndex);
//
// // highlight source & target node
// const link = data.links[hoveredLinkIndex];
// highlightNode(link.source);
// highlightNode(link.target);
// }
// });
//
// chart.on('mouseout', function () {
// // revert to normal
// resetHighlight();
// });
const ro = new ResizeObserver(() => chart.resize());
ro.observe(chartRef.current);

View file

@ -1,45 +1,44 @@
export function sankeyTooltip(echartNodes, nodeValues) {
return (params) => {
// sankeyUtils.ts
export function sankeyTooltip(echartNodes: any[], nodeValues: number[]) {
return (params: any) => {
if ('source' in params.data && 'target' in params.data) {
const sourceName = echartNodes[params.data.source].name;
const targetName = echartNodes[params.data.target].name;
const sourceValue = nodeValues[params.data.source];
return `
<div class="flex gap-2 w-fit px-2 bg-white items-center">
<div class="flex flex-col">
<div class="border-t border-l rounded-tl border-dotted border-gray-500" style="width: 8px; height: 30px"></div>
<div class="border-b border-l rounded-bl border-dotted border-gray-500 relative" style="width: 8px; height: 30px">
<div class="w-0 h-0 border-l-4 border-l-gray-500 border-y-4 border-y-transparent border-r-0 absolute -right-1 -bottom-1.5"></div>
<div class="flex gap-2 w-fit px-2 bg-white items-center rounded-xl">
<div class="flex flex-col">
<div class="flex flex-col text-sm">
<div class="font-semibold">
<span class="text-base" style="color:#394eff">&#8592;</span> ${sourceName}
</div>
<div class="text-black">
${sourceValue} <span class="text-disabled-text">Sessions</span>
</div>
<div class="font-semibold mt-2">
<span class="text-base" style="color:#394eff">&#8594;</span> ${targetName}
</div>
<div class="flex items-baseline gap-2 text-black">
<span>${params.data.value} ( ${params.data.percentage.toFixed(2)}% )</span>
<span class="text-disabled-text">Sessions</span>
</div>
</div>
</div>
<div class="flex flex-col">
<div class="font-semibold">${sourceName}</div>
<div>${sourceValue}</div>
<div class="font-semibold mt-2">${targetName}</div>
<div>
<span>${params.data.value}</span>
<span class="text-disabled-text">${params.data.percentage.toFixed(
2
)}%</span>
</div>
</div>
</div>
`;
//${sourceName} -> ${targetName}: ${params.data.value} sessions (${params.data.percentage.toFixed(2)}%)
`;
}
if ('name' in params.data) {
return `
<div class="flex flex-col bg-white">
<div class="font-semibold">${params.data.name}</div>
<div>${params.value} sessions</div>
<div class="flex flex-col">
<div class="font-semibold text-sm flex gap-1 items-center"><span class="text-base" style="color:#394eff; font-family: sans-serif;">&#9632;&#xFE0E;</span> ${params.data.name}</div>
<div class="text-black text-sm">${params.value} <span class="text-disabled-text">Sessions</span></div>
</div>
`;
`;
}
};
}
export const getEventPriority = (type: string) => {
export const getEventPriority = (type: string): number => {
switch (type) {
case 'DROP':
return 3;
@ -50,11 +49,9 @@ export const getEventPriority = (type: string) => {
}
};
export const getNodeName = (eventType: string, nodeName: string | null) => {
export const getNodeName = (eventType: string, nodeName: string | null): string => {
if (!nodeName) {
// only capitalize first
return eventType.charAt(0) + eventType.slice(1).toLowerCase();
}
return nodeName;
}
};

View file

@ -50,7 +50,7 @@ function WidgetOptions() {
>
<Space>
<Switch checked={metric.hideExcess} size="small" />
<span className="mr-4 color-gray-medium">Hide Minor Paths</span>
<span className="mr-4 color-gray-medium">Group Minor Paths</span>
</Space>
</a>
)}