fix: remove unwraps in storage.rs

deltadevsde · Jul 18, 2024 · 7be0f61 · 7be0f61
1 parent 6c3c22d
commit 7be0f61
Show file tree

Hide file tree

Showing 4 changed files with 202 additions and 182 deletions.
diff --git a/src/error.rs b/src/error.rs
@@ -61,6 +61,10 @@ pub enum DatabaseError {
     DeleteError(String),
     #[error(transparent)]
     GeneralError(#[from] GeneralError),
+    #[error("connecting to database: {0}")]
+    ConnectionError(String),
+    #[error("initializing database: {0}")]
+    InitializationError(String),
 }
 
 // Result alias for [`DataAvailabilityError`]

diff --git a/src/node_types/sequencer.rs b/src/node_types/sequencer.rs
@@ -199,20 +199,16 @@ impl Sequencer {
             Err(_) => 0,
         };
 
-        self.db.set_epoch(&epoch).map_err(DeimosError::Database)?;
-        self.db
-            .reset_epoch_operation_counter()
-            .map_err(DeimosError::Database)?;
+        self.db.set_epoch(&epoch)?;
+        self.db.reset_epoch_operation_counter()?;
 
         // add the commitment for the operations ran since the last epoch
         let current_commitment = self
             .create_tree()?
             .get_commitment()
             .map_err(DeimosError::MerkleTree)?;
 
-        self.db
-            .add_commitment(&epoch, &current_commitment)
-            .map_err(DeimosError::Database)?;
+        self.db.add_commitment(&epoch, &current_commitment)?;
 
         let proofs = match epoch > 0 {
             true => match self.db.get_proofs_in_epoch(&(epoch - 1)) {