Merge pull request #17 from S4N0I/master

Make node module context-aware to allow usage in worker threads

Makefile

@@ -8,6 +8,7 @@ build-wrapper:
 
 test: tests/data/text8-vector.json
 	node tests/smalltest.js
+	node tests/worker-thread-test.js
 	node tests/smalltest-manhattan.js
 	node tests/basictests.js basic-config.js
 

addon.cc

@@ -5,4 +5,6 @@ void InitAll(v8::Local<v8::Object> exports) {
   AnnoyIndexWrapper::Init(exports);
 }
 
-NODE_MODULE(addon, InitAll)
+NODE_MODULE_INIT() {
+    InitAll(exports);
+}

package.json

@@ -37,6 +37,7 @@
   "dependencies": {
     "bindings": "^1.2.1",
     "level": "^6.0.0",
-    "nan": "^2.14.0"
+    "nan": "^2.14.0",
+    "node-worker-threads-pool": "^1.5.1"
   }
 }

tests/testworker.js

@@ -0,0 +1,18 @@
+const { parentPort } = require('worker_threads');
+var Annoy = require('../index');
+
+var annoyPath = __dirname + '/data/test.annoy';
+
+const annoyIndex = new Annoy(10, 'Angular')
+
+if(!annoyIndex.load(annoyPath)) {
+    throw new Error('Loading annoy index failed!')
+} else {
+    console.log('Successfully loaded annoy index.')
+}
+
+parentPort.on('message', async (id) => {
+    // return the result to the main thread
+    const result = annoyIndex.getNNsByItem(id, 10, -1, false)
+    parentPort.postMessage(result);
+});

tests/worker-thread-test.js

@@ -0,0 +1,19 @@
+var test = require('tape');
+const { StaticPool } = require('node-worker-threads-pool')
+
+var workerPath = __dirname + '/testworker.js';
+
+test('Worker thread test', workerThreadTest);
+
+async function workerThreadTest(t) {
+    const workerPool = new StaticPool({
+        size: 2,
+        task: workerPath
+    });
+
+    const idsToLookUp = [0, 1]
+    const indexLookups = idsToLookUp.map(id => workerPool.exec(id))
+    console.log("Lookup results: ", (await Promise.all(indexLookups)))
+    workerPool.destroy()
+    t.end()
+}