add source to SocketServer sockets (#40)

Effect-TS · Dec 24, 2023 · 04b8b17 · 04b8b17
1 parent e97aef1
commit 04b8b17
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 2 deletions.
diff --git a/.changeset/serious-sheep-lick.md b/.changeset/serious-sheep-lick.md
@@ -0,0 +1,5 @@
+---
+"@effect/experimental": patch
+---
+
+add source to SocketServer sockets
diff --git a/docs/modules/Socket.ts.md b/docs/modules/Socket.ts.md
@@ -171,6 +171,7 @@ export interface Socket {
   readonly run: Effect.Effect<never, SocketError, void>
   readonly writer: Effect.Effect<Scope.Scope, never, (chunk: Uint8Array) => Effect.Effect<never, never, void>>
   readonly messages: Queue.Dequeue<Uint8Array>
+  readonly source?: unknown
 }
 ```
 

diff --git a/src/Socket.ts b/src/Socket.ts
@@ -43,6 +43,7 @@ export interface Socket {
   readonly run: Effect.Effect<never, SocketError, void>
   readonly writer: Effect.Effect<Scope.Scope, never, (chunk: Uint8Array) => Effect.Effect<never, never, void>>
   readonly messages: Queue.Dequeue<Uint8Array>
+  readonly source?: unknown
 }
 
 /**

diff --git a/src/SocketServer/Node.ts b/src/SocketServer/Node.ts
@@ -62,7 +62,10 @@ export const make = (
                 })
             )
           ),
-          Effect.flatMap((socket) => Queue.offer(queue, socket)),
+          Effect.flatMap((socket) => {
+            ;(socket as any).source = conn
+            return Queue.offer(queue, socket)
+          }),
           Effect.runFork
         )
       })
@@ -155,7 +158,10 @@ export const makeWebSocket = (
                 })
             )
           ),
-          Effect.flatMap((socket) => Queue.offer(queue, socket)),
+          Effect.flatMap((socket) => {
+            ;(socket as any).source = conn
+            return Queue.offer(queue, socket)
+          }),
           Effect.runFork
         )
       })