Jonathan Staab
41 lines
829 B
TypeScript
41 lines
829 B
TypeScript
import {Socket} from "./util/Socket"
|
|
import {EventEmitter} from 'events'
|
|
|
|
export class Pool extends EventEmitter {
|
|
data: Map<string, Socket>
|
|
constructor() {
|
|
super()
|
|
|
|
this.data = new Map()
|
|
}
|
|
has(url) {
|
|
return this.data.has(url)
|
|
}
|
|
get(url, {autoConnect = true} = {}) {
|
|
if (!this.data.has(url) && autoConnect) {
|
|
const socket = new Socket(url)
|
|
|
|
this.data.set(url, socket)
|
|
this.emit('init', {url})
|
|
|
|
socket.on('open', () => this.emit('open', {url}))
|
|
socket.on('close', () => this.emit('close', {url}))
|
|
}
|
|
|
|
return this.data.get(url)
|
|
}
|
|
remove(url) {
|
|
const socket = this.data.get(url)
|
|
|
|
if (socket) {
|
|
socket.removeAllListeners()
|
|
this.data.delete(url)
|
|
}
|
|
}
|
|
clear() {
|
|
for (const url of this.data.keys()) {
|
|
this.remove(url)
|
|
}
|
|
}
|
|
}
|