From 44c52a24824b14f62a403b20fbacdc5c961306f5 Mon Sep 17 00:00:00 2001
From: Till Faelligen <2353100+S7evinK@users.noreply.github.com>
Date: Thu, 19 Dec 2024 11:41:20 +0100
Subject: [PATCH] Only pull state we actually need

---
 roomserver/storage/shared/storage.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/roomserver/storage/shared/storage.go b/roomserver/storage/shared/storage.go
index a4eb0eb9..66a83f07 100644
--- a/roomserver/storage/shared/storage.go
+++ b/roomserver/storage/shared/storage.go
@@ -1487,7 +1487,9 @@ func (d *Database) GetBulkStateContent(ctx context.Context, roomIDs []string, tu
 		if roomInfo == nil || roomInfo.IsStub() {
 			continue
 		}
-		entries, err2 := d.loadStateAtSnapshot(ctx, roomInfo.StateSnapshotNID())
+		// We don't need a querier here, as we don't actually do state resolution
+		res := state.NewStateResolution(d, roomInfo, nil)
+		entries, err2 := res.LoadStateAtSnapshotForStringTuples(ctx, roomInfo.StateSnapshotNID(), tuples)
 		if err2 != nil {
 			return nil, fmt.Errorf("GetBulkStateContent: failed to load state for room %s : %w", roomID, err2)
 		}