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

store: Honor state keys for the state storage.

Browse source
This commit is contained in:
Damir Jelić 2020-12-19 14:44:46 +01:00
parent b05fed5a3b
commit 7abf0c8805
2 changed files with 45 additions and 31 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
matrix_sdk_base/src/client.rs
View file

@ -523,7 +523,7 @@ impl BaseClient {
) -> (
InviteState,
BTreeMap<UserId, StrippedMemberEvent>,
BTreeMap<String, AnyStrippedStateEvent>,
BTreeMap<String, BTreeMap<String, AnyStrippedStateEvent>>,
) {
events.into_iter().fold(
(InviteState::default(), BTreeMap::new(), BTreeMap::new()),
@ -544,7 +544,10 @@ impl BaseClient {
}
} else {
room_info.handle_state_event(&e);
state_events.insert(e.content().event_type().to_owned(), e);
state_events
.entry(e.content().event_type().to_owned())
.or_insert_with(BTreeMap::new)
.insert(e.state_key().to_owned(), e);
}
}
Err(err) => {
@ -566,7 +569,7 @@ impl BaseClient {
) -> (
State,
BTreeMap<UserId, MemberEvent>,
BTreeMap<String, AnySyncStateEvent>,
BTreeMap<String, BTreeMap<String, AnySyncStateEvent>>,
BTreeSet<UserId>,
) {
let mut state = State::default();
@ -610,7 +613,10 @@ impl BaseClient {
),
}
} else {
state_events.insert(event.content().event_type().to_owned(), event);
state_events
.entry(event.content().event_type().to_owned())
.or_insert_with(BTreeMap::new)
.insert(event.state_key().to_owned(), event);
}
}

62
matrix_sdk_base/src/store.rs
View file

@ -42,11 +42,11 @@ pub struct StateChanges {
pub presence: BTreeMap<UserId, PresenceEvent>,
pub members: BTreeMap<RoomId, BTreeMap<UserId, MemberEvent>>,
pub state: BTreeMap<RoomId, BTreeMap<String, AnySyncStateEvent>>,
pub state: BTreeMap<RoomId, BTreeMap<String, BTreeMap<String, AnySyncStateEvent>>>,
pub room_account_data: BTreeMap<RoomId, BTreeMap<String, AnyBasicEvent>>,
pub room_infos: BTreeMap<RoomId, RoomInfo>,
pub stripped_state: BTreeMap<RoomId, BTreeMap<String, AnyStrippedStateEvent>>,
pub stripped_state: BTreeMap<RoomId, BTreeMap<String, BTreeMap<String, AnyStrippedStateEvent>>>,
pub stripped_members: BTreeMap<RoomId, BTreeMap<UserId, StrippedMemberEvent>>,
pub invited_room_info: BTreeMap<RoomId, RoomInfo>,
}
@ -77,6 +77,8 @@ impl StateChanges {
self.stripped_state
.entry(room_id.to_owned())
.or_insert_with(BTreeMap::new)
.entry(event.content().event_type().to_string())
.or_insert_with(BTreeMap::new)
.insert(event.state_key().to_string(), event);
}
@ -92,7 +94,9 @@ impl StateChanges {
self.state
.entry(room_id.to_owned())
.or_insert_with(BTreeMap::new)
.insert(event.content().event_type().to_string(), event);
.entry(event.content().event_type().to_string())
.or_insert_with(BTreeMap::new)
.insert(event.state_key().to_string(), event);
}
}
@ -242,18 +246,20 @@ impl Store {
}
}
for (room, events) in &changes.state {
for event in events.values() {
state.insert(
format!(
"{}{}{}",
room.as_str(),
event.content().event_type(),
event.state_key(),
)
.as_bytes(),
serde_json::to_vec(&event).unwrap(),
)?;
for (room, event_types) in &changes.state {
for events in event_types.values() {
for event in events.values() {
state.insert(
format!(
"{}{}{}",
room.as_str(),
event.content().event_type(),
event.state_key(),
)
.as_bytes(),
serde_json::to_vec(&event).unwrap(),
)?;
}
}
}
@ -280,18 +286,20 @@ impl Store {
}
}
for (room, events) in &changes.stripped_state {
for event in events.values() {
stripped_state.insert(
format!(
"{}{}{}",
room.as_str(),
event.content().event_type(),
event.state_key(),
)
.as_bytes(),
serde_json::to_vec(&event).unwrap(),
)?;
for (room, event_types) in &changes.stripped_state {
for events in event_types.values() {
for event in events.values() {
stripped_state.insert(
format!(
"{}{}{}",
room.as_str(),
event.content().event_type(),
event.state_key(),
)
.as_bytes(),
serde_json::to_vec(&event).unwrap(),
)?;
}
}
}