use track caller for database::get (#89)

ontio · Sep 9, 2020 · 60f488f · 60f488f
1 parent d90e902
commit 60f488f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/ontio-std/src/database/mod.rs b/ontio-std/src/database/mod.rs
@@ -6,14 +6,14 @@ use super::abi::{Decoder, Encoder, Sink, Source};
 use super::prelude::*;
 use super::runtime;
 
+#[track_caller]
 pub fn get<K: AsRef<[u8]>, T>(key: K) -> Option<T>
 where
     for<'a> T: Decoder<'a> + 'static,
 {
-    runtime::storage_read(key.as_ref()).map(|val: Vec<u8>| {
-        let mut source = Source::new(&val);
-        source.read().unwrap()
-    })
+    let val = runtime::storage_read(key.as_ref())?;
+    let mut source = Source::new(&val);
+    Some(source.read().unwrap())
 }
 
 pub fn put<K: AsRef<[u8]>, T: Encoder>(key: K, val: T) {