1use socket2::SockRef;
2use tokio::net::TcpStream;
3use vector_config::configurable_component;
4
5#[configurable_component]
7#[derive(Clone, Copy, Debug, PartialEq, Eq)]
8#[serde(deny_unknown_fields)]
9#[configurable(metadata(docs::human_name = "Wait Time"))]
10pub struct TcpKeepaliveConfig {
11 #[configurable(metadata(docs::type_unit = "seconds"))]
13 pub time_secs: Option<u64>,
14}
15
16pub(crate) fn set_keepalive(
19 socket: &TcpStream,
20 params: &socket2::TcpKeepalive,
21) -> std::io::Result<()> {
22 SockRef::from(socket).set_tcp_keepalive(params)
23}
24
25pub(crate) fn set_receive_buffer_size(socket: &TcpStream, size: usize) -> std::io::Result<()> {
28 SockRef::from(socket).set_recv_buffer_size(size)
29}
30
31pub(crate) fn set_send_buffer_size(socket: &TcpStream, size: usize) -> std::io::Result<()> {
34 SockRef::from(socket).set_send_buffer_size(size)
35}