diff options
author | Erik Johnston <erik@matrix.org> | 2020-02-07 11:14:19 +0000 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-02-07 11:14:19 +0000 |
commit | de2d267375069c2d22bceb0d6ef9c6f5a77380e3 (patch) | |
tree | 12636dca8ea90ea5369207b7aea134521a1fc8f5 /synapse/replication/slave/storage/groups.py | |
parent | Admin api to add an email address (#6789) (diff) | |
download | synapse-de2d267375069c2d22bceb0d6ef9c6f5a77380e3.tar.xz |
Allow moving group read APIs to workers (#6866)
Diffstat (limited to 'synapse/replication/slave/storage/groups.py')
-rw-r--r-- | synapse/replication/slave/storage/groups.py | 14 |
1 files changed, 6 insertions, 8 deletions
diff --git a/synapse/replication/slave/storage/groups.py b/synapse/replication/slave/storage/groups.py index 69a4ae42f9..2d4fd08cf5 100644 --- a/synapse/replication/slave/storage/groups.py +++ b/synapse/replication/slave/storage/groups.py @@ -13,15 +13,14 @@ # See the License for the specific language governing permissions and # limitations under the License. -from synapse.storage import DataStore +from synapse.replication.slave.storage._base import BaseSlavedStore +from synapse.replication.slave.storage._slaved_id_tracker import SlavedIdTracker +from synapse.storage.data_stores.main.group_server import GroupServerWorkerStore from synapse.storage.database import Database from synapse.util.caches.stream_change_cache import StreamChangeCache -from ._base import BaseSlavedStore, __func__ -from ._slaved_id_tracker import SlavedIdTracker - -class SlavedGroupServerStore(BaseSlavedStore): +class SlavedGroupServerStore(GroupServerWorkerStore, BaseSlavedStore): def __init__(self, database: Database, db_conn, hs): super(SlavedGroupServerStore, self).__init__(database, db_conn, hs) @@ -35,9 +34,8 @@ class SlavedGroupServerStore(BaseSlavedStore): self._group_updates_id_gen.get_current_token(), ) - get_groups_changes_for_user = __func__(DataStore.get_groups_changes_for_user) - get_group_stream_token = __func__(DataStore.get_group_stream_token) - get_all_groups_for_user = __func__(DataStore.get_all_groups_for_user) + def get_group_stream_token(self): + return self._group_updates_id_gen.get_current_token() def stream_positions(self): result = super(SlavedGroupServerStore, self).stream_positions() |