moon
/
ditto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

subs: allow any object in place of the socket

This commit is contained in:
Alex Gleason 2023-08-26 15:55:16 -05:00
parent 67bba508af
commit c13b7f4af7
No known key found for this signature in database
GPG Key ID: 7211D1F99744FBB7
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/subs.ts
View File

@ -9,7 +9,7 @@ import type { EventData } from '@/types.ts';
* Subscriptions can be added, removed, and matched against events. * Subscriptions can be added, removed, and matched against events.
*/ */
class SubscriptionStore { class SubscriptionStore {
#store = new Map<WebSocket, Map<string, Subscription>>(); #store = new Map<unknown, Map<string, Subscription>>();
/** /**
* Add a subscription to the store, and then iterate over it. * Add a subscription to the store, and then iterate over it.
@ -20,7 +20,7 @@ class SubscriptionStore {
* } * }
* ``` * ```
*/ */
sub<K extends number>(socket: WebSocket, id: string, filters: DittoFilter<K>[]): Subscription<K> { sub<K extends number>(socket: unknown, id: string, filters: DittoFilter<K>[]): Subscription<K> {
let subs = this.#store.get(socket); let subs = this.#store.get(socket);
if (!subs) { if (!subs) {
@ -37,13 +37,13 @@ class SubscriptionStore {
} }
/** Remove a subscription from the store. */ /** Remove a subscription from the store. */
unsub(socket: WebSocket, id: string): void { unsub(socket: unknown, id: string): void {
this.#store.get(socket)?.get(id)?.close(); this.#store.get(socket)?.get(id)?.close();
this.#store.get(socket)?.delete(id); this.#store.get(socket)?.delete(id);
} }
/** Remove an entire socket. */ /** Remove an entire socket. */
close(socket: WebSocket): void { close(socket: unknown): void {
const subs = this.#store.get(socket); const subs = this.#store.get(socket);
if (subs) { if (subs) {