add promotion:start and promotion:end events Automattic#574

jesucarr · Apr 15, 2015 · 6db5386 · 6db5386
1 parent 1b052eb
commit 6db5386
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/lib/kue.js b/lib/kue.js
@@ -167,13 +167,23 @@ Queue.prototype._promote = function( promotionOptions ) {
             }
             if (typeof unlock === 'function') {
                 // If the lock is set successfully by this process, an unlock function is passed to our callback.
+                self.emit('promotion:start');
                 client.zrangebyscore(client.getKey('jobs:delayed'), 0, Date.now(), "LIMIT", 0, limit, function (err, ids) {
-                    if (err || !ids.length) return unlock();
+                    if (err || !ids.length) {
+                        self.emit('promotion:end');
+                        return unlock();
+                    }
+                    totalJobs = ids.length;
+                    jobsPromoted = 0;
                     ids.forEach( function(id){
                         Job.get(id, function(err, job) {
                             if (err) return;
                             events.emit(id, 'promotion');
                             job.inactive();
+                            jobsPromoted++;
+                            if (jobsPromoted === totalJobs) {
+                                self.emit('promotion:end');
+                            }
                         });
                     });
                     unlock();