mirrors/dendrite
1
0
Fork 0
mirror of https://github.com/matrix-org/dendrite.git synced 2026-01-16 10:33:11 -06:00
Code Issues Packages Projects Releases Wiki Activity

Track all changes

Browse source
This commit is contained in:
Neil Alexander 2022-10-10 16:07:45 +01:00
parent cf175bb400
commit aef6187da6
No known key found for this signature in database
GPG key ID: A02A2019A2BB0944
1 changed files with 16 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
federationapi/storage/shared/storage.go
View file

@ -68,6 +68,7 @@ func (d *Database) UpdateRoom(
removeHosts []string,
purgeRoomFirst bool,
) (joinedHosts []types.JoinedHost, err error) {
returnHosts := map[string]types.JoinedHost{}
err = d.Writer.Do(d.DB, nil, func(txn *sql.Tx) error {
if purgeRoomFirst {
// If the event is a create event then we'll delete all of the existing
@ -80,19 +81,30 @@ func (d *Database) UpdateRoom(
if joinedHosts, err = d.FederationJoinedHosts.SelectJoinedHostsWithTx(ctx, txn, roomID); err != nil {
return err
}
}
if err = d.FederationJoinedHosts.DeleteJoinedHosts(ctx, txn, removeHosts); err != nil {
return err
for _, add := range joinedHosts {
returnHosts[string(add.ServerName)] = add
}
if len(removeHosts) > 0 {
if err = d.FederationJoinedHosts.DeleteJoinedHosts(ctx, txn, removeHosts); err != nil {
return err
}
for _, remove := range removeHosts {
delete(returnHosts, remove)
}
}
}
for _, add := range addHosts {
err = d.FederationJoinedHosts.InsertJoinedHosts(ctx, txn, roomID, add.MemberEventID, add.ServerName)
if err != nil {
return err
}
joinedHosts = append(joinedHosts, add)
returnHosts[string(add.ServerName)] = add
}
return nil
})
for _, host := range returnHosts {
joinedHosts = append(joinedHosts, host)
}
return
}