bulltrackers-module 1.0.318 → 1.0.319

package/functions/core/utils/pubsub_utils.js

@@ -1,14 +1,8 @@
 /**
  * @fileoverview Core Pub/Sub utility functions.
- * REFACTORED: Hybrid module supporting both Stateless functions and Stateful Class.
- * Fixes "PubSubUtils is not a constructor" error.
+ * UPDATED: Added single-message publish method to the PubSubUtils class.
  */
 
-/**
- * Stateless Function: Publishes tasks in batches.
- * @param {object} dependencies - { pubsub, logger }
- * @param {object} config - { topicName, tasks, taskType, maxPubsubBatchSize }
- */
 async function batchPublishTasks(dependencies, config) {
     const { pubsub, logger } = dependencies;
     const { topicName, tasks, taskType, maxPubsubBatchSize = 500 } = config;
@@ -33,7 +27,6 @@ async function batchPublishTasks(dependencies, config) {
 
             await Promise.all(batchPromises);
             messagesPublished += batchTasks.length;
-            logger.log('TRACE', `[Core Utils] Published batch ${Math.ceil((i + 1) / maxPubsubBatchSize)} for ${taskType} (${batchTasks.length} messages)`);
         }
         logger.log('SUCCESS', `[Core Utils] Finished publishing ${messagesPublished} ${taskType} tasks to ${topicName}.`);
     } catch (error) {
@@ -42,39 +35,35 @@ async function batchPublishTasks(dependencies, config) {
     }
 }
 
-/**
- * Stateful Class Wrapper
- * Allows usage like: const utils = new PubSubUtils(deps); utils.batchPublishTasks(...)
- */
 class PubSubUtils {
     constructor(dependencies) {
         this.dependencies = dependencies;
     }
 
     /**
-     * Hybrid method: Supports both (config) and (dependencies, config) signatures.
+     * [NEW] Publishes a single JSON message to a topic.
      */
-    async batchPublishTasks(arg1, arg2) {
-        // If called as (dependencies, config), use passed dependencies (Stateless/Legacy style)
-        if (arg2) {
-            return batchPublishTasks(arg1, arg2);
+    async publish(topicName, message) {
+        const { pubsub, logger } = this.dependencies;
+        const topic = pubsub.topic(topicName);
+        const dataBuffer = Buffer.from(JSON.stringify(message));
+        
+        try {
+            await topic.publishMessage({ data: dataBuffer });
+        } catch (error) {
+            logger.log('ERROR', `[Core Utils] Failed to publish message to ${topicName}`, { error: error.message });
+            throw error;
         }
-        // If called as (config), use this.dependencies (Stateful style)
+    }
+
+    async batchPublishTasks(arg1, arg2) {
+        if (arg2) return batchPublishTasks(arg1, arg2);
         return batchPublishTasks(this.dependencies, arg1);
     }
 
-    /**
-     * Helper for Computation System (Dispatcher)
-     * Maps (topic, messages) -> batchPublishTasks
-     */
     async publishMessageBatch(topicName, messages) {
-        // Unpack {json: ...} structure if present
         const tasks = messages.map(m => m.json || m);
-        const config = {
-            topicName,
-            tasks,
-            taskType: 'computation-batch'
-        };
+        const config = { topicName, tasks, taskType: 'computation-batch' };
         return batchPublishTasks(this.dependencies, config);
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "bulltrackers-module",
-  "version": "1.0.318",
+  "version": "1.0.319",
   "description": "Helper Functions for Bulltrackers.",
   "main": "index.js",
   "files": [