hive-router 0.0.50

GraphQL router/gateway for Federation
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
58
59
60
61
62

use async_trait::async_trait;
use std::sync::Arc;

use crate::pipeline::error::PipelineError;
use crate::pipeline::persisted_documents::types::{ClientIdentity, PersistedDocumentId};
use file::FileResolverError;
use hive::HiveResolverError;

pub mod file;
pub mod hive;

pub use file::{FileManifestReloadTask, FileManifestResolver};
pub use hive::HiveCDNResolver;

#[derive(Debug, Clone, Copy)]
pub struct PersistedDocumentResolveInput<'a> {
    pub persisted_document_id: &'a PersistedDocumentId,
    pub client_identity: ClientIdentity<'a>,
}

#[derive(Debug, thiserror::Error)]
pub enum PersistedDocumentResolverError {
    #[error("Persisted document not found: {0}")]
    NotFound(String),
    #[error("Persisted documents configuration error: {0}")]
    Configuration(String),
    #[error("Persisted documents storage is not configured")]
    StorageNotConfigured,
    #[error("Hive Storage: {0}")]
    Hive(#[from] HiveResolverError),
    #[error("File Storage: {0}")]
    File(#[from] FileResolverError),
}

impl From<PersistedDocumentResolverError> for PipelineError {
    fn from(value: PersistedDocumentResolverError) -> Self {
        match value {
            PersistedDocumentResolverError::NotFound(document_id) => {
                PipelineError::PersistedDocumentNotFound(document_id)
            }
            PersistedDocumentResolverError::Hive(HiveResolverError::InvalidDocumentIdFormat(_))
            | PersistedDocumentResolverError::Hive(HiveResolverError::ClientIdentityMissing)
            | PersistedDocumentResolverError::Hive(HiveResolverError::ClientIdentityPartial) => {
                PipelineError::PersistedDocumentExtraction(value.to_string())
            }
            other => PipelineError::PersistedDocumentResolution(other.to_string()),
        }
    }
}

#[derive(Debug)]
pub struct ResolvedDocument {
    pub text: Arc<str>,
}

#[async_trait]
pub trait PersistedDocumentResolver: Send + Sync {
    async fn resolve(
        &self,
        input: PersistedDocumentResolveInput<'_>,
    ) -> Result<ResolvedDocument, PersistedDocumentResolverError>;
}