etcd-client 0.14.0

An etcd v3 API client
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

//! Authentication service.

use http::{header::AUTHORIZATION, HeaderValue, Request};
use std::sync::{Arc, RwLock};
use std::task::{Context, Poll};
use tower_service::Service;

#[derive(Debug, Clone)]
pub struct AuthService<S> {
    inner: S,
    token: Arc<RwLock<Option<HeaderValue>>>,
}

impl<S> AuthService<S> {
    #[inline]
    pub fn new(inner: S, token: Arc<RwLock<Option<HeaderValue>>>) -> Self {
        Self { inner, token }
    }
}

impl<S, Body, Response> Service<Request<Body>> for AuthService<S>
where
    S: Service<Request<Body>, Response = Response>,
{
    type Response = S::Response;
    type Error = S::Error;
    type Future = S::Future;

    #[inline]
    fn poll_ready(&mut self, cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
        self.inner.poll_ready(cx)
    }

    #[inline]
    fn call(&mut self, mut request: Request<Body>) -> Self::Future {
        if let Some(token) = self.token.read().unwrap().as_ref() {
            request.headers_mut().insert(AUTHORIZATION, token.clone());
        }

        self.inner.call(request)
    }
}