askForWork() performance over WAN

Closes iarna#19
nponeccop · May 15, 2016 · 57d912c · 57d912c
1 parent 8122aee
commit 57d912c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/worker.js b/worker.js
@@ -7,7 +7,7 @@ var WorkerTask = require('./task-worker');
 var ClientTask = require('./task-client');
 var emptyFunction = require('emptyfunction');
 
-exports.__construct = function (init) {
+exports.__construct = function (init) {	
     this._workers = {};
     this._workersCount = 0;
 
@@ -212,10 +212,10 @@ Worker.dispatchWorker = function (job,socket) {
         });
 
         task.writer.once('end', function () {
+            self.endWork(jobid);
             if (socket.connected) {
                 socket.workComplete(jobid,task.lastChunk);
             }
-            self.endWork(jobid);
         });
     }
     else {
@@ -229,14 +229,14 @@ Worker.dispatchWorker = function (job,socket) {
         });
 
         task.writer.once('end', function () {
+            self.endWork(jobid);
             if (socket.connected) {
                 if (task.lastChunk) addToBuffer(task.lastChunk);
                 socket.workComplete(jobid,buffer);
             }
-            self.endWork(jobid);
         });
     }
-    
+
     try {
         var handleReturnValue = function (value) {
             if (value && value.pipe) {