mirror of
https://git.anonymousland.org/anonymousland/synapse-product.git
synced 2024-09-08 02:43:04 +00:00
![Erik Johnston](/assets/img/avatar_default.png)
Currently we rely on `current_state_events` to figure out what rooms a user was in and their last membership event in there. However, if the server leaves the room then the table may be cleaned up and that information is lost. So lets add a table that separately holds that information.
161 lines
5.8 KiB
Python
161 lines
5.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright 2016 OpenMarket Ltd
|
|
# Copyright 2018 New Vector Ltd
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
import logging
|
|
|
|
from synapse.api.constants import EventTypes
|
|
from synapse.replication.tcp.streams.events import (
|
|
EventsStreamCurrentStateRow,
|
|
EventsStreamEventRow,
|
|
)
|
|
from synapse.storage.data_stores.main.event_federation import EventFederationWorkerStore
|
|
from synapse.storage.data_stores.main.event_push_actions import (
|
|
EventPushActionsWorkerStore,
|
|
)
|
|
from synapse.storage.data_stores.main.events_worker import EventsWorkerStore
|
|
from synapse.storage.data_stores.main.relations import RelationsWorkerStore
|
|
from synapse.storage.data_stores.main.roommember import RoomMemberWorkerStore
|
|
from synapse.storage.data_stores.main.signatures import SignatureWorkerStore
|
|
from synapse.storage.data_stores.main.state import StateGroupWorkerStore
|
|
from synapse.storage.data_stores.main.stream import StreamWorkerStore
|
|
from synapse.storage.data_stores.main.user_erasure_store import UserErasureWorkerStore
|
|
from synapse.storage.database import Database
|
|
|
|
from ._base import BaseSlavedStore
|
|
from ._slaved_id_tracker import SlavedIdTracker
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
# So, um, we want to borrow a load of functions intended for reading from
|
|
# a DataStore, but we don't want to take functions that either write to the
|
|
# DataStore or are cached and don't have cache invalidation logic.
|
|
#
|
|
# Rather than write duplicate versions of those functions, or lift them to
|
|
# a common base class, we going to grab the underlying __func__ object from
|
|
# the method descriptor on the DataStore and chuck them into our class.
|
|
|
|
|
|
class SlavedEventStore(
|
|
EventFederationWorkerStore,
|
|
RoomMemberWorkerStore,
|
|
EventPushActionsWorkerStore,
|
|
StreamWorkerStore,
|
|
StateGroupWorkerStore,
|
|
EventsWorkerStore,
|
|
SignatureWorkerStore,
|
|
UserErasureWorkerStore,
|
|
RelationsWorkerStore,
|
|
BaseSlavedStore,
|
|
):
|
|
def __init__(self, database: Database, db_conn, hs):
|
|
self._stream_id_gen = SlavedIdTracker(db_conn, "events", "stream_ordering")
|
|
self._backfill_id_gen = SlavedIdTracker(
|
|
db_conn, "events", "stream_ordering", step=-1
|
|
)
|
|
|
|
super(SlavedEventStore, self).__init__(database, db_conn, hs)
|
|
|
|
# Cached functions can't be accessed through a class instance so we need
|
|
# to reach inside the __dict__ to extract them.
|
|
|
|
def get_room_max_stream_ordering(self):
|
|
return self._stream_id_gen.get_current_token()
|
|
|
|
def get_room_min_stream_ordering(self):
|
|
return self._backfill_id_gen.get_current_token()
|
|
|
|
def stream_positions(self):
|
|
result = super(SlavedEventStore, self).stream_positions()
|
|
result["events"] = self._stream_id_gen.get_current_token()
|
|
result["backfill"] = -self._backfill_id_gen.get_current_token()
|
|
return result
|
|
|
|
def process_replication_rows(self, stream_name, token, rows):
|
|
if stream_name == "events":
|
|
self._stream_id_gen.advance(token)
|
|
for row in rows:
|
|
self._process_event_stream_row(token, row)
|
|
elif stream_name == "backfill":
|
|
self._backfill_id_gen.advance(-token)
|
|
for row in rows:
|
|
self.invalidate_caches_for_event(
|
|
-token,
|
|
row.event_id,
|
|
row.room_id,
|
|
row.type,
|
|
row.state_key,
|
|
row.redacts,
|
|
row.relates_to,
|
|
backfilled=True,
|
|
)
|
|
return super(SlavedEventStore, self).process_replication_rows(
|
|
stream_name, token, rows
|
|
)
|
|
|
|
def _process_event_stream_row(self, token, row):
|
|
data = row.data
|
|
|
|
if row.type == EventsStreamEventRow.TypeId:
|
|
self.invalidate_caches_for_event(
|
|
token,
|
|
data.event_id,
|
|
data.room_id,
|
|
data.type,
|
|
data.state_key,
|
|
data.redacts,
|
|
data.relates_to,
|
|
backfilled=False,
|
|
)
|
|
elif row.type == EventsStreamCurrentStateRow.TypeId:
|
|
if data.type == EventTypes.Member:
|
|
self.get_rooms_for_user_with_stream_ordering.invalidate(
|
|
(data.state_key,)
|
|
)
|
|
else:
|
|
raise Exception("Unknown events stream row type %s" % (row.type,))
|
|
|
|
def invalidate_caches_for_event(
|
|
self,
|
|
stream_ordering,
|
|
event_id,
|
|
room_id,
|
|
etype,
|
|
state_key,
|
|
redacts,
|
|
relates_to,
|
|
backfilled,
|
|
):
|
|
self._invalidate_get_event_cache(event_id)
|
|
|
|
self.get_latest_event_ids_in_room.invalidate((room_id,))
|
|
|
|
self.get_unread_event_push_actions_by_room_for_user.invalidate_many((room_id,))
|
|
|
|
if not backfilled:
|
|
self._events_stream_cache.entity_has_changed(room_id, stream_ordering)
|
|
|
|
if redacts:
|
|
self._invalidate_get_event_cache(redacts)
|
|
|
|
if etype == EventTypes.Member:
|
|
self._membership_stream_cache.entity_has_changed(state_key, stream_ordering)
|
|
self.get_invited_rooms_for_local_user.invalidate((state_key,))
|
|
|
|
if relates_to:
|
|
self.get_relations_for_event.invalidate_many((relates_to,))
|
|
self.get_aggregation_groups_for_event.invalidate_many((relates_to,))
|
|
self.get_applicable_edit.invalidate((relates_to,))
|