split API queue and sample queue

1 files changed, 9 insertions, 1 deletions

diff --git a/worker.js b/worker.js
index 6d6e206..f134f65 100644
--- a/worker.js
+++ b/worker.js
@@ -20,6 +20,7 @@ if (MADGLORY_TOKEN == undefined) throw "Need an API token";
             rabbit = await amqp.connect(RABBITMQ_URI);
             ch = await rabbit.createChannel();
             await ch.assertQueue("grab", {durable: true});
+            await ch.assertQueue("grab_sample", {durable: true});
             await ch.assertQueue("process", {durable: true});
             break;
         } catch (err) {
@@ -35,6 +36,13 @@ if (MADGLORY_TOKEN == undefined) throw "Need an API token";
             await getAPI(payload, "matches");
         if (msg.properties.type == "samples")
             await getAPI(payload, "samples");
+
+        console.log("done");
+        ch.ack(msg);
+    }, { noAck: false });
+
+    ch.consume("grab_sample", async (msg) => {
+        let payload = JSON.parse(msg.content.toString());
         if (msg.properties.type == "sample")
             await getSample(payload);
 
@@ -74,7 +82,7 @@ if (MADGLORY_TOKEN == undefined) throw "Need an API token";
                 if (where == "samples") {
                     // send to self
                     await Promise.all(datas
-                        .map((sample) => ch.sendToQueue("grab",
+                        .map((sample) => ch.sendToQueue("grab_sample",
                             new Buffer(JSON.stringify(sample.attributes.URL)),
                             { persistent: true, type: "sample" })
                     ));