Skip to content

Commit 5ff38e2

Browse files
authored
Preserve history visibility in HeaderedReverseTopologicalOrdering (#327)
* Preserve history visibility * Set the capacity when creating the map
1 parent b6003e5 commit 5ff38e2

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

stateresolutionv2.go

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -231,13 +231,16 @@ func ReverseTopologicalOrdering(input []*Event, order TopologicalOrder) []*Event
231231
func HeaderedReverseTopologicalOrdering(events []*HeaderedEvent, order TopologicalOrder) []*HeaderedEvent {
232232
r := stateResolverV2{}
233233
input := make([]*Event, len(events))
234+
hisVis := make(map[string]HistoryVisibility, len(events))
234235
for i := range events {
235236
unwrapped := events[i].Unwrap()
236237
input[i] = unwrapped
238+
hisVis[unwrapped.EventID()] = events[i].Visibility
237239
}
238240
result := make([]*HeaderedEvent, len(input))
239241
for i, e := range r.reverseTopologicalOrdering(input, order) {
240242
result[i] = e.Headered(e.roomVersion)
243+
result[i].Visibility = hisVis[e.EventID()]
241244
}
242245
return result
243246
}

0 commit comments

Comments
 (0)