kozabrada123/chorus
kozabrada123
/
chorus
mirror of https://github.com/polyphony-chat/chorus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Remove 'handle', add wasm friendly task spawning

This commit is contained in:
bitfl0wer 2023-11-22 14:23:33 +01:00
parent 93c12fff63
commit f8979b7feb
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/gateway/heartbeat.rs
View File

@ -4,7 +4,8 @@ use std::time::{self, Duration, Instant};
use tokio::sync::mpsc::{Receiver, Sender};
use safina_timer::sleep_until;
use tokio::task::{self, JoinHandle};
#[cfg(not(target_arch = "wasm32"))]
use tokio::task;
use super::*;
use crate::types;
@ -20,8 +21,6 @@ pub(super) struct HeartbeatHandler {
pub heartbeat_interval: Duration,
/// The send channel for the heartbeat thread
pub send: Sender<HeartbeatThreadCommunication>,
/// The handle of the thread
handle: JoinHandle<()>,
}
impl HeartbeatHandler {
@ -33,15 +32,18 @@ impl HeartbeatHandler {
let (send, receive) = tokio::sync::mpsc::channel(32);
let kill_receive = kill_rc.resubscribe();
// FIXME: Doesn't work in WASM
let handle: JoinHandle<()> = task::spawn(async move {
#[cfg(not(target_arch = "wasm32"))]
task::spawn(async move {
Self::heartbeat_task(websocket_tx, heartbeat_interval, receive, kill_receive).await;
});
#[cfg(target_arch = "wasm32")]
wasm_bindgen_futures::spawn_local(async move {
Self::heartbeat_task(websocket_tx, heartbeat_interval, receive, kill_receive).await;
});
Self {
heartbeat_interval,
send,
handle,
}
}