defer.gatherResults loop

2024-07-04 10:18:52 +02:00 · 2015-05-13 11:29:03 +01:00 · 2015-05-13 11:29:03 +01:00 · 619a21812b
parent fec4485e28
commit 619a21812b
1 changed files with 7 additions and 1 deletions
--- a/synapse/storage/state.py
+++ b/synapse/storage/state.py
@ -103,12 +103,18 @@ class StateStore(SQLBaseStore):
                for row in rows
            ]

-        for vals in states.values():
+        @defer.inlineCallbacks
+        def c(vals):
            vals[:] = yield self.runInteraction(
                "_get_state_groups_ev",
                fetch_events, vals
            )

+        yield defer.gatherResults(
+            [c(vals) for vals in states.values()],
+            consumeErrors=True,
+        )
+
        defer.returnValue(states)

    def _store_state_groups_txn(self, txn, event, context):