error handle of amqp manager

Wesnydy Ribeiro
1 parent df6397d6
Showing 1 changed file with 17 additions and 19 deletions Show diff stats
translate-api/helpers/amqpManager.js
@@ -13,20 +13,20 @@ var amqplib = require(&#39;amqplib/callback_api&#39;);
 /**
  * Function to send text to the queue.
  */
-exports.sendToQueue = function(body, id, queue, durability, res) {
+exports.sendToQueue = function(body, id, queue, durability, res, next) {
   amqplib.connect('amqp://localhost', function(err, conn) {
-    if(err) {
-      res.json({message : err});
-      throw err;
-    }
+    if (err)
+      return next(err, 'Cannot connect to RabbitMQ');
+
     conn.createChannel(function(err, ch) {
-      if(err) {
-          res.json({message : err});
-          throw err;
-      }
+      if (err)
+        return next(err, 'Channel creation failed.');
+
       ch.assertQueue(queue, {durable : durability});
       ch.sendToQueue(queue, new Buffer(body), {correlationId : id});
+      next(null, 'Message successfully sent');
+
       try {
         ch.close();
       }
@@ -41,23 +41,21 @@ exports.sendToQueue = function(body, id, queue, durability, res) {
 /**
  * Function to receive gloss from the queue.
  */
-exports.receiveFromQueue = function(id, queue, durability, res) {
+exports.receiveFromQueue = function(id, queue, durability, res, next) {
   amqplib.connect('amqp://localhost', function(err, conn) {
-    if(err) {
-      res.json({message : err});
-      throw err;
-    }
+    if (err)
+      return next(err, 'Cannot connect to RabbitMQ');
+
     conn.createChannel(function(err, ch) {
-      if(err){
-        res.json({message : err});
-        throw err;
-      }
+      if (err)
+        return next(err, 'Channel creation failed.');
+
       ch.assertQueue(queue, {durable : durability});
       ch.consume(queue, function(msg) {
         if (msg.properties.correlationId === id) {
           ch.ack(msg);
-          res.send(msg.content.toString())
+          next(null, msg.content.toString())
           try {
             ch.close();
           }