kubelet 0.7.0

A Kubernetes kubelet implementation in Rust
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

// This is heavily adapted from https://github.com/hyperium/tonic/blob/f1275b611e38ec5fe992b2f10552bf95e8448b17/examples/src/uds/client.rs

#[cfg_attr(target_family = "windows", path = "windows/mod.rs")]
#[cfg(target_family = "windows")]
pub mod windows;

use std::path::Path;

#[cfg(target_family = "windows")]
use crate::mio_uds_windows::UnixStream;
#[cfg(target_family = "unix")]
use tokio::net::UnixStream;
use tonic::transport::{Channel, Endpoint, Uri};
use tower::service_fn;

/// Returns a new UNIX socket channel suitable for use with tonic generated gRPC clients. Instead of
/// using `YourClient::connect`, you'll need to pass the returned channel to `YourClient::new`
pub async fn socket_channel<P: AsRef<Path>>(path: P) -> Result<Channel, tonic::transport::Error> {
    // Get an owned copy of the path so we can use it in the FnMut closure
    let p = path.as_ref().to_owned();

    // This is a dummy http endpoint needed for the Endpoint constructors, it is ignored by the
    // connector
    #[cfg(target_family = "unix")]
    let res = Endpoint::from_static("http://[::]:50051")
        .connect_with_connector(service_fn(move |_: Uri| {
            // Connect to a Uds socket
            UnixStream::connect(p.clone())
        }))
        .await;

    #[cfg(target_family = "windows")]
    let res = Endpoint::from_static("http://[::]:50051")
        .connect_with_connector(service_fn(move |_: Uri| {
            // Need to copy the path here again so this can be FnMut
            let path_copy = p.to_owned();
            // Connect to a Uds socket
            async move {
                tokio::task::spawn_blocking(move || {
                    let stream = UnixStream::connect(path_copy)?;
                    windows::UnixStream::new(stream)
                })
                .await?
            }
        }))
        .await;

    res
}