mirror of
https://mau.dev/maunium/synapse.git
synced 2024-07-16 16:12:44 +00:00
60a0f81c7a
synapse This is necessary for replicating the data in synapse to be visible to a separate service because presence and typing notifications aren't stored in a database so won't be visible to another process. This API can be used to either get the raw data by requesting the tables themselves or to just receive notifications for updates by following the streams meta-stream. Returns updates for each table requested a JSON array of arrays with a row for each row in the table. Each table is prefixed by a header row with the: name of the table, current stream_id position for the table, number of rows, number of columns and the names of the columns. This is followed by the rows that have been added to the server since the requester last asked. The API has a timeout and is hooked up to the notifier so that a slave can long poll for updates.
256 lines
9.3 KiB
Python
256 lines
9.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright 2014-2016 OpenMarket 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.
|
|
|
|
from ._base import SQLBaseStore
|
|
from synapse.api.constants import PresenceState
|
|
from synapse.util.caches.descriptors import cached, cachedInlineCallbacks
|
|
|
|
from collections import namedtuple
|
|
from twisted.internet import defer
|
|
|
|
|
|
class UserPresenceState(namedtuple("UserPresenceState",
|
|
("user_id", "state", "last_active_ts",
|
|
"last_federation_update_ts", "last_user_sync_ts",
|
|
"status_msg", "currently_active"))):
|
|
"""Represents the current presence state of the user.
|
|
|
|
user_id (str)
|
|
last_active (int): Time in msec that the user last interacted with server.
|
|
last_federation_update (int): Time in msec since either a) we sent a presence
|
|
update to other servers or b) we received a presence update, depending
|
|
on if is a local user or not.
|
|
last_user_sync (int): Time in msec that the user last *completed* a sync
|
|
(or event stream).
|
|
status_msg (str): User set status message.
|
|
"""
|
|
|
|
def copy_and_replace(self, **kwargs):
|
|
return self._replace(**kwargs)
|
|
|
|
@classmethod
|
|
def default(cls, user_id):
|
|
"""Returns a default presence state.
|
|
"""
|
|
return cls(
|
|
user_id=user_id,
|
|
state=PresenceState.OFFLINE,
|
|
last_active_ts=0,
|
|
last_federation_update_ts=0,
|
|
last_user_sync_ts=0,
|
|
status_msg=None,
|
|
currently_active=False,
|
|
)
|
|
|
|
|
|
class PresenceStore(SQLBaseStore):
|
|
@defer.inlineCallbacks
|
|
def update_presence(self, presence_states):
|
|
stream_ordering_manager = yield self._presence_id_gen.get_next_mult(
|
|
self, len(presence_states)
|
|
)
|
|
|
|
with stream_ordering_manager as stream_orderings:
|
|
yield self.runInteraction(
|
|
"update_presence",
|
|
self._update_presence_txn, stream_orderings, presence_states,
|
|
)
|
|
|
|
defer.returnValue((stream_orderings[-1], self._presence_id_gen.get_max_token()))
|
|
|
|
def _update_presence_txn(self, txn, stream_orderings, presence_states):
|
|
for stream_id, state in zip(stream_orderings, presence_states):
|
|
txn.call_after(
|
|
self.presence_stream_cache.entity_has_changed,
|
|
state.user_id, stream_id,
|
|
)
|
|
|
|
# Actually insert new rows
|
|
self._simple_insert_many_txn(
|
|
txn,
|
|
table="presence_stream",
|
|
values=[
|
|
{
|
|
"stream_id": stream_id,
|
|
"user_id": state.user_id,
|
|
"state": state.state,
|
|
"last_active_ts": state.last_active_ts,
|
|
"last_federation_update_ts": state.last_federation_update_ts,
|
|
"last_user_sync_ts": state.last_user_sync_ts,
|
|
"status_msg": state.status_msg,
|
|
"currently_active": state.currently_active,
|
|
}
|
|
for state in presence_states
|
|
],
|
|
)
|
|
|
|
# Delete old rows to stop database from getting really big
|
|
sql = (
|
|
"DELETE FROM presence_stream WHERE"
|
|
" stream_id < ?"
|
|
" AND user_id IN (%s)"
|
|
)
|
|
|
|
batches = (
|
|
presence_states[i:i + 50]
|
|
for i in xrange(0, len(presence_states), 50)
|
|
)
|
|
for states in batches:
|
|
args = [stream_id]
|
|
args.extend(s.user_id for s in states)
|
|
txn.execute(
|
|
sql % (",".join("?" for _ in states),),
|
|
args
|
|
)
|
|
|
|
def get_all_presence_updates(self, last_id, current_id):
|
|
def get_all_presence_updates_txn(txn):
|
|
sql = (
|
|
"SELECT stream_id, user_id, state, last_active_ts,"
|
|
" last_federation_update_ts, last_user_sync_ts, status_msg,"
|
|
" currently_active"
|
|
" FROM presence_stream"
|
|
" WHERE ? < stream_id AND stream_id <= ?"
|
|
)
|
|
txn.execute(sql, (last_id, current_id))
|
|
return txn.fetchall()
|
|
|
|
return self.runInteraction(
|
|
"get_all_presence_updates", get_all_presence_updates_txn
|
|
)
|
|
|
|
@defer.inlineCallbacks
|
|
def get_presence_for_users(self, user_ids):
|
|
rows = yield self._simple_select_many_batch(
|
|
table="presence_stream",
|
|
column="user_id",
|
|
iterable=user_ids,
|
|
keyvalues={},
|
|
retcols=(
|
|
"user_id",
|
|
"state",
|
|
"last_active_ts",
|
|
"last_federation_update_ts",
|
|
"last_user_sync_ts",
|
|
"status_msg",
|
|
"currently_active",
|
|
),
|
|
)
|
|
|
|
for row in rows:
|
|
row["currently_active"] = bool(row["currently_active"])
|
|
|
|
defer.returnValue([UserPresenceState(**row) for row in rows])
|
|
|
|
def get_current_presence_token(self):
|
|
return self._presence_id_gen.get_max_token()
|
|
|
|
def allow_presence_visible(self, observed_localpart, observer_userid):
|
|
return self._simple_insert(
|
|
table="presence_allow_inbound",
|
|
values={"observed_user_id": observed_localpart,
|
|
"observer_user_id": observer_userid},
|
|
desc="allow_presence_visible",
|
|
or_ignore=True,
|
|
)
|
|
|
|
def disallow_presence_visible(self, observed_localpart, observer_userid):
|
|
return self._simple_delete_one(
|
|
table="presence_allow_inbound",
|
|
keyvalues={"observed_user_id": observed_localpart,
|
|
"observer_user_id": observer_userid},
|
|
desc="disallow_presence_visible",
|
|
)
|
|
|
|
def is_presence_visible(self, observed_localpart, observer_userid):
|
|
return self._simple_select_one(
|
|
table="presence_allow_inbound",
|
|
keyvalues={"observed_user_id": observed_localpart,
|
|
"observer_user_id": observer_userid},
|
|
retcols=["observed_user_id"],
|
|
allow_none=True,
|
|
desc="is_presence_visible",
|
|
)
|
|
|
|
def add_presence_list_pending(self, observer_localpart, observed_userid):
|
|
return self._simple_insert(
|
|
table="presence_list",
|
|
values={"user_id": observer_localpart,
|
|
"observed_user_id": observed_userid,
|
|
"accepted": False},
|
|
desc="add_presence_list_pending",
|
|
)
|
|
|
|
@defer.inlineCallbacks
|
|
def set_presence_list_accepted(self, observer_localpart, observed_userid):
|
|
result = yield self._simple_update_one(
|
|
table="presence_list",
|
|
keyvalues={"user_id": observer_localpart,
|
|
"observed_user_id": observed_userid},
|
|
updatevalues={"accepted": True},
|
|
desc="set_presence_list_accepted",
|
|
)
|
|
self.get_presence_list_accepted.invalidate((observer_localpart,))
|
|
self.get_presence_list_observers_accepted.invalidate((observed_userid,))
|
|
defer.returnValue(result)
|
|
|
|
def get_presence_list(self, observer_localpart, accepted=None):
|
|
if accepted:
|
|
return self.get_presence_list_accepted(observer_localpart)
|
|
else:
|
|
keyvalues = {"user_id": observer_localpart}
|
|
if accepted is not None:
|
|
keyvalues["accepted"] = accepted
|
|
|
|
return self._simple_select_list(
|
|
table="presence_list",
|
|
keyvalues=keyvalues,
|
|
retcols=["observed_user_id", "accepted"],
|
|
desc="get_presence_list",
|
|
)
|
|
|
|
@cached()
|
|
def get_presence_list_accepted(self, observer_localpart):
|
|
return self._simple_select_list(
|
|
table="presence_list",
|
|
keyvalues={"user_id": observer_localpart, "accepted": True},
|
|
retcols=["observed_user_id", "accepted"],
|
|
desc="get_presence_list_accepted",
|
|
)
|
|
|
|
@cachedInlineCallbacks()
|
|
def get_presence_list_observers_accepted(self, observed_userid):
|
|
user_localparts = yield self._simple_select_onecol(
|
|
table="presence_list",
|
|
keyvalues={"observed_user_id": observed_userid, "accepted": True},
|
|
retcol="user_id",
|
|
desc="get_presence_list_accepted",
|
|
)
|
|
|
|
defer.returnValue([
|
|
"@%s:%s" % (u, self.hs.hostname,) for u in user_localparts
|
|
])
|
|
|
|
@defer.inlineCallbacks
|
|
def del_presence_list(self, observer_localpart, observed_userid):
|
|
yield self._simple_delete_one(
|
|
table="presence_list",
|
|
keyvalues={"user_id": observer_localpart,
|
|
"observed_user_id": observed_userid},
|
|
desc="del_presence_list",
|
|
)
|
|
self.get_presence_list_accepted.invalidate((observer_localpart,))
|
|
self.get_presence_list_observers_accepted.invalidate((observed_userid,))
|