pulse-sdk 0.0.4 → 0.0.5

package/dist/config.d.ts

@@ -9,6 +9,7 @@ export interface PulseConfig {
     eventTypes: string[];
     consumerName?: string;
     grouped?: boolean;
+    autoCommit?: boolean;
     topics?: TopicConfig[];
 }
 export declare function loadConfig(configPath?: string): PulseConfig;

package/dist/config.js

@@ -14,6 +14,7 @@ const DEFAULTS = {
     grpcUrl: 'localhost:50052',
     eventTypes: ['events'],
     grouped: true,
+    autoCommit: true,
 };
 function loadConfig(configPath) {
     const candidates = [
@@ -40,12 +41,16 @@ function loadConfig(configPath) {
         envCfg.consumerName = process.env.PULSE_CONSUMER_NAME;
     if (process.env.PULSE_GROUPED)
         envCfg.grouped = process.env.PULSE_GROUPED === 'true';
+    if (process.env.PULSE_AUTOCOMMIT)
+        envCfg.autoCommit = process.env.PULSE_AUTOCOMMIT === 'true';
     const merged = Object.assign({}, DEFAULTS, fileCfg, envCfg);
     // Ensure eventTypes array exists
     if (!merged.eventTypes)
         merged.eventTypes = DEFAULTS.eventTypes;
     if (merged.grouped === undefined)
         merged.grouped = true;
+    if (merged.autoCommit === undefined)
+        merged.autoCommit = true;
     return merged;
 }
 // Initialize topics from config using gRPC CreateTopic RPC.

package/dist/consumer.d.ts

@@ -1,5 +1,5 @@
 import { PulseConfig } from './config';
-export type EventHandler = (payload: any) => void;
+export type EventHandler = (payload: any) => any;
 export declare class Consumer {
     private config;
     private handlers;

package/dist/consumer.js

@@ -28,15 +28,25 @@ class Consumer {
             const handlers = this.handlers[topicName] || [];
             for (const h of handlers) {
                 const unregister = (0, consumerManager_1.registerSharedHandler)(this.config.grpcUrl, topicName, consumerName, (msg, stub, offset) => {
-                    // run handler with context so commit() works
+                    // run handler with context so commit() works and support auto-commit
                     // debug: console.log('consumer.wrapper.invoke', consumerName, topicName);
-                    (0, message_1.runWithContext)({ stub: stub || this.client, topic: topicName, consumerName, offset: offset ?? msg.offset }, () => {
-                        try {
-                            h(msg);
-                        }
-                        catch (e) { /* ignore */ }
-                    });
-                });
+                    (async () => {
+                        await (0, message_1.runWithContext)({ stub: stub || this.client, topic: topicName, consumerName, offset: offset ?? msg.offset }, async () => {
+                            try {
+                                const r = h(msg);
+                                if (r && typeof r.then === 'function')
+                                    await r;
+                            }
+                            catch (e) { /* ignore */ }
+                            if (this.config.autoCommit !== false) {
+                                try {
+                                    await (0, message_1.commit)();
+                                }
+                                catch (_) { /* ignore commit errors */ }
+                            }
+                        });
+                    })().catch(() => { });
+                }, { autoCommit: this.config.autoCommit });
                 this.unregisterFns.push(unregister);
             }
             // Return a promise that never resolves (stream runs until process exits)
@@ -69,12 +79,27 @@ class Consumer {
             const handlers = this.handlers[req.topic] || [];
             for (const h of handlers) {
                 // run handler within AsyncLocalStorage context so commit() can access stub and offset
-                (0, message_1.runWithContext)({ stub: this.client, topic: req.topic, consumerName, offset: message.offset }, () => {
+                (async () => {
                     try {
-                        h(message);
+                        await (0, message_1.runWithContext)({ stub: this.client, topic: req.topic, consumerName, offset: message.offset }, async () => {
+                            try {
+                                const r = h(message);
+                                if (r && typeof r.then === 'function')
+                                    await r;
+                            }
+                            catch (e) { /* handler error ignored here */ }
+                            if (this.config.autoCommit !== false) {
+                                try {
+                                    await (0, message_1.commit)();
+                                }
+                                catch (_) { /* ignore commit errors */ }
+                            }
+                        });
                     }
-                    catch (e) { /* handler error ignored here */ }
-                });
+                    catch (err) {
+                        // ignore
+                    }
+                })().catch(() => { });
             }
         });
         const p = new Promise((resolve, reject) => {

package/dist/consumerManager.d.ts

@@ -1,4 +1,6 @@
-type Handler = (msg: any) => void;
+type Handler = (msg: any) => any;
 export declare function shutdownAll(): Promise<void>;
-export declare function registerSharedHandler(grpcUrl: string, topic: string, consumerName: string, handler: Handler): () => void;
+export declare function registerSharedHandler(grpcUrl: string, topic: string, consumerName: string, handler: Handler, opts?: {
+    autoCommit?: boolean;
+}): () => void;
 export {};

package/dist/consumerManager.js

@@ -46,12 +46,12 @@ const registry = new Map();
 function keyFor(grpcUrl, topic, consumerName) {
     return `${grpcUrl}::${topic}::${consumerName}`;
 }
-function registerSharedHandler(grpcUrl, topic, consumerName, handler) {
+function registerSharedHandler(grpcUrl, topic, consumerName, handler, opts) {
     const k = keyFor(grpcUrl, topic, consumerName);
     let entry = registry.get(k);
     if (!entry) {
         const client = (0, client_1.createClient)(grpcUrl);
-        entry = { topic, consumerName, client, stream: null, handlers: new Set(), nextIndex: 0, grpcUrl };
+        entry = { topic, consumerName, client, stream: null, handlers: new Set(), nextIndex: 0, grpcUrl, autoCommit: opts?.autoCommit !== false };
         // add handler before starting the stream to avoid losing early messages
         entry.handlers.add(handler);
         registry.set(k, entry);
@@ -106,16 +106,25 @@ function startStream(entry) {
             entry.nextIndex = 0;
         const h = handlers[entry.nextIndex % handlers.length];
         entry.nextIndex = (entry.nextIndex + 1) % handlers.length;
-        try {
-            // run handler inside context providing the stub for commit()
-            // debug: console.log('consumerManager.dispatch', entry.topic, 'offset', message.offset, 'handlerIndex', entry.nextIndex);
-            (0, message_1.runWithContext)({ stub: entry.client, topic: entry.topic, consumerName: entry.consumerName, offset: message.offset }, () => {
-                h(message);
-            });
-        }
-        catch (e) {
-            console.warn('handler error', e);
-        }
+        // run handler inside context providing the stub for commit(); support async handlers
+        (async () => {
+            try {
+                await (0, message_1.runWithContext)({ stub: entry.client, topic: entry.topic, consumerName: entry.consumerName, offset: message.offset }, async () => {
+                    const r = h(message);
+                    if (r && typeof r.then === 'function')
+                        await r;
+                    if (entry.autoCommit) {
+                        try {
+                            await (0, message_1.commit)();
+                        }
+                        catch (err) { /* ignore commit errors here */ }
+                    }
+                });
+            }
+            catch (e) {
+                console.warn('handler error', e);
+            }
+        })().catch(() => { });
     });
     stream.on('error', (e) => {
         // Log once and clean up the registry entry to avoid reconnect storms and test leaks

package/dist/message.d.ts

@@ -11,7 +11,7 @@ interface MessageContext {
     offset: number;
     committed?: boolean;
 }
-export declare function runWithContext(ctx: MessageContext, fn: () => void): void;
+export declare function runWithContext(ctx: MessageContext, fn: () => void): unknown;
 export declare function getContext(): MessageContext | undefined;
 export declare function commit(): Promise<void>;
 export declare class Message implements IMessage {

package/dist/message.js

@@ -7,7 +7,7 @@ exports.commit = commit;
 const async_hooks_1 = require("async_hooks");
 const storage = new async_hooks_1.AsyncLocalStorage();
 function runWithContext(ctx, fn) {
-    storage.run(ctx, fn);
+    return storage.run(ctx, fn);
 }
 function getContext() {
     return storage.getStore();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pulse-sdk",
-  "version": "0.0.4",
+  "version": "0.0.5",
   "description": "Pulse SDK for Node.js/TypeScript",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",