ydb 0.12.0

Crate contains generated low-level grpc code from YDB API protobuf, used as base for ydb crate
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
51
52
53
54
55
56
57

use crate::client_common::{DBCredentials, TokenCache};
use crate::grpc_wrapper::raw_errors::{RawError, RawResult};
use crate::grpc_wrapper::runtime_interceptors::{
    GrpcInterceptor, InterceptorError, InterceptorRequest, InterceptorResult, RequestMetadata,
};
use http::HeaderValue;
use secrecy::ExposeSecret;

pub(crate) struct AuthGrpcInterceptor {
    db_name: HeaderValue,
    token_cache: TokenCache,
}
const VERSION_LABEL: &str = concat!("ydb-rust-sdk/", env!("CARGO_PKG_VERSION"));

impl AuthGrpcInterceptor {
    pub fn new(cred: DBCredentials) -> RawResult<AuthGrpcInterceptor> {
        let db_name = HeaderValue::from_str(cred.database.as_str()).map_err(|err| {
            RawError::custom(format!(
                "bad db name for set in headers '{}': {}",
                cred.database.as_str(),
                err
            ))
        })?;

        Ok(AuthGrpcInterceptor {
            db_name,
            token_cache: cred.token_cache,
        })
    }
}

impl GrpcInterceptor for AuthGrpcInterceptor {
    fn on_call(
        &self,
        _metadata: &mut RequestMetadata,
        mut req: InterceptorRequest,
    ) -> InterceptorResult<InterceptorRequest> {
        let token_secret = self.token_cache.token();
        let token_string = token_secret.expose_secret();
        let token = HeaderValue::from_str(token_string.as_str()).map_err(|err| {
            InterceptorError::custom(format!(
                "received bad token (len={}): {}",
                token_string.len(),
                err
            ))
        })?;

        req.headers_mut()
            .insert("x-ydb-database", self.db_name.clone());
        req.headers_mut().insert(
            "x-ydb-sdk-build-info",
            HeaderValue::from_static(VERSION_LABEL),
        );
        req.headers_mut().insert("x-ydb-auth-ticket", token);
        Ok(req)
    }
}