char/matrix-rust-sdk
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

fix(base): Flush when we're saving custom stuff in the sled store

Browse source
This commit is contained in:
Damir Jelić 2021-09-09 11:55:05 +02:00
parent 85fde6796a
commit b09b667782
1 changed files with 20 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
matrix_sdk_base/src/store/sled_store/mod.rs
View file

@ -771,7 +771,10 @@ impl SledStore {
} }
async fn set_custom_value(&self, key: &[u8], value: Vec<u8>) -> Result<Option<Vec<u8>>> { async fn set_custom_value(&self, key: &[u8], value: Vec<u8>) -> Result<Option<Vec<u8>>> {
Ok(self.custom.insert(key, value)?.map(|v| v.to_vec())) let ret = self.custom.insert(key, value)?.map(|v| v.to_vec());
self.inner.flush_async().await?;
Ok(ret)
} }
async fn remove_media_content(&self, request: &MediaRequest) -> Result<()> { async fn remove_media_content(&self, request: &MediaRequest) -> Result<()> {
@ -959,7 +962,7 @@ mod test {
}; };
use serde_json::json; use serde_json::json;
use super::{SledStore, StateChanges}; use super::{Result, SledStore, StateChanges};
use crate::{ use crate::{
deserialized_responses::MemberEvent, deserialized_responses::MemberEvent,
media::{MediaFormat, MediaRequest, MediaThumbnailSize, MediaType}, media::{MediaFormat, MediaRequest, MediaThumbnailSize, MediaType},
@ -1175,4 +1178,19 @@ mod test {
assert!(store.get_media_content(&request_file).await.unwrap().is_none()); assert!(store.get_media_content(&request_file).await.unwrap().is_none());
assert!(store.get_media_content(&request_thumbnail).await.unwrap().is_none()); assert!(store.get_media_content(&request_thumbnail).await.unwrap().is_none());
} }
#[async_test]
async fn test_custom_storage() -> Result<()> {
let key = "my_key";
let value = &[0, 1, 2, 3];
let store = SledStore::open().unwrap();
store.set_custom_value(key.as_bytes(), value.to_vec()).await?;
let read = store.get_custom_value(key.as_bytes()).await?;
assert_eq!(Some(value.as_ref()), read.as_deref());
Ok(())
}
} }