rosrust/tcpros/util/
tcpconnection.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
use log::error;
use std::net::{TcpListener, TcpStream};
use std::thread;

#[derive(Clone, Debug, PartialEq, Eq)]
pub enum Feedback {
    AcceptNextStream,
    StopAccepting,
}

pub fn iterate<F>(listener: TcpListener, tag: String, handler: F)
where
    F: Fn(TcpStream) -> Feedback + Send + 'static,
{
    thread::spawn(move || listener_thread(&listener, &tag, handler));
}

fn listener_thread<F>(connections: &TcpListener, tag: &str, handler: F)
where
    F: Fn(TcpStream) -> Feedback + Send + 'static,
{
    for stream in connections.incoming() {
        match stream {
            Ok(stream) => match handler(stream) {
                Feedback::AcceptNextStream => {}
                Feedback::StopAccepting => break,
            },
            Err(err) => {
                error!("TCP connection failed at {}: {}", tag, err);
            }
        }
    }
}