Clear storage before finalization

postech-dao · Apr 23, 2023 · cf03033 · cf03033
1 parent 300c6ca
commit cf03033
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/simperby/src/lib.rs b/simperby/src/lib.rs
@@ -116,6 +116,8 @@ impl Client {
                 let config = this.config.clone();
                 let auth = this.auth.clone();
                 drop(this);
+                storage::clear(&path).await?;
+                storage::init(&path, config.clone()).await?;
                 self.inner = Some(Self::open(&path, config, auth).await?.inner.unwrap());
                 return Ok(report);
             }

diff --git a/simperby/src/storage.rs b/simperby/src/storage.rs
@@ -84,3 +84,10 @@ pub(crate) async fn open(
     let consensus_state = StorageImpl::open(&consensus_state_path(path)).await?;
     Ok((governance_dms, consensus_dms, consensus_state, repository))
 }
+
+pub(crate) async fn clear(path: &str) -> Result<()> {
+    let _ = tokio::fs::remove_dir_all(&governance_dms_path(path)).await;
+    let _ = tokio::fs::remove_dir_all(&consensus_dms_path(path)).await;
+    let _ = tokio::fs::remove_dir_all(&consensus_state_path(path)).await;
+    Ok(())
+}