@jr200-labs/xstate-nats 0.9.13 → 0.9.14

package/dist/machines/subject.d.ts

@@ -1,9 +1,10 @@
 import { Subscription, PublishOptions, NatsConnection, RequestOptions } from '@nats-io/nats-core';
 import { RequestResult, SubjectSubscriptionConfig } from '../actions/subject';
+import { NatsTrafficMetrics, NatsTrafficRecordEvent, NatsTrafficResetEvent } from '../traffic';
 type InternalEvents = {
     type: 'ERROR';
     error: Error;
-};
+} | NatsTrafficRecordEvent;
 export type ExternalEvents = {
     type: 'SUBJECT.CONNECT';
     connection: NatsConnection;
@@ -37,7 +38,7 @@ export type ExternalEvents = {
     subject: string;
 } | {
     type: 'SUBJECT.UNSUBSCRIBE_ALL';
-};
+} | NatsTrafficResetEvent;
 export type Events = InternalEvents | ExternalEvents;
 export interface Context {
     uid: string;
@@ -45,9 +46,22 @@ export interface Context {
     subscriptions: Map<string, Subscription>;
     subscriptionConfigs: Map<string, SubjectSubscriptionConfig>;
     syncRequired: number;
+    trafficMetrics: NatsTrafficMetrics;
     error?: Error;
 }
-export declare const subjectManagerLogic: import("xstate").StateMachine<Context, InternalEvents | {
+export declare const subjectManagerLogic: import("xstate").StateMachine<Context, NatsTrafficRecordEvent | {
+    type: "METRICS.RESET_UPLOAD";
+    source?: import("..").NatsTrafficMetricSource;
+} | {
+    type: "METRICS.RESET_DOWNLOAD";
+    source?: import("..").NatsTrafficMetricSource;
+} | {
+    type: "METRICS.RESET_ALL";
+    source?: import("..").NatsTrafficMetricSource;
+} | {
+    type: "ERROR";
+    error: Error;
+} | {
     type: "SUBJECT.CONNECT";
     connection: NatsConnection;
 } | {

package/dist/machines/subject.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"subject.d.ts","sourceRoot":"","sources":["../../src/machines/subject.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAA;AAEjG,OAAO,EACL,aAAa,EACb,yBAAyB,EAI1B,MAAM,oBAAoB,CAAA;AAG3B,KAAK,cAAc,GAAG;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,KAAK,EAAE,KAAK,CAAA;CAAE,CAAA;AAGrD,MAAM,MAAM,cAAc,GACtB;IAAE,IAAI,EAAE,iBAAiB,CAAC;IAAC,UAAU,EAAE,cAAc,CAAA;CAAE,GACvD;IAAE,IAAI,EAAE,mBAAmB,CAAA;CAAE,GAC7B;IAAE,IAAI,EAAE,sBAAsB,CAAA;CAAE,GAChC;IACE,IAAI,EAAE,iBAAiB,CAAA;IACvB,OAAO,EAAE,MAAM,CAAA;IACf,OAAO,EAAE,GAAG,CAAA;IACZ,IAAI,CAAC,EAAE,cAAc,CAAA;IACrB,eAAe,CAAC,EAAE,CAAC,MAAM,EAAE;QAAE,EAAE,EAAE,IAAI,CAAA;KAAE,GAAG;QAAE,EAAE,EAAE,KAAK,CAAC;QAAC,KAAK,EAAE,KAAK,CAAA;KAAE,KAAK,IAAI,CAAA;CAC/E,GACD;IACE,IAAI,EAAE,iBAAiB,CAAA;IACvB,OAAO,EAAE,MAAM,CAAA;IACf,OAAO,EAAE,GAAG,CAAA;IACZ,IAAI,CAAC,EAAE,cAAc,CAAA;IACrB,QAAQ,EAAE,CAAC,IAAI,EAAE,GAAG,KAAK,IAAI,CAAA;IAC7B,eAAe,CAAC,EAAE,CAAC,MAAM,EAAE,aAAa,KAAK,IAAI,CAAA;CAClD,GACD;IAAE,IAAI,EAAE,mBAAmB,CAAC;IAAC,MAAM,EAAE,yBAAyB,CAAA;CAAE,GAChE;IAAE,IAAI,EAAE,qBAAqB,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GAChD;IAAE,IAAI,EAAE,yBAAyB,CAAA;CAAE,CAAA;AAEvC,MAAM,MAAM,MAAM,GAAG,cAAc,GAAG,cAAc,CAAA;AAEpD,MAAM,WAAW,OAAO;IACtB,GAAG,EAAE,MAAM,CAAA;IACX,gBAAgB,EAAE,cAAc,GAAG,IAAI,CAAA;IACvC,aAAa,EAAE,GAAG,CAAC,MAAM,EAAE,YAAY,CAAC,CAAA;IACxC,mBAAmB,EAAE,GAAG,CAAC,MAAM,EAAE,yBAAyB,CAAC,CAAA;IAC3D,YAAY,EAAE,MAAM,CAAA;IACpB,KAAK,CAAC,EAAE,KAAK,CAAA;CACd;AAED,eAAO,MAAM,mBAAmB;UAjCpB,iBAAiB;gBAAc,cAAc;;UAC7C,mBAAmB;;UACnB,sBAAsB;;UAEtB,iBAAiB;aACd,MAAM;aACN,GAAG;WACL,cAAc;sBACH,CAAC,MAAM,EAAE;QAAE,EAAE,EAAE,IAAI,CAAA;KAAE,GAAG;QAAE,EAAE,EAAE,KAAK,CAAC;QAAC,KAAK,EAAE,KAAK,CAAA;KAAE,KAAK,IAAI;;UAGxE,iBAAiB;aACd,MAAM;aACN,GAAG;WACL,cAAc;cACX,CAAC,IAAI,EAAE,GAAG,KAAK,IAAI;sBACX,CAAC,MAAM,EAAE,aAAa,KAAK,IAAI;;UAE3C,mBAAmB;YAAU,yBAAyB;;UACtD,qBAAqB;aAAW,MAAM;;UACtC,yBAAyB;;;;;;;;;;;;;EAgNnC,CAAA"}
+{"version":3,"file":"subject.d.ts","sourceRoot":"","sources":["../../src/machines/subject.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAA;AAEjG,OAAO,EACL,aAAa,EACb,yBAAyB,EAI1B,MAAM,oBAAoB,CAAA;AAC3B,OAAO,EAGL,kBAAkB,EAClB,sBAAsB,EACtB,qBAAqB,EAKtB,MAAM,YAAY,CAAA;AAGnB,KAAK,cAAc,GAAG;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,KAAK,EAAE,KAAK,CAAA;CAAE,GAAG,sBAAsB,CAAA;AAG9E,MAAM,MAAM,cAAc,GACtB;IAAE,IAAI,EAAE,iBAAiB,CAAC;IAAC,UAAU,EAAE,cAAc,CAAA;CAAE,GACvD;IAAE,IAAI,EAAE,mBAAmB,CAAA;CAAE,GAC7B;IAAE,IAAI,EAAE,sBAAsB,CAAA;CAAE,GAChC;IACE,IAAI,EAAE,iBAAiB,CAAA;IACvB,OAAO,EAAE,MAAM,CAAA;IACf,OAAO,EAAE,GAAG,CAAA;IACZ,IAAI,CAAC,EAAE,cAAc,CAAA;IACrB,eAAe,CAAC,EAAE,CAAC,MAAM,EAAE;QAAE,EAAE,EAAE,IAAI,CAAA;KAAE,GAAG;QAAE,EAAE,EAAE,KAAK,CAAC;QAAC,KAAK,EAAE,KAAK,CAAA;KAAE,KAAK,IAAI,CAAA;CAC/E,GACD;IACE,IAAI,EAAE,iBAAiB,CAAA;IACvB,OAAO,EAAE,MAAM,CAAA;IACf,OAAO,EAAE,GAAG,CAAA;IACZ,IAAI,CAAC,EAAE,cAAc,CAAA;IACrB,QAAQ,EAAE,CAAC,IAAI,EAAE,GAAG,KAAK,IAAI,CAAA;IAC7B,eAAe,CAAC,EAAE,CAAC,MAAM,EAAE,aAAa,KAAK,IAAI,CAAA;CAClD,GACD;IAAE,IAAI,EAAE,mBAAmB,CAAC;IAAC,MAAM,EAAE,yBAAyB,CAAA;CAAE,GAChE;IAAE,IAAI,EAAE,qBAAqB,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GAChD;IAAE,IAAI,EAAE,yBAAyB,CAAA;CAAE,GACnC,qBAAqB,CAAA;AAEzB,MAAM,MAAM,MAAM,GAAG,cAAc,GAAG,cAAc,CAAA;AAEpD,MAAM,WAAW,OAAO;IACtB,GAAG,EAAE,MAAM,CAAA;IACX,gBAAgB,EAAE,cAAc,GAAG,IAAI,CAAA;IACvC,aAAa,EAAE,GAAG,CAAC,MAAM,EAAE,YAAY,CAAC,CAAA;IACxC,mBAAmB,EAAE,GAAG,CAAC,MAAM,EAAE,yBAAyB,CAAC,CAAA;IAC3D,YAAY,EAAE,MAAM,CAAA;IACpB,cAAc,EAAE,kBAAkB,CAAA;IAClC,KAAK,CAAC,EAAE,KAAK,CAAA;CACd;AAED,eAAO,MAAM,mBAAmB;;;;;;;;;;UAvCF,OAAO;WAAS,KAAK;;UAIvC,iBAAiB;gBAAc,cAAc;;UAC7C,mBAAmB;;UACnB,sBAAsB;;UAEtB,iBAAiB;aACd,MAAM;aACN,GAAG;WACL,cAAc;sBACH,CAAC,MAAM,EAAE;QAAE,EAAE,EAAE,IAAI,CAAA;KAAE,GAAG;QAAE,EAAE,EAAE,KAAK,CAAC;QAAC,KAAK,EAAE,KAAK,CAAA;KAAE,KAAK,IAAI;;UAGxE,iBAAiB;aACd,MAAM;aACN,GAAG;WACL,cAAc;cACX,CAAC,IAAI,EAAE,GAAG,KAAK,IAAI;sBACX,CAAC,MAAM,EAAE,aAAa,KAAK,IAAI;;UAE3C,mBAAmB;YAAU,yBAAyB;;UACtD,qBAAqB;aAAW,MAAM;;UACtC,yBAAyB;;;;;;;;;;;;;EAoQnC,CAAA"}

package/dist/machines/subject.js

@@ -1,5 +1,6 @@
 import { assign, sendParent, setup } from 'xstate';
 import { subjectConsolidateState, subjectRequest, subjectPublish, } from '../actions/subject';
+import { byteLength, createEmptyTrafficMetrics, recordTrafficMetric, resetTrafficAll, resetTrafficDownload, resetTrafficUpload, } from '../traffic';
 export const subjectManagerLogic = setup({
     types: {
         context: {},
@@ -19,19 +20,49 @@ export const subjectManagerLogic = setup({
         subscriptionConfigs: new Map(),
         cachedConnection: null,
         syncRequired: 0,
+        trafficMetrics: createEmptyTrafficMetrics(),
         error: undefined,
     },
     on: {
+        'METRICS.RECORD': {
+            actions: assign({
+                trafficMetrics: ({ context, event }) => recordTrafficMetric(context.trafficMetrics, event),
+            }),
+        },
+        'METRICS.RESET_UPLOAD': {
+            actions: assign({
+                trafficMetrics: ({ context, event }) => resetTrafficUpload(context.trafficMetrics, event.source),
+            }),
+        },
+        'METRICS.RESET_DOWNLOAD': {
+            actions: assign({
+                trafficMetrics: ({ context, event }) => resetTrafficDownload(context.trafficMetrics, event.source),
+            }),
+        },
+        'METRICS.RESET_ALL': {
+            actions: assign({
+                trafficMetrics: ({ context, event }) => resetTrafficAll(context.trafficMetrics, event.source),
+            }),
+        },
         'SUBJECT.SUBSCRIBE': {
             actions: [
-                assign({
-                    subscriptionConfigs: ({ context, event }) => {
-                        const { config } = event;
-                        const newConfigs = new Map(context.subscriptionConfigs);
-                        newConfigs.set(config.subject, config);
-                        return newConfigs;
-                    },
-                    syncRequired: ({ context }) => context.syncRequired + 1,
+                assign(({ context, event, self }) => {
+                    if (event.type !== 'SUBJECT.SUBSCRIBE')
+                        return {};
+                    const config = {
+                        ...event.config,
+                        onDownloadBytes: (bytes) => self.send({
+                            type: 'METRICS.RECORD',
+                            source: 'nats.subscribe',
+                            downloadBytes: bytes,
+                        }),
+                    };
+                    const newConfigs = new Map(context.subscriptionConfigs);
+                    newConfigs.set(config.subject, config);
+                    return {
+                        subscriptionConfigs: newConfigs,
+                        syncRequired: context.syncRequired + 1,
+                    };
                 }),
             ],
         },
@@ -103,7 +134,12 @@ export const subjectManagerLogic = setup({
                     target: 'subject_disconnecting',
                 },
                 'SUBJECT.REQUEST': {
-                    actions: assign(({ event, context }) => {
+                    actions: assign(({ event, context, self }) => {
+                        self.send({
+                            type: 'METRICS.RECORD',
+                            source: 'nats.request',
+                            uploadBytes: byteLength(event.payload),
+                        });
                         subjectRequest({
                             input: {
                                 connection: context.cachedConnection,
@@ -112,6 +148,11 @@ export const subjectManagerLogic = setup({
                                 opts: event.opts,
                                 callback: event.callback,
                                 onRequestResult: event.onRequestResult,
+                                onDownloadBytes: (bytes) => self.send({
+                                    type: 'METRICS.RECORD',
+                                    source: 'nats.request',
+                                    downloadBytes: bytes,
+                                }),
                             },
                         });
                         return {};
@@ -119,7 +160,12 @@ export const subjectManagerLogic = setup({
                 },
                 'SUBJECT.PUBLISH': {
                     actions: [
-                        ({ context, event }) => {
+                        ({ context, event, self }) => {
+                            self.send({
+                                type: 'METRICS.RECORD',
+                                source: 'nats.publish',
+                                uploadBytes: byteLength(event.payload),
+                            });
                             subjectPublish({
                                 input: {
                                     connection: context.cachedConnection,

package/dist/machines/subject.js.map

@@ -1 +1 @@
-{"version":3,"file":"subject.js","sourceRoot":"","sources":["../../src/machines/subject.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,QAAQ,CAAA;AAClD,OAAO,EAGL,uBAAuB,EACvB,cAAc,EACd,cAAc,GACf,MAAM,oBAAoB,CAAA;AAwC3B,MAAM,CAAC,MAAM,mBAAmB,GAAG,KAAK,CAAC;IACvC,KAAK,EAAE;QACL,OAAO,EAAE,EAAa;QACtB,MAAM,EAAE,EAAY;KACrB;IACD,MAAM,EAAE;QACN,cAAc,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE;YAC9B,OAAO,OAAO,CAAC,YAAY,GAAG,CAAC,CAAA;QACjC,CAAC;KACF;CACF,CAAC,CAAC,aAAa,CAAC;IACf,stBAAstB;IACttB,OAAO,EAAE,cAAc;IACvB,OAAO,EAAE;QACP,GAAG,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QAC7B,aAAa,EAAE,IAAI,GAAG,EAAwB;QAC9C,mBAAmB,EAAE,IAAI,GAAG,EAAqC;QACjE,gBAAgB,EAAE,IAAI;QACtB,YAAY,EAAE,CAAC;QACf,KAAK,EAAE,SAAS;KACjB;IACD,EAAE,EAAE;QACF,mBAAmB,EAAE;YACnB,OAAO,EAAE;gBACP,MAAM,CAAC;oBACL,mBAAmB,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE;wBAC1C,MAAM,EAAE,MAAM,EAAE,GAAG,KAAK,CAAA;wBACxB,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAA;wBACvD,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAA;wBACtC,OAAO,UAAU,CAAA;oBACnB,CAAC;oBACD,YAAY,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,YAAY,GAAG,CAAC;iBACxD,CAAC;aACH;SACF;QACD,qBAAqB,EAAE;YACrB,OAAO,EAAE;gBACP,MAAM,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE;oBAC5B,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAA;oBACvD,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAA;oBAChC,OAAO;wBACL,mBAAmB,EAAE,UAAU;wBAC/B,YAAY,EAAE,OAAO,CAAC,YAAY,GAAG,CAAC;qBACvC,CAAA;gBACH,CAAC,CAAC;aACH;SACF;QACD,yBAAyB,EAAE;YACzB,OAAO,EAAE,MAAM,CAAC;gBACd,mBAAmB,EAAE,IAAI,GAAG,EAAE;gBAC9B,YAAY,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,YAAY,GAAG,CAAC;aACxD,CAAC;SACH;KACF;IACD,MAAM,EAAE;QACN,YAAY,EAAE;YACZ,EAAE,EAAE;gBACF,sBAAsB,EAAE,EAAE;gBAC1B,iBAAiB,EAAE;oBACjB,MAAM,EAAE,oBAAoB;oBAC5B,OAAO,EAAE;wBACP,MAAM,CAAC;4BACL,gBAAgB,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU;yBAClD,CAAC;qBACH;iBACF;aACF;SACF;QACD,qBAAqB,EAAE;YACrB,KAAK,EAAE;gBACL,mFAAmF;gBACnF,MAAM,CAAC;oBACL,gBAAgB,EAAE,IAAI;oBACtB,aAAa,EAAE,IAAI,GAAG,EAAwB;oBAC9C,YAAY,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAC5B,OAAO,CAAC,mBAAmB,CAAC,IAAI,GAAG,CAAC;wBAClC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC;wBACnC,CAAC,CAAC,OAAO,CAAC,YAAY;iBAC3B,CAAC;gBACF,UAAU,CAAC,EAAE,IAAI,EAAE,sBAAsB,EAAE,CAAC;aAC7C;YACD,MAAM,EAAE;gBACN,MAAM,EAAE,cAAc;aACvB;SACF;QACD,iBAAiB,EAAE;YACjB,KAAK,EAAE,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC,CAAC;YAClD,MAAM,EAAE;gBACN,MAAM,EAAE,iBAAiB;gBACzB,KAAK,EAAE,gBAAgB;aACxB;YACD,EAAE,EAAE;gBACF,iBAAiB,EAAE;oBACjB,MAAM,EAAE,oBAAoB;oBAC5B,OAAO,EAAE;wBACP,MAAM,CAAC;4BACL,gBAAgB,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU;yBAClD,CAAC;qBACH;iBACF;gBACD,sBAAsB,EAAE;oBACtB,MAAM,EAAE,uBAAuB;iBAChC;gBACD,iBAAiB,EAAE;oBACjB,OAAO,EAAE,MAAM,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE;wBACrC,cAAc,CAAC;4BACb,KAAK,EAAE;gCACL,UAAU,EAAE,OAAO,CAAC,gBAAiB;gCACrC,OAAO,EAAE,KAAK,CAAC,OAAO;gCACtB,OAAO,EAAE,KAAK,CAAC,OAAO;gCACtB,IAAI,EAAE,KAAK,CAAC,IAAI;gCAChB,QAAQ,EAAE,KAAK,CAAC,QAAQ;gCACxB,eAAe,EAAE,KAAK,CAAC,eAAe;6BACvC;yBACF,CAAC,CAAA;wBACF,OAAO,EAAE,CAAA;oBACX,CAAC,CAAC;iBACH;gBACD,iBAAiB,EAAE;oBACjB,OAAO,EAAE;wBACP,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE;4BACrB,cAAc,CAAC;gCACb,KAAK,EAAE;oCACL,UAAU,EAAE,OAAO,CAAC,gBAAiB;oCACrC,OAAO,EAAE,KAAK,CAAC,OAAO;oCACtB,OAAO,EAAE,KAAK,CAAC,OAAO;oCACtB,OAAO,EAAE,KAAK,CAAC,IAAI;oCACnB,eAAe,EAAE,KAAK,CAAC,eAAe;iCACvC;6BACF,CAAC,CAAA;wBACJ,CAAC;qBACF;iBACF;aACF;SACF;QACD,kBAAkB,EAAE;YAClB,EAAE,EAAE;gBACF,sBAAsB,EAAE;oBACtB,MAAM,EAAE,uBAAuB;iBAChC;aACF;YACD,MAAM,EAAE;gBACN;oBACE,MAAM,EAAE,iBAAiB;oBACzB,KAAK,EAAE,gBAAgB;iBACxB;gBACD;oBACE,MAAM,EAAE,mBAAmB;iBAC5B;aACF;SACF;QACD,eAAe,EAAE;YACf,EAAE,EAAE;gBACF,sBAAsB,EAAE;oBACtB,MAAM,EAAE,uBAAuB;iBAChC;aACF;YACD,KAAK,EAAE;gBACL,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE;oBACd,mEAAmE;oBACnE,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,YAAY,GAAG,CAAC,EAAE,CAAC,CAAC,CAAA;gBAC9D,CAAC;gBACD,MAAM,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE;oBACrB,MAAM,mBAAmB,GAAG,uBAAuB,CAAC;wBAClD,KAAK,EAAE;4BACL,UAAU,EAAE,OAAO,CAAC,gBAAiB;4BACrC,oBAAoB,EAAE,OAAO,CAAC,aAAa;4BAC3C,mBAAmB,EAAE,OAAO,CAAC,mBAAmB;yBACjD;qBACF,CAAC,CAAA;oBACF,OAAO;wBACL,GAAG,mBAAmB;qBACvB,CAAA;gBACH,CAAC,CAAC;aACH;YACD,MAAM,EAAE;gBACN,MAAM,EAAE,mBAAmB;aAC5B;SACF;QACD,aAAa,EAAE;YACb,EAAE,EAAE;gBACF,sBAAsB,EAAE;oBACtB,MAAM,EAAE,uBAAuB;iBAChC;gBACD,iBAAiB,EAAE;oBACjB,MAAM,EAAE,oBAAoB;oBAC5B,OAAO,EAAE;wBACP,MAAM,CAAC;4BACL,gBAAgB,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU;yBAClD,CAAC;qBACH;iBACF;aACF;SACF;KACF;CACF,CAAC,CAAA"}
+{"version":3,"file":"subject.js","sourceRoot":"","sources":["../../src/machines/subject.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,QAAQ,CAAA;AAClD,OAAO,EAGL,uBAAuB,EACvB,cAAc,EACd,cAAc,GACf,MAAM,oBAAoB,CAAA;AAC3B,OAAO,EACL,UAAU,EACV,yBAAyB,EAIzB,mBAAmB,EACnB,eAAe,EACf,oBAAoB,EACpB,kBAAkB,GACnB,MAAM,YAAY,CAAA;AA0CnB,MAAM,CAAC,MAAM,mBAAmB,GAAG,KAAK,CAAC;IACvC,KAAK,EAAE;QACL,OAAO,EAAE,EAAa;QACtB,MAAM,EAAE,EAAY;KACrB;IACD,MAAM,EAAE;QACN,cAAc,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE;YAC9B,OAAO,OAAO,CAAC,YAAY,GAAG,CAAC,CAAA;QACjC,CAAC;KACF;CACF,CAAC,CAAC,aAAa,CAAC;IACf,stBAAstB;IACttB,OAAO,EAAE,cAAc;IACvB,OAAO,EAAE;QACP,GAAG,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QAC7B,aAAa,EAAE,IAAI,GAAG,EAAwB;QAC9C,mBAAmB,EAAE,IAAI,GAAG,EAAqC;QACjE,gBAAgB,EAAE,IAAI;QACtB,YAAY,EAAE,CAAC;QACf,cAAc,EAAE,yBAAyB,EAAE;QAC3C,KAAK,EAAE,SAAS;KACjB;IACD,EAAE,EAAE;QACF,gBAAgB,EAAE;YAChB,OAAO,EAAE,MAAM,CAAC;gBACd,cAAc,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,mBAAmB,CAAC,OAAO,CAAC,cAAc,EAAE,KAAK,CAAC;aAC3F,CAAC;SACH;QACD,sBAAsB,EAAE;YACtB,OAAO,EAAE,MAAM,CAAC;gBACd,cAAc,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CACrC,kBAAkB,CAAC,OAAO,CAAC,cAAc,EAAE,KAAK,CAAC,MAAM,CAAC;aAC3D,CAAC;SACH;QACD,wBAAwB,EAAE;YACxB,OAAO,EAAE,MAAM,CAAC;gBACd,cAAc,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CACrC,oBAAoB,CAAC,OAAO,CAAC,cAAc,EAAE,KAAK,CAAC,MAAM,CAAC;aAC7D,CAAC;SACH;QACD,mBAAmB,EAAE;YACnB,OAAO,EAAE,MAAM,CAAC;gBACd,cAAc,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CACrC,eAAe,CAAC,OAAO,CAAC,cAAc,EAAE,KAAK,CAAC,MAAM,CAAC;aACxD,CAAC;SACH;QACD,mBAAmB,EAAE;YACnB,OAAO,EAAE;gBACP,MAAM,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,EAAE;oBAClC,IAAI,KAAK,CAAC,IAAI,KAAK,mBAAmB;wBAAE,OAAO,EAAE,CAAA;oBAEjD,MAAM,MAAM,GAA8B;wBACxC,GAAG,KAAK,CAAC,MAAM;wBACf,eAAe,EAAE,CAAC,KAAK,EAAE,EAAE,CACzB,IAAI,CAAC,IAAI,CAAC;4BACR,IAAI,EAAE,gBAAgB;4BACtB,MAAM,EAAE,gBAAgB;4BACxB,aAAa,EAAE,KAAK;yBACrB,CAAC;qBACL,CAAA;oBACD,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAA;oBACvD,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAA;oBACtC,OAAO;wBACL,mBAAmB,EAAE,UAAU;wBAC/B,YAAY,EAAE,OAAO,CAAC,YAAY,GAAG,CAAC;qBACvC,CAAA;gBACH,CAAC,CAAC;aACH;SACF;QACD,qBAAqB,EAAE;YACrB,OAAO,EAAE;gBACP,MAAM,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE;oBAC5B,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAA;oBACvD,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAA;oBAChC,OAAO;wBACL,mBAAmB,EAAE,UAAU;wBAC/B,YAAY,EAAE,OAAO,CAAC,YAAY,GAAG,CAAC;qBACvC,CAAA;gBACH,CAAC,CAAC;aACH;SACF;QACD,yBAAyB,EAAE;YACzB,OAAO,EAAE,MAAM,CAAC;gBACd,mBAAmB,EAAE,IAAI,GAAG,EAAE;gBAC9B,YAAY,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,YAAY,GAAG,CAAC;aACxD,CAAC;SACH;KACF;IACD,MAAM,EAAE;QACN,YAAY,EAAE;YACZ,EAAE,EAAE;gBACF,sBAAsB,EAAE,EAAE;gBAC1B,iBAAiB,EAAE;oBACjB,MAAM,EAAE,oBAAoB;oBAC5B,OAAO,EAAE;wBACP,MAAM,CAAC;4BACL,gBAAgB,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU;yBAClD,CAAC;qBACH;iBACF;aACF;SACF;QACD,qBAAqB,EAAE;YACrB,KAAK,EAAE;gBACL,mFAAmF;gBACnF,MAAM,CAAC;oBACL,gBAAgB,EAAE,IAAI;oBACtB,aAAa,EAAE,IAAI,GAAG,EAAwB;oBAC9C,YAAY,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAC5B,OAAO,CAAC,mBAAmB,CAAC,IAAI,GAAG,CAAC;wBAClC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC;wBACnC,CAAC,CAAC,OAAO,CAAC,YAAY;iBAC3B,CAAC;gBACF,UAAU,CAAC,EAAE,IAAI,EAAE,sBAAsB,EAAE,CAAC;aAC7C;YACD,MAAM,EAAE;gBACN,MAAM,EAAE,cAAc;aACvB;SACF;QACD,iBAAiB,EAAE;YACjB,KAAK,EAAE,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC,CAAC;YAClD,MAAM,EAAE;gBACN,MAAM,EAAE,iBAAiB;gBACzB,KAAK,EAAE,gBAAgB;aACxB;YACD,EAAE,EAAE;gBACF,iBAAiB,EAAE;oBACjB,MAAM,EAAE,oBAAoB;oBAC5B,OAAO,EAAE;wBACP,MAAM,CAAC;4BACL,gBAAgB,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU;yBAClD,CAAC;qBACH;iBACF;gBACD,sBAAsB,EAAE;oBACtB,MAAM,EAAE,uBAAuB;iBAChC;gBACD,iBAAiB,EAAE;oBACjB,OAAO,EAAE,MAAM,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE;wBAC3C,IAAI,CAAC,IAAI,CAAC;4BACR,IAAI,EAAE,gBAAgB;4BACtB,MAAM,EAAE,cAAc;4BACtB,WAAW,EAAE,UAAU,CAAC,KAAK,CAAC,OAAO,CAAC;yBACvC,CAAC,CAAA;wBACF,cAAc,CAAC;4BACb,KAAK,EAAE;gCACL,UAAU,EAAE,OAAO,CAAC,gBAAiB;gCACrC,OAAO,EAAE,KAAK,CAAC,OAAO;gCACtB,OAAO,EAAE,KAAK,CAAC,OAAO;gCACtB,IAAI,EAAE,KAAK,CAAC,IAAI;gCAChB,QAAQ,EAAE,KAAK,CAAC,QAAQ;gCACxB,eAAe,EAAE,KAAK,CAAC,eAAe;gCACtC,eAAe,EAAE,CAAC,KAAK,EAAE,EAAE,CACzB,IAAI,CAAC,IAAI,CAAC;oCACR,IAAI,EAAE,gBAAgB;oCACtB,MAAM,EAAE,cAAc;oCACtB,aAAa,EAAE,KAAK;iCACrB,CAAC;6BACL;yBACF,CAAC,CAAA;wBACF,OAAO,EAAE,CAAA;oBACX,CAAC,CAAC;iBACH;gBACD,iBAAiB,EAAE;oBACjB,OAAO,EAAE;wBACP,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,EAAE;4BAC3B,IAAI,CAAC,IAAI,CAAC;gCACR,IAAI,EAAE,gBAAgB;gCACtB,MAAM,EAAE,cAAc;gCACtB,WAAW,EAAE,UAAU,CAAC,KAAK,CAAC,OAAO,CAAC;6BACvC,CAAC,CAAA;4BACF,cAAc,CAAC;gCACb,KAAK,EAAE;oCACL,UAAU,EAAE,OAAO,CAAC,gBAAiB;oCACrC,OAAO,EAAE,KAAK,CAAC,OAAO;oCACtB,OAAO,EAAE,KAAK,CAAC,OAAO;oCACtB,OAAO,EAAE,KAAK,CAAC,IAAI;oCACnB,eAAe,EAAE,KAAK,CAAC,eAAe;iCACvC;6BACF,CAAC,CAAA;wBACJ,CAAC;qBACF;iBACF;aACF;SACF;QACD,kBAAkB,EAAE;YAClB,EAAE,EAAE;gBACF,sBAAsB,EAAE;oBACtB,MAAM,EAAE,uBAAuB;iBAChC;aACF;YACD,MAAM,EAAE;gBACN;oBACE,MAAM,EAAE,iBAAiB;oBACzB,KAAK,EAAE,gBAAgB;iBACxB;gBACD;oBACE,MAAM,EAAE,mBAAmB;iBAC5B;aACF;SACF;QACD,eAAe,EAAE;YACf,EAAE,EAAE;gBACF,sBAAsB,EAAE;oBACtB,MAAM,EAAE,uBAAuB;iBAChC;aACF;YACD,KAAK,EAAE;gBACL,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE;oBACd,mEAAmE;oBACnE,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,YAAY,GAAG,CAAC,EAAE,CAAC,CAAC,CAAA;gBAC9D,CAAC;gBACD,MAAM,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE;oBACrB,MAAM,mBAAmB,GAAG,uBAAuB,CAAC;wBAClD,KAAK,EAAE;4BACL,UAAU,EAAE,OAAO,CAAC,gBAAiB;4BACrC,oBAAoB,EAAE,OAAO,CAAC,aAAa;4BAC3C,mBAAmB,EAAE,OAAO,CAAC,mBAAmB;yBACjD;qBACF,CAAC,CAAA;oBACF,OAAO;wBACL,GAAG,mBAAmB;qBACvB,CAAA;gBACH,CAAC,CAAC;aACH;YACD,MAAM,EAAE;gBACN,MAAM,EAAE,mBAAmB;aAC5B;SACF;QACD,aAAa,EAAE;YACb,EAAE,EAAE;gBACF,sBAAsB,EAAE;oBACtB,MAAM,EAAE,uBAAuB;iBAChC;gBACD,iBAAiB,EAAE;oBACjB,MAAM,EAAE,oBAAoB;oBAC5B,OAAO,EAAE;wBACP,MAAM,CAAC;4BACL,gBAAgB,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU;yBAClD,CAAC;qBACH;iBACF;aACF;SACF;KACF;CACF,CAAC,CAAA"}

package/dist/traffic.d.ts

@@ -0,0 +1,33 @@
+export type NatsTrafficMetricSource = 'nats.publish' | 'nats.request' | 'nats.subscribe' | 'nats.kv';
+export type NatsTrafficDirection = 'upload' | 'download';
+export interface NatsTrafficMetricBucket {
+    uploadBytes: number;
+    downloadBytes: number;
+}
+export interface NatsTrafficMetrics extends NatsTrafficMetricBucket {
+    bySource: Partial<Record<NatsTrafficMetricSource, NatsTrafficMetricBucket>>;
+}
+export type NatsTrafficRecordEvent = {
+    type: 'METRICS.RECORD';
+    source: NatsTrafficMetricSource;
+    uploadBytes?: number;
+    downloadBytes?: number;
+};
+export type NatsTrafficResetEvent = {
+    type: 'METRICS.RESET_UPLOAD';
+    source?: NatsTrafficMetricSource;
+} | {
+    type: 'METRICS.RESET_DOWNLOAD';
+    source?: NatsTrafficMetricSource;
+} | {
+    type: 'METRICS.RESET_ALL';
+    source?: NatsTrafficMetricSource;
+};
+export type NatsTrafficEvent = NatsTrafficRecordEvent | NatsTrafficResetEvent;
+export declare function createEmptyTrafficMetrics(): NatsTrafficMetrics;
+export declare function byteLength(value: unknown): number;
+export declare function recordTrafficMetric(metrics: NatsTrafficMetrics, event: NatsTrafficRecordEvent): NatsTrafficMetrics;
+export declare function resetTrafficUpload(metrics: NatsTrafficMetrics, source?: NatsTrafficMetricSource): NatsTrafficMetrics;
+export declare function resetTrafficDownload(metrics: NatsTrafficMetrics, source?: NatsTrafficMetricSource): NatsTrafficMetrics;
+export declare function resetTrafficAll(metrics: NatsTrafficMetrics, source?: NatsTrafficMetricSource): NatsTrafficMetrics;
+//# sourceMappingURL=traffic.d.ts.map

package/dist/traffic.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"traffic.d.ts","sourceRoot":"","sources":["../src/traffic.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,uBAAuB,GAAG,cAAc,GAAG,cAAc,GAAG,gBAAgB,GAAG,SAAS,CAAA;AAEpG,MAAM,MAAM,oBAAoB,GAAG,QAAQ,GAAG,UAAU,CAAA;AAExD,MAAM,WAAW,uBAAuB;IACtC,WAAW,EAAE,MAAM,CAAA;IACnB,aAAa,EAAE,MAAM,CAAA;CACtB;AAED,MAAM,WAAW,kBAAmB,SAAQ,uBAAuB;IACjE,QAAQ,EAAE,OAAO,CAAC,MAAM,CAAC,uBAAuB,EAAE,uBAAuB,CAAC,CAAC,CAAA;CAC5E;AAED,MAAM,MAAM,sBAAsB,GAAG;IACnC,IAAI,EAAE,gBAAgB,CAAA;IACtB,MAAM,EAAE,uBAAuB,CAAA;IAC/B,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,aAAa,CAAC,EAAE,MAAM,CAAA;CACvB,CAAA;AAED,MAAM,MAAM,qBAAqB,GAC7B;IAAE,IAAI,EAAE,sBAAsB,CAAC;IAAC,MAAM,CAAC,EAAE,uBAAuB,CAAA;CAAE,GAClE;IAAE,IAAI,EAAE,wBAAwB,CAAC;IAAC,MAAM,CAAC,EAAE,uBAAuB,CAAA;CAAE,GACpE;IAAE,IAAI,EAAE,mBAAmB,CAAC;IAAC,MAAM,CAAC,EAAE,uBAAuB,CAAA;CAAE,CAAA;AAEnE,MAAM,MAAM,gBAAgB,GAAG,sBAAsB,GAAG,qBAAqB,CAAA;AAI7E,wBAAgB,yBAAyB,IAAI,kBAAkB,CAM9D;AAED,wBAAgB,UAAU,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,CAYjD;AAED,wBAAgB,mBAAmB,CACjC,OAAO,EAAE,kBAAkB,EAC3B,KAAK,EAAE,sBAAsB,GAC5B,kBAAkB,CAkBpB;AAED,wBAAgB,kBAAkB,CAChC,OAAO,EAAE,kBAAkB,EAC3B,MAAM,CAAC,EAAE,uBAAuB,GAC/B,kBAAkB,CAEpB;AAED,wBAAgB,oBAAoB,CAClC,OAAO,EAAE,kBAAkB,EAC3B,MAAM,CAAC,EAAE,uBAAuB,GAC/B,kBAAkB,CAEpB;AAED,wBAAgB,eAAe,CAC7B,OAAO,EAAE,kBAAkB,EAC3B,MAAM,CAAC,EAAE,uBAAuB,GAC/B,kBAAkB,CAYpB"}

package/dist/traffic.js

@@ -0,0 +1,99 @@
+const textEncoder = new TextEncoder();
+export function createEmptyTrafficMetrics() {
+    return {
+        uploadBytes: 0,
+        downloadBytes: 0,
+        bySource: {},
+    };
+}
+export function byteLength(value) {
+    if (value == null)
+        return 0;
+    if (value instanceof Uint8Array)
+        return value.byteLength;
+    if (value instanceof ArrayBuffer)
+        return value.byteLength;
+    if (ArrayBuffer.isView(value))
+        return value.byteLength;
+    if (typeof value === 'string')
+        return textEncoder.encode(value).byteLength;
+    try {
+        return textEncoder.encode(JSON.stringify(value)).byteLength;
+    }
+    catch {
+        return 0;
+    }
+}
+export function recordTrafficMetric(metrics, event) {
+    const uploadBytes = normaliseByteCount(event.uploadBytes);
+    const downloadBytes = normaliseByteCount(event.downloadBytes);
+    if (uploadBytes === 0 && downloadBytes === 0)
+        return metrics;
+    const currentSource = metrics.bySource[event.source] ?? { uploadBytes: 0, downloadBytes: 0 };
+    return {
+        uploadBytes: metrics.uploadBytes + uploadBytes,
+        downloadBytes: metrics.downloadBytes + downloadBytes,
+        bySource: {
+            ...metrics.bySource,
+            [event.source]: {
+                uploadBytes: currentSource.uploadBytes + uploadBytes,
+                downloadBytes: currentSource.downloadBytes + downloadBytes,
+            },
+        },
+    };
+}
+export function resetTrafficUpload(metrics, source) {
+    return resetTrafficDirection(metrics, 'upload', source);
+}
+export function resetTrafficDownload(metrics, source) {
+    return resetTrafficDirection(metrics, 'download', source);
+}
+export function resetTrafficAll(metrics, source) {
+    if (!source)
+        return createEmptyTrafficMetrics();
+    const sourceMetrics = metrics.bySource[source] ?? { uploadBytes: 0, downloadBytes: 0 };
+    const bySource = { ...metrics.bySource };
+    delete bySource[source];
+    return {
+        uploadBytes: metrics.uploadBytes - sourceMetrics.uploadBytes,
+        downloadBytes: metrics.downloadBytes - sourceMetrics.downloadBytes,
+        bySource,
+    };
+}
+function resetTrafficDirection(metrics, direction, source) {
+    const key = direction === 'upload' ? 'uploadBytes' : 'downloadBytes';
+    if (!source) {
+        const bySource = Object.fromEntries(Object.entries(metrics.bySource).map(([sourceName, sourceMetrics]) => [
+            sourceName,
+            {
+                ...sourceMetrics,
+                [key]: 0,
+            },
+        ]));
+        return {
+            ...metrics,
+            [key]: 0,
+            bySource,
+        };
+    }
+    const sourceMetrics = metrics.bySource[source];
+    if (!sourceMetrics)
+        return metrics;
+    return {
+        ...metrics,
+        [key]: metrics[key] - sourceMetrics[key],
+        bySource: {
+            ...metrics.bySource,
+            [source]: {
+                ...sourceMetrics,
+                [key]: 0,
+            },
+        },
+    };
+}
+function normaliseByteCount(value) {
+    if (!Number.isFinite(value) || value === undefined || value <= 0)
+        return 0;
+    return Math.floor(value);
+}
+//# sourceMappingURL=traffic.js.map

package/dist/traffic.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"traffic.js","sourceRoot":"","sources":["../src/traffic.ts"],"names":[],"mappings":"AA2BA,MAAM,WAAW,GAAG,IAAI,WAAW,EAAE,CAAA;AAErC,MAAM,UAAU,yBAAyB;IACvC,OAAO;QACL,WAAW,EAAE,CAAC;QACd,aAAa,EAAE,CAAC;QAChB,QAAQ,EAAE,EAAE;KACb,CAAA;AACH,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,KAAc;IACvC,IAAI,KAAK,IAAI,IAAI;QAAE,OAAO,CAAC,CAAA;IAC3B,IAAI,KAAK,YAAY,UAAU;QAAE,OAAO,KAAK,CAAC,UAAU,CAAA;IACxD,IAAI,KAAK,YAAY,WAAW;QAAE,OAAO,KAAK,CAAC,UAAU,CAAA;IACzD,IAAI,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC,UAAU,CAAA;IACtD,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,UAAU,CAAA;IAE1E,IAAI,CAAC;QACH,OAAO,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,UAAU,CAAA;IAC7D,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,CAAC,CAAA;IACV,CAAC;AACH,CAAC;AAED,MAAM,UAAU,mBAAmB,CACjC,OAA2B,EAC3B,KAA6B;IAE7B,MAAM,WAAW,GAAG,kBAAkB,CAAC,KAAK,CAAC,WAAW,CAAC,CAAA;IACzD,MAAM,aAAa,GAAG,kBAAkB,CAAC,KAAK,CAAC,aAAa,CAAC,CAAA;IAE7D,IAAI,WAAW,KAAK,CAAC,IAAI,aAAa,KAAK,CAAC;QAAE,OAAO,OAAO,CAAA;IAE5D,MAAM,aAAa,GAAG,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,WAAW,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,CAAA;IAC5F,OAAO;QACL,WAAW,EAAE,OAAO,CAAC,WAAW,GAAG,WAAW;QAC9C,aAAa,EAAE,OAAO,CAAC,aAAa,GAAG,aAAa;QACpD,QAAQ,EAAE;YACR,GAAG,OAAO,CAAC,QAAQ;YACnB,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE;gBACd,WAAW,EAAE,aAAa,CAAC,WAAW,GAAG,WAAW;gBACpD,aAAa,EAAE,aAAa,CAAC,aAAa,GAAG,aAAa;aAC3D;SACF;KACF,CAAA;AACH,CAAC;AAED,MAAM,UAAU,kBAAkB,CAChC,OAA2B,EAC3B,MAAgC;IAEhC,OAAO,qBAAqB,CAAC,OAAO,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAA;AACzD,CAAC;AAED,MAAM,UAAU,oBAAoB,CAClC,OAA2B,EAC3B,MAAgC;IAEhC,OAAO,qBAAqB,CAAC,OAAO,EAAE,UAAU,EAAE,MAAM,CAAC,CAAA;AAC3D,CAAC;AAED,MAAM,UAAU,eAAe,CAC7B,OAA2B,EAC3B,MAAgC;IAEhC,IAAI,CAAC,MAAM;QAAE,OAAO,yBAAyB,EAAE,CAAA;IAE/C,MAAM,aAAa,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,EAAE,WAAW,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,CAAA;IACtF,MAAM,QAAQ,GAAG,EAAE,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAA;IACxC,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAA;IAEvB,OAAO;QACL,WAAW,EAAE,OAAO,CAAC,WAAW,GAAG,aAAa,CAAC,WAAW;QAC5D,aAAa,EAAE,OAAO,CAAC,aAAa,GAAG,aAAa,CAAC,aAAa;QAClE,QAAQ;KACT,CAAA;AACH,CAAC;AAED,SAAS,qBAAqB,CAC5B,OAA2B,EAC3B,SAA+B,EAC/B,MAAgC;IAEhC,MAAM,GAAG,GAAG,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,eAAe,CAAA;IAEpE,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,QAAQ,GAAG,MAAM,CAAC,WAAW,CACjC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,EAAE,aAAa,CAAC,EAAE,EAAE,CAAC;YACpE,UAAU;YACV;gBACE,GAAG,aAAa;gBAChB,CAAC,GAAG,CAAC,EAAE,CAAC;aACT;SACF,CAAC,CAC+B,CAAA;QAEnC,OAAO;YACL,GAAG,OAAO;YACV,CAAC,GAAG,CAAC,EAAE,CAAC;YACR,QAAQ;SACT,CAAA;IACH,CAAC;IAED,MAAM,aAAa,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAA;IAC9C,IAAI,CAAC,aAAa;QAAE,OAAO,OAAO,CAAA;IAElC,OAAO;QACL,GAAG,OAAO;QACV,CAAC,GAAG,CAAC,EAAE,OAAO,CAAC,GAAG,CAAC,GAAG,aAAa,CAAC,GAAG,CAAC;QACxC,QAAQ,EAAE;YACR,GAAG,OAAO,CAAC,QAAQ;YACnB,CAAC,MAAM,CAAC,EAAE;gBACR,GAAG,aAAa;gBAChB,CAAC,GAAG,CAAC,EAAE,CAAC;aACT;SACF;KACF,CAAA;AACH,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAyB;IACnD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,IAAI,CAAC;QAAE,OAAO,CAAC,CAAA;IAC1E,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;AAC1B,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@jr200-labs/xstate-nats",
-  "version": "0.9.13",
+  "version": "0.9.14",
   "description": "XState machine for NATS",
   "license": "MIT",
   "author": "Jayshan Raghunandan",

package/src/actions/kv.test.ts

@@ -181,6 +181,65 @@ describe('kvConsolidateState', () => {
     expect(result.subscriptions.has('Pair(bucket, key1)')).toBe(true)
   })
 
+  it('should report KV watch entry download bytes', async () => {
+    const rawValue = '{"value":"hello"}'
+    let delivered = false
+    let resolveIterator: () => void
+    const iteratorDone = new Promise<void>((resolve) => {
+      resolveIterator = resolve
+    })
+    const mockWatcher = {
+      [Symbol.asyncIterator]: () => ({
+        next: () => {
+          if (!delivered) {
+            delivered = true
+            return Promise.resolve({
+              value: {
+                operation: 'PUT',
+                string: () => rawValue,
+              },
+              done: false,
+            })
+          }
+          resolveIterator()
+          return new Promise(() => {})
+        },
+      }),
+    }
+    const mockKv = {
+      watch: vi.fn().mockResolvedValue(mockWatcher),
+    }
+    const mockKvm = {
+      open: vi.fn().mockResolvedValue(mockKv),
+    }
+    const onDownloadBytes = vi.fn()
+
+    const actor = createActor(kvConsolidateState, {
+      input: {
+        kvm: mockKvm as any,
+        connection: {} as any,
+        currentState: new Map(),
+        targetState: new Map([
+          [
+            'Pair(bucket, key1)',
+            {
+              bucket: 'bucket',
+              key: 'key1',
+              callback: vi.fn(),
+              onDownloadBytes,
+            },
+          ],
+        ]) as any,
+      },
+    })
+
+    actor.start()
+    await iteratorDone
+    await new Promise((r) => setTimeout(r, 10))
+    expect(onDownloadBytes).toHaveBeenCalledWith(new TextEncoder().encode(rawValue).byteLength)
+    actor.stop()
+  })
+
   it('should keep existing subscriptions that are still in target', async () => {
     const existingWatcher = { stop: vi.fn() } as any
     const currentState = new Map([['Pair(bucket, key1)', existingWatcher]])

package/src/actions/kv.ts

@@ -3,6 +3,7 @@ import { Kvm, KvWatchEntry, KvWatchOptions } from '@nats-io/kv'
 import { Pair } from '../utils'
 import { fromPromise } from 'xstate'
 import { recordError, withSpan } from '../telemetry'
+import { byteLength } from '../traffic'
 
 export class KvSubscriptionKey extends Pair<string, string> {}
 
@@ -12,6 +13,7 @@ export type KvSubscriptionConfig = {
   callback: (data: any) => void
   opts?: KvWatchOptions
   replayOnReconnect?: boolean
+  onDownloadBytes?: (bytes: number) => void
 }
 
 export const kvConsolidateState = fromPromise(
@@ -80,13 +82,19 @@ export const kvConsolidateState = fromPromise(
                     },
                     (span) => {
                       let parsedValue
+                      let rawValue: string
                       try {
-                        parsedValue = JSON.parse(e.string())
+                        rawValue = e.string()
+                        parsedValue = JSON.parse(rawValue)
                       } catch {
-                        parsedValue = e.string()
+                        rawValue = e.string()
+                        parsedValue = rawValue
                       }
 
                       try {
+                        // KvWatchEntry does not expose the raw wire payload on this path; the
+                        // UTF-8 byte length of string() is the measurable application payload.
+                        config.onDownloadBytes?.(byteLength(rawValue))
                         config.callback({
                           bucket: config.bucket,
                           key: config.key,

package/src/actions/subject.test.ts

@@ -199,6 +199,47 @@ describe('subjectConsolidateState', () => {
     expect(callback).toHaveBeenCalledWith({ id: 2 })
   })
 
+  it('should report subscription message download bytes', async () => {
+    const connection = createMockConnection()
+    const payload = new TextEncoder().encode('{"id":1}')
+    let resolveIterator: () => void
+    const iteratorDone = new Promise<void>((r) => (resolveIterator = r))
+    let delivered = false
+
+    const mockSub = {
+      unsubscribe: vi.fn(),
+      [Symbol.asyncIterator]: () => ({
+        next: () => {
+          if (!delivered) {
+            delivered = true
+            return Promise.resolve({
+              value: { data: payload, json: () => ({ id: 1 }), string: () => '{"id":1}' },
+              done: false,
+            })
+          }
+          resolveIterator!()
+          return new Promise(() => {})
+        },
+      }),
+    } as any
+    connection.subscribe.mockReturnValue(mockSub)
+
+    const onDownloadBytes = vi.fn()
+    subjectConsolidateState({
+      input: {
+        connection,
+        currentSubscriptions: new Map(),
+        targetSubscriptions: new Map([
+          ['test.subject', { subject: 'test.subject', callback: vi.fn(), onDownloadBytes }],
+        ]),
+      },
+    })
+
+    await iteratorDone
+    await new Promise((r) => setTimeout(r, 10))
+    expect(onDownloadBytes).toHaveBeenCalledWith(payload.byteLength)
+  })
+
   it('should handle callback errors in the message loop', async () => {
     const connection = createMockConnection()
     let resolveIterator: () => void

package/src/actions/subject.ts

@@ -13,11 +13,13 @@ import {
   recordError,
   withSpan,
 } from '../telemetry'
+import { byteLength } from '../traffic'
 
 export type SubjectSubscriptionConfig = {
   subject: string
   callback: (data: any) => void
   opts?: SubscriptionOptions
+  onDownloadBytes?: (bytes: number) => void
 }
 
 export type RequestResult = { ok: true } | { ok: false; error: Error }
@@ -70,15 +72,17 @@ export const subjectConsolidateState = ({
           try {
             for await (const msg of sub) {
               const parentCtx = extractContextFromHeaders((msg as Msg).headers)
+              const payloadBytes = (msg as Msg).data?.length ?? 0
               await withSpan(
                 'xstate.nats.message',
                 'xstate.nats.error',
                 {
                   subject,
-                  'payload.bytes': (msg as Msg).data?.length,
+                  'payload.bytes': payloadBytes,
                 },
                 (span) => {
                   try {
+                    subscriptionConfig.onDownloadBytes?.(payloadBytes)
                     subscriptionConfig?.callback(parseNatsResult(msg))
                   } catch (callbackError) {
                     // Record on span AND preserve the existing console.error
@@ -117,19 +121,15 @@ export const subjectRequest = ({
     opts?: RequestOptions
     callback: (data: any) => void
     onRequestResult?: (result: RequestResult) => void
+    onDownloadBytes?: (bytes: number) => void
   }
 }) => {
-  const { connection, subject, payload, opts, callback, onRequestResult } = input
+  const { connection, subject, payload, opts, callback, onRequestResult, onDownloadBytes } = input
   if (!connection) {
     throw new Error('NATS connection is not available')
   }
 
-  const payloadBytes =
-    payload instanceof Uint8Array
-      ? payload.byteLength
-      : typeof payload === 'string'
-        ? payload.length
-        : undefined
+  const payloadBytes = byteLength(payload)
 
   void withSpan(
     'xstate.nats.request',
@@ -151,6 +151,7 @@ export const subjectRequest = ({
       return connection
         .request(subject, payload, requestOpts)
         .then((msg: Msg) => {
+          onDownloadBytes?.(msg.data?.length ?? 0)
           callback(parseNatsResult(msg))
           onRequestResult?.({ ok: true })
         })
@@ -185,12 +186,7 @@ export const subjectPublish = ({
     throw new Error('NATS connection is not available')
   }
 
-  const payloadBytes =
-    payload instanceof Uint8Array
-      ? payload.byteLength
-      : typeof payload === 'string'
-        ? payload.length
-        : undefined
+  const payloadBytes = byteLength(payload)
 
   try {
     withSpan(

package/src/index.ts

@@ -12,6 +12,20 @@ export {
 export { KvSubscriptionKey, type KvSubscriptionConfig } from './actions/kv'
 export { parseNatsResult } from './actions/connection'
 export { type AuthConfig } from './actions/types'
+export {
+  byteLength as natsTrafficByteLength,
+  createEmptyTrafficMetrics,
+  recordTrafficMetric,
+  resetTrafficAll,
+  resetTrafficDownload,
+  resetTrafficUpload,
+  type NatsTrafficEvent,
+  type NatsTrafficMetricBucket,
+  type NatsTrafficMetrics,
+  type NatsTrafficMetricSource,
+  type NatsTrafficRecordEvent,
+  type NatsTrafficResetEvent,
+} from './traffic'
 
 export {
   type NatsConnectionConfig,