From 2c489fa582e9910f0a7174ef675034d33f3fcaaa Mon Sep 17 00:00:00 2001 From: Tulir Asokan Date: Mon, 27 Jan 2025 12:48:57 +0200 Subject: [PATCH] hicli/init: add safety for too many empty rooms --- pkg/hicli/init.go | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/pkg/hicli/init.go b/pkg/hicli/init.go index 5b08de5..c7951d8 100644 --- a/pkg/hicli/init.go +++ b/pkg/hicli/init.go @@ -120,8 +120,11 @@ func (h *HiClient) GetInitialSync(ctx context.Context, batchSize int) iter.Seq[* payload := SyncComplete{ Rooms: make(map[id.RoomID]*SyncRoom, len(rooms)), } - for _, room := range rooms { + for roomIdx, room := range rooms { if room.SortingTimestamp == rooms[len(rooms)-1].SortingTimestamp { + if roomIdx == 0 { + batchSize *= 2 + } break } maxTS = room.SortingTimestamp.Time