sozu-client 0.4.2-rc.1

This library provides a client to interact with Sōzu.
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

//! # Socket module
//!
//! This module provides helpers to interact with Unix sockets

use std::{io, path::PathBuf, time::Duration};

use mio::net::UnixStream;
use tokio::time::sleep;
use tracing::{debug, trace};

// -----------------------------------------------------------------------------
// Error

#[derive(thiserror::Error, Debug)]
pub enum Error {
    #[error("failed to wait for the socket to be ready, {0}")]
    Wait(io::Error),
}

// -----------------------------------------------------------------------------
// helpers

#[tracing::instrument]
pub async fn connect(path: &PathBuf) -> Result<UnixStream, Error> {
    loop {
        return match UnixStream::connect(path) {
            Ok(stream) => {
                debug!(
                    path = path.display().to_string(),
                    "Successfully connected to socket"
                );
                Ok(stream)
            }
            Err(err)
                if matches!(
                    err.kind(),
                    io::ErrorKind::NotFound | io::ErrorKind::ConnectionRefused
                ) =>
            {
                trace!(
                    path = path.display().to_string(),
                    "Try to connect to socket"
                );
                sleep(Duration::from_millis(100)).await;
                continue;
            }
            Err(err) => Err(Error::Wait(err)),
        };
    }
}