mirror/matrix-rust-sdk
1
0
Fork 0
mirror of https://github.com/matrix-org/matrix-rust-sdk.git synced 2026-05-19 06:04:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

feat(indexeddb): add IndexedDB-backed impl for EventCacheStore::load_all_chunks_metadata

Browse Source 
Signed-off-by: Michael Goldenberg <m@mgoldenberg.net>
This commit is contained in:
Michael Goldenberg
2025-07-14 23:02:29 -04:00
committed by Ivan Enderlin
parent 03a76fbaf5
commit 74972d8db7
3 changed files with 54 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
crates/matrix-sdk-indexeddb/src/event_cache_store/integration_tests.rs
View File

@@ -728,6 +728,13 @@ macro_rules! event_cache_store_integration_tests {
event_cache_store.test_rebuild_empty_linked_chunk().await;
}
#[async_test]
async fn test_load_all_chunks_metadata() {
let event_cache_store =
get_event_cache_store().await.unwrap().into_event_cache_store();
event_cache_store.test_load_all_chunks_metadata().await;
}
#[async_test]
async fn test_clear_all_linked_chunks() {
let event_cache_store =

25
crates/matrix-sdk-indexeddb/src/event_cache_store/mod.rs
View File

@@ -287,10 +287,27 @@ impl_event_cache_store! {
&self,
linked_chunk_id: LinkedChunkId<'_>,
) -> Result<Vec<ChunkMetadata>, IndexeddbEventCacheStoreError> {
self.memory_store
.load_all_chunks_metadata(linked_chunk_id)
.await
.map_err(IndexeddbEventCacheStoreError::MemoryStore)
let linked_chunk_id = linked_chunk_id.to_owned();
let room_id = linked_chunk_id.room_id();
let transaction = self.transaction(
&[keys::LINKED_CHUNKS, keys::EVENTS, keys::GAPS],
IdbTransactionMode::Readwrite,
)?;
let mut raw_chunks = Vec::new();
let chunks = transaction.get_chunks_in_room(room_id).await?;
for chunk in chunks {
let chunk_id = ChunkIdentifier::new(chunk.identifier);
let num_items = transaction.get_events_count_by_chunk(room_id, &chunk_id).await?;
raw_chunks.push(ChunkMetadata {
num_items,
previous: chunk.previous.map(ChunkIdentifier::new),
identifier: ChunkIdentifier::new(chunk.identifier),
next: chunk.next.map(ChunkIdentifier::new),
});
}
Ok(raw_chunks)
}
async fn load_last_chunk(

26
crates/matrix-sdk-indexeddb/src/event_cache_store/transaction.rs
View File

@@ -580,6 +580,17 @@ impl<'a> IndexeddbEventCacheStoreTransaction<'a> {
self.get_items_by_key_components::<Event, IndexedEventPositionKey>(room_id, range).await
}
/// Query IndexedDB for number of events in the given position range in the
/// given room.
pub async fn get_events_count_by_position(
&self,
room_id: &RoomId,
range: impl Into<IndexedKeyRange<&Position>>,
) -> Result<usize, IndexeddbEventCacheStoreTransactionError> {
self.get_items_count_by_key_components::<Event, IndexedEventPositionKey>(room_id, range)
.await
}
/// Query IndexedDB for events in the given chunk in the given room.
pub async fn get_events_by_chunk(
&self,
@@ -594,6 +605,21 @@ impl<'a> IndexeddbEventCacheStoreTransaction<'a> {
self.get_events_by_position(room_id, range).await
}
/// Query IndexedDB for number of events in the given chunk in the given
/// room.
pub async fn get_events_count_by_chunk(
&self,
room_id: &RoomId,
chunk_id: &ChunkIdentifier,
) -> Result<usize, IndexeddbEventCacheStoreTransactionError> {
let mut lower = IndexedEventPositionKey::lower_key_components();
lower.chunk_identifier = chunk_id.index();
let mut upper = IndexedEventPositionKey::upper_key_components();
upper.chunk_identifier = chunk_id.index();
let range = IndexedKeyRange::Bound(&lower, &upper);
self.get_events_count_by_position(room_id, range).await
}
/// Puts an event in the given room. If an event with the same key already
/// exists, it will be overwritten.
pub async fn put_event(