Rebuild indices on insert

tshemsedinov · tshemsedinov · commit 2a8becdcb906 · 2019-04-05T18:02:32.000+03:00
diff --git a/JavaScript/3-index.js b/JavaScript/3-index.js
@@ -80,7 +80,19 @@ class Graph {
   }
   insert(records) {
     for (const record of records) {
-      this.add(record);
+      const vertex = this.add(record);
+      const keys = Object.keys(record);
+      for (const [key, idx] of this.indices) {
+        if (keys.includes(key)) {
+          const value = record[key];
+          let records = idx.get(value);
+          if (!records) {
+            records = new Set();
+            idx.set(value, records);
+          }
+          records.add(vertex);
+        }
+      }
     }
   }
   index(key) {
@@ -104,7 +116,7 @@ class Graph {
 
 // Usage
 
-const graph = new Graph('name');
+const graph = new Graph('name').index('city');
 
 graph.insert([
   { name: 'Marcus Aurelius', city: 'Rome', born: 121, dynasty: 'Antonine' },
@@ -114,6 +126,8 @@ graph.insert([
   { name: 'Trajan', city: 'Sevilla', born: 98, dynasty: 'Nerva–Trajan' }
 ]);
 
+graph.index('dynasty');
+
 graph.link('Marcus Aurelius').to('Lucius Verus');
 graph.link('Lucius Verus').to('Trajan', 'Marcus Aurelius', 'Marcus Aurelius');
 graph.link('Antoninus Pius').to('Marcus Aurelius', 'Lucius Verus');
@@ -123,12 +137,9 @@ graph.link('Trajan').to('Lucius Verus', 'Marcus Aurelius');
 console.dir({ graph }, { depth: null });
 
 const res = graph
-  .index('city').index('dynasty')
   .select({ city: 'Rome', dynasty: 'Antonine' })
   .linked('Trajan');
 
-console.dir({ graph }, { depth: null });
-
 console.log('\nQuery result:\n');
 for (const item of res.vertices) {
   console.dir(item.data);
