MeexReay/bRAC
1
0
Fork 0
mirror of https://github.com/MeexReay/bRAC.git synced 2025-05-06 13:38:04 +03:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix recv_loop perfomance

Browse Source
This commit is contained in:
MeexReay 2025-02-10 14:52:59 +03:00
parent ce63efeba6
commit f06e67db95
2 changed files with 13 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/main.rs
View File

@ -149,7 +149,7 @@ fn main() {
}
if args.read_messages {
print!("{}", read_messages(&config.host, config.max_messages).expect("Error reading messages").0.join("\n"));
print!("{}", read_messages(&config.host, config.max_messages, 0).ok().flatten().expect("Error reading messages").0.join("\n"));
return;
}

16
src/rac.rs
View File

@ -33,7 +33,7 @@ fn skip_null(stream: &mut TcpStream) -> Result<Vec<u8>, Box<dyn Error>> {
}
}
pub fn read_messages(host: &str, max_messages: usize) -> Result<(Vec<String>, usize), Box<dyn Error>> {
pub fn read_messages(host: &str, max_messages: usize, last_size: usize) -> Result<Option<(Vec<String>, usize)>, Box<dyn Error>> {
let mut stream = TcpStream::connect(host)?;
stream.write_all(&[0x00])?;
@ -56,6 +56,10 @@ pub fn read_messages(host: &str, max_messages: usize) -> Result<(Vec<String>, us
.parse()?
};
if last_size == packet_size {
return Ok(None);
}
stream.write_all(&[0x01])?;
let packet_data = {
@ -75,20 +79,18 @@ pub fn read_messages(host: &str, max_messages: usize) -> Result<(Vec<String>, us
.map(|o| o.to_string())
.collect();
Ok((lines, packet_size))
Ok(Some((lines, packet_size)))
}
fn recv_loop(config: Arc<Config>, host: &str, cache: Arc<(RwLock<Vec<String>>, AtomicUsize)>, input: Arc<RwLock<String>>, disable_formatting: bool) -> Result<(), Box<dyn Error>> {
while let Ok(data) = read_messages(host, config.max_messages) {
if data.1 == cache.1.load(Ordering::SeqCst) {
continue
}
while let Ok(data) = read_messages(host, config.max_messages, cache.1.load(Ordering::SeqCst)) {
if let Some(data) = data {
*cache.0.write().unwrap() = data.0.clone();
cache.1.store(data.1, Ordering::SeqCst);
print_console(config.clone(), data.0, &input.read().unwrap(), disable_formatting)?;
thread::sleep(Duration::from_millis(config.update_time as u64));
}
}
Ok(())
}