objstore 0.1.0-alpha.1

Core objstore crate: common traits, types, and APIs used by storage backend implementations.
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

use std::sync::Arc;

use crate::{ObjStoreProvider, store::DynObjStore};

#[derive(Clone, Debug)]
pub struct ObjStoreBuilder {
    providers: Vec<Arc<dyn ObjStoreProvider>>,
}

impl Default for ObjStoreBuilder {
    fn default() -> Self {
        Self::new()
    }
}

impl ObjStoreBuilder {
    pub fn new() -> Self {
        Self {
            providers: Vec::new(),
        }
    }

    pub fn register_provider<P: ObjStoreProvider + 'static>(&mut self, provider: P) {
        self.providers.push(Arc::new(provider));
    }

    pub fn with_provider(mut self, provider: Arc<dyn ObjStoreProvider>) -> Self {
        self.providers.push(provider);
        self
    }

    pub fn build(&self, uri: &str) -> Result<DynObjStore, anyhow::Error> {
        let url = url::Url::parse(uri).map_err(|e| anyhow::anyhow!("Invalid URL: {}", e))?;

        for provider in &self.providers {
            if provider.url_scheme() == url.scheme() {
                return provider.build(&url);
            }
        }
        Err(anyhow::anyhow!(
            "No suitable provider found for URI: {}",
            url
        ))
    }
}