Merge pull request #25 from BorisPetelj/master

Fire blocked/unblocked events
jwalton · Mar 20, 2018 · b5c02fa · b5c02fa
2 parents b1e056d + bbd741e
commit b5c02fa
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 0 deletions.
diff --git a/src/AmqpConnectionManager.coffee b/src/AmqpConnectionManager.coffee
@@ -102,6 +102,13 @@ class AmqpConnectionManager extends EventEmitter
             .then (connection) =>
                 @_currentConnection = connection
 
+                #emit 'blocked' when RabbitMQ server decides to block the connection (resources running low)
+                connection.on 'blocked', (reason) =>
+                    @emit 'blocked', {reason}
+
+                connection.on 'unblocked', () =>
+                    @emit 'unblocked'
+
                 # Reconnect if the broker goes away.
                 connection.on 'error', (err) =>
                     Promise.resolve()

diff --git a/test/AmqpConnectionManagerTest.coffee b/test/AmqpConnectionManagerTest.coffee
@@ -221,3 +221,29 @@ describe 'AmqpConnectionManager', ->
                     expect(connectsSeen).to.equal 1
 
                 .then resolve, reject
+
+    it 'should detect connection block/unblock', ->
+        new Promise (resolve, reject) ->
+            amqp = new AmqpConnectionManager('amqp://localhost')
+            connectsSeen = 0
+            blockSeen = 0
+            unblockSeen = 0
+
+            amqp.on 'blocked', ({err}) ->
+                blockSeen++
+
+            amqp.on 'unblocked', () ->
+                unblockSeen++
+
+            amqp.once 'connect', ({connection, url}) ->
+                connectsSeen++
+                # Close the connection nicely
+                amqplib.simulateRemoteBlock()
+                amqplib.simulateRemoteUnblock()
+
+                Promise.resolve().then ->
+                    expect(connectsSeen).to.equal 1
+                    expect(blockSeen).to.equal 1
+                    expect(unblockSeen).to.equal 1
+
+                .then resolve, reject
diff --git a/test/fixtures.coffee b/test/fixtures.coffee
@@ -13,6 +13,12 @@ class exports.FakeAmqp
     simulateRemoteClose: ->
         @connection.emit 'close', new Error("Connection closed")
 
+    simulateRemoteBlock: ->
+        @connection.emit 'blocked', new Error("Connection blocked")
+
+    simulateRemoteUnblock: ->
+        @connection.emit 'unblocked'
+
     reset: ->
         @connection = null
         @url = null