etcdv3client 0.3.0-beta.2

a simple etcdv3 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

use std::convert::TryFrom;
use std::sync::{Arc, RwLock};

use tonic::{metadata::MetadataValue, service::Interceptor};

pub(crate) const TOKEN_FIELD_NAME: &str = "token";

#[derive(Debug, Clone)]
pub(crate) struct CredentialInterceptor {
    token: Arc<RwLock<String>>,
}

impl CredentialInterceptor {
    pub(crate) fn new(token: Arc<RwLock<String>>) -> Self {
        CredentialInterceptor { token }
    }

    pub fn empty() -> Self {
        Self::new(Arc::new(RwLock::new(String::new())))
    }
}

impl Interceptor for CredentialInterceptor {
    fn call(
        &mut self,
        mut request: tonic::Request<()>,
    ) -> std::result::Result<tonic::Request<()>, tonic::Status> {
        let token = self.token.read().unwrap();
        if !token.is_empty() {
            let token = MetadataValue::try_from(token.as_str()).unwrap();
            request.metadata_mut().insert(TOKEN_FIELD_NAME, token);
        }

        Ok(request)
    }
}