playov2-js-utilities 0.3.6 → 0.3.7

package/lib/index.js

@@ -9,5 +9,6 @@ module.exports = {
     NotificationTemplates: require('./notification-templates'),
     NotificationConfig: require('./notification.config'),
     httpRequest: require('./request'),
-    Cron: require('./cron')
+    Cron: require('./cron'),
+    MessagePublisher: require('./message_publisher')
 };

package/lib/message_publisher/index.js

@@ -0,0 +1,70 @@
+
+
+const { CloudTasksClient } = require('@google-cloud/tasks');
+
+const PLAYO_LOGGER = require("@playo/logger");
+const LOGGER = new PLAYO_LOGGER("playo-message-publisher");
+
+// Instantiates a client.
+const client = new CloudTasksClient();
+
+const PROJECT = process.env.GCP_PROJECT;
+const LOCATION = process.env.GCP_LOCATION;
+
+if (!PROJECT || !LOCATION) {
+    console.warn('You need to pass - GCP_PROJECT, GCP_LOCATION in env for message queue to work!');
+}
+/**
+ *
+ * @param {String} queueId - Name of the queue being targeted
+ * @param {Object} payload - {
+ *                              httpMethod: String <POST, GET, PUT>
+ *                              url: String,
+ *                              headers: Object,
+ *                              body: Object, // not needed if GET request
+ *                              ...
+ *                                                       
+ *                           }
+ * @param {Object} messageProcessingProperties - {
+ *                                                  delay 
+ *                                                }
+ * @param {String} requestId 
+ */
+async function createHttpTask(queueId, payload, messageProcessingProperties, requestId) {
+
+    try {
+        const project = PROJECT;
+        const queue = queueId;
+        const location = LOCATION;
+
+        const { delay = 0 } = messageProcessingProperties
+
+        const parent = client.queuePath(project, location, queue);
+
+        const task = { httpRequest: payload };
+
+        if (payload.body) {
+            task.httpRequest.body = Buffer.from(payload.body).toString('base64');
+        }
+
+        if (delay) {
+            // The time when the task is scheduled to be attempted.
+            task.scheduleTime = {
+                seconds: delay + Date.now() / 1000,
+            };
+        }
+
+        // Send create task request.
+        LOGGER.info(requestId, task, `Sending task to queue ${queueId}}`)
+
+        const request = { parent: parent, task: task };
+        const [response] = await client.createTask(request);
+        LOGGER.info(requestId, { name: response.name }, `Added task to queue ${queueId}}`)
+    } catch (err) {
+        LOGGER.error(requestId, err, err.message)
+    }
+};
+
+module.exports = {
+    createHttpTask
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "playov2-js-utilities",
-  "version": "0.3.6",
+  "version": "0.3.7",
   "description": "Private package for JS utility functions",
   "main": "index.js",
   "scripts": {
@@ -21,6 +21,7 @@
     "jest": "^27.1.0"
   },
   "dependencies": {
+    "@google-cloud/tasks": "^2.5.0",
     "@playo/logger": "^0.9.18",
     "axios": "^0.24.0",
     "cron": "^1.8.2"