Minor Adjustments

lib/server/rabbit/workers/KubeJobsWorker.js

@@ -0,0 +1,48 @@
+// Imports
+import { Worker } from "rabbiteer";
+
+// Constants
+const jobQueueName = "KubeJobs";
+const jobQueueRoutingName = "KubeJobsRouting";
+const exchangeName = "KubeJobsExchange";
+const setQueues = {
+  SET1Sec: 1000,
+  SET5Sec: 5000,
+  SET10Sec: 10_000,
+  SET30Sec: 30_000,
+  SET1Min: 60_000,
+  SET5Min: 60_000 * 5,
+  SET10Min: 60_000 * 10,
+  SET15Min: 60_000 * 15,
+  SET30Min: 60_000 * 30,
+  SET1Hr: 360_000,
+  SET2Hr: 360_000 * 2,
+  SET3Hr: 360_000 * 3,
+  SET4Hr: 360_000 * 4,
+};
+
+// Class
+export default class KubeJobsWorker extends Worker {
+  constructor(skio) {
+    super(jobQueueName);
+    this.skio = skio;
+  }
+
+  async configure(ch) {
+    await ch.assertExchange(exchangeName, "direct");
+    await ch.assertQueue(this.queue, this.queueOptions);
+    await ch.bindQueue(this.queue, exchangeName, jobQueueRoutingName);
+    await this.configureSetQueues(ch);
+    await ch.consume(this.queue, (msg) => this.consume(msg, () => ch.ack(msg)));
+  }
+
+  // Configure set queues that will all filter into this queue
+  async configureSetQueues(ch) {
+    await Promise.all(Object.keys(setQueues).map((k)=>
+        ch.assertQueue(k, { messageTtl: setQueues[k], deadLetterExchange: exchangeName, deadLetterRoutingKey: jobQueueRoutingName })))
+  }
+
+  onMessage(string) {
+    console.log(string);
+  }
+}