M-Labs/sfkv
8
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

write_value(): take value_buf for serialization from outside

Browse Source 
the caller should know the required size of that.
This commit is contained in:
Astro 2020-12-12 01:07:24 +01:00
parent c3e294d211
commit 08262fc3e9
2 changed files with 14 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/lib.rs
View File

@ -186,9 +186,12 @@ impl<B: StoreBackend> Store<B> {
/// encode with `postcard`, write at `key`
#[cfg(feature = "postcard-values")]
pub fn write_value<T: Serialize>(&mut self, key: &str, value: &T) -> Result<(), Error<B::Error>> {
let mut buf = [0; 64];
let data = postcard::to_slice(value, &mut buf)
pub fn write_value<T, V>(&mut self, key: &str, value: &T, mut value_buf: V) -> Result<(), Error<B::Error>>
where
T: Serialize,
V: AsMut<[u8]>,
{
let data = postcard::to_slice(value, value_buf.as_mut())
.map_err(Error::Encode)?;
self.write(key, data)
}

16
src/test.rs
View File

@ -83,7 +83,7 @@ fn write_read_str_utf8() {
#[cfg(feature = "postcard-values")]
fn write_read_value_str() {
let mut store = make_store();
store.write_value("foo", &"bar").unwrap();
store.write_value("foo", &"bar", [0; 4]).unwrap();
assert_eq!(store.read_value("foo"), Ok(Some("bar")));
}
@ -91,7 +91,7 @@ fn write_read_value_str() {
#[cfg(feature = "postcard-values")]
fn write_read_value_u32() {
let mut store = make_store();
store.write_value("foo", &42005u32).unwrap();
store.write_value("foo", &42005u32, [0; 4]).unwrap();
assert_eq!(store.read_value("foo"), Ok(Some(42005u32)));
}
@ -99,7 +99,7 @@ fn write_read_value_u32() {
#[cfg(feature = "postcard-values")]
fn write_read_value_i64() {
let mut store = make_store();
store.write_value("foo", &-99999999i64).unwrap();
store.write_value("foo", &-99999999i64, &mut [0; 8]).unwrap();
assert_eq!(store.read_value("foo"), Ok(Some(-99999999i64)));
}
@ -107,7 +107,7 @@ fn write_read_value_i64() {
#[cfg(feature = "postcard-values")]
fn write_read_value_f32() {
let mut store = make_store();
store.write_value("foo", &3.75e17f32).unwrap();
store.write_value("foo", &3.75e17f32, &mut [0; 4]).unwrap();
assert_eq!(store.read_value("foo"), Ok(Some(3.75e17f32)));
}
@ -115,7 +115,7 @@ fn write_read_value_f32() {
#[cfg(feature = "postcard-values")]
fn write_read_value_f64() {
let mut store = make_store();
store.write_value("foo", &-1.999e-13f64).unwrap();
store.write_value("foo", &-1.999e-13f64, [0; 8]).unwrap();
assert_eq!(store.read_value("foo"), Ok(Some(-1.999e-13f64)));
}
@ -134,11 +134,11 @@ fn write_many() {
#[cfg(feature = "postcard-values")]
fn write_value_many() {
let mut store = make_store();
store.write_value("quux", &(false, 3u8, -42i16)).unwrap();
store.write_value("quux", &(false, 3u8, -42i16), [0; 4]).unwrap();
for i in 0..1000u32 {
store.write_value("bar", &i).unwrap();
store.write_value("bar", &i, [0; 4]).unwrap();
assert_eq!(store.read_value("bar"), Ok(Some(i)));
store.write_value("foo", &(i % 2 == 0)).unwrap();
store.write_value("foo", &(i % 2 == 0), [0]).unwrap();
assert_eq!(store.read_value("foo"), Ok(Some(i % 2 == 0)));
}
assert_eq!(store.read_value("quux"), Ok(Some((false, 3u8, -42i16))));