chore(web): refactor websocket state machine

1 files changed, 9 insertions, 1 deletions

diff --git a/language-web/src/main/js/editor/PendingRequest.ts b/language-web/src/main/js/editor/PendingRequest.ts
index 784f06ec..49d4c36c 100644
--- a/language-web/src/main/js/editor/PendingRequest.ts
+++ b/language-web/src/main/js/editor/PendingRequest.ts
@@ -9,16 +9,24 @@ export class PendingRequest {
 
   private readonly rejectCallback: (reason?: unknown) => void;
 
+  private readonly timeoutCallback: () => void;
+
   private resolved = false;
 
   private timeoutId: NodeJS.Timeout;
 
-  constructor(resolve: (value: unknown) => void, reject: (reason?: unknown) => void) {
+  constructor(
+    resolve: (value: unknown) => void,
+    reject: (reason?: unknown) => void,
+    timeout: () => void,
+  ) {
     this.resolveCallback = resolve;
     this.rejectCallback = reject;
+    this.timeoutCallback = timeout;
     this.timeoutId = setTimeout(() => {
       if (!this.resolved) {
         this.reject(new Error('Request timed out'));
+        this.timeoutCallback();
       }
     }, REQUEST_TIMEOUT_MS);
   }