IntenseWebs/LibreQoS
3
0
Fork 0
mirror of https://github.com/LibreQoE/LibreQoS.git synced 2025-02-25 18:55:32 -06:00
Code Issues Packages Projects Releases Wiki Activity

Simplify the submission socket system.

Browse Source
This commit is contained in:
Herbert Wolverson
2023-08-18 20:26:36 +00:00
parent c51f00eb85
commit 4da8f89965
1 changed files with 7 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
src/rust/lts_client/src/submission_queue/comm_channel/mod.rs
View File

@@ -12,38 +12,21 @@ pub(crate) enum SenderChannelMessage {
}
pub(crate) async fn start_communication_channel(mut rx: Receiver<SenderChannelMessage>) {
let mut connected = false;
let mut stream: Option<TcpStream> = None;
// let mut connected = false;
// let mut stream: Option<TcpStream> = None;
loop {
match rx.try_recv() {
Ok(SenderChannelMessage::QueueReady) => {
// If not connected, see if we are allowed to connect and get a target
if !connected || stream.is_none() {
log::info!("Establishing LTS TCP channel.");
stream = connect_if_permitted().await;
if stream.is_some() {
connected = true;
}
}
let mut stream = connect_if_permitted().await;
// If we're still not connected, skip - otherwise, send the
// queued data
if let Some(tcpstream) = &mut stream {
if connected && tcpstream.writable().await.is_ok() {
// Send the data
let all_good = send_queue(tcpstream).await;
if all_good.is_err() {
log::error!("Stream fail during send. Will re-send");
connected = false;
stream = None;
}
} else {
stream = None;
connected = false;
// Send the data
let all_good = send_queue(tcpstream).await;
if all_good.is_err() {
log::error!("Stream fail during send. Will re-send");
}
} else {
connected = false;
stream = None;
}
}
Ok(SenderChannelMessage::Quit) => {