feat(useWebSocket): allow different heartbeat response message (#3950)

lavolpecheprogramma · antfu · web-flow · commit adbe017803c6 · 2024-07-17T13:52:24.000Z
Co-authored-by: Anthony Fu &lt;github@antfu.me&gt;
Co-authored-by: Anthony Fu &lt;anthonyfu117@hotmail.com&gt;
diff --git a/packages/core/useWebSocket/index.ts b/packages/core/useWebSocket/index.ts
@@ -27,6 +27,11 @@ export interface UseWebSocketOptions {
      */
     message?: string | ArrayBuffer | Blob
 
+    /**
+     * Response message for the heartbeat, if undefined the message will be used
+     */
+    responseMessage?: string | ArrayBuffer | Blob
+
     /**
      * Interval, in milliseconds
      *
@@ -251,8 +256,9 @@ export function useWebSocket<Data = any>(
         resetHeartbeat()
         const {
           message = DEFAULT_PING_MESSAGE,
+          responseMessage = message,
         } = resolveNestedOptions(options.heartbeat)
-        if (e.data === message)
+        if (e.data === responseMessage)
           return
       }
 
