Pull prev txn from in memory

synapse/storage/transactions.py

@@ -58,6 +58,8 @@ class TransactionStore(SQLBaseStore):
         # Newly delivered transactions that *were* persisted while in flight
         self.update_delivered_transactions = {}
 
+        self.last_transaction = {}
+
         reactor.addSystemEventTrigger("before", "shutdown", self._persist_in_mem_txns)
         hs.get_clock().looping_call(
             self._persist_in_mem_txns,
@@ -159,6 +161,10 @@ class TransactionStore(SQLBaseStore):
 
         self.inflight_transactions.setdefault(destination, {})[transaction_id] = txn_row
 
+        prev_txn = self.last_transaction.get(destination)
+        if prev_txn:
+            return defer.succeed(prev_txn)
+        else:
             return self.runInteraction(
                 "_get_prevs_txn",
                 self._get_prevs_txn,
@@ -196,6 +202,8 @@ class TransactionStore(SQLBaseStore):
             destination, {}
         ).pop(transaction_id, None)
 
+        self.last_transaction[destination] = transaction_id
+
         if txn_row:
             d = self.new_delivered_transactions.setdefault(destination, {})
             d[transaction_id] = txn_row._replace(