@heritageai/messaging 1.0.2 → 1.0.4

package/dist/nats-wrapper.d.ts

@@ -1,8 +1,8 @@
-import nats from 'node-nats-streaming';
+import nats from "node-nats-streaming";
 declare class NatsWrapper {
     private _client?;
     get client(): nats.Stan;
-    connect(clusterId: string, clientId: string, url: string): Promise<void>;
+    connect(clusterId: string, clientId: string, url: string, retries?: number, delayMs?: number): Promise<void>;
 }
 export declare const natsWrapper: NatsWrapper;
 export {};

package/dist/nats-wrapper.js

@@ -4,27 +4,37 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.natsWrapper = void 0;
-// src/nats-wrapper.ts
 const node_nats_streaming_1 = __importDefault(require("node-nats-streaming"));
 class NatsWrapper {
     get client() {
         if (!this._client) {
-            throw new Error('Cannot access NATS client before connecting');
+            throw new Error("Cannot access NATS client before connecting");
         }
         return this._client;
     }
-    connect(clusterId, clientId, url) {
-        this._client = node_nats_streaming_1.default.connect(clusterId, clientId, { url });
+    connect(clusterId, clientId, url, retries = 5, delayMs = 3000) {
         return new Promise((resolve, reject) => {
-            // @ts-ignore
-            this.client.on('connect', () => {
-                console.log('Connected to NATS');
-                resolve();
-            });
-            // @ts-ignore
-            this.client.on('error', (err) => {
-                reject(err);
-            });
+            const attempt = (remaining) => {
+                console.log(`🔌 Connecting to NATS (${remaining} retries left)...`);
+                this._client = node_nats_streaming_1.default.connect(clusterId, clientId, { url });
+                // @ts-ignore
+                this._client.on("connect", () => {
+                    console.log("✅ Connected to NATS");
+                    resolve();
+                });
+                // @ts-ignore
+                this._client.on("error", (err) => {
+                    console.error("❌ NATS connection error:", err.message);
+                    this._client?.close();
+                    this._client = undefined;
+                    if (remaining <= 0) {
+                        reject(err);
+                        return;
+                    }
+                    setTimeout(() => attempt(remaining - 1), delayMs);
+                });
+            };
+            attempt(retries);
         });
     }
 }

package/package.json

@@ -2,7 +2,7 @@
   "author": "future_platform",
   "name": "@heritageai/messaging",
   "license": "ISC",
-  "version": "1.0.2",
+  "version": "1.0.4",
   "description": "",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/nats-wrapper.ts

@@ -1,30 +1,51 @@
-// src/nats-wrapper.ts
-import nats, { Stan } from 'node-nats-streaming';
+import nats, { Stan } from "node-nats-streaming";
 
 class NatsWrapper {
   private _client?: Stan;
 
   get client() {
     if (!this._client) {
-      throw new Error('Cannot access NATS client before connecting');
+      throw new Error("Cannot access NATS client before connecting");
     }
     return this._client;
   }
 
-  connect(clusterId: string, clientId: string, url: string): Promise<void> {
-    this._client = nats.connect(clusterId, clientId, { url });
-
+  connect(
+    clusterId: string,
+    clientId: string,
+    url: string,
+    retries = 5,
+    delayMs = 3000,
+  ): Promise<void> {
     return new Promise((resolve, reject) => {
-      // @ts-ignore
-      this.client.on('connect', () => {
-        console.log('Connected to NATS');
-        resolve();
-      });
-
-      // @ts-ignore
-      this.client.on('error', (err) => {
-        reject(err);
-      });
+      const attempt = (remaining: number) => {
+        console.log(`🔌 Connecting to NATS (${remaining} retries left)...`);
+
+        this._client = nats.connect(clusterId, clientId, { url });
+
+        // @ts-ignore
+        this._client.on("connect", () => {
+          console.log("✅ Connected to NATS");
+          resolve();
+        });
+
+        // @ts-ignore
+        this._client.on("error", (err) => {
+          console.error("❌ NATS connection error:", err.message);
+
+          this._client?.close();
+          this._client = undefined;
+
+          if (remaining <= 0) {
+            reject(err);
+            return;
+          }
+
+          setTimeout(() => attempt(remaining - 1), delayMs);
+        });
+      };
+
+      attempt(retries);
     });
   }
 }