52 lines
No EOL
1.5 KiB
TypeScript
52 lines
No EOL
1.5 KiB
TypeScript
type WsOpts = {
|
|
url: string;
|
|
reconnectInterval?: number;
|
|
onOpen?: () => void;
|
|
onMessage: (message: string) => void;
|
|
onError?: (error: Event) => void;
|
|
onClose?: () => void;
|
|
}
|
|
|
|
export function createWebSocketClient(opts: WsOpts) {
|
|
let socket: WebSocket | null = null;
|
|
const connect = (tries: number) => {
|
|
|
|
if(tries > 50) {
|
|
console.error('failed to connect to websocket after 50 tries, please reload the page');
|
|
return;
|
|
}
|
|
|
|
socket = new WebSocket(opts.url);
|
|
// Handle connection open
|
|
socket.onopen = () => {
|
|
};
|
|
// Handle incoming messages
|
|
socket.onmessage = (event) => {
|
|
opts.onMessage(event.data)
|
|
};
|
|
// Handle connection errors
|
|
socket.onerror = (error) => {
|
|
try {
|
|
socket?.close()
|
|
} catch(ex) {
|
|
// noop
|
|
}
|
|
let interval = tries * (opts.reconnectInterval || 100);
|
|
setTimeout(() => connect(tries + 1), interval);
|
|
};
|
|
// Handle connection close and attempt reconnection
|
|
socket.onclose = () => {
|
|
let interval = tries * (opts.reconnectInterval || 100);
|
|
setTimeout(() => connect(tries + 1), interval);
|
|
};
|
|
};
|
|
connect(1);
|
|
const sendMessage = (message: string) => {
|
|
if (socket && socket.readyState === WebSocket.OPEN) {
|
|
socket.send(message);
|
|
} else {
|
|
setTimeout(() => sendMessage(message), 100);
|
|
}
|
|
};
|
|
return { sendMessage };
|
|
} |