diff options
author | Patrick Cloke <clokep@users.noreply.github.com> | 2020-08-28 07:54:27 -0400 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-08-28 07:54:27 -0400 |
commit | 5c03134d0f8dd157ea1800ce1a4bcddbdb73ddf1 (patch) | |
tree | 98d41759fd18e423fd820d04d2a19cbe89f9dbd7 /synapse/federation | |
parent | Define StateMap as immutable and add a MutableStateMap type. (#8183) (diff) | |
download | synapse-5c03134d0f8dd157ea1800ce1a4bcddbdb73ddf1.tar.xz |
Convert additional database code to async/await. (#8195)
Diffstat (limited to 'synapse/federation')
-rw-r--r-- | synapse/federation/persistence.py | 19 |
1 files changed, 11 insertions, 8 deletions
diff --git a/synapse/federation/persistence.py b/synapse/federation/persistence.py index 769cd5de28..de1fe7da38 100644 --- a/synapse/federation/persistence.py +++ b/synapse/federation/persistence.py @@ -20,6 +20,7 @@ These actions are mostly only used by the :py:mod:`.replication` module. """ import logging +from typing import Optional, Tuple from synapse.federation.units import Transaction from synapse.logging.utils import log_function @@ -36,25 +37,27 @@ class TransactionActions(object): self.store = datastore @log_function - def have_responded(self, origin, transaction): - """ Have we already responded to a transaction with the same id and + async def have_responded( + self, origin: str, transaction: Transaction + ) -> Optional[Tuple[int, JsonDict]]: + """Have we already responded to a transaction with the same id and origin? Returns: - Deferred: Results in `None` if we have not previously responded to - this transaction or a 2-tuple of `(int, dict)` representing the - response code and response body. + `None` if we have not previously responded to this transaction or a + 2-tuple of `(int, dict)` representing the response code and response body. """ - if not transaction.transaction_id: + transaction_id = transaction.transaction_id # type: ignore + if not transaction_id: raise RuntimeError("Cannot persist a transaction with no transaction_id") - return self.store.get_received_txn_response(transaction.transaction_id, origin) + return await self.store.get_received_txn_response(transaction_id, origin) @log_function async def set_response( self, origin: str, transaction: Transaction, code: int, response: JsonDict ) -> None: - """ Persist how we responded to a transaction. + """Persist how we responded to a transaction. """ transaction_id = transaction.transaction_id # type: ignore if not transaction_id: |