tornado-packages/ethers.js
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Initial signature change to support custom WebSockets for reconnecting providers.

Browse Source
This commit is contained in:
Richard Moore 2022-09-30 20:04:03 -04:00
parent db308812b2
commit 87e1f52702
1 changed files with 27 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
src.ts/providers/provider-websocket.ts
View File

@ -17,24 +17,34 @@ export interface WebSocketLike {
close(code?: number, reason?: string): void; close(code?: number, reason?: string): void;
} }
export type WebSocketCreator = () => WebSocketLike;
export class WebSocketProvider extends SocketProvider { export class WebSocketProvider extends SocketProvider {
#connect: null | WebSocketCreator;
#websocket: null | WebSocketLike; #websocket: null | WebSocketLike;
get websocket(): WebSocketLike { get websocket(): WebSocketLike {
if (this.#websocket == null) { throw new Error("websocket closed"); } if (this.#websocket == null) { throw new Error("websocket closed"); }
return this.#websocket; return this.#websocket;
} }
constructor(url: string | WebSocketLike, network?: Networkish) { constructor(url: string | WebSocketLike | WebSocketCreator, network?: Networkish) {
super(network); super(network);
if (typeof(url) === "string") { if (typeof(url) === "string") {
this.#websocket = new _WebSocket(url); this.#connect = () => { return new _WebSocket(url); };
this.#websocket = this.#connect();
} else if (typeof(url) === "function") {
this.#connect = url;
this.#websocket = url();
} else { } else {
this.#connect = null;
this.#websocket = url; this.#websocket = url;
} }
this.websocket.onopen = async () => { this.websocket.onopen = async () => {
try { try {
await this._start() await this._start()
this.resume();
} catch (error) { } catch (error) {
console.log("failed to start WebsocketProvider", error); console.log("failed to start WebsocketProvider", error);
// @TODO: now what? Attempt reconnect? // @TODO: now what? Attempt reconnect?
@ -44,6 +54,21 @@ export class WebSocketProvider extends SocketProvider {
this.websocket.onmessage = (message: { data: string }) => { this.websocket.onmessage = (message: { data: string }) => {
this._processMessage(message.data); this._processMessage(message.data);
}; };
/*
this.websocket.onclose = (event) => {
// @TODO: What event.code should we reconnect on?
const reconnect = false;
if (reconnect) {
this.pause(true);
if (this.#connect) {
this.#websocket = this.#connect();
this.#websocket.onopen = ...
// @TODO: this requires the super class to rebroadcast; move it there
}
this._reconnect();
}
};
*/
} }
async _write(message: string): Promise<void> { async _write(message: string): Promise<void> {