diff options
| author | Erik Johnston <erik@matrix.org> | 2020-05-07 13:51:08 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-05-07 13:51:08 +0100 |
| commit | d7983b63a6746d92225295f1e9d521f847cf8ba7 (patch) | |
| tree | 13d581210d94c26bd75036592996f6f53f7d4bb2 /synapse/replication/tcp/streams/federation.py | |
| parent | Merge pull request #7398 from Starbix/alpine-3.11 (diff) | |
| download | synapse-d7983b63a6746d92225295f1e9d521f847cf8ba7.tar.xz | |
Support any process writing to cache invalidation stream. (#7436)
Diffstat (limited to 'synapse/replication/tcp/streams/federation.py')
| -rw-r--r-- | synapse/replication/tcp/streams/federation.py | 12 |
1 files changed, 9 insertions, 3 deletions
diff --git a/synapse/replication/tcp/streams/federation.py b/synapse/replication/tcp/streams/federation.py index b0505b8a2c..9bcd13b009 100644 --- a/synapse/replication/tcp/streams/federation.py +++ b/synapse/replication/tcp/streams/federation.py @@ -15,7 +15,11 @@ # limitations under the License. from collections import namedtuple -from synapse.replication.tcp.streams._base import Stream, make_http_update_function +from synapse.replication.tcp.streams._base import ( + Stream, + current_token_without_instance, + make_http_update_function, +) class FederationStream(Stream): @@ -41,7 +45,9 @@ class FederationStream(Stream): # will be a real FederationSender, which has stubs for current_token and # get_replication_rows.) federation_sender = hs.get_federation_sender() - current_token = federation_sender.get_current_token + current_token = current_token_without_instance( + federation_sender.get_current_token + ) update_function = federation_sender.get_replication_rows elif hs.should_send_federation(): @@ -58,7 +64,7 @@ class FederationStream(Stream): super().__init__(hs.get_instance_name(), current_token, update_function) @staticmethod - def _stub_current_token(): + def _stub_current_token(instance_name: str) -> int: # dummy current-token method for use on workers return 0 |