Refactor heartbeat task to support WebAssembly

bitfl0wer · Nov 20, 2023 · f0a4fe6 · f0a4fe6
1 parent 325c3d3
commit f0a4fe6
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/gateway/heartbeat.rs b/src/gateway/heartbeat.rs
@@ -33,9 +33,14 @@ impl HeartbeatHandler {
         let (send, receive) = tokio::sync::mpsc::channel(32);
         let kill_receive = kill_rc.resubscribe();
 
+        #[cfg(not(target_arch = "wasm32"))]
         let handle: JoinHandle<()> = task::spawn(async move {
             Self::heartbeat_task(websocket_tx, heartbeat_interval, receive, kill_receive).await;
         });
+        #[cfg(target_arch = "wasm32")]
+        let handle: JoinHandle<()> = task::spawn_local(move || {
+            Self::heartbeat_task(websocket_tx, heartbeat_interval, receive, kill_receive);
+        });
 
         Self {
             heartbeat_interval,