Nickiel/church_controller_rust
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

propogated all stop flag for each client

Browse source
This commit is contained in:
Nickiel12 2021-12-18 17:44:56 -08:00
parent 6917e51400
commit 1dc31de486
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/modules/socket_handler.rs
View file

@ -22,7 +22,7 @@ impl Socket {
listener.set_nonblocking(true).unwrap();
while thread_stop_flag.get() {
for (stream, _addr) in listener.accept() {
Socket::handle_client(stream, messenger_tx.clone());
Socket::handle_client(stream, messenger_tx.clone(), thread_stop_flag.clone());
}
thread::sleep(Duration::from_millis(100));
}
@ -31,14 +31,14 @@ impl Socket {
(tx, handle)
}
pub fn handle_client(mut stream: TcpStream, update_tx: Sender<String>) {
pub fn handle_client(mut stream: TcpStream, update_tx: Sender<String>, program_shutdown_flag: sync_flag::SyncFlagRx) {
let mut buffer = [0; 1024];
loop {
let read_size = stream.read(&mut buffer).unwrap();
//Tcp is supposed to have a 0 byte read if closed by client
if read_size == 0 {
if read_size == 0 || !program_shutdown_flag.get() {
stream.shutdown(Shutdown::Both).unwrap();
thread::sleep(Duration::from_millis(75));
break;