hoernschen/dendrite
1
0
Fork 0
mirror of https://github.com/hoernschen/dendrite.git synced 2024-12-27 07:28:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Limit number of servers

Browse source
This commit is contained in:
Neil Alexander 2021-06-22 11:22:13 +01:00
parent 161857d9b7
commit 1607b8ef78
No known key found for this signature in database
GPG key ID: A02A2019A2BB0944
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
federationapi/routing/send.go
View file

@ -483,12 +483,14 @@ func (t *txnReq) getServers(ctx context.Context, roomID string, eventOrigin goma
return t.servers return t.servers
} }
dedupe := map[gomatrixserverlib.ServerName]struct{}{} dedupe := map[gomatrixserverlib.ServerName]struct{}{}
count := 1
t.servers = []gomatrixserverlib.ServerName{t.Origin} // transaction origin t.servers = []gomatrixserverlib.ServerName{t.Origin} // transaction origin
dedupe[t.Origin] = struct{}{} dedupe[t.Origin] = struct{}{}
if eventOrigin != "" { if eventOrigin != "" {
if _, ok := dedupe[eventOrigin]; !ok { if _, ok := dedupe[eventOrigin]; !ok {
t.servers = append(t.servers, eventOrigin) // event origin, if specified t.servers = append(t.servers, eventOrigin) // event origin, if specified
dedupe[eventOrigin] = struct{}{} dedupe[eventOrigin] = struct{}{}
count++
} }
} }
serverReq := &api.QueryServerJoinedToRoomRequest{ serverReq := &api.QueryServerJoinedToRoomRequest{
@ -500,6 +502,10 @@ func (t *txnReq) getServers(ctx context.Context, roomID string, eventOrigin goma
if _, ok := dedupe[server]; !ok { if _, ok := dedupe[server]; !ok {
t.servers = append(t.servers, server) t.servers = append(t.servers, server)
dedupe[server] = struct{}{} dedupe[server] = struct{}{}
count++
if count == 10 {
break
}
} }
} }
util.GetLogger(ctx).Infof("Found %d server(s) to query for missing events in %q", len(t.servers), roomID) util.GetLogger(ctx).Infof("Found %d server(s) to query for missing events in %q", len(t.servers), roomID)