Always store consumer function.

taariq · Jun 11, 2011 · 19a0fb6 · 19a0fb6
1 parent 2066aef
commit 19a0fb6
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/lib/stores/memory.js b/lib/stores/memory.js
@@ -263,13 +263,12 @@ Client.prototype.count = function (fn) {
  */
 
 Client.prototype.consume = function (fn) {
+  this.consumer = fn;
   this.paused = false;
 
   if (this.buffer.length) {
     fn(this.buffer, null);
     this.buffer = [];
-  } else {
-    this.consumer = fn;
   }
 
   return this;