banyan_utils/
sqlite.rs

1//! helper methods to work with ipfs/ipld
2use anyhow::{anyhow, Result};
3use banyan::store::{BlockWriter, ReadOnlyStore};
4use ipfs_sqlite_block_store::BlockStore;
5use libipld::{codec::References, store::StoreParams, Block, Cid, Ipld};
6use parking_lot::Mutex;
7use std::sync::Arc;
8
9use crate::tags::Sha256Digest;
10
11#[derive(Clone)]
12pub struct SqliteStore<S: StoreParams>(Arc<Mutex<BlockStore<S>>>);
13
14impl<S: StoreParams> SqliteStore<S> {
15    pub fn new(store: BlockStore<S>) -> anyhow::Result<Self> {
16        Ok(SqliteStore(Arc::new(Mutex::new(store))))
17    }
18}
19
20impl<S: StoreParams> ReadOnlyStore<Sha256Digest> for SqliteStore<S>
21where
22    Ipld: References<S::Codecs>,
23{
24    fn get(&self, link: &Sha256Digest) -> Result<Box<[u8]>> {
25        let cid = Cid::from(*link);
26        let block = self.0.lock().get_block(&cid)?;
27        if let Some(block) = block {
28            Ok(block.into())
29        } else {
30            Err(anyhow!("block not found!"))
31        }
32    }
33}
34
35impl<S: StoreParams> BlockWriter<Sha256Digest> for SqliteStore<S>
36where
37    Ipld: References<S::Codecs>,
38{
39    fn put(&mut self, data: Vec<u8>) -> Result<Sha256Digest> {
40        let digest = Sha256Digest::new(&data);
41        let cid = digest.into();
42        let block = Block::new_unchecked(cid, data);
43        self.0.lock().put_block(&block, None)?;
44        Ok(digest)
45    }
46}
banyan_utils/sqlite.rs

banyan_utils/
sqlite.rs
