Updated query to accept message functions allowing more sophisticated…

… message behaviour
nactio · Dec 5, 2017 · c52ce89 · c52ce89
1 parent b41b6e5
commit c52ce89
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 1 deletion.
diff --git a/lib/actor.js b/lib/actor.js
@@ -86,6 +86,9 @@ class Actor {
     deffered.promise.then(() => clearTimeout(timeoutHandle)).catch(() => { });
 
     let tempReference = new TemporaryReference(deffered);
+    if (typeof (message) === 'function') {
+      message = message(tempReference);
+    }
     this.dispatch(message, tempReference);
     return deffered.promise;
   }

diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "nact",
-  "version": "4.0.1",
+  "version": "4.1.0",
   "description": "nact ⇒ node.js + actors = your services have never been so µ",
   "main": "lib/index.js",
   "scripts": {

diff --git a/test/actor.js b/test/actor.js
@@ -384,6 +384,15 @@ describe('Actor', function () {
       );
       (await query(actor, 'test', 50)).should.equal('done');
     });
+
+    it(`should accept a message function which takes in the temporary actor reference`, async function () {
+      let actor = spawnStateless(
+        system,
+        async (msg, ctx) => { dispatch(msg, 'done', ctx.self); },
+        'test'
+      );
+      (await query(actor, (sender) => sender, 50)).should.equal('done');
+    });
   });
 
   describe('#state$', function () {