fixing dequeue to decrement bytes for the buffer

twitvid · Dec 26, 2012 · ad4eeb5 · ad4eeb5
1 parent 8fb0727
commit ad4eeb5
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/lib/message_batch_stream.js b/lib/message_batch_stream.js
@@ -42,6 +42,20 @@ MessageBatchStream.prototype.enQueue = function(message) {
 	this.countBytes += this.sizeOfMsg(message);
 };
 
+/**
+ * Dequeue a batch of items
+ * @return {Array}
+ */
+MessageBatchStream.prototype.deQueueBatch = function() {
+	var subBuffer = AccStream.prototype.deQueueBatch.call(this);
+	for (var i = 0; i < subBuffer.length; i++) {
+		var message = subBuffer[i];
+		this.countBytes -= this.sizeOfMsg(message);
+	}
+	this.countBytes = Math.max(this.countBytes, 0);
+	return subBuffer;
+};
+
 /**
 * Get the size of a message Object
 * @param {Object} message