Jon Staab
93 lines
1.7 KiB
TypeScript
93 lines
1.7 KiB
TypeScript
import {remove} from "@welshman/lib"
|
|
import {normalizeRelayUrl} from "@welshman/util"
|
|
import {Socket} from "./socket.js"
|
|
import {defaultSocketPolicies} from "./policy.js"
|
|
|
|
export const makeSocket = (url: string, policies = defaultSocketPolicies) => {
|
|
const socket = new Socket(url)
|
|
|
|
for (const applyPolicy of policies) {
|
|
applyPolicy(socket)
|
|
}
|
|
|
|
return socket
|
|
}
|
|
|
|
export type PoolSubscription = (socket: Socket) => void
|
|
|
|
export type PoolOptions = {
|
|
makeSocket?: (url: string) => Socket
|
|
}
|
|
|
|
export let poolSingleton: Pool
|
|
|
|
export class Pool {
|
|
_data = new Map<string, Socket>()
|
|
_subs: PoolSubscription[] = []
|
|
|
|
static get() {
|
|
if (!poolSingleton) {
|
|
poolSingleton = new Pool()
|
|
}
|
|
|
|
return poolSingleton
|
|
}
|
|
|
|
constructor(readonly options: PoolOptions = {}) {}
|
|
|
|
has(url: string) {
|
|
return this._data.has(normalizeRelayUrl(url))
|
|
}
|
|
|
|
makeSocket(url: string) {
|
|
if (this.options.makeSocket) {
|
|
return this.options.makeSocket(url)
|
|
}
|
|
|
|
return makeSocket(url)
|
|
}
|
|
|
|
get(_url: string): Socket {
|
|
const url = normalizeRelayUrl(_url)
|
|
const socket = this._data.get(url)
|
|
|
|
if (socket) {
|
|
return socket
|
|
}
|
|
|
|
const newSocket = this.makeSocket(url)
|
|
|
|
this._data.set(url, newSocket)
|
|
|
|
for (const cb of this._subs) {
|
|
cb(newSocket)
|
|
}
|
|
|
|
return newSocket
|
|
}
|
|
|
|
subscribe(cb: PoolSubscription) {
|
|
this._subs.push(cb)
|
|
|
|
return () => {
|
|
this._subs = remove(cb, this._subs)
|
|
}
|
|
}
|
|
|
|
remove(url: string) {
|
|
const socket = this._data.get(url)
|
|
|
|
if (socket) {
|
|
socket.cleanup()
|
|
|
|
this._data.delete(url)
|
|
}
|
|
}
|
|
|
|
clear() {
|
|
for (const url of this._data.keys()) {
|
|
this.remove(url)
|
|
}
|
|
}
|
|
}
|