rosrust/tcpros/util/
tcpconnection.rs

1use log::error;
2use std::net::{TcpListener, TcpStream};
3use std::thread;
4
5#[derive(Clone, Debug, PartialEq, Eq)]
6pub enum Feedback {
7    AcceptNextStream,
8    StopAccepting,
9}
10
11pub fn iterate<F>(listener: TcpListener, tag: String, handler: F)
12where
13    F: Fn(TcpStream) -> Feedback + Send + 'static,
14{
15    thread::spawn(move || listener_thread(&listener, &tag, handler));
16}
17
18fn listener_thread<F>(connections: &TcpListener, tag: &str, handler: F)
19where
20    F: Fn(TcpStream) -> Feedback + Send + 'static,
21{
22    for stream in connections.incoming() {
23        match stream {
24            Ok(stream) => match handler(stream) {
25                Feedback::AcceptNextStream => {}
26                Feedback::StopAccepting => break,
27            },
28            Err(err) => {
29                error!("TCP connection failed at {}: {}", tag, err);
30            }
31        }
32    }
33}