Skip to content

Commit d971b81

Browse files
committed
fixup! Add relations to the cache for polls too, add doc comment
1 parent 9a0747a commit d971b81

File tree

2 files changed

+30
-3
lines changed

2 files changed

+30
-3
lines changed

crates/matrix-sdk/src/event_cache/mod.rs

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -774,6 +774,7 @@ impl RoomEventCacheInner {
774774
.await
775775
}
776776

777+
/// Appends events with relationships to others into the relations cache.
777778
fn append_relations(
778779
&self,
779780
mut cache: RwLockWriteGuard<'_, AllEventsCache>,
@@ -834,6 +835,34 @@ impl RoomEventCacheInner {
834835
}
835836
}
836837
}
838+
Some(AnyMessageLikeEventContent::PollResponse(c)) => {
839+
cache
840+
.relations
841+
.entry(c.relates_to.event_id)
842+
.or_default()
843+
.insert(ev.event_id().to_owned());
844+
}
845+
Some(AnyMessageLikeEventContent::PollEnd(c)) => {
846+
cache
847+
.relations
848+
.entry(c.relates_to.event_id)
849+
.or_default()
850+
.insert(ev.event_id().to_owned());
851+
}
852+
Some(AnyMessageLikeEventContent::UnstablePollResponse(c)) => {
853+
cache
854+
.relations
855+
.entry(c.relates_to.event_id)
856+
.or_default()
857+
.insert(ev.event_id().to_owned());
858+
}
859+
Some(AnyMessageLikeEventContent::UnstablePollEnd(c)) => {
860+
cache
861+
.relations
862+
.entry(c.relates_to.event_id)
863+
.or_default()
864+
.insert(ev.event_id().to_owned());
865+
}
837866
Some(AnyMessageLikeEventContent::Reaction(c)) => {
838867
cache
839868
.relations

crates/matrix-sdk/tests/integration/room/common.rs

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,9 +14,7 @@ use matrix_sdk_test::{
1414
use ruma::{
1515
event_id,
1616
events::{
17-
room::{
18-
member::MembershipState, message::RoomMessageEventContent,
19-
},
17+
room::{member::MembershipState, message::RoomMessageEventContent},
2018
AnyStateEvent, AnySyncStateEvent, AnyTimelineEvent, StateEventType,
2119
},
2220
room_id,

0 commit comments

Comments
 (0)