Jonathan Staab
50 lines
1.1 KiB
TypeScript
50 lines
1.1 KiB
TypeScript
import {Connection} from "./Connection"
|
|
import {Emitter} from './util/Emitter'
|
|
|
|
export class Pool extends Emitter {
|
|
data: Map<string, Connection>
|
|
constructor() {
|
|
super()
|
|
|
|
this.data = new Map()
|
|
}
|
|
has(url: string) {
|
|
return this.data.has(url)
|
|
}
|
|
get(url: string, {autoConnect = true, reconnectAfter = 3000} = {}): Connection {
|
|
let connection = this.data.get(url)
|
|
|
|
if (autoConnect) {
|
|
if (!connection) {
|
|
connection = new Connection(url)
|
|
|
|
this.data.set(url, connection)
|
|
this.emit('init', connection)
|
|
|
|
connection.on('open', () => this.emit('open', connection))
|
|
connection.on('close', () => this.emit('close', connection))
|
|
}
|
|
|
|
connection.ensureConnected({
|
|
shouldReconnect: connection.meta.lastClose < Date.now() - reconnectAfter,
|
|
})
|
|
}
|
|
|
|
return connection!
|
|
}
|
|
remove(url: string) {
|
|
const connection = this.data.get(url)
|
|
|
|
if (connection) {
|
|
connection.destroy()
|
|
|
|
this.data.delete(url)
|
|
}
|
|
}
|
|
clear() {
|
|
for (const url of this.data.keys()) {
|
|
this.remove(url)
|
|
}
|
|
}
|
|
}
|