qdone 2.1.0 → 2.1.1

package/commonjs/src/sqs.js

@@ -0,0 +1,97 @@
+"use strict";
+/**
+ * Functions that deal with SQS
+ */
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getQueueAttributes = exports.getMatchingQueues = exports.setSQSClient = exports.getSQSClient = void 0;
+const client_sqs_1 = require("@aws-sdk/client-sqs");
+const path_1 = require("path");
+const debug_1 = __importDefault(require("debug"));
+const debug = (0, debug_1.default)('qdone:sqs');
+/**
+ * Utility function to return an instantiated, shared SQSClient.
+ */
+let client;
+function getSQSClient() {
+    if (client)
+        return client;
+    client = new client_sqs_1.SQSClient();
+    return client;
+}
+exports.getSQSClient = getSQSClient;
+/**
+ * Utility function to set the client explicitly, used in testing.
+ */
+function setSQSClient(explicitClient) {
+    client = explicitClient;
+}
+exports.setSQSClient = setSQSClient;
+/**
+ * Returns qrls for queues matching the given prefix and regex.
+ */
+async function getMatchingQueues(prefix, regex) {
+    const input = { QueueNamePrefix: prefix, MaxResults: 1000 };
+    const client = getSQSClient();
+    async function processQueues(nextToken) {
+        if (nextToken)
+            input.NextToken = nextToken;
+        const command = new client_sqs_1.ListQueuesCommand(input);
+        // debug({ nextToken, input, command })
+        const result = await client.send(command);
+        // debug({ result })
+        const { QueueUrls: qrls, NextToken: nextToken2 } = result;
+        // debug({ qrls, nextToken2 })
+        return (qrls || []).filter(q => regex.test(q)).concat(nextToken2 ? await processQueues(nextToken2) : []);
+    }
+    return processQueues();
+}
+exports.getMatchingQueues = getMatchingQueues;
+/**
+ * Gets attributes on every queue in parallel.
+ */
+async function getQueueAttributes(qrls) {
+    const promises = [];
+    // debug({ qrls })
+    for (const qrl of qrls) {
+        const input = {
+            QueueUrl: qrl,
+            AttributeNames: [
+                'ApproximateNumberOfMessages',
+                'ApproximateNumberOfMessagesNotVisible',
+                'ApproximateNumberOfMessagesDelayed'
+            ]
+        };
+        const command = new client_sqs_1.GetQueueAttributesCommand(input);
+        // debug({ input, command })
+        promises.push((async () => {
+            const queue = (0, path_1.basename)(qrl);
+            try {
+                const result = await client.send(command);
+                // debug({ queue, result })
+                return { queue, result };
+            }
+            catch (e) {
+                if (e instanceof client_sqs_1.QueueDoesNotExist) {
+                    // For queues that have been deleted in the meantime for whatever
+                    // reason, just show as having no messages instead of failing the
+                    // whole batch
+                    return {
+                        queue,
+                        Attributes: {
+                            ApproximateNumberOfMessages: '0',
+                            ApproximateNumberOfMessagesNotVisible: '0',
+                            ApproximateNumberOfMessagesDelayed: '0'
+                        }
+                    };
+                }
+                throw e;
+            }
+        })());
+    }
+    return Promise.all(promises);
+}
+exports.getQueueAttributes = getQueueAttributes;
+debug('loaded');

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "qdone",
-  "version": "2.1.0",
+  "version": "2.1.1",
   "description": "A distributed scheduler for SQS",
   "type": "module",
   "main": "./index.js",
@@ -57,8 +57,12 @@
     "publish-latest": "npm publish --tag latest",
     "publish-next": "npm publish --tag next"
   },
-  "author": "Ryan Witt",
-  "license": "MIT",
+  "author": "SureDone, Inc.",
+  "contributors": [
+    "Ryan Witt <onecreativenerd@gmail.com>",
+    "Shane O'Hanlon <shane.c.ohanlon@gmail.com>"
+  ],
+  "license": "ISC",
   "repository": {
     "type": "git",
     "url": "git+https://github.com/suredone/qdone.git"