SBIN2010 commented on code in PR #45908: URL: https://github.com/apache/superset/pull/45908#discussion_r4428845779
########## superset-frontend/plugins/plugin-chart-echarts/src/Sankey/crossFilter.ts: ########## @@ -0,0 +1,358 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, + * software distributed under the License is distributed on an + * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY + * KIND, either express or implied. See the License for the + * specific language governing permissions and limitations + * under the License. + */ +import { GenericDataType } from '@apache-superset/core/common'; +import { + DataRecordValue, + getColumnLabel, + getTimeFormatter, + QueryFormColumn, + QueryObjectFilterClause, +} from '@superset-ui/core'; +import { NULL_STRING } from '../constants'; +import { formatSeriesName } from '../utils/series'; + +/** + * Display text for an empty string. Matches `optionLabel` in + * `src/utils/common.ts`; this plugin cannot import that module. + */ +export const EMPTY_STRING = '<empty string>'; + +export type SankeyFilterColumn = 'source' | 'target'; + +/** Fields ECharts keeps on a node and returns from a click. */ +export type SankeyNodeDatum = { + name: string; + filterColumn: SankeyFilterColumn; + rawValue: DataRecordValue; +}; + +/** Fields ECharts keeps on a link and returns from a click. */ +export type SankeyLinkDatum = { + source: string; + target: string; + value: number; + sourceValue: DataRecordValue; + targetValue: DataRecordValue; +}; + +type NamedLink = { + source: string; + target: string; +}; + +type ParsedSelection = + | { kind: 'node'; column: SankeyFilterColumn; name: string } + | { kind: 'link'; source: string; target: string }; + +/** + * Label used as the ECharts node name. `null` and `''` stay distinct so a + * click can emit `IS NULL` or `IN ['']`. + */ +export function sankeyNodeName(value: DataRecordValue): string { + if (value == null) { + return NULL_STRING; + } + if (typeof value === 'bigint') { + return value.toString(); + } + const text = String(value); + return text === '' ? EMPTY_STRING : text; +} + +export function sankeyNodeKey( + column: SankeyFilterColumn, + name: string, +): string { + return JSON.stringify(['node', column, name]); +} + +export function sankeyLinkKey(source: string, target: string): string { + return JSON.stringify(['link', source, target]); +} + +export function isSankeyLinkDatum(data: unknown): data is SankeyLinkDatum { + return ( + typeof data === 'object' && + data !== null && + 'sourceValue' in data && + 'targetValue' in data + ); +} + +export function isSankeyNodeDatum(data: unknown): data is SankeyNodeDatum { + return ( + typeof data === 'object' && + data !== null && + 'filterColumn' in data && + 'rawValue' in data && + !isSankeyLinkDatum(data) + ); +} + +export function getSelectedSankeyKeys(selectedValues: unknown): string[] { + if (!selectedValues) { + return []; + } + if (Array.isArray(selectedValues)) { + return selectedValues.map(value => String(value)); + } + if (typeof selectedValues === 'object') { + return Object.values(selectedValues as Record<string, unknown>).map(value => + String(value), + ); + } + return []; +} + +function parseSelection(key: string): ParsedSelection | undefined { + try { + const parsed: unknown = JSON.parse(key); + if (!Array.isArray(parsed)) { + return undefined; + } + if ( + parsed[0] === 'node' && + (parsed[1] === 'source' || parsed[1] === 'target') && + typeof parsed[2] === 'string' + ) { + return { kind: 'node', column: parsed[1], name: parsed[2] }; + } + if ( + parsed[0] === 'link' && + typeof parsed[1] === 'string' && + typeof parsed[2] === 'string' + ) { + return { kind: 'link', source: parsed[1], target: parsed[2] }; + } + return undefined; + } catch { + return undefined; + } +} + +function nodeStaysBright( + name: string, + selection: ParsedSelection, + links: NamedLink[], +): boolean { + if (selection.kind === 'link') { + return name === selection.source || name === selection.target; + } + if (name === selection.name) { + return true; + } + // Keep the other end of each matching flow readable. + return links.some(link => + selection.column === 'source' + ? link.source === selection.name && link.target === name + : link.target === selection.name && link.source === name, + ); Review Comment: In `getSankey`, nodes and connections are traversed exactly once. The key point is that connections are indexed by their endpoints, with subsequent lookups performed using a set. This ensures that the full list of connections is not re-examined at each step. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected] --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
