From 9a57d84952d21ee86ae4c5363b84d0932c56fdc1 Mon Sep 17 00:00:00 2001
From: LV-426 <lv-426@taproot.org.il>
Date: Sun, 7 Dec 2014 17:41:28 +0200
Subject: __addNode(): assert id present on insertion,              introduce
 id_to_node_map

---
 src/model/graph.js | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

(limited to 'src/model/graph.js')

diff --git a/src/model/graph.js b/src/model/graph.js
index 55b72302..fca38ef4 100644
--- a/src/model/graph.js
+++ b/src/model/graph.js
@@ -8,6 +8,7 @@ var debug = false;
 function Graph() {
 
     var nodes = [],
+        id_to_node_map = {},
         links = [],
         diffBus = new Bacon.Bus();
 
@@ -54,13 +55,16 @@ function Graph() {
             node = model_core.create_node_from_spec(spec);
         }
 
-        existing_node = findNode(node.id, null);
+        existing_node = find_node__by_id(node.id);
         if (existing_node) {
-            console.debug('__addNode: id collision: existing-node.id: \'' + existing_node.id + '\', ' + 'new-node.id: \'' + node.id + '\'');
+            console.log('__addNode: id collision: existing-node.id: \'' + existing_node.id + '\', ' + 'new-node.id: \'' + node.id + '\'');
             return existing_node;
         }
 
+        util.assert(undefined != node.id, '__addNode: node id missing');
         nodes.push(node);
+        id_to_node_map[node.id] = node;
+        console.log('__addNode: node added: id: ' + node.id);
 
         if (notify) {
             diffBus.push({nodes: {add: [node]}});
@@ -80,6 +84,9 @@ function Graph() {
             var index = findNodeIndex(n.id, n.state);
             if (index !== undefined) {
                 nodes.splice(index, 1);
+
+                util.assert(undefined != n.id, '_removeNodes: node id missing');
+                delete id_to_node_map[n.id];
             }
         }
         if (ns.length > 0) {
-- 
cgit v1.3.1