artemist/vapore
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Use correct time for heartbeats

Browse source
This commit is contained in:
Artemis Tosini 2024-08-28 02:17:58 +00:00
parent 07d4a5d9b4
commit a42b72d963
Signed by: artemist
GPG key ID: EE5227935FE3FF18
2 changed files with 21 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
daemon/src/connection.rs
View file

@ -182,7 +182,6 @@ impl Context {
let message = maybe_message.ok_or_eyre("Socket was closed while trying to recieve")?; let message = maybe_message.ok_or_eyre("Socket was closed while trying to recieve")?;
if let Err(err) = self.as_mut().handle_receive(message) { if let Err(err) = self.as_mut().handle_receive(message) {
log::warn!("Got error while processing message: {:?}", err); log::warn!("Got error while processing message: {:?}", err);
} }
} }
@ -245,14 +244,17 @@ impl CMSession {
inner: inner_wrapped, inner: inner_wrapped,
}; };
let session_cloned = session.clone();
tokio::spawn(async move { session_cloned.send_heartbeat_task().await });
Ok(session) Ok(session)
} }
async fn send_heartbeat_task(mut self) -> eyre::Result<()> { pub fn begin_heartbeat(&self, interval: u32) {
let mut interval = tokio::time::interval(time::Duration::from_secs(5)); log::debug!("Starting heartbeats every {} seconds", interval);
let cloned = self.clone();
tokio::spawn(async move { cloned.send_heartbeat_task(interval).await });
}
async fn send_heartbeat_task(mut self, interval_secs: u32) -> eyre::Result<()> {
let mut interval = tokio::time::interval(time::Duration::from_secs(interval_secs as u64));
loop { loop {
interval.tick().await; interval.tick().await;
self.send_notification(EMsg::k_EMsgClientHeartBeat, CMsgClientHeartBeat::default())? self.send_notification(EMsg::k_EMsgClientHeartBeat, CMsgClientHeartBeat::default())?

13
daemon/src/main.rs
View file

@ -163,6 +163,19 @@ pub async fn main() -> eyre::Result<()> {
log::debug!("Got logon response: {:#x?}", response); log::debug!("Got logon response: {:#x?}", response);
if response.body.eresult != Some(0x01) {
eyre::bail!("Login failed");
};
session.set_steam_id(response.header.steamid());
session.set_client_session_id(response.header.client_sessionid());
if let Some(heartbeat_seconds) = response.body.heartbeat_seconds {
if heartbeat_seconds >= 5 {
session.begin_heartbeat(heartbeat_seconds as u32);
}
}
Ok(()) Ok(())
} }