yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-08-31 04:14:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

create export data on every click (#16344)

Browse Source
This commit is contained in:
Karan Hotchandani 2024-05-20 12:22:49 +05:30 committed by GitHub
parent 034877b1c3
commit 25b9576114
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
openmetadata-ui/src/main/resources/ui/src/context/LineageProvider/LineageProvider.tsx
View File

@ -175,7 +175,6 @@ const LineageProvider = ({ children }: LineageProviderProps) => {
const [childMap, setChildMap] = useState<EntityReferenceChild>();
const [paginationData, setPaginationData] = useState({});
const { showModal } = useEntityExportModalProvider();
const [exportResult, setExportResult] = useState<string>('');
const initLineageChildMaps = useCallback(
(
@ -228,11 +227,10 @@ const LineageProvider = ({ children }: LineageProviderProps) => {
entityType !== EntityType.PIPELINE &&
entityType !== EntityType.STORED_PROCEDURE
) {
const { map: childMapObj, exportResult } = getChildMap(
const { map: childMapObj } = getChildMap(
{ ...res, nodes: allNodes },
decodedFqn
);
setExportResult(exportResult);
setChildMap(childMapObj);
const { nodes: newNodes, edges: newEdges } = getPaginatedChildMap(
{
@ -250,8 +248,6 @@ const LineageProvider = ({ children }: LineageProviderProps) => {
edges: [...(res.edges ?? []), ...newEdges],
});
} else {
const csv = getExportData(allNodes);
setExportResult(csv);
setEntityLineage({
...res,
nodes: allNodes,
@ -281,9 +277,20 @@ const LineageProvider = ({ children }: LineageProviderProps) => {
const exportLineageData = useCallback(
async (_: string) => {
if (
entityType === EntityType.PIPELINE ||
entityType === EntityType.STORED_PROCEDURE
) {
// Since pipeline is an edge, we cannot create a tree, hence we take the nodes from the lineage response
// to get the export data.
return getExportData(entityLineage.nodes ?? []);
}
const { exportResult } = getChildMap(entityLineage, decodedFqn);
return exportResult;
},
[exportResult]
[entityType, decodedFqn, entityLineage]
);
const onExportClick = useCallback(() => {
@ -293,7 +300,7 @@ const LineageProvider = ({ children }: LineageProviderProps) => {
onExport: exportLineageData,
});
}
}, [decodedFqn, exportResult]);
}, [entityType, decodedFqn, entityLineage]);
const loadChildNodesHandler = useCallback(
async (node: SourceType, direction: EdgeTypeEnum) => {