Merge pull request #97 from jessegranger/master

Emit 'response_processed' event once per batch

README.md

@@ -111,6 +111,7 @@ Each consumer is an [`EventEmitter`](http://nodejs.org/api/events.html) and emit
 |`processing_error`|`err`, `message`|Fired when an error occurs processing the message.|
 |`message_received`|`message`|Fired when a message is received.|
 |`message_processed`|`message`|Fired when a message is successfully processed and removed from the queue.|
+|`response_processed`|None|Fired after one batch of items (up to `batchSize`) has been successfully processed.|
 |`stopped`|None|Fired when the consumer finally stops its work.|
 |`empty`|None|Fired when the queue is empty (All messages have been consumed).|
 

index.js

@@ -130,6 +130,7 @@ Consumer.prototype._handleSqsResponse = function (err, response) {
   if (response && response.Messages && response.Messages.length > 0) {
     async.each(response.Messages, this._processMessageBound, function () {
       // start polling again once all of the messages have been processed
+      consumer.emit('response_processed');
       consumer._poll();
     });
   } else if (response && !response.Messages) {

test/index.js

@@ -435,6 +435,37 @@ describe('Consumer', function () {
 
       consumer.start();
     });
+
+    it('fires response_processed event for each batch', function (done) {
+      sqs.receiveMessage.yieldsAsync(null, {
+        Messages: [
+          {
+            ReceiptHandle: 'receipt-handle-1',
+            MessageId: '1',
+            Body: 'body-1'
+          },
+          {
+            ReceiptHandle: 'receipt-handle-2',
+            MessageId: '2',
+            Body: 'body-2'
+          }
+        ]
+      });
+      handleMessage.yields(null);
+
+      consumer = new Consumer({
+        queueUrl: 'some-queue-url',
+        messageAttributeNames: ['attribute-1', 'attribute-2'],
+        region: 'some-region',
+        handleMessage: handleMessage,
+        batchSize: 2,
+        sqs: sqs
+      });
+
+      consumer.on('response_processed', done);
+      consumer.start();
+
+    });
   });
 
   describe('.stop', function () {