fix(devtools-shared): delay unestablished ws connections from closing

refinedev · May 8, 2024 · 9d9f375 · 9d9f375
1 parent 97f245f
commit 9d9f375
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/packages/devtools-shared/src/context.tsx b/packages/devtools-shared/src/context.tsx
@@ -69,7 +69,16 @@ export const DevToolsContextProvider: React.FC<Props> = ({
     return () => {
       unsubscribe();
 
-      wsInstance.close(1000, window.location.origin);
+      // In strict mode, the WebSocket instance might not be connected yet
+      // so we need to wait for it to connect before closing it
+      // otherwise it will log an unnecessary error in the console
+      if (wsInstance.readyState === WebSocket.CONNECTING) {
+        wsInstance.addEventListener("open", () => {
+          wsInstance.close(1000, window.location.origin);
+        });
+      } else {
+        wsInstance.close(1000, window.location.origin);
+      }
     };
   }, []);