Adapt SnapshotSource interface

stellar · Mar 6, 2023 · af12bcb · af12bcb · latobarita · Mar 6, 2023
1 parent 040a29c
commit af12bcb
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 12 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/src/rust/Cargo.toml b/src/rust/Cargo.toml
@@ -18,9 +18,9 @@ rustc-simple-version = "0.1.0"
 [dependencies.soroban-env-host]
 version = "0.0.14"
 git = "https://github.com/stellar/rs-soroban-env"
-rev = "d06aaddca61f011cc64ec098b464233423197c3a"
+rev = "5695440da452837555d8f7f259cc33341fdf07b0"
 features = ["vm"]
 
 [dependencies.soroban-test-wasms]
 git = "https://github.com/stellar/rs-soroban-env"
-rev = "d06aaddca61f011cc64ec098b464233423197c3a"
+rev = "5695440da452837555d8f7f259cc33341fdf07b0"
diff --git a/src/rust/src/contract.rs b/src/rust/src/contract.rs
@@ -383,14 +383,14 @@ impl Pfc {
 }
 
 impl SnapshotSource for Pfc {
-    fn get(&self, key: &LedgerKey) -> Result<LedgerEntry, HostError> {
-        let kv = xdr_to_vec_u8(key)?;
+    fn get(&self, key: &Rc<LedgerKey>) -> Result<Rc<LedgerEntry>, HostError> {
+        let kv = xdr_to_vec_u8(key.as_ref())?;
         let lev = self.with_cb(|cb| cb.get_ledger_entry(&kv))?;
-        xdr_from_cxx_buf(&lev)
+        xdr_from_cxx_buf(&lev).map(|le| Rc::new(le))
     }
 
-    fn has(&self, key: &LedgerKey) -> Result<bool, HostError> {
-        let kv = xdr_to_vec_u8(key)?;
+    fn has(&self, key: &Rc<LedgerKey>) -> Result<bool, HostError> {
+        let kv = xdr_to_vec_u8(key.as_ref())?;
         self.with_cb(|cb| cb.has_ledger_entry(&kv))
     }
 }