@toa.io/extensions.realtime 0.0.0 → 0.22.0

package/components/streams/source/create.ts

@@ -1,14 +1,31 @@
 import { type Readable } from 'node:stream'
+import { type Operation } from '@toa.io/types'
 import { type Context } from './types'
-import { addStream } from './lib/streams'
+import { Stream } from './lib/stream'
 
-export async function effect (input: Input, context: Context): Promise<Readable> {
-  const key = input.key
+export class Effect implements Operation {
+  private readonly streams = new Map<string, Stream>()
 
-  if (!(key in context.state))
-    addStream(key, context.state)
+  public mount (context: Context): void {
+    context.state.streams = this.streams
+  }
 
-  return context.state[key].fork()
+  public async execute (input: Input, context: Context): Promise<Readable> {
+    const key = input.key
+
+    if (!this.streams.has(key))
+      this.createStream(key)
+
+    return (this.streams.get(key) as Stream).fork()
+  }
+
+  private createStream (key: string): void {
+    const stream = new Stream()
+
+    this.streams.set(key, stream)
+
+    stream.once('close', () => this.streams.delete(key))
+  }
 }
 
 interface Input {

package/components/streams/source/lib/{streams.ts → stream.ts}

@@ -1,13 +1,5 @@
 import { PassThrough, Readable } from 'node:stream'
 
-export function addStream (key: string, map: Record<string, Stream>): void {
-  const stream = new Stream()
-
-  map[key] = stream
-
-  stream.once('close', () => delete map[key])
-}
-
 export class Stream extends Readable {
   private forks: number = 0
 

package/components/streams/source/push.ts

@@ -1,5 +1,5 @@
 import { type Context, type PushInput } from './types'
 
 export async function effect ({ key, event, data }: PushInput, context: Context): Promise<void> {
-  context.state[key]?.push({ event, data })
+  context.state.streams.get(key)?.push({ event, data })
 }

package/components/streams/source/types.ts

@@ -1,7 +1,9 @@
-import { type Stream } from './lib/streams'
+import { type Stream } from './lib/stream'
 
 export interface Context {
-  state: Record<string, Stream>
+  state: {
+    streams: Map<string, Stream>
+  }
 }
 
 export interface PushInput {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@toa.io/extensions.realtime",
-  "version": "0.0.0",
+  "version": "0.22.0",
   "description": "Toa Realtime",
   "author": "temich <tema.gurtovoy@gmail.com>",
   "homepage": "https://github.com/toa-io/toa#readme",
@@ -24,8 +24,8 @@
     "testEnvironment": "node"
   },
   "scripts": {
-    "transpile": "npx tsc && npx tsc -p ./components/streams",
+    "transpile": "rm -rf transpiled && npx tsc && rm -rf ./components/streams/operations && npx tsc -p ./components/streams",
     "features": "npx cucumber-js"
   },
-  "gitHead": "28fc4b45c224c3683acaaf0e4abd1eb04e07b408"
+  "gitHead": "c463348c8eb54a43f7755ec8aeba9acafb56d53b"
 }