74 lines
1.6 KiB
JavaScript
74 lines
1.6 KiB
JavaScript
"use strict"
|
|
|
|
import Time from '../common/Time.js'
|
|
|
|
const CODE_CUSTOM_DISCONNECT = 4000
|
|
|
|
/**
|
|
* Wrapper around ws that
|
|
* - buffers 'send' until a connection is available
|
|
* - automatically tries to reconnect on close
|
|
*/
|
|
export default class WsWrapper {
|
|
// actual ws handle
|
|
handle = null
|
|
|
|
// timeout for automatic reconnect
|
|
reconnectTimeout = null
|
|
|
|
// buffer for 'send'
|
|
sendBuffer = []
|
|
|
|
constructor(addr, protocols) {
|
|
this.addr = addr
|
|
this.protocols = protocols
|
|
|
|
this.onopen = () => {}
|
|
this.onclose = () => {}
|
|
this.onmessage = () => {}
|
|
}
|
|
|
|
send (txt) {
|
|
if (this.handle) {
|
|
this.handle.send(txt)
|
|
} else {
|
|
this.sendBuffer.push(txt)
|
|
}
|
|
}
|
|
|
|
connect() {
|
|
let ws = new WebSocket(this.addr, this.protocols)
|
|
ws.onopen = (e) => {
|
|
if (this.reconnectTimeout) {
|
|
clearTimeout(this.reconnectTimeout)
|
|
}
|
|
this.handle = ws
|
|
// should have a queue worker
|
|
while (this.sendBuffer.length > 0) {
|
|
this.handle.send(this.sendBuffer.shift())
|
|
}
|
|
this.onopen(e)
|
|
}
|
|
ws.onmessage = (e) => {
|
|
this.onmessage(e)
|
|
}
|
|
ws.onerror = (e) => {
|
|
this.handle = null
|
|
this.reconnectTimeout = setTimeout(() => { this.connect() }, 1 * Time.SEC)
|
|
this.onclose(e)
|
|
}
|
|
ws.onclose = (e) => {
|
|
this.handle = null
|
|
if (e.code !== CODE_CUSTOM_DISCONNECT) {
|
|
this.reconnectTimeout = setTimeout(() => { this.connect() }, 1 * Time.SEC)
|
|
}
|
|
this.onclose(e)
|
|
}
|
|
}
|
|
|
|
disconnect() {
|
|
if (this.handle) {
|
|
this.handle.close(CODE_CUSTOM_DISCONNECT)
|
|
}
|
|
}
|
|
}
|