edma_storage 0.0.2

EDMA: Embedded Database Management for All
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
58
59
60
61
62
63
64
65

pub mod tx;
pub mod ty;

use async_trait::async_trait;
use redb::Database;
pub use tx::*;
pub use ty::*;

use crate::{
	util::{generate_path, get_absolute_path},
	DBTransaction, DatastoreAdapter, Error, StorageAdapter, StorageAdapterName, StorageVariant,
};
pub struct ReDBAdapter(StorageAdapter<DBType>);

#[cfg(feature = "test-suite")]
crate::full_adapter_test_impl!(ReDBAdapter::default());

impl ReDBAdapter {
	impl_new_type_adapter!(DBType);

	pub fn new(path: &str) -> Result<ReDBAdapter, Error> {
		let path = &path["redb:".len()..];
		let abs_path = get_absolute_path(path);
		let db_instance = unsafe { Database::create(abs_path.as_str()).unwrap() };

		Ok(ReDBAdapter(StorageAdapter::<DBType>::new(
			StorageAdapterName::ReDB,
			path.to_string(),
			db_instance,
			StorageVariant::KeyValueStore,
		)?))
	}
}

#[async_trait]
impl DatastoreAdapter for ReDBAdapter {
	type Transaction = ReDBTransaction;

	fn default() -> Self {
		let path = &generate_path("redb", None);
		ReDBAdapter::new(path).unwrap()
	}

	fn spawn(&self) -> Self {
		ReDBAdapter::default()
	}

	fn path(&self) -> &str {
		&self.0.path
	}

	async fn transaction(&self, w: bool) -> Result<Self::Transaction, Error> {
		let inner = self.get_initialized_inner().unwrap();
		let db = &inner.db_instance;
		let tx = db.begin_write().unwrap();

		let tx = unsafe { extend_tx_lifetime(tx) };

		Ok(DBTransaction::<DBType, TxType>::new(tx, db.clone(), w).unwrap())
	}
}

unsafe fn extend_tx_lifetime(tx: redb::WriteTransaction<'_>) -> redb::WriteTransaction<'static> {
	std::mem::transmute::<redb::WriteTransaction<'_>, redb::WriteTransaction<'static>>(tx)
}