monoio 0.2.4

A thread per core runtime based on iouring.
Documentation 
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

use std::{cell::Cell, io, os::fd::AsRawFd};

#[cfg(feature = "unstable")]
#[thread_local]
pub(crate) static TFO_CONNECT_AVAILABLE: Cell<bool> = Cell::new(true);

#[cfg(not(feature = "unstable"))]
thread_local! {
    pub(crate) static TFO_CONNECT_AVAILABLE: Cell<bool> = const { Cell::new(true) };
}

/// Call before listen.
pub(crate) fn set_tcp_fastopen<S: AsRawFd>(fd: &S, fast_open: i32) -> io::Result<()> {
    crate::syscall!(setsockopt(
        fd.as_raw_fd(),
        libc::SOL_TCP,
        libc::TCP_FASTOPEN,
        &fast_open as *const _ as *const libc::c_void,
        std::mem::size_of::<libc::c_int>() as libc::socklen_t
    ))?;
    Ok(())
}

/// Call before connect.
/// Linux 4.1+ only.
pub(crate) fn set_tcp_fastopen_connect<S: AsRawFd>(fd: &S) -> io::Result<()> {
    const ENABLED: libc::c_int = 0x1;

    crate::syscall!(setsockopt(
        fd.as_raw_fd(),
        libc::SOL_TCP,
        libc::TCP_FASTOPEN_CONNECT,
        &ENABLED as *const _ as *const libc::c_void,
        std::mem::size_of::<libc::c_int>() as libc::socklen_t
    ))?;
    Ok(())
}

pub(crate) fn try_set_tcp_fastopen_connect<S: AsRawFd>(fd: &S) {
    if !TFO_CONNECT_AVAILABLE.get() {
        return;
    }
    match set_tcp_fastopen_connect(fd) {
        Ok(_) => (),
        Err(e) if e.raw_os_error() == Some(libc::ENOPROTOOPT) => {
            TFO_CONNECT_AVAILABLE.set(false);
        }
        Err(_e) => {
            #[cfg(all(debug_assertions, feature = "debug"))]
            tracing::warn!("set_tcp_fastopen_connect failed: {}", _e);
        }
    }
}