rustdesk/rustdesk-server
1
0 Fork 0
Code Releases 1 Activity

style: optimize online query

Browse Source
This commit is contained in:
lejianwen committed 2025-01-22 19:44:16 +08:00
1 parent 9ee99f1412
commit d095ac8cec
2 files changed
+20 -26
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/rendezvous_server.rs
+18 -25
View File
@@ -624,19 +624,7 @@ impl RendezvousServer {
}
}
Some(rendezvous_message::Union::OnlineRequest(or)) => {
let mut states = BytesMut::zeroed((or.peers.len() + 7) / 8);
for (i, peer_id) in or.peers.iter().enumerate() {
if let Some(peer) = self.pm.get_in_memory(peer_id).await {
let elapsed = peer.read().await.last_reg_time.elapsed().as_millis() as i32;
// bytes index from left to right
let states_idx = i / 8;
let bit_idx = 7 - i % 8;
if elapsed < REG_TIMEOUT {
states[states_idx] |= 0x01 << bit_idx;
}
}
}
let mut states = self.peers_online_state(or.peers).await;
let mut msg_out = RendezvousMessage::new();
msg_out.set_online_response(OnlineResponse {
states: states.into(),
@@ -650,6 +638,22 @@ impl RendezvousServer {
false
}
async fn peers_online_state(&mut self, peers: Vec<String>) -> BytesMut {
let mut states = BytesMut::zeroed((peers.len() + 7) / 8);
for (i, peer_id) in peers.iter().enumerate() {
if let Some(peer) = self.pm.get_in_memory(peer_id).await {
let elapsed = peer.read().await.last_reg_time.elapsed().as_millis() as i32;
// bytes index from left to right
let states_idx = i / 8;
let bit_idx = 7 - i % 8;
if elapsed < REG_TIMEOUT {
states[states_idx] |= 0x01 << bit_idx;
}
}
}
states
}
#[inline]
async fn update_addr(
&mut self,
@@ -882,18 +886,7 @@ impl RendezvousServer {
stream: &mut FramedStream,
peers: Vec<String>,
) -> ResultType<()> {
let mut states = BytesMut::zeroed((peers.len() + 7) / 8);
for (i, peer_id) in peers.iter().enumerate() {
if let Some(peer) = self.pm.get_in_memory(peer_id).await {
let elapsed = peer.read().await.last_reg_time.elapsed().as_millis() as i32;
// bytes index from left to right
let states_idx = i / 8;
let bit_idx = 7 - i % 8;
if elapsed < REG_TIMEOUT {
states[states_idx] |= 0x01 << bit_idx;
}
}
}
let mut states = self.peers_online_state(peers).await;
let mut msg_out = RendezvousMessage::new();
msg_out.set_online_response(OnlineResponse {