nmtjs 0.15.2 → 0.16.0-beta.1

package/src/runtime/{pubsub → subscription}/manager.ts

@@ -11,27 +11,27 @@ import type { Container, Logger } from '@nmtjs/core'
 import type { t } from '@nmtjs/type'
 import { isAbortError } from '@nmtjs/common'
 
-import { pubSubAdapter } from '../injectables.ts'
+import { subscriptionAdapter } from '../injectables.ts'
 
-export type PubSubAdapterEvent = { channel: string; payload: any }
+export type SubscriptionAdapterEvent = { channel: string; payload: any }
 
-export interface PubSubAdapterType {
+export interface SubscriptionAdapterType {
   publish(channel: string, payload: any): Promise<boolean>
   subscribe(
     channel: string,
     signal?: AbortSignal,
-  ): AsyncGenerator<PubSubAdapterEvent>
+  ): AsyncGenerator<SubscriptionAdapterEvent>
   initialize(): Promise<void>
   dispose(): Promise<void>
 }
 
-export type PubSubChannel = {
+export type SubscriptionChannel = {
   stream: Readable
   subscription: TAnySubscriptionContract
   event: TAnyEventContract
 }
 
-export type PubSubSubscribe = <
+export type SubscribeFn = <
   Contract extends TAnySubscriptionContract,
   Events extends {
     [K in keyof Contract['events']]?: true
@@ -63,7 +63,7 @@ export type PubSubSubscribe = <
   }
 >
 
-export type PubSubPublish = <
+export type PublishFn = <
   S extends TAnySubscriptionContract,
   E extends S['events'][keyof S['events']],
 >(
@@ -72,23 +72,24 @@ export type PubSubPublish = <
   data: t.infer.decode.input<E['payload']>,
 ) => Promise<boolean>
 
-export type PubSubManagerOptions = { logger: Logger; container: Container }
+export type SubscriptionManagerOptions = {
+  logger: Logger
+  container: Container
+}
 
-export class PubSubManager {
-  readonly subscriptions = new Map<string, PubSubChannel>()
+export class SubscriptionManager {
+  readonly subscriptions = new Map<string, SubscriptionChannel>()
+  protected readonly logger: Logger
 
-  constructor(protected readonly options: PubSubManagerOptions) {}
+  constructor(protected readonly options: SubscriptionManagerOptions) {
+    this.logger = options.logger.child({ component: SubscriptionManager.name })
+  }
 
   protected get adapter() {
-    return this.options.container.resolve(pubSubAdapter)
+    return this.options.container.resolve(subscriptionAdapter)
   }
 
-  subscribe: PubSubSubscribe = async (
-    subscription,
-    events,
-    options,
-    signal,
-  ) => {
+  subscribe: SubscribeFn = async (subscription, events, options, signal) => {
     const adapter = await this.adapter
 
     const eventKeys =
@@ -96,6 +97,11 @@ export class PubSubManager {
         ? Object.keys(subscription.events)
         : Object.keys(events)
 
+    this.logger.debug(
+      { subscription: subscription.name, eventCount: eventKeys.length },
+      'Opening subscription',
+    )
+
     const streams = Array(eventKeys.length)
 
     for (const index in eventKeys) {
@@ -103,44 +109,115 @@ export class PubSubManager {
       const channel = getChannelName(event, options)
       if (this.subscriptions.has(channel)) {
         streams[index] = this.subscriptions.get(channel)!.stream
+        this.logger.trace(
+          {
+            channel,
+            event: event.name,
+            activeChannels: this.subscriptions.size,
+          },
+          'Reusing pubsub channel stream',
+        )
       } else {
         const iterable = adapter.subscribe(channel, signal)
         const stream = this.createEventStream(iterable)
-        stream.on('close', () => this.subscriptions.delete(channel))
+        stream.on('close', () => {
+          this.subscriptions.delete(channel)
+          this.logger.debug(
+            {
+              channel,
+              event: event.name,
+              activeChannels: this.subscriptions.size,
+            },
+            'Pubsub channel stream closed',
+          )
+        })
+        stream.on('error', (error) => {
+          this.logger.warn(
+            { channel, event: event.name, error },
+            'Pubsub channel stream failed',
+          )
+        })
         streams[index] = stream
         this.subscriptions.set(channel, { subscription, event, stream })
+        this.logger.debug(
+          {
+            channel,
+            event: event.name,
+            activeChannels: this.subscriptions.size,
+          },
+          'Creating channel stream',
+        )
       }
     }
 
-    return mergeEventStreams(streams, signal)
+    const mergedStream = mergeEventStreams(streams, signal)
+
+    mergedStream.once('close', () => {
+      this.logger.debug(
+        { subscription: subscription.name, eventCount: eventKeys.length },
+        'Pubsub subscription stream closed',
+      )
+    })
+    mergedStream.once('error', (error) => {
+      this.logger.warn(
+        {
+          subscription: subscription.name,
+          eventCount: eventKeys.length,
+          error,
+        },
+        'Pubsub subscription stream failed',
+      )
+    })
+
+    return mergedStream
   }
 
-  publish: PubSubPublish = async (event, options, data) => {
+  publish: PublishFn = async (event, options, data) => {
     const adapter = await this.adapter
 
     const channel = getChannelName(event, options)
 
+    this.logger.trace({ channel, event: event.name }, 'Publishing pubsub event')
+
     try {
       const payload = event.payload.encode(data)
-      return await adapter.publish(channel, payload)
+      const published = await adapter.publish(channel, payload)
+
+      if (published) {
+        this.logger.trace(
+          { channel, event: event.name },
+          'Published pubsub event',
+        )
+      } else {
+        this.logger.warn(
+          { channel, event: event.name },
+          'Pubsub adapter reported publish failure',
+        )
+      }
+
+      return published
     } catch (error: any) {
-      this.options.logger.error(
-        `Failed to publish event "${event.name}" on channel "${channel}": ${error.message}`,
+      this.logger.error(
+        { channel, event: event.name, error },
+        'Failed to publish pubsub event',
       )
-      return Promise.reject(error)
+      throw error
     }
   }
 
   private createEventStream(
-    iterable: AsyncGenerator<PubSubAdapterEvent>,
+    iterable: AsyncGenerator<SubscriptionAdapterEvent>,
   ): Readable {
     const { subscriptions } = this
+    const logger = this.logger
+
     return new Readable({
       objectMode: true,
       read() {
         iterable.next().then(
           ({ value, done }) => {
             if (done) {
+              logger.trace('Pubsub adapter stream ended')
               this.push(null)
             } else {
               const subscription = subscriptions.get(value.channel)
@@ -148,17 +225,32 @@ export class PubSubManager {
                 const { event } = subscription
                 try {
                   const data = event.payload.decode(value.payload)
+                  logger.trace(
+                    { channel: value.channel, event: event.name },
+                    'Received event',
+                  )
                   this.push({ event: event.name, data })
                 } catch (error: any) {
+                  logger.warn(
+                    { channel: value.channel, event: event.name, error },
+                    'Failed to decode pubsub event payload',
+                  )
                   this.destroy(error)
                 }
+              } else {
+                logger.trace(
+                  { channel: value.channel },
+                  'Dropped pubsub event for inactive channel',
+                )
               }
             }
           },
           (error) => {
             if (isAbortError(error)) {
+              logger.trace('Pubsub adapter stream aborted')
               this.push(null)
             } else {
+              logger.warn({ error }, 'Pubsub adapter stream failed')
               this.destroy(error)
             }
           },

package/src/runtime/subscription/redis.ts

@@ -0,0 +1,157 @@
+import EventEmitter, { on } from 'node:events'
+
+import type { Logger } from '@nmtjs/core'
+import { isAbortError } from '@nmtjs/common'
+import { createFactoryInjectable } from '@nmtjs/core'
+
+import type { RuntimePlugin } from '../plugin.ts'
+import type { Store } from '../types.ts'
+import type {
+  SubscriptionAdapterEvent,
+  SubscriptionAdapterType,
+} from './manager.ts'
+import { storeConfig, subscriptionAdapter } from '../injectables.ts'
+import { createStoreClient } from '../store/index.ts'
+
+export class RedisSubscriptionAdapter implements SubscriptionAdapterType {
+  protected readonly events = new EventEmitter()
+  protected readonly listeners = new Map<string, number>()
+  protected readonly logger?: Logger
+  protected subscriberClient?: Store
+
+  constructor(
+    protected readonly client: Store,
+    logger?: Logger,
+  ) {
+    this.logger = logger?.child({ component: RedisSubscriptionAdapter.name })
+  }
+
+  async initialize() {
+    this.logger?.debug('Initializing Redis adapter')
+
+    // Create a dedicated subscriber client (Redis requires separate clients for pub/sub)
+    this.subscriberClient = this.client.duplicate()
+
+    // Set up message handler
+    this.subscriberClient.on('message', (channel: string, message: string) => {
+      try {
+        const parsed = JSON.parse(message)
+        this.logger?.trace({ channel }, 'Received Redis message')
+        this.events.emit(channel, parsed)
+      } catch (error) {
+        this.logger?.warn({ channel, error }, 'Failed to parse Redis message')
+      }
+    })
+
+    this.logger?.debug('Redis adapter initialized')
+  }
+
+  async dispose() {
+    this.logger?.debug('Disposing Redis adapter')
+
+    if (this.subscriberClient) {
+      await this.subscriberClient.quit()
+      this.subscriberClient = undefined
+    }
+
+    this.logger?.debug('Redis adapter disposed')
+  }
+
+  async publish(channel: string, payload: any): Promise<boolean> {
+    this.logger?.trace({ channel }, 'Publishing Redis message')
+
+    try {
+      await this.client.publish(channel, JSON.stringify(payload))
+      this.logger?.trace({ channel }, 'Published Redis message')
+      return true
+    } catch (error) {
+      this.logger?.warn({ channel, error }, 'Failed to publish Redis message')
+      return false
+    }
+  }
+
+  async *subscribe(
+    channel: string,
+    signal?: AbortSignal,
+  ): AsyncGenerator<SubscriptionAdapterEvent> {
+    if (!this.subscriberClient) {
+      throw new Error('RedisSubscriptionAdapter not initialized')
+    }
+
+    this.logger?.debug({ channel }, 'Opening Redis channel listener')
+
+    if (!this.listeners.has(channel)) {
+      await this.subscriberClient.subscribe(channel)
+      this.listeners.set(channel, 1)
+      this.logger?.debug({ channel, listeners: 1 }, 'Subscribed Redis channel')
+    } else {
+      const listeners = this.listeners.get(channel)! + 1
+      this.listeners.set(channel, listeners)
+      this.logger?.trace(
+        { channel, listeners },
+        'Reusing Redis channel listener',
+      )
+    }
+
+    try {
+      signal?.throwIfAborted()
+      for await (const [payload] of on(this.events, channel, { signal })) {
+        this.logger?.trace({ channel }, 'Delivering Redis message')
+        yield { channel, payload }
+      }
+    } catch (error: any) {
+      if (isAbortError(error)) {
+        this.logger?.trace({ channel }, 'Redis channel listener aborted')
+        throw error
+      }
+
+      this.logger?.warn({ channel, error }, 'Redis channel listener failed')
+    } finally {
+      const count = this.listeners.get(channel)
+      if (count !== undefined) {
+        if (count > 1) {
+          const listeners = count - 1
+          this.listeners.set(channel, listeners)
+          this.logger?.trace(
+            { channel, listeners },
+            'Detached Redis channel listener',
+          )
+        } else {
+          await this.subscriberClient?.unsubscribe(channel)
+          this.listeners.delete(channel)
+          this.logger?.debug(
+            { channel, listeners: 0 },
+            'Unsubscribed Redis channel',
+          )
+        }
+      }
+    }
+  }
+}
+
+export const RedisSubscriptionAdapterPlugin = (): RuntimePlugin => {
+  return {
+    name: 'redis-subscription-adapter',
+    hooks: {
+      'lifecycle:beforeInitialize': async (ctx) => {
+        const adapter = await ctx.container.resolve(
+          createFactoryInjectable({
+            dependencies: { config: storeConfig },
+            factory: async ({ config }) => {
+              const connection = await createStoreClient(config)
+              const adapter = new RedisSubscriptionAdapter(
+                connection,
+                ctx.logger,
+              )
+              await adapter.initialize()
+              return { adapter, connection }
+            },
+            pick: ({ adapter }) => adapter,
+            dispose: ({ connection }) => connection.quit(),
+          }),
+        )
+        ctx.container.provide(subscriptionAdapter, adapter)
+      },
+    },
+  }
+}

package/src/runtime/workers/base.ts

@@ -6,11 +6,11 @@ import type { BaseRuntimeOptions } from '../runtime.ts'
 import type { ServerConfig } from '../server/config.ts'
 import * as injectables from '../injectables.ts'
 import { JobManager } from '../jobs/manager.ts'
-import { PubSubManager } from '../pubsub/manager.ts'
 import { BaseRuntime } from '../runtime.ts'
+import { SubscriptionManager } from '../subscription/manager.ts'
 
 export abstract class BaseWorkerRuntime extends BaseRuntime {
-  pubsub: PubSubManager
+  subscriptionManager: SubscriptionManager
   jobManager?: JobManager
 
   constructor(
@@ -20,7 +20,7 @@ export abstract class BaseWorkerRuntime extends BaseRuntime {
   ) {
     super(options)
 
-    this.pubsub = new PubSubManager({
+    this.subscriptionManager = new SubscriptionManager({
       logger: this.logger,
       container: this.container,
     })
@@ -38,12 +38,12 @@ export abstract class BaseWorkerRuntime extends BaseRuntime {
       provision(CoreInjectables.logger, this.logger),
       provision(injectables.workerType, this.workerType),
       provision(
-        injectables.pubSubPublish,
-        this.pubsub.publish.bind(this.pubsub),
+        injectables.publish,
+        this.subscriptionManager.publish.bind(this.subscriptionManager),
       ),
       provision(
-        injectables.pubSubSubscribe,
-        this.pubsub.subscribe.bind(this.pubsub),
+        injectables.subscribe,
+        this.subscriptionManager.subscribe.bind(this.subscriptionManager),
       ),
     ]
 

package/dist/runtime/pubsub/manager.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"manager.js","sourceRoot":"","sources":["../../../src/runtime/pubsub/manager.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,aAAa,CAAA;AAChC,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AACxC,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAA;AASnD,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAA;AAE5C,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAA;AA+DjD,MAAM,OAAO,aAAa;IAGO,OAAO;IAF7B,aAAa,GAAG,IAAI,GAAG,EAAyB,CAAA;IAEzD,YAA+B,OAA6B,EAAE;uBAA/B,OAAO;IAAyB,CAAC;IAEhE,IAAc,OAAO,GAAG;QACtB,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC,aAAa,CAAC,CAAA;IAAA,CACrD;IAED,SAAS,GAAoB,KAAK,EAChC,YAAY,EACZ,MAAM,EACN,OAAO,EACP,MAAM,EACN,EAAE,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,CAAA;QAElC,MAAM,SAAS,GACb,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,KAAK,CAAC;YAC9B,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC;YAClC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;QAEzB,MAAM,OAAO,GAAG,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,CAAA;QAEvC,KAAK,MAAM,KAAK,IAAI,SAAS,EAAE,CAAC;YAC9B,MAAM,KAAK,GAAG,YAAY,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAA;YACnD,MAAM,OAAO,GAAG,cAAc,CAAC,KAAK,EAAE,OAAO,CAAC,CAAA;YAC9C,IAAI,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;gBACpC,OAAO,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,OAAO,CAAE,CAAC,MAAM,CAAA;YAC1D,CAAC;iBAAM,CAAC;gBACN,MAAM,QAAQ,GAAG,OAAO,CAAC,SAAS,CAAC,OAAO,EAAE,MAAM,CAAC,CAAA;gBACnD,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAA;gBAC/C,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAA;gBAC5D,OAAO,CAAC,KAAK,CAAC,GAAG,MAAM,CAAA;gBACvB,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAA;YAClE,CAAC;QACH,CAAC;QAED,OAAO,iBAAiB,CAAC,OAAO,EAAE,MAAM,CAAC,CAAA;IAAA,CAC1C,CAAA;IAED,OAAO,GAAkB,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,CAAC;QACvD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,CAAA;QAElC,MAAM,OAAO,GAAG,cAAc,CAAC,KAAK,EAAE,OAAO,CAAC,CAAA;QAE9C,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;YAC1C,OAAO,MAAM,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,CAAA;QAChD,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CACvB,4BAA4B,KAAK,CAAC,IAAI,iBAAiB,OAAO,MAAM,KAAK,CAAC,OAAO,EAAE,CACpF,CAAA;YACD,OAAO,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QAC9B,CAAC;IAAA,CACF,CAAA;IAEO,iBAAiB,CACvB,QAA4C,EAClC;QACV,MAAM,EAAE,aAAa,EAAE,GAAG,IAAI,CAAA;QAC9B,OAAO,IAAI,QAAQ,CAAC;YAClB,UAAU,EAAE,IAAI;YAChB,IAAI,GAAG;gBACL,QAAQ,CAAC,IAAI,EAAE,CAAC,IAAI,CAClB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;oBACnB,IAAI,IAAI,EAAE,CAAC;wBACT,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;oBACjB,CAAC;yBAAM,CAAC;wBACN,MAAM,YAAY,GAAG,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAA;wBACrD,IAAI,YAAY,EAAE,CAAC;4BACjB,MAAM,EAAE,KAAK,EAAE,GAAG,YAAY,CAAA;4BAC9B,IAAI,CAAC;gCACH,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAA;gCAChD,IAAI,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;4BACxC,CAAC;4BAAC,OAAO,KAAU,EAAE,CAAC;gCACpB,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;4BACrB,CAAC;wBACH,CAAC;oBACH,CAAC;gBAAA,CACF,EACD,CAAC,KAAK,EAAE,EAAE,CAAC;oBACT,IAAI,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC;wBACxB,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;oBACjB,CAAC;yBAAM,CAAC;wBACN,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;oBACrB,CAAC;gBAAA,CACF,CACF,CAAA;YAAA,CACF;SACF,CAAC,CAAA;IAAA,CACH;CACF;AAED,SAAS,MAAM,CAAC,GAAG,IAAS,EAAE;IAC5B,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;AAAA,CACtC;AAED,SAAS,cAAc,CACrB,QAAW,EACX,OAAqB,EACrB;IACA,MAAM,GAAG,GAAG,OAAO,CAAC,CAAC,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAA;IACrD,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,iCAAiC,CAAC,CAAA;IACxD,OAAO,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,GAAG,CAAC,CAAA;AAAA,CAClC;AAED,SAAS,iBAAiB,CAAC,OAA0C,EAAU;IAC7E,MAAM,IAAI,GAAG,UAAU,CAAC,MAAM,CAAC,CAAA;IAC/B,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC;SACvC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;SACxC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,IAAI,KAAK,EAAE,CAAC;SACxC,IAAI,CAAC,GAAG,CAAC,CAAA;IACZ,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAA;IACvB,OAAO,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAA;AAAA,CAChC;AAED,SAAS,iBAAiB,CACxB,OAAmB,EACnB,MAAoB,EACV;IACV,MAAM,WAAW,GAAG,IAAI,WAAW,CAAC;QAClC,MAAM;QACN,UAAU,EAAE,IAAI;QAChB,kBAAkB,EAAE,IAAI;QACxB,kBAAkB,EAAE,IAAI;KACzB,CAAC,CAAA;IAEF,IAAI,KAAK,GAAG,CAAC,CAAA;IAEb,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAA;QACxC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC;YACvB,KAAK,EAAE,CAAA;YACP,IAAI,KAAK,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC;gBAC7B,WAAW,CAAC,GAAG,EAAE,CAAA;YACnB,CAAC;QAAA,CACF,CAAC,CAAA;IACJ,CAAC;IAED,OAAO,WAAW,CAAA;AAAA,CACnB"}

package/dist/runtime/pubsub/redis.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"redis.js","sourceRoot":"","sources":["../../../src/runtime/pubsub/redis.ts"],"names":[],"mappings":"AAAA,OAAO,YAAY,EAAE,EAAE,EAAE,EAAE,MAAM,aAAa,CAAA;AAE9C,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAA;AAC5C,OAAO,EAAE,uBAAuB,EAAE,SAAS,EAAE,MAAM,aAAa,CAAA;AAKhE,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAA;AAC9D,OAAO,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAA;AAErD,MAAM,OAAO,kBAAkB;IAKE,MAAM;IAJlB,MAAM,GAAG,IAAI,YAAY,EAAE,CAAA;IAC3B,SAAS,GAAG,IAAI,GAAG,EAAkB,CAAA;IAC9C,gBAAgB,CAAQ;IAElC,YAA+B,MAAa,EAAE;sBAAf,MAAM;IAAU,CAAC;IAEhD,KAAK,CAAC,UAAU,GAAG;QACjB,qFAAqF;QACrF,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAA;QAE/C,yBAAyB;QACzB,IAAI,CAAC,gBAAgB,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,OAAe,EAAE,OAAe,EAAE,EAAE,CAAC;YACxE,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAA;gBAClC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAA;YACnC,CAAC;YAAC,MAAM,CAAC,CAAA,CAAC;QAAA,CACX,CAAC,CAAA;IAAA,CACH;IAED,KAAK,CAAC,OAAO,GAAG;QACd,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1B,MAAM,IAAI,CAAC,gBAAgB,CAAC,IAAI,EAAE,CAAA;YAClC,IAAI,CAAC,gBAAgB,GAAG,SAAS,CAAA;QACnC,CAAC;IAAA,CACF;IAED,KAAK,CAAC,OAAO,CAAC,OAAe,EAAE,OAAY,EAAoB;QAC7D,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAA;YAC3D,OAAO,IAAI,CAAA;QACb,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAA;QACd,CAAC;IAAA,CACF;IAED,KAAK,CAAC,CAAC,SAAS,CACd,OAAe,EACf,MAAoB,EACgB;QACpC,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC3B,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAA;QACvD,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;YACjC,MAAM,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,OAAO,CAAC,CAAA;YAC9C,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAA;QAChC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAE,GAAG,CAAC,CAAC,CAAA;QAC/D,CAAC;QAED,IAAI,CAAC;YACH,MAAM,EAAE,cAAc,EAAE,CAAA;YACxB,IAAI,KAAK,EAAE,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;gBACnE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAA;YAC5B,CAAC;QACH,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,IAAI,YAAY,CAAC,KAAK,CAAC;gBAAE,MAAM,KAAK,CAAA;QACtC,CAAC;gBAAS,CAAC;YACT,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;YACzC,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACxB,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;oBACd,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,GAAG,CAAC,CAAC,CAAA;gBACxC,CAAC;qBAAM,CAAC;oBACN,MAAM,IAAI,CAAC,gBAAgB,EAAE,WAAW,CAAC,OAAO,CAAC,CAAA;oBACjD,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;gBAChC,CAAC;YACH,CAAC;QACH,CAAC;IAAA,CACF;CACF;AAED,MAAM,CAAC,MAAM,wBAAwB,GAAG,GAAkB,EAAE,CAAC;IAC3D,OAAO;QACL,IAAI,EAAE,sBAAsB;QAC5B,KAAK,EAAE;YACL,4BAA4B,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC;gBAC3C,MAAM,OAAO,GAAG,MAAM,GAAG,CAAC,SAAS,CAAC,OAAO,CACzC,uBAAuB,CAAC;oBACtB,YAAY,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE;oBACrC,OAAO,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;wBAC7B,MAAM,UAAU,GAAG,MAAM,iBAAiB,CAAC,MAAM,CAAC,CAAA;wBAClD,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,UAAU,CAAC,CAAA;wBAClD,MAAM,OAAO,CAAC,UAAU,EAAE,CAAA;wBAC1B,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,CAAA;oBAAA,CAC/B;oBACD,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,OAAO;oBAC9B,OAAO,EAAE,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,UAAU,CAAC,IAAI,EAAE;iBAC/C,CAAC,CACH,CAAA;gBACD,GAAG,CAAC,SAAS,CAAC,OAAO,CAAC,aAAa,EAAE,OAAO,CAAC,CAAA;YAAA,CAC9C;SACF;KACF,CAAA;AAAA,CACF,CAAA"}

package/src/runtime/pubsub/redis.ts

@@ -1,106 +0,0 @@
-import EventEmitter, { on } from 'node:events'
-
-import { isAbortError } from '@nmtjs/common'
-import { createFactoryInjectable, provision } from '@nmtjs/core'
-
-import type { RuntimePlugin } from '../plugin.ts'
-import type { Store } from '../types.ts'
-import type { PubSubAdapterEvent, PubSubAdapterType } from './manager.ts'
-import { pubSubAdapter, storeConfig } from '../injectables.ts'
-import { createStoreClient } from '../store/index.ts'
-
-export class RedisPubSubAdapter implements PubSubAdapterType {
-  protected readonly events = new EventEmitter()
-  protected readonly listeners = new Map<string, number>()
-  protected subscriberClient?: Store
-
-  constructor(protected readonly client: Store) {}
-
-  async initialize() {
-    // Create a dedicated subscriber client (Redis requires separate clients for pub/sub)
-    this.subscriberClient = this.client.duplicate()
-
-    // Set up message handler
-    this.subscriberClient.on('message', (channel: string, message: string) => {
-      try {
-        const parsed = JSON.parse(message)
-        this.events.emit(channel, parsed)
-      } catch {}
-    })
-  }
-
-  async dispose() {
-    if (this.subscriberClient) {
-      await this.subscriberClient.quit()
-      this.subscriberClient = undefined
-    }
-  }
-
-  async publish(channel: string, payload: any): Promise<boolean> {
-    try {
-      await this.client.publish(channel, JSON.stringify(payload))
-      return true
-    } catch {
-      return false
-    }
-  }
-
-  async *subscribe(
-    channel: string,
-    signal?: AbortSignal,
-  ): AsyncGenerator<PubSubAdapterEvent> {
-    if (!this.subscriberClient) {
-      throw new Error('RedisPubSubAdapter not initialized')
-    }
-
-    if (!this.listeners.has(channel)) {
-      await this.subscriberClient.subscribe(channel)
-      this.listeners.set(channel, 1)
-    } else {
-      this.listeners.set(channel, this.listeners.get(channel)! + 1)
-    }
-
-    try {
-      signal?.throwIfAborted()
-      for await (const [payload] of on(this.events, channel, { signal })) {
-        yield { channel, payload }
-      }
-    } catch (error: any) {
-      if (isAbortError(error)) throw error
-    } finally {
-      const count = this.listeners.get(channel)
-      if (count !== undefined) {
-        if (count > 1) {
-          this.listeners.set(channel, count - 1)
-        } else {
-          await this.subscriberClient?.unsubscribe(channel)
-          this.listeners.delete(channel)
-        }
-      }
-    }
-  }
-}
-
-export const RedisPubSubAdapterPlugin = (): RuntimePlugin => {
-  return {
-    name: 'pubsub-redis-adapter',
-    hooks: {
-      'lifecycle:beforeInitialize': async (ctx) => {
-        const adapter = await ctx.container.resolve(
-          createFactoryInjectable({
-            dependencies: { config: storeConfig },
-            factory: async ({ config }) => {
-              const connection = await createStoreClient(config)
-              const adapter = new RedisPubSubAdapter(connection)
-              await adapter.initialize()
-              return { adapter, connection }
-            },
-            pick: ({ adapter }) => adapter,
-            dispose: ({ connection }) => connection.quit(),
-          }),
-        )
-        ctx.container.provide(pubSubAdapter, adapter)
-      },
-    },
-  }
-}