implement missing trait function; add missing match arm (#5)

dyedgreen · Jun 12, 2023 · b3d7b52 · b3d7b52
1 parent 7c41433
commit b3d7b52
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/src/error.rs b/src/error.rs
@@ -78,6 +78,7 @@ impl From<SanakirjaError> for Error {
             SanakirjaError::IO(err) => Self::IO(err),
             SanakirjaError::Poison => Self::Poison,
             SanakirjaError::VersionMismatch | SanakirjaError::CRC(_) => Self::Corruption,
+            SanakirjaError::Corrupt(_) => Self::Corruption,
         }
     }
 }

diff --git a/src/store/txn.rs b/src/store/txn.rs
@@ -1,5 +1,5 @@
 use crate::Error;
-use sanakirja::{AllocPage, Commit, Env, LoadPage, MutTxn, RootDb, RootPage, Txn};
+use sanakirja::{AllocPage, Commit, Env, LoadPage, MutTxn, RootDb, RootPage, Txn, MutPage};
 use std::borrow::Borrow;
 
 pub(crate) enum DynTxn<E: Borrow<Env>> {
@@ -105,4 +105,11 @@ impl<E: Borrow<Env>> AllocPage for DynTxn<E> {
             DynTxn::MutTxn(txn) => Ok(txn.decr_rc_owned(off)?),
         }
     }
+
+    fn alloc_contiguous(&mut self, off: u64) -> Result<MutPage, <Self as LoadPage>::Error> {
+        match self {
+            DynTxn::Txn(_) => Err(Error::ReadOnlyWrite),
+            DynTxn::MutTxn(txn) => Ok(txn.alloc_contiguous(off)?),
+        }
+    }
 }