From 19f858b73b9baf093e53fbd9bc1e3961deb89dad Mon Sep 17 00:00:00 2001
From: Nikolay Rozhkov <nironame@gmail.com>
Date: Sun, 18 Jun 2023 23:53:57 +0300
Subject: [PATCH] Fixed nodes duplicates

---
 demos/sankey.html                                      | 10 ++++++----
 packages/mermaid/src/diagrams/sankey/sankeyDB.ts       | 10 ++++++----
 packages/mermaid/src/diagrams/sankey/sankeyRenderer.ts |  1 -
 3 files changed, 12 insertions(+), 9 deletions(-)

diff --git a/demos/sankey.html b/demos/sankey.html
index b06ead15e..ebacae649 100644
--- a/demos/sankey.html
+++ b/demos/sankey.html
@@ -17,10 +17,12 @@
     <h2>Simple flow</h2>
     <pre class="mermaid">
       sankey
-      node_a->10->node_b;
-      node_c->20->node_b;
-      node_b->15->node_d;
-      node_a-> 7->node_d;
+      a->28->b->10->c
+      a->11->b
+      k->18->b->35->d->13->e->43->dd11
+      d->5->e
+      d->10->c
+      k->25->e
     </pre>
     <script type="module">
       import mermaid from './mermaid.esm.mjs';
diff --git a/packages/mermaid/src/diagrams/sankey/sankeyDB.ts b/packages/mermaid/src/diagrams/sankey/sankeyDB.ts
index 23ef95528..8c20a465d 100644
--- a/packages/mermaid/src/diagrams/sankey/sankeyDB.ts
+++ b/packages/mermaid/src/diagrams/sankey/sankeyDB.ts
@@ -92,12 +92,14 @@ class Node {
  */
 const addNode = function(ID: string): Node {
   ID = common.sanitizeText(ID, configApi.getConfig());
+  let node: Node;
   if (nodesHash[ID] === undefined) {
-    nodesHash[ID] = new Node(ID);
+    node = new Node(ID);
+    nodesHash[ID] = node;
+    nodes.push(node);
+  } else {
+    node = nodesHash[ID];
   }
-  const node = nodesHash[ID];
-  nodes.push(node);
-  // debugger;
   return node;
 };
 
diff --git a/packages/mermaid/src/diagrams/sankey/sankeyRenderer.ts b/packages/mermaid/src/diagrams/sankey/sankeyRenderer.ts
index b0fac5c12..54b129098 100644
--- a/packages/mermaid/src/diagrams/sankey/sankeyRenderer.ts
+++ b/packages/mermaid/src/diagrams/sankey/sankeyRenderer.ts
@@ -99,7 +99,6 @@ export const draw = function (text: string, id: string, _version: string, diagOb
     graph.links.push({source: link.source.ID, target: link.target.ID, value: link.value});
   });
 
-
   debugger;
   // const graph = {
   //   nodes: [