brainos-grpcadapter 0.5.0

gRPC adapter for Brain OS
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

//! gRPC service-impl structs and their shared constructor/principal-resolution
//! glue. The actual `tonic::async_trait` impls live in `handlers::memory` and
//! `handlers::agent`.

use std::sync::Arc;

use tonic::Request;

use crate::auth::resolve_principal_from_metadata;

/// gRPC implementation of `MemoryService`.
pub struct MemoryServiceImpl {
    pub(crate) processor: Arc<signal::SignalProcessor>,
    pub(crate) api_keys: Arc<Vec<brain::ApiKeyConfig>>,
}

impl MemoryServiceImpl {
    pub fn new(processor: Arc<signal::SignalProcessor>) -> Self {
        let api_keys = Arc::new(processor.config().access.api_keys.clone());
        Self {
            processor,
            api_keys,
        }
    }

    /// Resolve the `Principal` from the request metadata. Returns `None`
    /// for keys without `agent_id` (back-compat).
    pub(crate) async fn resolve_principal<T>(
        &self,
        req: &Request<T>,
    ) -> Option<identity::Principal> {
        resolve_principal_from_metadata(req, &self.api_keys, self.processor.as_ref()).await
    }
}

/// gRPC implementation of `AgentService`.
pub struct AgentServiceImpl {
    pub(crate) processor: Arc<signal::SignalProcessor>,
    pub(crate) api_keys: Arc<Vec<brain::ApiKeyConfig>>,
}

impl AgentServiceImpl {
    pub fn new(processor: Arc<signal::SignalProcessor>) -> Self {
        let api_keys = Arc::new(processor.config().access.api_keys.clone());
        Self {
            processor,
            api_keys,
        }
    }

    pub(crate) async fn resolve_principal<T>(
        &self,
        req: &Request<T>,
    ) -> Option<identity::Principal> {
        resolve_principal_from_metadata(req, &self.api_keys, self.processor.as_ref()).await
    }
}