From 1607b8ef782c151bf353ea5205aebdc91a8b2d4d Mon Sep 17 00:00:00 2001
From: Neil Alexander <neilalexander@users.noreply.github.com>
Date: Tue, 22 Jun 2021 11:22:13 +0100
Subject: [PATCH] Limit number of servers

---
 federationapi/routing/send.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/federationapi/routing/send.go b/federationapi/routing/send.go
index fee9e2ac..36f98f31 100644
--- a/federationapi/routing/send.go
+++ b/federationapi/routing/send.go
@@ -483,12 +483,14 @@ func (t *txnReq) getServers(ctx context.Context, roomID string, eventOrigin goma
 		return t.servers
 	}
 	dedupe := map[gomatrixserverlib.ServerName]struct{}{}
+	count := 1
 	t.servers = []gomatrixserverlib.ServerName{t.Origin} // transaction origin
 	dedupe[t.Origin] = struct{}{}
 	if eventOrigin != "" {
 		if _, ok := dedupe[eventOrigin]; !ok {
 			t.servers = append(t.servers, eventOrigin) // event origin, if specified
 			dedupe[eventOrigin] = struct{}{}
+			count++
 		}
 	}
 	serverReq := &api.QueryServerJoinedToRoomRequest{
@@ -500,6 +502,10 @@ func (t *txnReq) getServers(ctx context.Context, roomID string, eventOrigin goma
 			if _, ok := dedupe[server]; !ok {
 				t.servers = append(t.servers, server)
 				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)