apollo_proof_manager: impl ProofStorage

Author: einat-starkware

crates/apollo_proof_manager/src/proof_storage.rs

@@ -140,4 +140,39 @@ impl FsProofStorage {
 
         Ok(Proof(Arc::new(proof_data)))
     }
+
+    fn write_proof_atomically(&self, facts_hash: Felt, proof: Proof) -> FsProofStorageResult<()> {
+        // Write proof to a temporary directory.
+        let (_tmp_root, tmp_dir) = self.create_tmp_dir(facts_hash)?;
+        self.write_proof_to_file(&tmp_dir, &proof)?;
+
+        // Atomically rename directory to persistent one.
+        let persistent_dir = self.get_persistent_dir_with_create(facts_hash)?;
+        std::fs::rename(tmp_dir, persistent_dir)?;
+        Ok(())
+    }
+}
+
+impl ProofStorage for FsProofStorage {
+    type Error = FsProofStorageError;
+
+    fn set_proof(&self, facts_hash: Felt, proof: Proof) -> Result<(), Self::Error> {
+        self.write_proof_atomically(facts_hash, proof)?;
+        Ok(())
+    }
+
+    fn get_proof(&self, facts_hash: Felt) -> Result<Option<Proof>, Self::Error> {
+        if !self.contains_proof(facts_hash)? {
+            return Ok(None);
+        }
+
+        match self.read_proof_from_file(facts_hash) {
+            Ok(proof) => Ok(Some(proof)),
+            Err(e) => Err(e),
+        }
+    }
+
+    fn contains_proof(&self, facts_hash: Felt) -> Result<bool, Self::Error> {
+        Ok(self.get_persistent_dir(facts_hash).exists())
+    }
 }