mirror of
https://git.anonymousland.org/anonymousland/synapse.git
synced 2024-09-07 16:11:58 +00:00
Merge worker apps into one. (#6964)
This commit is contained in:
parent
4aea0bd292
commit
bbf8886a05
1
changelog.d/6964.misc
Normal file
1
changelog.d/6964.misc
Normal file
@ -0,0 +1 @@
|
|||||||
|
Merge worker apps together.
|
@ -13,161 +13,11 @@
|
|||||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
# See the License for the specific language governing permissions and
|
# See the License for the specific language governing permissions and
|
||||||
# limitations under the License.
|
# limitations under the License.
|
||||||
import logging
|
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
from twisted.internet import defer, reactor
|
from synapse.app.generic_worker import start
|
||||||
from twisted.web.resource import NoResource
|
from synapse.util.logcontext import LoggingContext
|
||||||
|
|
||||||
import synapse
|
|
||||||
from synapse import events
|
|
||||||
from synapse.app import _base
|
|
||||||
from synapse.config._base import ConfigError
|
|
||||||
from synapse.config.homeserver import HomeServerConfig
|
|
||||||
from synapse.config.logger import setup_logging
|
|
||||||
from synapse.http.site import SynapseSite
|
|
||||||
from synapse.logging.context import LoggingContext, run_in_background
|
|
||||||
from synapse.metrics import METRICS_PREFIX, MetricsResource, RegistryProxy
|
|
||||||
from synapse.replication.slave.storage.appservice import SlavedApplicationServiceStore
|
|
||||||
from synapse.replication.slave.storage.directory import DirectoryStore
|
|
||||||
from synapse.replication.slave.storage.events import SlavedEventStore
|
|
||||||
from synapse.replication.slave.storage.registration import SlavedRegistrationStore
|
|
||||||
from synapse.replication.tcp.client import ReplicationClientHandler
|
|
||||||
from synapse.server import HomeServer
|
|
||||||
from synapse.util.httpresourcetree import create_resource_tree
|
|
||||||
from synapse.util.manhole import manhole
|
|
||||||
from synapse.util.versionstring import get_version_string
|
|
||||||
|
|
||||||
logger = logging.getLogger("synapse.app.appservice")
|
|
||||||
|
|
||||||
|
|
||||||
class AppserviceSlaveStore(
|
|
||||||
DirectoryStore,
|
|
||||||
SlavedEventStore,
|
|
||||||
SlavedApplicationServiceStore,
|
|
||||||
SlavedRegistrationStore,
|
|
||||||
):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
class AppserviceServer(HomeServer):
|
|
||||||
DATASTORE_CLASS = AppserviceSlaveStore
|
|
||||||
|
|
||||||
def _listen_http(self, listener_config):
|
|
||||||
port = listener_config["port"]
|
|
||||||
bind_addresses = listener_config["bind_addresses"]
|
|
||||||
site_tag = listener_config.get("tag", port)
|
|
||||||
resources = {}
|
|
||||||
for res in listener_config["resources"]:
|
|
||||||
for name in res["names"]:
|
|
||||||
if name == "metrics":
|
|
||||||
resources[METRICS_PREFIX] = MetricsResource(RegistryProxy)
|
|
||||||
|
|
||||||
root_resource = create_resource_tree(resources, NoResource())
|
|
||||||
|
|
||||||
_base.listen_tcp(
|
|
||||||
bind_addresses,
|
|
||||||
port,
|
|
||||||
SynapseSite(
|
|
||||||
"synapse.access.http.%s" % (site_tag,),
|
|
||||||
site_tag,
|
|
||||||
listener_config,
|
|
||||||
root_resource,
|
|
||||||
self.version_string,
|
|
||||||
),
|
|
||||||
)
|
|
||||||
|
|
||||||
logger.info("Synapse appservice now listening on port %d", port)
|
|
||||||
|
|
||||||
def start_listening(self, listeners):
|
|
||||||
for listener in listeners:
|
|
||||||
if listener["type"] == "http":
|
|
||||||
self._listen_http(listener)
|
|
||||||
elif listener["type"] == "manhole":
|
|
||||||
_base.listen_tcp(
|
|
||||||
listener["bind_addresses"],
|
|
||||||
listener["port"],
|
|
||||||
manhole(
|
|
||||||
username="matrix", password="rabbithole", globals={"hs": self}
|
|
||||||
),
|
|
||||||
)
|
|
||||||
elif listener["type"] == "metrics":
|
|
||||||
if not self.get_config().enable_metrics:
|
|
||||||
logger.warning(
|
|
||||||
(
|
|
||||||
"Metrics listener configured, but "
|
|
||||||
"enable_metrics is not True!"
|
|
||||||
)
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
_base.listen_metrics(listener["bind_addresses"], listener["port"])
|
|
||||||
else:
|
|
||||||
logger.warning("Unrecognized listener type: %s", listener["type"])
|
|
||||||
|
|
||||||
self.get_tcp_replication().start_replication(self)
|
|
||||||
|
|
||||||
def build_tcp_replication(self):
|
|
||||||
return ASReplicationHandler(self)
|
|
||||||
|
|
||||||
|
|
||||||
class ASReplicationHandler(ReplicationClientHandler):
|
|
||||||
def __init__(self, hs):
|
|
||||||
super(ASReplicationHandler, self).__init__(hs.get_datastore())
|
|
||||||
self.appservice_handler = hs.get_application_service_handler()
|
|
||||||
|
|
||||||
async def on_rdata(self, stream_name, token, rows):
|
|
||||||
await super(ASReplicationHandler, self).on_rdata(stream_name, token, rows)
|
|
||||||
|
|
||||||
if stream_name == "events":
|
|
||||||
max_stream_id = self.store.get_room_max_stream_ordering()
|
|
||||||
run_in_background(self._notify_app_services, max_stream_id)
|
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
|
||||||
def _notify_app_services(self, room_stream_id):
|
|
||||||
try:
|
|
||||||
yield self.appservice_handler.notify_interested_services(room_stream_id)
|
|
||||||
except Exception:
|
|
||||||
logger.exception("Error notifying application services of event")
|
|
||||||
|
|
||||||
|
|
||||||
def start(config_options):
|
|
||||||
try:
|
|
||||||
config = HomeServerConfig.load_config("Synapse appservice", config_options)
|
|
||||||
except ConfigError as e:
|
|
||||||
sys.stderr.write("\n" + str(e) + "\n")
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
assert config.worker_app == "synapse.app.appservice"
|
|
||||||
|
|
||||||
events.USE_FROZEN_DICTS = config.use_frozen_dicts
|
|
||||||
|
|
||||||
if config.notify_appservices:
|
|
||||||
sys.stderr.write(
|
|
||||||
"\nThe appservices must be disabled in the main synapse process"
|
|
||||||
"\nbefore they can be run in a separate worker."
|
|
||||||
"\nPlease add ``notify_appservices: false`` to the main config"
|
|
||||||
"\n"
|
|
||||||
)
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
# Force the pushers to start since they will be disabled in the main config
|
|
||||||
config.notify_appservices = True
|
|
||||||
|
|
||||||
ps = AppserviceServer(
|
|
||||||
config.server_name,
|
|
||||||
config=config,
|
|
||||||
version_string="Synapse/" + get_version_string(synapse),
|
|
||||||
)
|
|
||||||
|
|
||||||
setup_logging(ps, config, use_worker_options=True)
|
|
||||||
|
|
||||||
ps.setup()
|
|
||||||
reactor.addSystemEventTrigger(
|
|
||||||
"before", "startup", _base.start, ps, config.worker_listeners
|
|
||||||
)
|
|
||||||
|
|
||||||
_base.start_worker_reactor("synapse-appservice", config)
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
with LoggingContext("main"):
|
with LoggingContext("main"):
|
||||||
|
@ -13,195 +13,11 @@
|
|||||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
# See the License for the specific language governing permissions and
|
# See the License for the specific language governing permissions and
|
||||||
# limitations under the License.
|
# limitations under the License.
|
||||||
import logging
|
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
from twisted.internet import reactor
|
from synapse.app.generic_worker import start
|
||||||
from twisted.web.resource import NoResource
|
from synapse.util.logcontext import LoggingContext
|
||||||
|
|
||||||
import synapse
|
|
||||||
from synapse import events
|
|
||||||
from synapse.app import _base
|
|
||||||
from synapse.config._base import ConfigError
|
|
||||||
from synapse.config.homeserver import HomeServerConfig
|
|
||||||
from synapse.config.logger import setup_logging
|
|
||||||
from synapse.http.server import JsonResource
|
|
||||||
from synapse.http.site import SynapseSite
|
|
||||||
from synapse.logging.context import LoggingContext
|
|
||||||
from synapse.metrics import METRICS_PREFIX, MetricsResource, RegistryProxy
|
|
||||||
from synapse.replication.slave.storage._base import BaseSlavedStore
|
|
||||||
from synapse.replication.slave.storage.account_data import SlavedAccountDataStore
|
|
||||||
from synapse.replication.slave.storage.appservice import SlavedApplicationServiceStore
|
|
||||||
from synapse.replication.slave.storage.client_ips import SlavedClientIpStore
|
|
||||||
from synapse.replication.slave.storage.deviceinbox import SlavedDeviceInboxStore
|
|
||||||
from synapse.replication.slave.storage.devices import SlavedDeviceStore
|
|
||||||
from synapse.replication.slave.storage.directory import DirectoryStore
|
|
||||||
from synapse.replication.slave.storage.events import SlavedEventStore
|
|
||||||
from synapse.replication.slave.storage.groups import SlavedGroupServerStore
|
|
||||||
from synapse.replication.slave.storage.keys import SlavedKeyStore
|
|
||||||
from synapse.replication.slave.storage.profile import SlavedProfileStore
|
|
||||||
from synapse.replication.slave.storage.push_rule import SlavedPushRuleStore
|
|
||||||
from synapse.replication.slave.storage.receipts import SlavedReceiptsStore
|
|
||||||
from synapse.replication.slave.storage.registration import SlavedRegistrationStore
|
|
||||||
from synapse.replication.slave.storage.room import RoomStore
|
|
||||||
from synapse.replication.slave.storage.transactions import SlavedTransactionStore
|
|
||||||
from synapse.replication.tcp.client import ReplicationClientHandler
|
|
||||||
from synapse.rest.client.v1.login import LoginRestServlet
|
|
||||||
from synapse.rest.client.v1.push_rule import PushRuleRestServlet
|
|
||||||
from synapse.rest.client.v1.room import (
|
|
||||||
JoinedRoomMemberListRestServlet,
|
|
||||||
PublicRoomListRestServlet,
|
|
||||||
RoomEventContextServlet,
|
|
||||||
RoomMemberListRestServlet,
|
|
||||||
RoomMessageListRestServlet,
|
|
||||||
RoomStateRestServlet,
|
|
||||||
)
|
|
||||||
from synapse.rest.client.v1.voip import VoipRestServlet
|
|
||||||
from synapse.rest.client.v2_alpha import groups
|
|
||||||
from synapse.rest.client.v2_alpha.account import ThreepidRestServlet
|
|
||||||
from synapse.rest.client.v2_alpha.keys import KeyChangesServlet, KeyQueryServlet
|
|
||||||
from synapse.rest.client.v2_alpha.register import RegisterRestServlet
|
|
||||||
from synapse.rest.client.versions import VersionsRestServlet
|
|
||||||
from synapse.server import HomeServer
|
|
||||||
from synapse.storage.data_stores.main.monthly_active_users import (
|
|
||||||
MonthlyActiveUsersWorkerStore,
|
|
||||||
)
|
|
||||||
from synapse.util.httpresourcetree import create_resource_tree
|
|
||||||
from synapse.util.manhole import manhole
|
|
||||||
from synapse.util.versionstring import get_version_string
|
|
||||||
|
|
||||||
logger = logging.getLogger("synapse.app.client_reader")
|
|
||||||
|
|
||||||
|
|
||||||
class ClientReaderSlavedStore(
|
|
||||||
SlavedDeviceInboxStore,
|
|
||||||
SlavedDeviceStore,
|
|
||||||
SlavedReceiptsStore,
|
|
||||||
SlavedPushRuleStore,
|
|
||||||
SlavedGroupServerStore,
|
|
||||||
SlavedAccountDataStore,
|
|
||||||
SlavedEventStore,
|
|
||||||
SlavedKeyStore,
|
|
||||||
RoomStore,
|
|
||||||
DirectoryStore,
|
|
||||||
SlavedApplicationServiceStore,
|
|
||||||
SlavedRegistrationStore,
|
|
||||||
SlavedTransactionStore,
|
|
||||||
SlavedProfileStore,
|
|
||||||
SlavedClientIpStore,
|
|
||||||
MonthlyActiveUsersWorkerStore,
|
|
||||||
BaseSlavedStore,
|
|
||||||
):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
class ClientReaderServer(HomeServer):
|
|
||||||
DATASTORE_CLASS = ClientReaderSlavedStore
|
|
||||||
|
|
||||||
def _listen_http(self, listener_config):
|
|
||||||
port = listener_config["port"]
|
|
||||||
bind_addresses = listener_config["bind_addresses"]
|
|
||||||
site_tag = listener_config.get("tag", port)
|
|
||||||
resources = {}
|
|
||||||
for res in listener_config["resources"]:
|
|
||||||
for name in res["names"]:
|
|
||||||
if name == "metrics":
|
|
||||||
resources[METRICS_PREFIX] = MetricsResource(RegistryProxy)
|
|
||||||
elif name == "client":
|
|
||||||
resource = JsonResource(self, canonical_json=False)
|
|
||||||
|
|
||||||
PublicRoomListRestServlet(self).register(resource)
|
|
||||||
RoomMemberListRestServlet(self).register(resource)
|
|
||||||
JoinedRoomMemberListRestServlet(self).register(resource)
|
|
||||||
RoomStateRestServlet(self).register(resource)
|
|
||||||
RoomEventContextServlet(self).register(resource)
|
|
||||||
RoomMessageListRestServlet(self).register(resource)
|
|
||||||
RegisterRestServlet(self).register(resource)
|
|
||||||
LoginRestServlet(self).register(resource)
|
|
||||||
ThreepidRestServlet(self).register(resource)
|
|
||||||
KeyQueryServlet(self).register(resource)
|
|
||||||
KeyChangesServlet(self).register(resource)
|
|
||||||
VoipRestServlet(self).register(resource)
|
|
||||||
PushRuleRestServlet(self).register(resource)
|
|
||||||
VersionsRestServlet(self).register(resource)
|
|
||||||
|
|
||||||
groups.register_servlets(self, resource)
|
|
||||||
|
|
||||||
resources.update({"/_matrix/client": resource})
|
|
||||||
|
|
||||||
root_resource = create_resource_tree(resources, NoResource())
|
|
||||||
|
|
||||||
_base.listen_tcp(
|
|
||||||
bind_addresses,
|
|
||||||
port,
|
|
||||||
SynapseSite(
|
|
||||||
"synapse.access.http.%s" % (site_tag,),
|
|
||||||
site_tag,
|
|
||||||
listener_config,
|
|
||||||
root_resource,
|
|
||||||
self.version_string,
|
|
||||||
),
|
|
||||||
)
|
|
||||||
|
|
||||||
logger.info("Synapse client reader now listening on port %d", port)
|
|
||||||
|
|
||||||
def start_listening(self, listeners):
|
|
||||||
for listener in listeners:
|
|
||||||
if listener["type"] == "http":
|
|
||||||
self._listen_http(listener)
|
|
||||||
elif listener["type"] == "manhole":
|
|
||||||
_base.listen_tcp(
|
|
||||||
listener["bind_addresses"],
|
|
||||||
listener["port"],
|
|
||||||
manhole(
|
|
||||||
username="matrix", password="rabbithole", globals={"hs": self}
|
|
||||||
),
|
|
||||||
)
|
|
||||||
elif listener["type"] == "metrics":
|
|
||||||
if not self.get_config().enable_metrics:
|
|
||||||
logger.warning(
|
|
||||||
(
|
|
||||||
"Metrics listener configured, but "
|
|
||||||
"enable_metrics is not True!"
|
|
||||||
)
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
_base.listen_metrics(listener["bind_addresses"], listener["port"])
|
|
||||||
else:
|
|
||||||
logger.warning("Unrecognized listener type: %s", listener["type"])
|
|
||||||
|
|
||||||
self.get_tcp_replication().start_replication(self)
|
|
||||||
|
|
||||||
def build_tcp_replication(self):
|
|
||||||
return ReplicationClientHandler(self.get_datastore())
|
|
||||||
|
|
||||||
|
|
||||||
def start(config_options):
|
|
||||||
try:
|
|
||||||
config = HomeServerConfig.load_config("Synapse client reader", config_options)
|
|
||||||
except ConfigError as e:
|
|
||||||
sys.stderr.write("\n" + str(e) + "\n")
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
assert config.worker_app == "synapse.app.client_reader"
|
|
||||||
|
|
||||||
events.USE_FROZEN_DICTS = config.use_frozen_dicts
|
|
||||||
|
|
||||||
ss = ClientReaderServer(
|
|
||||||
config.server_name,
|
|
||||||
config=config,
|
|
||||||
version_string="Synapse/" + get_version_string(synapse),
|
|
||||||
)
|
|
||||||
|
|
||||||
setup_logging(ss, config, use_worker_options=True)
|
|
||||||
|
|
||||||
ss.setup()
|
|
||||||
reactor.addSystemEventTrigger(
|
|
||||||
"before", "startup", _base.start, ss, config.worker_listeners
|
|
||||||
)
|
|
||||||
|
|
||||||
_base.start_worker_reactor("synapse-client-reader", config)
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
with LoggingContext("main"):
|
with LoggingContext("main"):
|
||||||
|
@ -13,191 +13,11 @@
|
|||||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
# See the License for the specific language governing permissions and
|
# See the License for the specific language governing permissions and
|
||||||
# limitations under the License.
|
# limitations under the License.
|
||||||
import logging
|
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
from twisted.internet import reactor
|
from synapse.app.generic_worker import start
|
||||||
from twisted.web.resource import NoResource
|
from synapse.util.logcontext import LoggingContext
|
||||||
|
|
||||||
import synapse
|
|
||||||
from synapse import events
|
|
||||||
from synapse.app import _base
|
|
||||||
from synapse.config._base import ConfigError
|
|
||||||
from synapse.config.homeserver import HomeServerConfig
|
|
||||||
from synapse.config.logger import setup_logging
|
|
||||||
from synapse.http.server import JsonResource
|
|
||||||
from synapse.http.site import SynapseSite
|
|
||||||
from synapse.logging.context import LoggingContext
|
|
||||||
from synapse.metrics import METRICS_PREFIX, MetricsResource, RegistryProxy
|
|
||||||
from synapse.replication.slave.storage._base import BaseSlavedStore
|
|
||||||
from synapse.replication.slave.storage.account_data import SlavedAccountDataStore
|
|
||||||
from synapse.replication.slave.storage.appservice import SlavedApplicationServiceStore
|
|
||||||
from synapse.replication.slave.storage.client_ips import SlavedClientIpStore
|
|
||||||
from synapse.replication.slave.storage.devices import SlavedDeviceStore
|
|
||||||
from synapse.replication.slave.storage.directory import DirectoryStore
|
|
||||||
from synapse.replication.slave.storage.events import SlavedEventStore
|
|
||||||
from synapse.replication.slave.storage.profile import SlavedProfileStore
|
|
||||||
from synapse.replication.slave.storage.push_rule import SlavedPushRuleStore
|
|
||||||
from synapse.replication.slave.storage.pushers import SlavedPusherStore
|
|
||||||
from synapse.replication.slave.storage.receipts import SlavedReceiptsStore
|
|
||||||
from synapse.replication.slave.storage.registration import SlavedRegistrationStore
|
|
||||||
from synapse.replication.slave.storage.room import RoomStore
|
|
||||||
from synapse.replication.slave.storage.transactions import SlavedTransactionStore
|
|
||||||
from synapse.replication.tcp.client import ReplicationClientHandler
|
|
||||||
from synapse.rest.client.v1.profile import (
|
|
||||||
ProfileAvatarURLRestServlet,
|
|
||||||
ProfileDisplaynameRestServlet,
|
|
||||||
ProfileRestServlet,
|
|
||||||
)
|
|
||||||
from synapse.rest.client.v1.room import (
|
|
||||||
JoinRoomAliasServlet,
|
|
||||||
RoomMembershipRestServlet,
|
|
||||||
RoomSendEventRestServlet,
|
|
||||||
RoomStateEventRestServlet,
|
|
||||||
)
|
|
||||||
from synapse.server import HomeServer
|
|
||||||
from synapse.storage.data_stores.main.monthly_active_users import (
|
|
||||||
MonthlyActiveUsersWorkerStore,
|
|
||||||
)
|
|
||||||
from synapse.storage.data_stores.main.user_directory import UserDirectoryStore
|
|
||||||
from synapse.util.httpresourcetree import create_resource_tree
|
|
||||||
from synapse.util.manhole import manhole
|
|
||||||
from synapse.util.versionstring import get_version_string
|
|
||||||
|
|
||||||
logger = logging.getLogger("synapse.app.event_creator")
|
|
||||||
|
|
||||||
|
|
||||||
class EventCreatorSlavedStore(
|
|
||||||
# FIXME(#3714): We need to add UserDirectoryStore as we write directly
|
|
||||||
# rather than going via the correct worker.
|
|
||||||
UserDirectoryStore,
|
|
||||||
DirectoryStore,
|
|
||||||
SlavedTransactionStore,
|
|
||||||
SlavedProfileStore,
|
|
||||||
SlavedAccountDataStore,
|
|
||||||
SlavedPusherStore,
|
|
||||||
SlavedReceiptsStore,
|
|
||||||
SlavedPushRuleStore,
|
|
||||||
SlavedDeviceStore,
|
|
||||||
SlavedClientIpStore,
|
|
||||||
SlavedApplicationServiceStore,
|
|
||||||
SlavedEventStore,
|
|
||||||
SlavedRegistrationStore,
|
|
||||||
RoomStore,
|
|
||||||
MonthlyActiveUsersWorkerStore,
|
|
||||||
BaseSlavedStore,
|
|
||||||
):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
class EventCreatorServer(HomeServer):
|
|
||||||
DATASTORE_CLASS = EventCreatorSlavedStore
|
|
||||||
|
|
||||||
def _listen_http(self, listener_config):
|
|
||||||
port = listener_config["port"]
|
|
||||||
bind_addresses = listener_config["bind_addresses"]
|
|
||||||
site_tag = listener_config.get("tag", port)
|
|
||||||
resources = {}
|
|
||||||
for res in listener_config["resources"]:
|
|
||||||
for name in res["names"]:
|
|
||||||
if name == "metrics":
|
|
||||||
resources[METRICS_PREFIX] = MetricsResource(RegistryProxy)
|
|
||||||
elif name == "client":
|
|
||||||
resource = JsonResource(self, canonical_json=False)
|
|
||||||
RoomSendEventRestServlet(self).register(resource)
|
|
||||||
RoomMembershipRestServlet(self).register(resource)
|
|
||||||
RoomStateEventRestServlet(self).register(resource)
|
|
||||||
JoinRoomAliasServlet(self).register(resource)
|
|
||||||
ProfileAvatarURLRestServlet(self).register(resource)
|
|
||||||
ProfileDisplaynameRestServlet(self).register(resource)
|
|
||||||
ProfileRestServlet(self).register(resource)
|
|
||||||
resources.update(
|
|
||||||
{
|
|
||||||
"/_matrix/client/r0": resource,
|
|
||||||
"/_matrix/client/unstable": resource,
|
|
||||||
"/_matrix/client/v2_alpha": resource,
|
|
||||||
"/_matrix/client/api/v1": resource,
|
|
||||||
}
|
|
||||||
)
|
|
||||||
|
|
||||||
root_resource = create_resource_tree(resources, NoResource())
|
|
||||||
|
|
||||||
_base.listen_tcp(
|
|
||||||
bind_addresses,
|
|
||||||
port,
|
|
||||||
SynapseSite(
|
|
||||||
"synapse.access.http.%s" % (site_tag,),
|
|
||||||
site_tag,
|
|
||||||
listener_config,
|
|
||||||
root_resource,
|
|
||||||
self.version_string,
|
|
||||||
),
|
|
||||||
)
|
|
||||||
|
|
||||||
logger.info("Synapse event creator now listening on port %d", port)
|
|
||||||
|
|
||||||
def start_listening(self, listeners):
|
|
||||||
for listener in listeners:
|
|
||||||
if listener["type"] == "http":
|
|
||||||
self._listen_http(listener)
|
|
||||||
elif listener["type"] == "manhole":
|
|
||||||
_base.listen_tcp(
|
|
||||||
listener["bind_addresses"],
|
|
||||||
listener["port"],
|
|
||||||
manhole(
|
|
||||||
username="matrix", password="rabbithole", globals={"hs": self}
|
|
||||||
),
|
|
||||||
)
|
|
||||||
elif listener["type"] == "metrics":
|
|
||||||
if not self.get_config().enable_metrics:
|
|
||||||
logger.warning(
|
|
||||||
(
|
|
||||||
"Metrics listener configured, but "
|
|
||||||
"enable_metrics is not True!"
|
|
||||||
)
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
_base.listen_metrics(listener["bind_addresses"], listener["port"])
|
|
||||||
else:
|
|
||||||
logger.warning("Unrecognized listener type: %s", listener["type"])
|
|
||||||
|
|
||||||
self.get_tcp_replication().start_replication(self)
|
|
||||||
|
|
||||||
def build_tcp_replication(self):
|
|
||||||
return ReplicationClientHandler(self.get_datastore())
|
|
||||||
|
|
||||||
|
|
||||||
def start(config_options):
|
|
||||||
try:
|
|
||||||
config = HomeServerConfig.load_config("Synapse event creator", config_options)
|
|
||||||
except ConfigError as e:
|
|
||||||
sys.stderr.write("\n" + str(e) + "\n")
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
assert config.worker_app == "synapse.app.event_creator"
|
|
||||||
|
|
||||||
assert config.worker_replication_http_port is not None
|
|
||||||
|
|
||||||
# This should only be done on the user directory worker or the master
|
|
||||||
config.update_user_directory = False
|
|
||||||
|
|
||||||
events.USE_FROZEN_DICTS = config.use_frozen_dicts
|
|
||||||
|
|
||||||
ss = EventCreatorServer(
|
|
||||||
config.server_name,
|
|
||||||
config=config,
|
|
||||||
version_string="Synapse/" + get_version_string(synapse),
|
|
||||||
)
|
|
||||||
|
|
||||||
setup_logging(ss, config, use_worker_options=True)
|
|
||||||
|
|
||||||
ss.setup()
|
|
||||||
reactor.addSystemEventTrigger(
|
|
||||||
"before", "startup", _base.start, ss, config.worker_listeners
|
|
||||||
)
|
|
||||||
|
|
||||||
_base.start_worker_reactor("synapse-event-creator", config)
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
with LoggingContext("main"):
|
with LoggingContext("main"):
|
||||||
|
@ -13,177 +13,11 @@
|
|||||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
# See the License for the specific language governing permissions and
|
# See the License for the specific language governing permissions and
|
||||||
# limitations under the License.
|
# limitations under the License.
|
||||||
import logging
|
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
from twisted.internet import reactor
|
from synapse.app.generic_worker import start
|
||||||
from twisted.web.resource import NoResource
|
from synapse.util.logcontext import LoggingContext
|
||||||
|
|
||||||
import synapse
|
|
||||||
from synapse import events
|
|
||||||
from synapse.api.urls import FEDERATION_PREFIX, SERVER_KEY_V2_PREFIX
|
|
||||||
from synapse.app import _base
|
|
||||||
from synapse.config._base import ConfigError
|
|
||||||
from synapse.config.homeserver import HomeServerConfig
|
|
||||||
from synapse.config.logger import setup_logging
|
|
||||||
from synapse.federation.transport.server import TransportLayerServer
|
|
||||||
from synapse.http.site import SynapseSite
|
|
||||||
from synapse.logging.context import LoggingContext
|
|
||||||
from synapse.metrics import METRICS_PREFIX, MetricsResource, RegistryProxy
|
|
||||||
from synapse.replication.slave.storage._base import BaseSlavedStore
|
|
||||||
from synapse.replication.slave.storage.account_data import SlavedAccountDataStore
|
|
||||||
from synapse.replication.slave.storage.appservice import SlavedApplicationServiceStore
|
|
||||||
from synapse.replication.slave.storage.devices import SlavedDeviceStore
|
|
||||||
from synapse.replication.slave.storage.directory import DirectoryStore
|
|
||||||
from synapse.replication.slave.storage.events import SlavedEventStore
|
|
||||||
from synapse.replication.slave.storage.groups import SlavedGroupServerStore
|
|
||||||
from synapse.replication.slave.storage.keys import SlavedKeyStore
|
|
||||||
from synapse.replication.slave.storage.profile import SlavedProfileStore
|
|
||||||
from synapse.replication.slave.storage.push_rule import SlavedPushRuleStore
|
|
||||||
from synapse.replication.slave.storage.pushers import SlavedPusherStore
|
|
||||||
from synapse.replication.slave.storage.receipts import SlavedReceiptsStore
|
|
||||||
from synapse.replication.slave.storage.registration import SlavedRegistrationStore
|
|
||||||
from synapse.replication.slave.storage.room import RoomStore
|
|
||||||
from synapse.replication.slave.storage.transactions import SlavedTransactionStore
|
|
||||||
from synapse.replication.tcp.client import ReplicationClientHandler
|
|
||||||
from synapse.rest.key.v2 import KeyApiV2Resource
|
|
||||||
from synapse.server import HomeServer
|
|
||||||
from synapse.storage.data_stores.main.monthly_active_users import (
|
|
||||||
MonthlyActiveUsersWorkerStore,
|
|
||||||
)
|
|
||||||
from synapse.util.httpresourcetree import create_resource_tree
|
|
||||||
from synapse.util.manhole import manhole
|
|
||||||
from synapse.util.versionstring import get_version_string
|
|
||||||
|
|
||||||
logger = logging.getLogger("synapse.app.federation_reader")
|
|
||||||
|
|
||||||
|
|
||||||
class FederationReaderSlavedStore(
|
|
||||||
SlavedAccountDataStore,
|
|
||||||
SlavedProfileStore,
|
|
||||||
SlavedApplicationServiceStore,
|
|
||||||
SlavedPusherStore,
|
|
||||||
SlavedPushRuleStore,
|
|
||||||
SlavedReceiptsStore,
|
|
||||||
SlavedEventStore,
|
|
||||||
SlavedKeyStore,
|
|
||||||
SlavedRegistrationStore,
|
|
||||||
SlavedGroupServerStore,
|
|
||||||
SlavedDeviceStore,
|
|
||||||
RoomStore,
|
|
||||||
DirectoryStore,
|
|
||||||
SlavedTransactionStore,
|
|
||||||
MonthlyActiveUsersWorkerStore,
|
|
||||||
BaseSlavedStore,
|
|
||||||
):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
class FederationReaderServer(HomeServer):
|
|
||||||
DATASTORE_CLASS = FederationReaderSlavedStore
|
|
||||||
|
|
||||||
def _listen_http(self, listener_config):
|
|
||||||
port = listener_config["port"]
|
|
||||||
bind_addresses = listener_config["bind_addresses"]
|
|
||||||
site_tag = listener_config.get("tag", port)
|
|
||||||
resources = {}
|
|
||||||
for res in listener_config["resources"]:
|
|
||||||
for name in res["names"]:
|
|
||||||
if name == "metrics":
|
|
||||||
resources[METRICS_PREFIX] = MetricsResource(RegistryProxy)
|
|
||||||
elif name == "federation":
|
|
||||||
resources.update({FEDERATION_PREFIX: TransportLayerServer(self)})
|
|
||||||
if name == "openid" and "federation" not in res["names"]:
|
|
||||||
# Only load the openid resource separately if federation resource
|
|
||||||
# is not specified since federation resource includes openid
|
|
||||||
# resource.
|
|
||||||
resources.update(
|
|
||||||
{
|
|
||||||
FEDERATION_PREFIX: TransportLayerServer(
|
|
||||||
self, servlet_groups=["openid"]
|
|
||||||
)
|
|
||||||
}
|
|
||||||
)
|
|
||||||
|
|
||||||
if name in ["keys", "federation"]:
|
|
||||||
resources[SERVER_KEY_V2_PREFIX] = KeyApiV2Resource(self)
|
|
||||||
|
|
||||||
root_resource = create_resource_tree(resources, NoResource())
|
|
||||||
|
|
||||||
_base.listen_tcp(
|
|
||||||
bind_addresses,
|
|
||||||
port,
|
|
||||||
SynapseSite(
|
|
||||||
"synapse.access.http.%s" % (site_tag,),
|
|
||||||
site_tag,
|
|
||||||
listener_config,
|
|
||||||
root_resource,
|
|
||||||
self.version_string,
|
|
||||||
),
|
|
||||||
reactor=self.get_reactor(),
|
|
||||||
)
|
|
||||||
|
|
||||||
logger.info("Synapse federation reader now listening on port %d", port)
|
|
||||||
|
|
||||||
def start_listening(self, listeners):
|
|
||||||
for listener in listeners:
|
|
||||||
if listener["type"] == "http":
|
|
||||||
self._listen_http(listener)
|
|
||||||
elif listener["type"] == "manhole":
|
|
||||||
_base.listen_tcp(
|
|
||||||
listener["bind_addresses"],
|
|
||||||
listener["port"],
|
|
||||||
manhole(
|
|
||||||
username="matrix", password="rabbithole", globals={"hs": self}
|
|
||||||
),
|
|
||||||
)
|
|
||||||
elif listener["type"] == "metrics":
|
|
||||||
if not self.get_config().enable_metrics:
|
|
||||||
logger.warning(
|
|
||||||
(
|
|
||||||
"Metrics listener configured, but "
|
|
||||||
"enable_metrics is not True!"
|
|
||||||
)
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
_base.listen_metrics(listener["bind_addresses"], listener["port"])
|
|
||||||
else:
|
|
||||||
logger.warning("Unrecognized listener type: %s", listener["type"])
|
|
||||||
|
|
||||||
self.get_tcp_replication().start_replication(self)
|
|
||||||
|
|
||||||
def build_tcp_replication(self):
|
|
||||||
return ReplicationClientHandler(self.get_datastore())
|
|
||||||
|
|
||||||
|
|
||||||
def start(config_options):
|
|
||||||
try:
|
|
||||||
config = HomeServerConfig.load_config(
|
|
||||||
"Synapse federation reader", config_options
|
|
||||||
)
|
|
||||||
except ConfigError as e:
|
|
||||||
sys.stderr.write("\n" + str(e) + "\n")
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
assert config.worker_app == "synapse.app.federation_reader"
|
|
||||||
|
|
||||||
events.USE_FROZEN_DICTS = config.use_frozen_dicts
|
|
||||||
|
|
||||||
ss = FederationReaderServer(
|
|
||||||
config.server_name,
|
|
||||||
config=config,
|
|
||||||
version_string="Synapse/" + get_version_string(synapse),
|
|
||||||
)
|
|
||||||
|
|
||||||
setup_logging(ss, config, use_worker_options=True)
|
|
||||||
|
|
||||||
ss.setup()
|
|
||||||
reactor.addSystemEventTrigger(
|
|
||||||
"before", "startup", _base.start, ss, config.worker_listeners
|
|
||||||
)
|
|
||||||
|
|
||||||
_base.start_worker_reactor("synapse-federation-reader", config)
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
with LoggingContext("main"):
|
with LoggingContext("main"):
|
||||||
|
@ -13,308 +13,11 @@
|
|||||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
# See the License for the specific language governing permissions and
|
# See the License for the specific language governing permissions and
|
||||||
# limitations under the License.
|
# limitations under the License.
|
||||||
import logging
|
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
from twisted.internet import defer, reactor
|
from synapse.app.generic_worker import start
|
||||||
from twisted.web.resource import NoResource
|
from synapse.util.logcontext import LoggingContext
|
||||||
|
|
||||||
import synapse
|
|
||||||
from synapse import events
|
|
||||||
from synapse.app import _base
|
|
||||||
from synapse.config._base import ConfigError
|
|
||||||
from synapse.config.homeserver import HomeServerConfig
|
|
||||||
from synapse.config.logger import setup_logging
|
|
||||||
from synapse.federation import send_queue
|
|
||||||
from synapse.http.site import SynapseSite
|
|
||||||
from synapse.logging.context import LoggingContext, run_in_background
|
|
||||||
from synapse.metrics import METRICS_PREFIX, MetricsResource, RegistryProxy
|
|
||||||
from synapse.metrics.background_process_metrics import run_as_background_process
|
|
||||||
from synapse.replication.slave.storage.deviceinbox import SlavedDeviceInboxStore
|
|
||||||
from synapse.replication.slave.storage.devices import SlavedDeviceStore
|
|
||||||
from synapse.replication.slave.storage.events import SlavedEventStore
|
|
||||||
from synapse.replication.slave.storage.presence import SlavedPresenceStore
|
|
||||||
from synapse.replication.slave.storage.receipts import SlavedReceiptsStore
|
|
||||||
from synapse.replication.slave.storage.registration import SlavedRegistrationStore
|
|
||||||
from synapse.replication.slave.storage.transactions import SlavedTransactionStore
|
|
||||||
from synapse.replication.tcp.client import ReplicationClientHandler
|
|
||||||
from synapse.replication.tcp.streams._base import (
|
|
||||||
DeviceListsStream,
|
|
||||||
ReceiptsStream,
|
|
||||||
ToDeviceStream,
|
|
||||||
)
|
|
||||||
from synapse.server import HomeServer
|
|
||||||
from synapse.storage.database import Database
|
|
||||||
from synapse.types import ReadReceipt
|
|
||||||
from synapse.util.async_helpers import Linearizer
|
|
||||||
from synapse.util.httpresourcetree import create_resource_tree
|
|
||||||
from synapse.util.manhole import manhole
|
|
||||||
from synapse.util.versionstring import get_version_string
|
|
||||||
|
|
||||||
logger = logging.getLogger("synapse.app.federation_sender")
|
|
||||||
|
|
||||||
|
|
||||||
class FederationSenderSlaveStore(
|
|
||||||
SlavedDeviceInboxStore,
|
|
||||||
SlavedTransactionStore,
|
|
||||||
SlavedReceiptsStore,
|
|
||||||
SlavedEventStore,
|
|
||||||
SlavedRegistrationStore,
|
|
||||||
SlavedDeviceStore,
|
|
||||||
SlavedPresenceStore,
|
|
||||||
):
|
|
||||||
def __init__(self, database: Database, db_conn, hs):
|
|
||||||
super(FederationSenderSlaveStore, self).__init__(database, db_conn, hs)
|
|
||||||
|
|
||||||
# We pull out the current federation stream position now so that we
|
|
||||||
# always have a known value for the federation position in memory so
|
|
||||||
# that we don't have to bounce via a deferred once when we start the
|
|
||||||
# replication streams.
|
|
||||||
self.federation_out_pos_startup = self._get_federation_out_pos(db_conn)
|
|
||||||
|
|
||||||
def _get_federation_out_pos(self, db_conn):
|
|
||||||
sql = "SELECT stream_id FROM federation_stream_position WHERE type = ?"
|
|
||||||
sql = self.database_engine.convert_param_style(sql)
|
|
||||||
|
|
||||||
txn = db_conn.cursor()
|
|
||||||
txn.execute(sql, ("federation",))
|
|
||||||
rows = txn.fetchall()
|
|
||||||
txn.close()
|
|
||||||
|
|
||||||
return rows[0][0] if rows else -1
|
|
||||||
|
|
||||||
|
|
||||||
class FederationSenderServer(HomeServer):
|
|
||||||
DATASTORE_CLASS = FederationSenderSlaveStore
|
|
||||||
|
|
||||||
def _listen_http(self, listener_config):
|
|
||||||
port = listener_config["port"]
|
|
||||||
bind_addresses = listener_config["bind_addresses"]
|
|
||||||
site_tag = listener_config.get("tag", port)
|
|
||||||
resources = {}
|
|
||||||
for res in listener_config["resources"]:
|
|
||||||
for name in res["names"]:
|
|
||||||
if name == "metrics":
|
|
||||||
resources[METRICS_PREFIX] = MetricsResource(RegistryProxy)
|
|
||||||
|
|
||||||
root_resource = create_resource_tree(resources, NoResource())
|
|
||||||
|
|
||||||
_base.listen_tcp(
|
|
||||||
bind_addresses,
|
|
||||||
port,
|
|
||||||
SynapseSite(
|
|
||||||
"synapse.access.http.%s" % (site_tag,),
|
|
||||||
site_tag,
|
|
||||||
listener_config,
|
|
||||||
root_resource,
|
|
||||||
self.version_string,
|
|
||||||
),
|
|
||||||
)
|
|
||||||
|
|
||||||
logger.info("Synapse federation_sender now listening on port %d", port)
|
|
||||||
|
|
||||||
def start_listening(self, listeners):
|
|
||||||
for listener in listeners:
|
|
||||||
if listener["type"] == "http":
|
|
||||||
self._listen_http(listener)
|
|
||||||
elif listener["type"] == "manhole":
|
|
||||||
_base.listen_tcp(
|
|
||||||
listener["bind_addresses"],
|
|
||||||
listener["port"],
|
|
||||||
manhole(
|
|
||||||
username="matrix", password="rabbithole", globals={"hs": self}
|
|
||||||
),
|
|
||||||
)
|
|
||||||
elif listener["type"] == "metrics":
|
|
||||||
if not self.get_config().enable_metrics:
|
|
||||||
logger.warning(
|
|
||||||
(
|
|
||||||
"Metrics listener configured, but "
|
|
||||||
"enable_metrics is not True!"
|
|
||||||
)
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
_base.listen_metrics(listener["bind_addresses"], listener["port"])
|
|
||||||
else:
|
|
||||||
logger.warning("Unrecognized listener type: %s", listener["type"])
|
|
||||||
|
|
||||||
self.get_tcp_replication().start_replication(self)
|
|
||||||
|
|
||||||
def build_tcp_replication(self):
|
|
||||||
return FederationSenderReplicationHandler(self)
|
|
||||||
|
|
||||||
|
|
||||||
class FederationSenderReplicationHandler(ReplicationClientHandler):
|
|
||||||
def __init__(self, hs):
|
|
||||||
super(FederationSenderReplicationHandler, self).__init__(hs.get_datastore())
|
|
||||||
self.send_handler = FederationSenderHandler(hs, self)
|
|
||||||
|
|
||||||
async def on_rdata(self, stream_name, token, rows):
|
|
||||||
await super(FederationSenderReplicationHandler, self).on_rdata(
|
|
||||||
stream_name, token, rows
|
|
||||||
)
|
|
||||||
self.send_handler.process_replication_rows(stream_name, token, rows)
|
|
||||||
|
|
||||||
def get_streams_to_replicate(self):
|
|
||||||
args = super(
|
|
||||||
FederationSenderReplicationHandler, self
|
|
||||||
).get_streams_to_replicate()
|
|
||||||
args.update(self.send_handler.stream_positions())
|
|
||||||
return args
|
|
||||||
|
|
||||||
def on_remote_server_up(self, server: str):
|
|
||||||
"""Called when get a new REMOTE_SERVER_UP command."""
|
|
||||||
|
|
||||||
# Let's wake up the transaction queue for the server in case we have
|
|
||||||
# pending stuff to send to it.
|
|
||||||
self.send_handler.wake_destination(server)
|
|
||||||
|
|
||||||
|
|
||||||
def start(config_options):
|
|
||||||
try:
|
|
||||||
config = HomeServerConfig.load_config(
|
|
||||||
"Synapse federation sender", config_options
|
|
||||||
)
|
|
||||||
except ConfigError as e:
|
|
||||||
sys.stderr.write("\n" + str(e) + "\n")
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
assert config.worker_app == "synapse.app.federation_sender"
|
|
||||||
|
|
||||||
events.USE_FROZEN_DICTS = config.use_frozen_dicts
|
|
||||||
|
|
||||||
if config.send_federation:
|
|
||||||
sys.stderr.write(
|
|
||||||
"\nThe send_federation must be disabled in the main synapse process"
|
|
||||||
"\nbefore they can be run in a separate worker."
|
|
||||||
"\nPlease add ``send_federation: false`` to the main config"
|
|
||||||
"\n"
|
|
||||||
)
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
# Force the pushers to start since they will be disabled in the main config
|
|
||||||
config.send_federation = True
|
|
||||||
|
|
||||||
ss = FederationSenderServer(
|
|
||||||
config.server_name,
|
|
||||||
config=config,
|
|
||||||
version_string="Synapse/" + get_version_string(synapse),
|
|
||||||
)
|
|
||||||
|
|
||||||
setup_logging(ss, config, use_worker_options=True)
|
|
||||||
|
|
||||||
ss.setup()
|
|
||||||
reactor.addSystemEventTrigger(
|
|
||||||
"before", "startup", _base.start, ss, config.worker_listeners
|
|
||||||
)
|
|
||||||
|
|
||||||
_base.start_worker_reactor("synapse-federation-sender", config)
|
|
||||||
|
|
||||||
|
|
||||||
class FederationSenderHandler(object):
|
|
||||||
"""Processes the replication stream and forwards the appropriate entries
|
|
||||||
to the federation sender.
|
|
||||||
"""
|
|
||||||
|
|
||||||
def __init__(self, hs: FederationSenderServer, replication_client):
|
|
||||||
self.store = hs.get_datastore()
|
|
||||||
self._is_mine_id = hs.is_mine_id
|
|
||||||
self.federation_sender = hs.get_federation_sender()
|
|
||||||
self.replication_client = replication_client
|
|
||||||
|
|
||||||
self.federation_position = self.store.federation_out_pos_startup
|
|
||||||
self._fed_position_linearizer = Linearizer(name="_fed_position_linearizer")
|
|
||||||
|
|
||||||
self._last_ack = self.federation_position
|
|
||||||
|
|
||||||
self._room_serials = {}
|
|
||||||
self._room_typing = {}
|
|
||||||
|
|
||||||
def on_start(self):
|
|
||||||
# There may be some events that are persisted but haven't been sent,
|
|
||||||
# so send them now.
|
|
||||||
self.federation_sender.notify_new_events(
|
|
||||||
self.store.get_room_max_stream_ordering()
|
|
||||||
)
|
|
||||||
|
|
||||||
def wake_destination(self, server: str):
|
|
||||||
self.federation_sender.wake_destination(server)
|
|
||||||
|
|
||||||
def stream_positions(self):
|
|
||||||
return {"federation": self.federation_position}
|
|
||||||
|
|
||||||
def process_replication_rows(self, stream_name, token, rows):
|
|
||||||
# The federation stream contains things that we want to send out, e.g.
|
|
||||||
# presence, typing, etc.
|
|
||||||
if stream_name == "federation":
|
|
||||||
send_queue.process_rows_for_federation(self.federation_sender, rows)
|
|
||||||
run_in_background(self.update_token, token)
|
|
||||||
|
|
||||||
# We also need to poke the federation sender when new events happen
|
|
||||||
elif stream_name == "events":
|
|
||||||
self.federation_sender.notify_new_events(token)
|
|
||||||
|
|
||||||
# ... and when new receipts happen
|
|
||||||
elif stream_name == ReceiptsStream.NAME:
|
|
||||||
run_as_background_process(
|
|
||||||
"process_receipts_for_federation", self._on_new_receipts, rows
|
|
||||||
)
|
|
||||||
|
|
||||||
# ... as well as device updates and messages
|
|
||||||
elif stream_name == DeviceListsStream.NAME:
|
|
||||||
hosts = {row.destination for row in rows}
|
|
||||||
for host in hosts:
|
|
||||||
self.federation_sender.send_device_messages(host)
|
|
||||||
|
|
||||||
elif stream_name == ToDeviceStream.NAME:
|
|
||||||
# The to_device stream includes stuff to be pushed to both local
|
|
||||||
# clients and remote servers, so we ignore entities that start with
|
|
||||||
# '@' (since they'll be local users rather than destinations).
|
|
||||||
hosts = {row.entity for row in rows if not row.entity.startswith("@")}
|
|
||||||
for host in hosts:
|
|
||||||
self.federation_sender.send_device_messages(host)
|
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
|
||||||
def _on_new_receipts(self, rows):
|
|
||||||
"""
|
|
||||||
Args:
|
|
||||||
rows (iterable[synapse.replication.tcp.streams.ReceiptsStreamRow]):
|
|
||||||
new receipts to be processed
|
|
||||||
"""
|
|
||||||
for receipt in rows:
|
|
||||||
# we only want to send on receipts for our own users
|
|
||||||
if not self._is_mine_id(receipt.user_id):
|
|
||||||
continue
|
|
||||||
receipt_info = ReadReceipt(
|
|
||||||
receipt.room_id,
|
|
||||||
receipt.receipt_type,
|
|
||||||
receipt.user_id,
|
|
||||||
[receipt.event_id],
|
|
||||||
receipt.data,
|
|
||||||
)
|
|
||||||
yield self.federation_sender.send_read_receipt(receipt_info)
|
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
|
||||||
def update_token(self, token):
|
|
||||||
try:
|
|
||||||
self.federation_position = token
|
|
||||||
|
|
||||||
# We linearize here to ensure we don't have races updating the token
|
|
||||||
with (yield self._fed_position_linearizer.queue(None)):
|
|
||||||
if self._last_ack < self.federation_position:
|
|
||||||
yield self.store.update_federation_out_pos(
|
|
||||||
"federation", self.federation_position
|
|
||||||
)
|
|
||||||
|
|
||||||
# We ACK this token over replication so that the master can drop
|
|
||||||
# its in memory queues
|
|
||||||
self.replication_client.send_federation_ack(
|
|
||||||
self.federation_position
|
|
||||||
)
|
|
||||||
self._last_ack = self.federation_position
|
|
||||||
except Exception:
|
|
||||||
logger.exception("Error updating federation stream position")
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
with LoggingContext("main"):
|
with LoggingContext("main"):
|
||||||
|
@ -13,241 +13,11 @@
|
|||||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
# See the License for the specific language governing permissions and
|
# See the License for the specific language governing permissions and
|
||||||
# limitations under the License.
|
# limitations under the License.
|
||||||
import logging
|
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
from twisted.internet import defer, reactor
|
from synapse.app.generic_worker import start
|
||||||
from twisted.web.resource import NoResource
|
from synapse.util.logcontext import LoggingContext
|
||||||
|
|
||||||
import synapse
|
|
||||||
from synapse import events
|
|
||||||
from synapse.api.errors import HttpResponseException, SynapseError
|
|
||||||
from synapse.app import _base
|
|
||||||
from synapse.config._base import ConfigError
|
|
||||||
from synapse.config.homeserver import HomeServerConfig
|
|
||||||
from synapse.config.logger import setup_logging
|
|
||||||
from synapse.http.server import JsonResource
|
|
||||||
from synapse.http.servlet import RestServlet, parse_json_object_from_request
|
|
||||||
from synapse.http.site import SynapseSite
|
|
||||||
from synapse.logging.context import LoggingContext
|
|
||||||
from synapse.metrics import METRICS_PREFIX, MetricsResource, RegistryProxy
|
|
||||||
from synapse.replication.slave.storage._base import BaseSlavedStore
|
|
||||||
from synapse.replication.slave.storage.appservice import SlavedApplicationServiceStore
|
|
||||||
from synapse.replication.slave.storage.client_ips import SlavedClientIpStore
|
|
||||||
from synapse.replication.slave.storage.devices import SlavedDeviceStore
|
|
||||||
from synapse.replication.slave.storage.registration import SlavedRegistrationStore
|
|
||||||
from synapse.replication.tcp.client import ReplicationClientHandler
|
|
||||||
from synapse.rest.client.v2_alpha._base import client_patterns
|
|
||||||
from synapse.server import HomeServer
|
|
||||||
from synapse.util.httpresourcetree import create_resource_tree
|
|
||||||
from synapse.util.manhole import manhole
|
|
||||||
from synapse.util.versionstring import get_version_string
|
|
||||||
|
|
||||||
logger = logging.getLogger("synapse.app.frontend_proxy")
|
|
||||||
|
|
||||||
|
|
||||||
class PresenceStatusStubServlet(RestServlet):
|
|
||||||
PATTERNS = client_patterns("/presence/(?P<user_id>[^/]*)/status")
|
|
||||||
|
|
||||||
def __init__(self, hs):
|
|
||||||
super(PresenceStatusStubServlet, self).__init__()
|
|
||||||
self.http_client = hs.get_simple_http_client()
|
|
||||||
self.auth = hs.get_auth()
|
|
||||||
self.main_uri = hs.config.worker_main_http_uri
|
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
|
||||||
def on_GET(self, request, user_id):
|
|
||||||
# Pass through the auth headers, if any, in case the access token
|
|
||||||
# is there.
|
|
||||||
auth_headers = request.requestHeaders.getRawHeaders("Authorization", [])
|
|
||||||
headers = {"Authorization": auth_headers}
|
|
||||||
|
|
||||||
try:
|
|
||||||
result = yield self.http_client.get_json(
|
|
||||||
self.main_uri + request.uri.decode("ascii"), headers=headers
|
|
||||||
)
|
|
||||||
except HttpResponseException as e:
|
|
||||||
raise e.to_synapse_error()
|
|
||||||
|
|
||||||
return 200, result
|
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
|
||||||
def on_PUT(self, request, user_id):
|
|
||||||
yield self.auth.get_user_by_req(request)
|
|
||||||
return 200, {}
|
|
||||||
|
|
||||||
|
|
||||||
class KeyUploadServlet(RestServlet):
|
|
||||||
PATTERNS = client_patterns("/keys/upload(/(?P<device_id>[^/]+))?$")
|
|
||||||
|
|
||||||
def __init__(self, hs):
|
|
||||||
"""
|
|
||||||
Args:
|
|
||||||
hs (synapse.server.HomeServer): server
|
|
||||||
"""
|
|
||||||
super(KeyUploadServlet, self).__init__()
|
|
||||||
self.auth = hs.get_auth()
|
|
||||||
self.store = hs.get_datastore()
|
|
||||||
self.http_client = hs.get_simple_http_client()
|
|
||||||
self.main_uri = hs.config.worker_main_http_uri
|
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
|
||||||
def on_POST(self, request, device_id):
|
|
||||||
requester = yield self.auth.get_user_by_req(request, allow_guest=True)
|
|
||||||
user_id = requester.user.to_string()
|
|
||||||
body = parse_json_object_from_request(request)
|
|
||||||
|
|
||||||
if device_id is not None:
|
|
||||||
# passing the device_id here is deprecated; however, we allow it
|
|
||||||
# for now for compatibility with older clients.
|
|
||||||
if requester.device_id is not None and device_id != requester.device_id:
|
|
||||||
logger.warning(
|
|
||||||
"Client uploading keys for a different device "
|
|
||||||
"(logged in as %s, uploading for %s)",
|
|
||||||
requester.device_id,
|
|
||||||
device_id,
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
device_id = requester.device_id
|
|
||||||
|
|
||||||
if device_id is None:
|
|
||||||
raise SynapseError(
|
|
||||||
400, "To upload keys, you must pass device_id when authenticating"
|
|
||||||
)
|
|
||||||
|
|
||||||
if body:
|
|
||||||
# They're actually trying to upload something, proxy to main synapse.
|
|
||||||
# Pass through the auth headers, if any, in case the access token
|
|
||||||
# is there.
|
|
||||||
auth_headers = request.requestHeaders.getRawHeaders(b"Authorization", [])
|
|
||||||
headers = {"Authorization": auth_headers}
|
|
||||||
result = yield self.http_client.post_json_get_json(
|
|
||||||
self.main_uri + request.uri.decode("ascii"), body, headers=headers
|
|
||||||
)
|
|
||||||
|
|
||||||
return 200, result
|
|
||||||
else:
|
|
||||||
# Just interested in counts.
|
|
||||||
result = yield self.store.count_e2e_one_time_keys(user_id, device_id)
|
|
||||||
return 200, {"one_time_key_counts": result}
|
|
||||||
|
|
||||||
|
|
||||||
class FrontendProxySlavedStore(
|
|
||||||
SlavedDeviceStore,
|
|
||||||
SlavedClientIpStore,
|
|
||||||
SlavedApplicationServiceStore,
|
|
||||||
SlavedRegistrationStore,
|
|
||||||
BaseSlavedStore,
|
|
||||||
):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
class FrontendProxyServer(HomeServer):
|
|
||||||
DATASTORE_CLASS = FrontendProxySlavedStore
|
|
||||||
|
|
||||||
def _listen_http(self, listener_config):
|
|
||||||
port = listener_config["port"]
|
|
||||||
bind_addresses = listener_config["bind_addresses"]
|
|
||||||
site_tag = listener_config.get("tag", port)
|
|
||||||
resources = {}
|
|
||||||
for res in listener_config["resources"]:
|
|
||||||
for name in res["names"]:
|
|
||||||
if name == "metrics":
|
|
||||||
resources[METRICS_PREFIX] = MetricsResource(RegistryProxy)
|
|
||||||
elif name == "client":
|
|
||||||
resource = JsonResource(self, canonical_json=False)
|
|
||||||
KeyUploadServlet(self).register(resource)
|
|
||||||
|
|
||||||
# If presence is disabled, use the stub servlet that does
|
|
||||||
# not allow sending presence
|
|
||||||
if not self.config.use_presence:
|
|
||||||
PresenceStatusStubServlet(self).register(resource)
|
|
||||||
|
|
||||||
resources.update(
|
|
||||||
{
|
|
||||||
"/_matrix/client/r0": resource,
|
|
||||||
"/_matrix/client/unstable": resource,
|
|
||||||
"/_matrix/client/v2_alpha": resource,
|
|
||||||
"/_matrix/client/api/v1": resource,
|
|
||||||
}
|
|
||||||
)
|
|
||||||
|
|
||||||
root_resource = create_resource_tree(resources, NoResource())
|
|
||||||
|
|
||||||
_base.listen_tcp(
|
|
||||||
bind_addresses,
|
|
||||||
port,
|
|
||||||
SynapseSite(
|
|
||||||
"synapse.access.http.%s" % (site_tag,),
|
|
||||||
site_tag,
|
|
||||||
listener_config,
|
|
||||||
root_resource,
|
|
||||||
self.version_string,
|
|
||||||
),
|
|
||||||
reactor=self.get_reactor(),
|
|
||||||
)
|
|
||||||
|
|
||||||
logger.info("Synapse client reader now listening on port %d", port)
|
|
||||||
|
|
||||||
def start_listening(self, listeners):
|
|
||||||
for listener in listeners:
|
|
||||||
if listener["type"] == "http":
|
|
||||||
self._listen_http(listener)
|
|
||||||
elif listener["type"] == "manhole":
|
|
||||||
_base.listen_tcp(
|
|
||||||
listener["bind_addresses"],
|
|
||||||
listener["port"],
|
|
||||||
manhole(
|
|
||||||
username="matrix", password="rabbithole", globals={"hs": self}
|
|
||||||
),
|
|
||||||
)
|
|
||||||
elif listener["type"] == "metrics":
|
|
||||||
if not self.get_config().enable_metrics:
|
|
||||||
logger.warning(
|
|
||||||
(
|
|
||||||
"Metrics listener configured, but "
|
|
||||||
"enable_metrics is not True!"
|
|
||||||
)
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
_base.listen_metrics(listener["bind_addresses"], listener["port"])
|
|
||||||
else:
|
|
||||||
logger.warning("Unrecognized listener type: %s", listener["type"])
|
|
||||||
|
|
||||||
self.get_tcp_replication().start_replication(self)
|
|
||||||
|
|
||||||
def build_tcp_replication(self):
|
|
||||||
return ReplicationClientHandler(self.get_datastore())
|
|
||||||
|
|
||||||
|
|
||||||
def start(config_options):
|
|
||||||
try:
|
|
||||||
config = HomeServerConfig.load_config("Synapse frontend proxy", config_options)
|
|
||||||
except ConfigError as e:
|
|
||||||
sys.stderr.write("\n" + str(e) + "\n")
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
assert config.worker_app == "synapse.app.frontend_proxy"
|
|
||||||
|
|
||||||
assert config.worker_main_http_uri is not None
|
|
||||||
|
|
||||||
events.USE_FROZEN_DICTS = config.use_frozen_dicts
|
|
||||||
|
|
||||||
ss = FrontendProxyServer(
|
|
||||||
config.server_name,
|
|
||||||
config=config,
|
|
||||||
version_string="Synapse/" + get_version_string(synapse),
|
|
||||||
)
|
|
||||||
|
|
||||||
setup_logging(ss, config, use_worker_options=True)
|
|
||||||
|
|
||||||
ss.setup()
|
|
||||||
reactor.addSystemEventTrigger(
|
|
||||||
"before", "startup", _base.start, ss, config.worker_listeners
|
|
||||||
)
|
|
||||||
|
|
||||||
_base.start_worker_reactor("synapse-frontend-proxy", config)
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
with LoggingContext("main"):
|
with LoggingContext("main"):
|
||||||
|
917
synapse/app/generic_worker.py
Normal file
917
synapse/app/generic_worker.py
Normal file
@ -0,0 +1,917 @@
|
|||||||
|
#!/usr/bin/env python
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
# Copyright 2016 OpenMarket Ltd
|
||||||
|
# Copyright 2020 The Matrix.org Foundation C.I.C.
|
||||||
|
#
|
||||||
|
# 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 contextlib
|
||||||
|
import logging
|
||||||
|
import sys
|
||||||
|
|
||||||
|
from twisted.internet import defer, reactor
|
||||||
|
from twisted.web.resource import NoResource
|
||||||
|
|
||||||
|
import synapse
|
||||||
|
import synapse.events
|
||||||
|
from synapse.api.constants import EventTypes
|
||||||
|
from synapse.api.errors import HttpResponseException, SynapseError
|
||||||
|
from synapse.api.urls import (
|
||||||
|
CLIENT_API_PREFIX,
|
||||||
|
FEDERATION_PREFIX,
|
||||||
|
LEGACY_MEDIA_PREFIX,
|
||||||
|
MEDIA_PREFIX,
|
||||||
|
SERVER_KEY_V2_PREFIX,
|
||||||
|
)
|
||||||
|
from synapse.app import _base
|
||||||
|
from synapse.config._base import ConfigError
|
||||||
|
from synapse.config.homeserver import HomeServerConfig
|
||||||
|
from synapse.config.logger import setup_logging
|
||||||
|
from synapse.federation import send_queue
|
||||||
|
from synapse.federation.transport.server import TransportLayerServer
|
||||||
|
from synapse.handlers.presence import PresenceHandler, get_interested_parties
|
||||||
|
from synapse.http.server import JsonResource
|
||||||
|
from synapse.http.servlet import RestServlet, parse_json_object_from_request
|
||||||
|
from synapse.http.site import SynapseSite
|
||||||
|
from synapse.logging.context import LoggingContext, run_in_background
|
||||||
|
from synapse.metrics import METRICS_PREFIX, MetricsResource, RegistryProxy
|
||||||
|
from synapse.metrics.background_process_metrics import run_as_background_process
|
||||||
|
from synapse.replication.slave.storage._base import BaseSlavedStore, __func__
|
||||||
|
from synapse.replication.slave.storage.account_data import SlavedAccountDataStore
|
||||||
|
from synapse.replication.slave.storage.appservice import SlavedApplicationServiceStore
|
||||||
|
from synapse.replication.slave.storage.client_ips import SlavedClientIpStore
|
||||||
|
from synapse.replication.slave.storage.deviceinbox import SlavedDeviceInboxStore
|
||||||
|
from synapse.replication.slave.storage.devices import SlavedDeviceStore
|
||||||
|
from synapse.replication.slave.storage.directory import DirectoryStore
|
||||||
|
from synapse.replication.slave.storage.events import SlavedEventStore
|
||||||
|
from synapse.replication.slave.storage.filtering import SlavedFilteringStore
|
||||||
|
from synapse.replication.slave.storage.groups import SlavedGroupServerStore
|
||||||
|
from synapse.replication.slave.storage.keys import SlavedKeyStore
|
||||||
|
from synapse.replication.slave.storage.presence import SlavedPresenceStore
|
||||||
|
from synapse.replication.slave.storage.profile import SlavedProfileStore
|
||||||
|
from synapse.replication.slave.storage.push_rule import SlavedPushRuleStore
|
||||||
|
from synapse.replication.slave.storage.pushers import SlavedPusherStore
|
||||||
|
from synapse.replication.slave.storage.receipts import SlavedReceiptsStore
|
||||||
|
from synapse.replication.slave.storage.registration import SlavedRegistrationStore
|
||||||
|
from synapse.replication.slave.storage.room import RoomStore
|
||||||
|
from synapse.replication.slave.storage.transactions import SlavedTransactionStore
|
||||||
|
from synapse.replication.tcp.client import ReplicationClientHandler
|
||||||
|
from synapse.replication.tcp.streams._base import (
|
||||||
|
DeviceListsStream,
|
||||||
|
ReceiptsStream,
|
||||||
|
ToDeviceStream,
|
||||||
|
)
|
||||||
|
from synapse.replication.tcp.streams.events import EventsStreamEventRow, EventsStreamRow
|
||||||
|
from synapse.rest.admin import register_servlets_for_media_repo
|
||||||
|
from synapse.rest.client.v1 import events
|
||||||
|
from synapse.rest.client.v1.initial_sync import InitialSyncRestServlet
|
||||||
|
from synapse.rest.client.v1.login import LoginRestServlet
|
||||||
|
from synapse.rest.client.v1.profile import (
|
||||||
|
ProfileAvatarURLRestServlet,
|
||||||
|
ProfileDisplaynameRestServlet,
|
||||||
|
ProfileRestServlet,
|
||||||
|
)
|
||||||
|
from synapse.rest.client.v1.push_rule import PushRuleRestServlet
|
||||||
|
from synapse.rest.client.v1.room import (
|
||||||
|
JoinedRoomMemberListRestServlet,
|
||||||
|
JoinRoomAliasServlet,
|
||||||
|
PublicRoomListRestServlet,
|
||||||
|
RoomEventContextServlet,
|
||||||
|
RoomInitialSyncRestServlet,
|
||||||
|
RoomMemberListRestServlet,
|
||||||
|
RoomMembershipRestServlet,
|
||||||
|
RoomMessageListRestServlet,
|
||||||
|
RoomSendEventRestServlet,
|
||||||
|
RoomStateEventRestServlet,
|
||||||
|
RoomStateRestServlet,
|
||||||
|
)
|
||||||
|
from synapse.rest.client.v1.voip import VoipRestServlet
|
||||||
|
from synapse.rest.client.v2_alpha import groups, sync, user_directory
|
||||||
|
from synapse.rest.client.v2_alpha._base import client_patterns
|
||||||
|
from synapse.rest.client.v2_alpha.account import ThreepidRestServlet
|
||||||
|
from synapse.rest.client.v2_alpha.keys import KeyChangesServlet, KeyQueryServlet
|
||||||
|
from synapse.rest.client.v2_alpha.register import RegisterRestServlet
|
||||||
|
from synapse.rest.client.versions import VersionsRestServlet
|
||||||
|
from synapse.rest.key.v2 import KeyApiV2Resource
|
||||||
|
from synapse.server import HomeServer
|
||||||
|
from synapse.storage.data_stores.main.media_repository import MediaRepositoryStore
|
||||||
|
from synapse.storage.data_stores.main.monthly_active_users import (
|
||||||
|
MonthlyActiveUsersWorkerStore,
|
||||||
|
)
|
||||||
|
from synapse.storage.data_stores.main.presence import UserPresenceState
|
||||||
|
from synapse.storage.data_stores.main.user_directory import UserDirectoryStore
|
||||||
|
from synapse.types import ReadReceipt
|
||||||
|
from synapse.util.async_helpers import Linearizer
|
||||||
|
from synapse.util.httpresourcetree import create_resource_tree
|
||||||
|
from synapse.util.manhole import manhole
|
||||||
|
from synapse.util.stringutils import random_string
|
||||||
|
from synapse.util.versionstring import get_version_string
|
||||||
|
|
||||||
|
logger = logging.getLogger("synapse.app.generic_worker")
|
||||||
|
|
||||||
|
|
||||||
|
class PresenceStatusStubServlet(RestServlet):
|
||||||
|
"""If presence is disabled this servlet can be used to stub out setting
|
||||||
|
presence status, while proxying the getters to the master instance.
|
||||||
|
"""
|
||||||
|
|
||||||
|
PATTERNS = client_patterns("/presence/(?P<user_id>[^/]*)/status")
|
||||||
|
|
||||||
|
def __init__(self, hs):
|
||||||
|
super(PresenceStatusStubServlet, self).__init__()
|
||||||
|
self.http_client = hs.get_simple_http_client()
|
||||||
|
self.auth = hs.get_auth()
|
||||||
|
self.main_uri = hs.config.worker_main_http_uri
|
||||||
|
|
||||||
|
async def on_GET(self, request, user_id):
|
||||||
|
# Pass through the auth headers, if any, in case the access token
|
||||||
|
# is there.
|
||||||
|
auth_headers = request.requestHeaders.getRawHeaders("Authorization", [])
|
||||||
|
headers = {"Authorization": auth_headers}
|
||||||
|
|
||||||
|
try:
|
||||||
|
result = await self.http_client.get_json(
|
||||||
|
self.main_uri + request.uri.decode("ascii"), headers=headers
|
||||||
|
)
|
||||||
|
except HttpResponseException as e:
|
||||||
|
raise e.to_synapse_error()
|
||||||
|
|
||||||
|
return 200, result
|
||||||
|
|
||||||
|
async def on_PUT(self, request, user_id):
|
||||||
|
await self.auth.get_user_by_req(request)
|
||||||
|
return 200, {}
|
||||||
|
|
||||||
|
|
||||||
|
class KeyUploadServlet(RestServlet):
|
||||||
|
"""An implementation of the `KeyUploadServlet` that responds to read only
|
||||||
|
requests, but otherwise proxies through to the master instance.
|
||||||
|
"""
|
||||||
|
|
||||||
|
PATTERNS = client_patterns("/keys/upload(/(?P<device_id>[^/]+))?$")
|
||||||
|
|
||||||
|
def __init__(self, hs):
|
||||||
|
"""
|
||||||
|
Args:
|
||||||
|
hs (synapse.server.HomeServer): server
|
||||||
|
"""
|
||||||
|
super(KeyUploadServlet, self).__init__()
|
||||||
|
self.auth = hs.get_auth()
|
||||||
|
self.store = hs.get_datastore()
|
||||||
|
self.http_client = hs.get_simple_http_client()
|
||||||
|
self.main_uri = hs.config.worker_main_http_uri
|
||||||
|
|
||||||
|
async def on_POST(self, request, device_id):
|
||||||
|
requester = await self.auth.get_user_by_req(request, allow_guest=True)
|
||||||
|
user_id = requester.user.to_string()
|
||||||
|
body = parse_json_object_from_request(request)
|
||||||
|
|
||||||
|
if device_id is not None:
|
||||||
|
# passing the device_id here is deprecated; however, we allow it
|
||||||
|
# for now for compatibility with older clients.
|
||||||
|
if requester.device_id is not None and device_id != requester.device_id:
|
||||||
|
logger.warning(
|
||||||
|
"Client uploading keys for a different device "
|
||||||
|
"(logged in as %s, uploading for %s)",
|
||||||
|
requester.device_id,
|
||||||
|
device_id,
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
device_id = requester.device_id
|
||||||
|
|
||||||
|
if device_id is None:
|
||||||
|
raise SynapseError(
|
||||||
|
400, "To upload keys, you must pass device_id when authenticating"
|
||||||
|
)
|
||||||
|
|
||||||
|
if body:
|
||||||
|
# They're actually trying to upload something, proxy to main synapse.
|
||||||
|
# Pass through the auth headers, if any, in case the access token
|
||||||
|
# is there.
|
||||||
|
auth_headers = request.requestHeaders.getRawHeaders(b"Authorization", [])
|
||||||
|
headers = {"Authorization": auth_headers}
|
||||||
|
result = await self.http_client.post_json_get_json(
|
||||||
|
self.main_uri + request.uri.decode("ascii"), body, headers=headers
|
||||||
|
)
|
||||||
|
|
||||||
|
return 200, result
|
||||||
|
else:
|
||||||
|
# Just interested in counts.
|
||||||
|
result = await self.store.count_e2e_one_time_keys(user_id, device_id)
|
||||||
|
return 200, {"one_time_key_counts": result}
|
||||||
|
|
||||||
|
|
||||||
|
UPDATE_SYNCING_USERS_MS = 10 * 1000
|
||||||
|
|
||||||
|
|
||||||
|
class GenericWorkerPresence(object):
|
||||||
|
def __init__(self, hs):
|
||||||
|
self.hs = hs
|
||||||
|
self.is_mine_id = hs.is_mine_id
|
||||||
|
self.http_client = hs.get_simple_http_client()
|
||||||
|
self.store = hs.get_datastore()
|
||||||
|
self.user_to_num_current_syncs = {}
|
||||||
|
self.clock = hs.get_clock()
|
||||||
|
self.notifier = hs.get_notifier()
|
||||||
|
|
||||||
|
active_presence = self.store.take_presence_startup_info()
|
||||||
|
self.user_to_current_state = {state.user_id: state for state in active_presence}
|
||||||
|
|
||||||
|
# user_id -> last_sync_ms. Lists the users that have stopped syncing
|
||||||
|
# but we haven't notified the master of that yet
|
||||||
|
self.users_going_offline = {}
|
||||||
|
|
||||||
|
self._send_stop_syncing_loop = self.clock.looping_call(
|
||||||
|
self.send_stop_syncing, UPDATE_SYNCING_USERS_MS
|
||||||
|
)
|
||||||
|
|
||||||
|
self.process_id = random_string(16)
|
||||||
|
logger.info("Presence process_id is %r", self.process_id)
|
||||||
|
|
||||||
|
def send_user_sync(self, user_id, is_syncing, last_sync_ms):
|
||||||
|
if self.hs.config.use_presence:
|
||||||
|
self.hs.get_tcp_replication().send_user_sync(
|
||||||
|
user_id, is_syncing, last_sync_ms
|
||||||
|
)
|
||||||
|
|
||||||
|
def mark_as_coming_online(self, user_id):
|
||||||
|
"""A user has started syncing. Send a UserSync to the master, unless they
|
||||||
|
had recently stopped syncing.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
user_id (str)
|
||||||
|
"""
|
||||||
|
going_offline = self.users_going_offline.pop(user_id, None)
|
||||||
|
if not going_offline:
|
||||||
|
# Safe to skip because we haven't yet told the master they were offline
|
||||||
|
self.send_user_sync(user_id, True, self.clock.time_msec())
|
||||||
|
|
||||||
|
def mark_as_going_offline(self, user_id):
|
||||||
|
"""A user has stopped syncing. We wait before notifying the master as
|
||||||
|
its likely they'll come back soon. This allows us to avoid sending
|
||||||
|
a stopped syncing immediately followed by a started syncing notification
|
||||||
|
to the master
|
||||||
|
|
||||||
|
Args:
|
||||||
|
user_id (str)
|
||||||
|
"""
|
||||||
|
self.users_going_offline[user_id] = self.clock.time_msec()
|
||||||
|
|
||||||
|
def send_stop_syncing(self):
|
||||||
|
"""Check if there are any users who have stopped syncing a while ago
|
||||||
|
and haven't come back yet. If there are poke the master about them.
|
||||||
|
"""
|
||||||
|
now = self.clock.time_msec()
|
||||||
|
for user_id, last_sync_ms in list(self.users_going_offline.items()):
|
||||||
|
if now - last_sync_ms > UPDATE_SYNCING_USERS_MS:
|
||||||
|
self.users_going_offline.pop(user_id, None)
|
||||||
|
self.send_user_sync(user_id, False, last_sync_ms)
|
||||||
|
|
||||||
|
def set_state(self, user, state, ignore_status_msg=False):
|
||||||
|
# TODO Hows this supposed to work?
|
||||||
|
return defer.succeed(None)
|
||||||
|
|
||||||
|
get_states = __func__(PresenceHandler.get_states)
|
||||||
|
get_state = __func__(PresenceHandler.get_state)
|
||||||
|
current_state_for_users = __func__(PresenceHandler.current_state_for_users)
|
||||||
|
|
||||||
|
def user_syncing(self, user_id, affect_presence):
|
||||||
|
if affect_presence:
|
||||||
|
curr_sync = self.user_to_num_current_syncs.get(user_id, 0)
|
||||||
|
self.user_to_num_current_syncs[user_id] = curr_sync + 1
|
||||||
|
|
||||||
|
# If we went from no in flight sync to some, notify replication
|
||||||
|
if self.user_to_num_current_syncs[user_id] == 1:
|
||||||
|
self.mark_as_coming_online(user_id)
|
||||||
|
|
||||||
|
def _end():
|
||||||
|
# We check that the user_id is in user_to_num_current_syncs because
|
||||||
|
# user_to_num_current_syncs may have been cleared if we are
|
||||||
|
# shutting down.
|
||||||
|
if affect_presence and user_id in self.user_to_num_current_syncs:
|
||||||
|
self.user_to_num_current_syncs[user_id] -= 1
|
||||||
|
|
||||||
|
# If we went from one in flight sync to non, notify replication
|
||||||
|
if self.user_to_num_current_syncs[user_id] == 0:
|
||||||
|
self.mark_as_going_offline(user_id)
|
||||||
|
|
||||||
|
@contextlib.contextmanager
|
||||||
|
def _user_syncing():
|
||||||
|
try:
|
||||||
|
yield
|
||||||
|
finally:
|
||||||
|
_end()
|
||||||
|
|
||||||
|
return defer.succeed(_user_syncing())
|
||||||
|
|
||||||
|
@defer.inlineCallbacks
|
||||||
|
def notify_from_replication(self, states, stream_id):
|
||||||
|
parties = yield get_interested_parties(self.store, states)
|
||||||
|
room_ids_to_states, users_to_states = parties
|
||||||
|
|
||||||
|
self.notifier.on_new_event(
|
||||||
|
"presence_key",
|
||||||
|
stream_id,
|
||||||
|
rooms=room_ids_to_states.keys(),
|
||||||
|
users=users_to_states.keys(),
|
||||||
|
)
|
||||||
|
|
||||||
|
@defer.inlineCallbacks
|
||||||
|
def process_replication_rows(self, token, rows):
|
||||||
|
states = [
|
||||||
|
UserPresenceState(
|
||||||
|
row.user_id,
|
||||||
|
row.state,
|
||||||
|
row.last_active_ts,
|
||||||
|
row.last_federation_update_ts,
|
||||||
|
row.last_user_sync_ts,
|
||||||
|
row.status_msg,
|
||||||
|
row.currently_active,
|
||||||
|
)
|
||||||
|
for row in rows
|
||||||
|
]
|
||||||
|
|
||||||
|
for state in states:
|
||||||
|
self.user_to_current_state[state.user_id] = state
|
||||||
|
|
||||||
|
stream_id = token
|
||||||
|
yield self.notify_from_replication(states, stream_id)
|
||||||
|
|
||||||
|
def get_currently_syncing_users(self):
|
||||||
|
if self.hs.config.use_presence:
|
||||||
|
return [
|
||||||
|
user_id
|
||||||
|
for user_id, count in self.user_to_num_current_syncs.items()
|
||||||
|
if count > 0
|
||||||
|
]
|
||||||
|
else:
|
||||||
|
return set()
|
||||||
|
|
||||||
|
|
||||||
|
class GenericWorkerTyping(object):
|
||||||
|
def __init__(self, hs):
|
||||||
|
self._latest_room_serial = 0
|
||||||
|
self._reset()
|
||||||
|
|
||||||
|
def _reset(self):
|
||||||
|
"""
|
||||||
|
Reset the typing handler's data caches.
|
||||||
|
"""
|
||||||
|
# map room IDs to serial numbers
|
||||||
|
self._room_serials = {}
|
||||||
|
# map room IDs to sets of users currently typing
|
||||||
|
self._room_typing = {}
|
||||||
|
|
||||||
|
def stream_positions(self):
|
||||||
|
# We must update this typing token from the response of the previous
|
||||||
|
# sync. In particular, the stream id may "reset" back to zero/a low
|
||||||
|
# value which we *must* use for the next replication request.
|
||||||
|
return {"typing": self._latest_room_serial}
|
||||||
|
|
||||||
|
def process_replication_rows(self, token, rows):
|
||||||
|
if self._latest_room_serial > token:
|
||||||
|
# The master has gone backwards. To prevent inconsistent data, just
|
||||||
|
# clear everything.
|
||||||
|
self._reset()
|
||||||
|
|
||||||
|
# Set the latest serial token to whatever the server gave us.
|
||||||
|
self._latest_room_serial = token
|
||||||
|
|
||||||
|
for row in rows:
|
||||||
|
self._room_serials[row.room_id] = token
|
||||||
|
self._room_typing[row.room_id] = row.user_ids
|
||||||
|
|
||||||
|
|
||||||
|
class GenericWorkerSlavedStore(
|
||||||
|
# FIXME(#3714): We need to add UserDirectoryStore as we write directly
|
||||||
|
# rather than going via the correct worker.
|
||||||
|
UserDirectoryStore,
|
||||||
|
SlavedDeviceInboxStore,
|
||||||
|
SlavedDeviceStore,
|
||||||
|
SlavedReceiptsStore,
|
||||||
|
SlavedPushRuleStore,
|
||||||
|
SlavedGroupServerStore,
|
||||||
|
SlavedAccountDataStore,
|
||||||
|
SlavedPusherStore,
|
||||||
|
SlavedEventStore,
|
||||||
|
SlavedKeyStore,
|
||||||
|
RoomStore,
|
||||||
|
DirectoryStore,
|
||||||
|
SlavedApplicationServiceStore,
|
||||||
|
SlavedRegistrationStore,
|
||||||
|
SlavedTransactionStore,
|
||||||
|
SlavedProfileStore,
|
||||||
|
SlavedClientIpStore,
|
||||||
|
SlavedPresenceStore,
|
||||||
|
SlavedFilteringStore,
|
||||||
|
MonthlyActiveUsersWorkerStore,
|
||||||
|
MediaRepositoryStore,
|
||||||
|
BaseSlavedStore,
|
||||||
|
):
|
||||||
|
def __init__(self, database, db_conn, hs):
|
||||||
|
super(GenericWorkerSlavedStore, self).__init__(database, db_conn, hs)
|
||||||
|
|
||||||
|
# We pull out the current federation stream position now so that we
|
||||||
|
# always have a known value for the federation position in memory so
|
||||||
|
# that we don't have to bounce via a deferred once when we start the
|
||||||
|
# replication streams.
|
||||||
|
self.federation_out_pos_startup = self._get_federation_out_pos(db_conn)
|
||||||
|
|
||||||
|
def _get_federation_out_pos(self, db_conn):
|
||||||
|
sql = "SELECT stream_id FROM federation_stream_position WHERE type = ?"
|
||||||
|
sql = self.database_engine.convert_param_style(sql)
|
||||||
|
|
||||||
|
txn = db_conn.cursor()
|
||||||
|
txn.execute(sql, ("federation",))
|
||||||
|
rows = txn.fetchall()
|
||||||
|
txn.close()
|
||||||
|
|
||||||
|
return rows[0][0] if rows else -1
|
||||||
|
|
||||||
|
|
||||||
|
class GenericWorkerServer(HomeServer):
|
||||||
|
DATASTORE_CLASS = GenericWorkerSlavedStore
|
||||||
|
|
||||||
|
def _listen_http(self, listener_config):
|
||||||
|
port = listener_config["port"]
|
||||||
|
bind_addresses = listener_config["bind_addresses"]
|
||||||
|
site_tag = listener_config.get("tag", port)
|
||||||
|
resources = {}
|
||||||
|
for res in listener_config["resources"]:
|
||||||
|
for name in res["names"]:
|
||||||
|
if name == "metrics":
|
||||||
|
resources[METRICS_PREFIX] = MetricsResource(RegistryProxy)
|
||||||
|
elif name == "client":
|
||||||
|
resource = JsonResource(self, canonical_json=False)
|
||||||
|
|
||||||
|
PublicRoomListRestServlet(self).register(resource)
|
||||||
|
RoomMemberListRestServlet(self).register(resource)
|
||||||
|
JoinedRoomMemberListRestServlet(self).register(resource)
|
||||||
|
RoomStateRestServlet(self).register(resource)
|
||||||
|
RoomEventContextServlet(self).register(resource)
|
||||||
|
RoomMessageListRestServlet(self).register(resource)
|
||||||
|
RegisterRestServlet(self).register(resource)
|
||||||
|
LoginRestServlet(self).register(resource)
|
||||||
|
ThreepidRestServlet(self).register(resource)
|
||||||
|
KeyQueryServlet(self).register(resource)
|
||||||
|
KeyChangesServlet(self).register(resource)
|
||||||
|
VoipRestServlet(self).register(resource)
|
||||||
|
PushRuleRestServlet(self).register(resource)
|
||||||
|
VersionsRestServlet(self).register(resource)
|
||||||
|
RoomSendEventRestServlet(self).register(resource)
|
||||||
|
RoomMembershipRestServlet(self).register(resource)
|
||||||
|
RoomStateEventRestServlet(self).register(resource)
|
||||||
|
JoinRoomAliasServlet(self).register(resource)
|
||||||
|
ProfileAvatarURLRestServlet(self).register(resource)
|
||||||
|
ProfileDisplaynameRestServlet(self).register(resource)
|
||||||
|
ProfileRestServlet(self).register(resource)
|
||||||
|
KeyUploadServlet(self).register(resource)
|
||||||
|
|
||||||
|
sync.register_servlets(self, resource)
|
||||||
|
events.register_servlets(self, resource)
|
||||||
|
InitialSyncRestServlet(self).register(resource)
|
||||||
|
RoomInitialSyncRestServlet(self).register(resource)
|
||||||
|
|
||||||
|
user_directory.register_servlets(self, resource)
|
||||||
|
|
||||||
|
# If presence is disabled, use the stub servlet that does
|
||||||
|
# not allow sending presence
|
||||||
|
if not self.config.use_presence:
|
||||||
|
PresenceStatusStubServlet(self).register(resource)
|
||||||
|
|
||||||
|
groups.register_servlets(self, resource)
|
||||||
|
|
||||||
|
resources.update({CLIENT_API_PREFIX: resource})
|
||||||
|
elif name == "federation":
|
||||||
|
resources.update({FEDERATION_PREFIX: TransportLayerServer(self)})
|
||||||
|
elif name == "media":
|
||||||
|
media_repo = self.get_media_repository_resource()
|
||||||
|
|
||||||
|
# We need to serve the admin servlets for media on the
|
||||||
|
# worker.
|
||||||
|
admin_resource = JsonResource(self, canonical_json=False)
|
||||||
|
register_servlets_for_media_repo(self, admin_resource)
|
||||||
|
|
||||||
|
resources.update(
|
||||||
|
{
|
||||||
|
MEDIA_PREFIX: media_repo,
|
||||||
|
LEGACY_MEDIA_PREFIX: media_repo,
|
||||||
|
"/_synapse/admin": admin_resource,
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
if name == "openid" and "federation" not in res["names"]:
|
||||||
|
# Only load the openid resource separately if federation resource
|
||||||
|
# is not specified since federation resource includes openid
|
||||||
|
# resource.
|
||||||
|
resources.update(
|
||||||
|
{
|
||||||
|
FEDERATION_PREFIX: TransportLayerServer(
|
||||||
|
self, servlet_groups=["openid"]
|
||||||
|
)
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
if name in ["keys", "federation"]:
|
||||||
|
resources[SERVER_KEY_V2_PREFIX] = KeyApiV2Resource(self)
|
||||||
|
|
||||||
|
root_resource = create_resource_tree(resources, NoResource())
|
||||||
|
|
||||||
|
_base.listen_tcp(
|
||||||
|
bind_addresses,
|
||||||
|
port,
|
||||||
|
SynapseSite(
|
||||||
|
"synapse.access.http.%s" % (site_tag,),
|
||||||
|
site_tag,
|
||||||
|
listener_config,
|
||||||
|
root_resource,
|
||||||
|
self.version_string,
|
||||||
|
),
|
||||||
|
reactor=self.get_reactor(),
|
||||||
|
)
|
||||||
|
|
||||||
|
logger.info("Synapse worker now listening on port %d", port)
|
||||||
|
|
||||||
|
def start_listening(self, listeners):
|
||||||
|
for listener in listeners:
|
||||||
|
if listener["type"] == "http":
|
||||||
|
self._listen_http(listener)
|
||||||
|
elif listener["type"] == "manhole":
|
||||||
|
_base.listen_tcp(
|
||||||
|
listener["bind_addresses"],
|
||||||
|
listener["port"],
|
||||||
|
manhole(
|
||||||
|
username="matrix", password="rabbithole", globals={"hs": self}
|
||||||
|
),
|
||||||
|
)
|
||||||
|
elif listener["type"] == "metrics":
|
||||||
|
if not self.get_config().enable_metrics:
|
||||||
|
logger.warning(
|
||||||
|
(
|
||||||
|
"Metrics listener configured, but "
|
||||||
|
"enable_metrics is not True!"
|
||||||
|
)
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
_base.listen_metrics(listener["bind_addresses"], listener["port"])
|
||||||
|
else:
|
||||||
|
logger.warning("Unrecognized listener type: %s", listener["type"])
|
||||||
|
|
||||||
|
self.get_tcp_replication().start_replication(self)
|
||||||
|
|
||||||
|
def remove_pusher(self, app_id, push_key, user_id):
|
||||||
|
self.get_tcp_replication().send_remove_pusher(app_id, push_key, user_id)
|
||||||
|
|
||||||
|
def build_tcp_replication(self):
|
||||||
|
return GenericWorkerReplicationHandler(self)
|
||||||
|
|
||||||
|
def build_presence_handler(self):
|
||||||
|
return GenericWorkerPresence(self)
|
||||||
|
|
||||||
|
def build_typing_handler(self):
|
||||||
|
return GenericWorkerTyping(self)
|
||||||
|
|
||||||
|
|
||||||
|
class GenericWorkerReplicationHandler(ReplicationClientHandler):
|
||||||
|
def __init__(self, hs):
|
||||||
|
super(GenericWorkerReplicationHandler, self).__init__(hs.get_datastore())
|
||||||
|
|
||||||
|
self.store = hs.get_datastore()
|
||||||
|
self.typing_handler = hs.get_typing_handler()
|
||||||
|
# NB this is a SynchrotronPresence, not a normal PresenceHandler
|
||||||
|
self.presence_handler = hs.get_presence_handler()
|
||||||
|
self.notifier = hs.get_notifier()
|
||||||
|
|
||||||
|
self.notify_pushers = hs.config.start_pushers
|
||||||
|
self.pusher_pool = hs.get_pusherpool()
|
||||||
|
|
||||||
|
if hs.config.send_federation:
|
||||||
|
self.send_handler = FederationSenderHandler(hs, self)
|
||||||
|
else:
|
||||||
|
self.send_handler = None
|
||||||
|
|
||||||
|
async def on_rdata(self, stream_name, token, rows):
|
||||||
|
await super(GenericWorkerReplicationHandler, self).on_rdata(
|
||||||
|
stream_name, token, rows
|
||||||
|
)
|
||||||
|
run_in_background(self.process_and_notify, stream_name, token, rows)
|
||||||
|
|
||||||
|
def get_streams_to_replicate(self):
|
||||||
|
args = super(GenericWorkerReplicationHandler, self).get_streams_to_replicate()
|
||||||
|
args.update(self.typing_handler.stream_positions())
|
||||||
|
if self.send_handler:
|
||||||
|
args.update(self.send_handler.stream_positions())
|
||||||
|
return args
|
||||||
|
|
||||||
|
def get_currently_syncing_users(self):
|
||||||
|
return self.presence_handler.get_currently_syncing_users()
|
||||||
|
|
||||||
|
async def process_and_notify(self, stream_name, token, rows):
|
||||||
|
try:
|
||||||
|
if self.send_handler:
|
||||||
|
self.send_handler.process_replication_rows(stream_name, token, rows)
|
||||||
|
|
||||||
|
if stream_name == "events":
|
||||||
|
# We shouldn't get multiple rows per token for events stream, so
|
||||||
|
# we don't need to optimise this for multiple rows.
|
||||||
|
for row in rows:
|
||||||
|
if row.type != EventsStreamEventRow.TypeId:
|
||||||
|
continue
|
||||||
|
assert isinstance(row, EventsStreamRow)
|
||||||
|
|
||||||
|
event = await self.store.get_event(
|
||||||
|
row.data.event_id, allow_rejected=True
|
||||||
|
)
|
||||||
|
if event.rejected_reason:
|
||||||
|
continue
|
||||||
|
|
||||||
|
extra_users = ()
|
||||||
|
if event.type == EventTypes.Member:
|
||||||
|
extra_users = (event.state_key,)
|
||||||
|
max_token = self.store.get_room_max_stream_ordering()
|
||||||
|
self.notifier.on_new_room_event(
|
||||||
|
event, token, max_token, extra_users
|
||||||
|
)
|
||||||
|
|
||||||
|
await self.pusher_pool.on_new_notifications(token, token)
|
||||||
|
elif stream_name == "push_rules":
|
||||||
|
self.notifier.on_new_event(
|
||||||
|
"push_rules_key", token, users=[row.user_id for row in rows]
|
||||||
|
)
|
||||||
|
elif stream_name in ("account_data", "tag_account_data"):
|
||||||
|
self.notifier.on_new_event(
|
||||||
|
"account_data_key", token, users=[row.user_id for row in rows]
|
||||||
|
)
|
||||||
|
elif stream_name == "receipts":
|
||||||
|
self.notifier.on_new_event(
|
||||||
|
"receipt_key", token, rooms=[row.room_id for row in rows]
|
||||||
|
)
|
||||||
|
await self.pusher_pool.on_new_receipts(
|
||||||
|
token, token, {row.room_id for row in rows}
|
||||||
|
)
|
||||||
|
elif stream_name == "typing":
|
||||||
|
self.typing_handler.process_replication_rows(token, rows)
|
||||||
|
self.notifier.on_new_event(
|
||||||
|
"typing_key", token, rooms=[row.room_id for row in rows]
|
||||||
|
)
|
||||||
|
elif stream_name == "to_device":
|
||||||
|
entities = [row.entity for row in rows if row.entity.startswith("@")]
|
||||||
|
if entities:
|
||||||
|
self.notifier.on_new_event("to_device_key", token, users=entities)
|
||||||
|
elif stream_name == "device_lists":
|
||||||
|
all_room_ids = set()
|
||||||
|
for row in rows:
|
||||||
|
room_ids = await self.store.get_rooms_for_user(row.user_id)
|
||||||
|
all_room_ids.update(room_ids)
|
||||||
|
self.notifier.on_new_event("device_list_key", token, rooms=all_room_ids)
|
||||||
|
elif stream_name == "presence":
|
||||||
|
await self.presence_handler.process_replication_rows(token, rows)
|
||||||
|
elif stream_name == "receipts":
|
||||||
|
self.notifier.on_new_event(
|
||||||
|
"groups_key", token, users=[row.user_id for row in rows]
|
||||||
|
)
|
||||||
|
elif stream_name == "pushers":
|
||||||
|
for row in rows:
|
||||||
|
if row.deleted:
|
||||||
|
self.stop_pusher(row.user_id, row.app_id, row.pushkey)
|
||||||
|
else:
|
||||||
|
await self.start_pusher(row.user_id, row.app_id, row.pushkey)
|
||||||
|
except Exception:
|
||||||
|
logger.exception("Error processing replication")
|
||||||
|
|
||||||
|
def stop_pusher(self, user_id, app_id, pushkey):
|
||||||
|
if not self.notify_pushers:
|
||||||
|
return
|
||||||
|
|
||||||
|
key = "%s:%s" % (app_id, pushkey)
|
||||||
|
pushers_for_user = self.pusher_pool.pushers.get(user_id, {})
|
||||||
|
pusher = pushers_for_user.pop(key, None)
|
||||||
|
if pusher is None:
|
||||||
|
return
|
||||||
|
logger.info("Stopping pusher %r / %r", user_id, key)
|
||||||
|
pusher.on_stop()
|
||||||
|
|
||||||
|
async def start_pusher(self, user_id, app_id, pushkey):
|
||||||
|
if not self.notify_pushers:
|
||||||
|
return
|
||||||
|
|
||||||
|
key = "%s:%s" % (app_id, pushkey)
|
||||||
|
logger.info("Starting pusher %r / %r", user_id, key)
|
||||||
|
return await self.pusher_pool.start_pusher_by_id(app_id, pushkey, user_id)
|
||||||
|
|
||||||
|
def on_remote_server_up(self, server: str):
|
||||||
|
"""Called when get a new REMOTE_SERVER_UP command."""
|
||||||
|
|
||||||
|
# Let's wake up the transaction queue for the server in case we have
|
||||||
|
# pending stuff to send to it.
|
||||||
|
if self.send_handler:
|
||||||
|
self.send_handler.wake_destination(server)
|
||||||
|
|
||||||
|
|
||||||
|
class FederationSenderHandler(object):
|
||||||
|
"""Processes the replication stream and forwards the appropriate entries
|
||||||
|
to the federation sender.
|
||||||
|
"""
|
||||||
|
|
||||||
|
def __init__(self, hs: GenericWorkerServer, replication_client):
|
||||||
|
self.store = hs.get_datastore()
|
||||||
|
self._is_mine_id = hs.is_mine_id
|
||||||
|
self.federation_sender = hs.get_federation_sender()
|
||||||
|
self.replication_client = replication_client
|
||||||
|
|
||||||
|
self.federation_position = self.store.federation_out_pos_startup
|
||||||
|
self._fed_position_linearizer = Linearizer(name="_fed_position_linearizer")
|
||||||
|
|
||||||
|
self._last_ack = self.federation_position
|
||||||
|
|
||||||
|
self._room_serials = {}
|
||||||
|
self._room_typing = {}
|
||||||
|
|
||||||
|
def on_start(self):
|
||||||
|
# There may be some events that are persisted but haven't been sent,
|
||||||
|
# so send them now.
|
||||||
|
self.federation_sender.notify_new_events(
|
||||||
|
self.store.get_room_max_stream_ordering()
|
||||||
|
)
|
||||||
|
|
||||||
|
def wake_destination(self, server: str):
|
||||||
|
self.federation_sender.wake_destination(server)
|
||||||
|
|
||||||
|
def stream_positions(self):
|
||||||
|
return {"federation": self.federation_position}
|
||||||
|
|
||||||
|
def process_replication_rows(self, stream_name, token, rows):
|
||||||
|
# The federation stream contains things that we want to send out, e.g.
|
||||||
|
# presence, typing, etc.
|
||||||
|
if stream_name == "federation":
|
||||||
|
send_queue.process_rows_for_federation(self.federation_sender, rows)
|
||||||
|
run_in_background(self.update_token, token)
|
||||||
|
|
||||||
|
# We also need to poke the federation sender when new events happen
|
||||||
|
elif stream_name == "events":
|
||||||
|
self.federation_sender.notify_new_events(token)
|
||||||
|
|
||||||
|
# ... and when new receipts happen
|
||||||
|
elif stream_name == ReceiptsStream.NAME:
|
||||||
|
run_as_background_process(
|
||||||
|
"process_receipts_for_federation", self._on_new_receipts, rows
|
||||||
|
)
|
||||||
|
|
||||||
|
# ... as well as device updates and messages
|
||||||
|
elif stream_name == DeviceListsStream.NAME:
|
||||||
|
hosts = {row.destination for row in rows}
|
||||||
|
for host in hosts:
|
||||||
|
self.federation_sender.send_device_messages(host)
|
||||||
|
|
||||||
|
elif stream_name == ToDeviceStream.NAME:
|
||||||
|
# The to_device stream includes stuff to be pushed to both local
|
||||||
|
# clients and remote servers, so we ignore entities that start with
|
||||||
|
# '@' (since they'll be local users rather than destinations).
|
||||||
|
hosts = {row.entity for row in rows if not row.entity.startswith("@")}
|
||||||
|
for host in hosts:
|
||||||
|
self.federation_sender.send_device_messages(host)
|
||||||
|
|
||||||
|
async def _on_new_receipts(self, rows):
|
||||||
|
"""
|
||||||
|
Args:
|
||||||
|
rows (iterable[synapse.replication.tcp.streams.ReceiptsStreamRow]):
|
||||||
|
new receipts to be processed
|
||||||
|
"""
|
||||||
|
for receipt in rows:
|
||||||
|
# we only want to send on receipts for our own users
|
||||||
|
if not self._is_mine_id(receipt.user_id):
|
||||||
|
continue
|
||||||
|
receipt_info = ReadReceipt(
|
||||||
|
receipt.room_id,
|
||||||
|
receipt.receipt_type,
|
||||||
|
receipt.user_id,
|
||||||
|
[receipt.event_id],
|
||||||
|
receipt.data,
|
||||||
|
)
|
||||||
|
await self.federation_sender.send_read_receipt(receipt_info)
|
||||||
|
|
||||||
|
async def update_token(self, token):
|
||||||
|
try:
|
||||||
|
self.federation_position = token
|
||||||
|
|
||||||
|
# We linearize here to ensure we don't have races updating the token
|
||||||
|
with (await self._fed_position_linearizer.queue(None)):
|
||||||
|
if self._last_ack < self.federation_position:
|
||||||
|
await self.store.update_federation_out_pos(
|
||||||
|
"federation", self.federation_position
|
||||||
|
)
|
||||||
|
|
||||||
|
# We ACK this token over replication so that the master can drop
|
||||||
|
# its in memory queues
|
||||||
|
self.replication_client.send_federation_ack(
|
||||||
|
self.federation_position
|
||||||
|
)
|
||||||
|
self._last_ack = self.federation_position
|
||||||
|
except Exception:
|
||||||
|
logger.exception("Error updating federation stream position")
|
||||||
|
|
||||||
|
|
||||||
|
def start(config_options):
|
||||||
|
try:
|
||||||
|
config = HomeServerConfig.load_config("Synapse worker", config_options)
|
||||||
|
except ConfigError as e:
|
||||||
|
sys.stderr.write("\n" + str(e) + "\n")
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
# For backwards compatibility let any of the old app names.
|
||||||
|
assert config.worker_app in (
|
||||||
|
"synapse.app.appservice",
|
||||||
|
"synapse.app.client_reader",
|
||||||
|
"synapse.app.event_creator",
|
||||||
|
"synapse.app.federation_reader",
|
||||||
|
"synapse.app.federation_sender",
|
||||||
|
"synapse.app.frontend_proxy",
|
||||||
|
"synapse.app.generic_worker",
|
||||||
|
"synapse.app.media_repository",
|
||||||
|
"synapse.app.pusher",
|
||||||
|
"synapse.app.synchrotron",
|
||||||
|
"synapse.app.user_dir",
|
||||||
|
)
|
||||||
|
|
||||||
|
if config.worker_app == "synapse.app.appservice":
|
||||||
|
if config.notify_appservices:
|
||||||
|
sys.stderr.write(
|
||||||
|
"\nThe appservices must be disabled in the main synapse process"
|
||||||
|
"\nbefore they can be run in a separate worker."
|
||||||
|
"\nPlease add ``notify_appservices: false`` to the main config"
|
||||||
|
"\n"
|
||||||
|
)
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
# Force the appservice to start since they will be disabled in the main config
|
||||||
|
config.notify_appservices = True
|
||||||
|
|
||||||
|
if config.worker_app == "synapse.app.pusher":
|
||||||
|
if config.start_pushers:
|
||||||
|
sys.stderr.write(
|
||||||
|
"\nThe pushers must be disabled in the main synapse process"
|
||||||
|
"\nbefore they can be run in a separate worker."
|
||||||
|
"\nPlease add ``start_pushers: false`` to the main config"
|
||||||
|
"\n"
|
||||||
|
)
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
# Force the pushers to start since they will be disabled in the main config
|
||||||
|
config.start_pushers = True
|
||||||
|
|
||||||
|
if config.worker_app == "synapse.app.user_dir":
|
||||||
|
if config.update_user_directory:
|
||||||
|
sys.stderr.write(
|
||||||
|
"\nThe update_user_directory must be disabled in the main synapse process"
|
||||||
|
"\nbefore they can be run in a separate worker."
|
||||||
|
"\nPlease add ``update_user_directory: false`` to the main config"
|
||||||
|
"\n"
|
||||||
|
)
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
# Force the pushers to start since they will be disabled in the main config
|
||||||
|
config.update_user_directory = True
|
||||||
|
|
||||||
|
if config.worker_app == "synapse.app.federation_sender":
|
||||||
|
if config.send_federation:
|
||||||
|
sys.stderr.write(
|
||||||
|
"\nThe send_federation must be disabled in the main synapse process"
|
||||||
|
"\nbefore they can be run in a separate worker."
|
||||||
|
"\nPlease add ``send_federation: false`` to the main config"
|
||||||
|
"\n"
|
||||||
|
)
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
# Force the pushers to start since they will be disabled in the main config
|
||||||
|
config.send_federation = True
|
||||||
|
|
||||||
|
synapse.events.USE_FROZEN_DICTS = config.use_frozen_dicts
|
||||||
|
|
||||||
|
ss = GenericWorkerServer(
|
||||||
|
config.server_name,
|
||||||
|
config=config,
|
||||||
|
version_string="Synapse/" + get_version_string(synapse),
|
||||||
|
)
|
||||||
|
|
||||||
|
setup_logging(ss, config, use_worker_options=True)
|
||||||
|
|
||||||
|
ss.setup()
|
||||||
|
reactor.addSystemEventTrigger(
|
||||||
|
"before", "startup", _base.start, ss, config.worker_listeners
|
||||||
|
)
|
||||||
|
|
||||||
|
_base.start_worker_reactor("synapse-generic-worker", config)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
with LoggingContext("main"):
|
||||||
|
start(sys.argv[1:])
|
@ -13,162 +13,11 @@
|
|||||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
# See the License for the specific language governing permissions and
|
# See the License for the specific language governing permissions and
|
||||||
# limitations under the License.
|
# limitations under the License.
|
||||||
import logging
|
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
from twisted.internet import reactor
|
from synapse.app.generic_worker import start
|
||||||
from twisted.web.resource import NoResource
|
from synapse.util.logcontext import LoggingContext
|
||||||
|
|
||||||
import synapse
|
|
||||||
from synapse import events
|
|
||||||
from synapse.api.urls import LEGACY_MEDIA_PREFIX, MEDIA_PREFIX
|
|
||||||
from synapse.app import _base
|
|
||||||
from synapse.config._base import ConfigError
|
|
||||||
from synapse.config.homeserver import HomeServerConfig
|
|
||||||
from synapse.config.logger import setup_logging
|
|
||||||
from synapse.http.server import JsonResource
|
|
||||||
from synapse.http.site import SynapseSite
|
|
||||||
from synapse.logging.context import LoggingContext
|
|
||||||
from synapse.metrics import METRICS_PREFIX, MetricsResource, RegistryProxy
|
|
||||||
from synapse.replication.slave.storage._base import BaseSlavedStore
|
|
||||||
from synapse.replication.slave.storage.appservice import SlavedApplicationServiceStore
|
|
||||||
from synapse.replication.slave.storage.client_ips import SlavedClientIpStore
|
|
||||||
from synapse.replication.slave.storage.registration import SlavedRegistrationStore
|
|
||||||
from synapse.replication.slave.storage.room import RoomStore
|
|
||||||
from synapse.replication.slave.storage.transactions import SlavedTransactionStore
|
|
||||||
from synapse.replication.tcp.client import ReplicationClientHandler
|
|
||||||
from synapse.rest.admin import register_servlets_for_media_repo
|
|
||||||
from synapse.server import HomeServer
|
|
||||||
from synapse.storage.data_stores.main.media_repository import MediaRepositoryStore
|
|
||||||
from synapse.util.httpresourcetree import create_resource_tree
|
|
||||||
from synapse.util.manhole import manhole
|
|
||||||
from synapse.util.versionstring import get_version_string
|
|
||||||
|
|
||||||
logger = logging.getLogger("synapse.app.media_repository")
|
|
||||||
|
|
||||||
|
|
||||||
class MediaRepositorySlavedStore(
|
|
||||||
RoomStore,
|
|
||||||
SlavedApplicationServiceStore,
|
|
||||||
SlavedRegistrationStore,
|
|
||||||
SlavedClientIpStore,
|
|
||||||
SlavedTransactionStore,
|
|
||||||
BaseSlavedStore,
|
|
||||||
MediaRepositoryStore,
|
|
||||||
):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
class MediaRepositoryServer(HomeServer):
|
|
||||||
DATASTORE_CLASS = MediaRepositorySlavedStore
|
|
||||||
|
|
||||||
def _listen_http(self, listener_config):
|
|
||||||
port = listener_config["port"]
|
|
||||||
bind_addresses = listener_config["bind_addresses"]
|
|
||||||
site_tag = listener_config.get("tag", port)
|
|
||||||
resources = {}
|
|
||||||
for res in listener_config["resources"]:
|
|
||||||
for name in res["names"]:
|
|
||||||
if name == "metrics":
|
|
||||||
resources[METRICS_PREFIX] = MetricsResource(RegistryProxy)
|
|
||||||
elif name == "media":
|
|
||||||
media_repo = self.get_media_repository_resource()
|
|
||||||
|
|
||||||
# We need to serve the admin servlets for media on the
|
|
||||||
# worker.
|
|
||||||
admin_resource = JsonResource(self, canonical_json=False)
|
|
||||||
register_servlets_for_media_repo(self, admin_resource)
|
|
||||||
|
|
||||||
resources.update(
|
|
||||||
{
|
|
||||||
MEDIA_PREFIX: media_repo,
|
|
||||||
LEGACY_MEDIA_PREFIX: media_repo,
|
|
||||||
"/_synapse/admin": admin_resource,
|
|
||||||
}
|
|
||||||
)
|
|
||||||
|
|
||||||
root_resource = create_resource_tree(resources, NoResource())
|
|
||||||
|
|
||||||
_base.listen_tcp(
|
|
||||||
bind_addresses,
|
|
||||||
port,
|
|
||||||
SynapseSite(
|
|
||||||
"synapse.access.http.%s" % (site_tag,),
|
|
||||||
site_tag,
|
|
||||||
listener_config,
|
|
||||||
root_resource,
|
|
||||||
self.version_string,
|
|
||||||
),
|
|
||||||
)
|
|
||||||
|
|
||||||
logger.info("Synapse media repository now listening on port %d", port)
|
|
||||||
|
|
||||||
def start_listening(self, listeners):
|
|
||||||
for listener in listeners:
|
|
||||||
if listener["type"] == "http":
|
|
||||||
self._listen_http(listener)
|
|
||||||
elif listener["type"] == "manhole":
|
|
||||||
_base.listen_tcp(
|
|
||||||
listener["bind_addresses"],
|
|
||||||
listener["port"],
|
|
||||||
manhole(
|
|
||||||
username="matrix", password="rabbithole", globals={"hs": self}
|
|
||||||
),
|
|
||||||
)
|
|
||||||
elif listener["type"] == "metrics":
|
|
||||||
if not self.get_config().enable_metrics:
|
|
||||||
logger.warning(
|
|
||||||
(
|
|
||||||
"Metrics listener configured, but "
|
|
||||||
"enable_metrics is not True!"
|
|
||||||
)
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
_base.listen_metrics(listener["bind_addresses"], listener["port"])
|
|
||||||
else:
|
|
||||||
logger.warning("Unrecognized listener type: %s", listener["type"])
|
|
||||||
|
|
||||||
self.get_tcp_replication().start_replication(self)
|
|
||||||
|
|
||||||
def build_tcp_replication(self):
|
|
||||||
return ReplicationClientHandler(self.get_datastore())
|
|
||||||
|
|
||||||
|
|
||||||
def start(config_options):
|
|
||||||
try:
|
|
||||||
config = HomeServerConfig.load_config(
|
|
||||||
"Synapse media repository", config_options
|
|
||||||
)
|
|
||||||
except ConfigError as e:
|
|
||||||
sys.stderr.write("\n" + str(e) + "\n")
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
assert config.worker_app == "synapse.app.media_repository"
|
|
||||||
|
|
||||||
if config.enable_media_repo:
|
|
||||||
_base.quit_with_error(
|
|
||||||
"enable_media_repo must be disabled in the main synapse process\n"
|
|
||||||
"before the media repo can be run in a separate worker.\n"
|
|
||||||
"Please add ``enable_media_repo: false`` to the main config\n"
|
|
||||||
)
|
|
||||||
|
|
||||||
events.USE_FROZEN_DICTS = config.use_frozen_dicts
|
|
||||||
|
|
||||||
ss = MediaRepositoryServer(
|
|
||||||
config.server_name,
|
|
||||||
config=config,
|
|
||||||
version_string="Synapse/" + get_version_string(synapse),
|
|
||||||
)
|
|
||||||
|
|
||||||
setup_logging(ss, config, use_worker_options=True)
|
|
||||||
|
|
||||||
ss.setup()
|
|
||||||
reactor.addSystemEventTrigger(
|
|
||||||
"before", "startup", _base.start, ss, config.worker_listeners
|
|
||||||
)
|
|
||||||
|
|
||||||
_base.start_worker_reactor("synapse-media-repository", config)
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
with LoggingContext("main"):
|
with LoggingContext("main"):
|
||||||
|
@ -13,213 +13,12 @@
|
|||||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
# See the License for the specific language governing permissions and
|
# See the License for the specific language governing permissions and
|
||||||
# limitations under the License.
|
# limitations under the License.
|
||||||
import logging
|
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
from twisted.internet import defer, reactor
|
from synapse.app.generic_worker import start
|
||||||
from twisted.web.resource import NoResource
|
from synapse.util.logcontext import LoggingContext
|
||||||
|
|
||||||
import synapse
|
|
||||||
from synapse import events
|
|
||||||
from synapse.app import _base
|
|
||||||
from synapse.config._base import ConfigError
|
|
||||||
from synapse.config.homeserver import HomeServerConfig
|
|
||||||
from synapse.config.logger import setup_logging
|
|
||||||
from synapse.http.site import SynapseSite
|
|
||||||
from synapse.logging.context import LoggingContext, run_in_background
|
|
||||||
from synapse.metrics import METRICS_PREFIX, MetricsResource, RegistryProxy
|
|
||||||
from synapse.replication.slave.storage._base import __func__
|
|
||||||
from synapse.replication.slave.storage.account_data import SlavedAccountDataStore
|
|
||||||
from synapse.replication.slave.storage.events import SlavedEventStore
|
|
||||||
from synapse.replication.slave.storage.pushers import SlavedPusherStore
|
|
||||||
from synapse.replication.slave.storage.receipts import SlavedReceiptsStore
|
|
||||||
from synapse.replication.slave.storage.room import RoomStore
|
|
||||||
from synapse.replication.tcp.client import ReplicationClientHandler
|
|
||||||
from synapse.server import HomeServer
|
|
||||||
from synapse.storage import DataStore
|
|
||||||
from synapse.util.httpresourcetree import create_resource_tree
|
|
||||||
from synapse.util.manhole import manhole
|
|
||||||
from synapse.util.versionstring import get_version_string
|
|
||||||
|
|
||||||
logger = logging.getLogger("synapse.app.pusher")
|
|
||||||
|
|
||||||
|
|
||||||
class PusherSlaveStore(
|
|
||||||
SlavedEventStore,
|
|
||||||
SlavedPusherStore,
|
|
||||||
SlavedReceiptsStore,
|
|
||||||
SlavedAccountDataStore,
|
|
||||||
RoomStore,
|
|
||||||
):
|
|
||||||
update_pusher_last_stream_ordering_and_success = __func__(
|
|
||||||
DataStore.update_pusher_last_stream_ordering_and_success
|
|
||||||
)
|
|
||||||
|
|
||||||
update_pusher_failing_since = __func__(DataStore.update_pusher_failing_since)
|
|
||||||
|
|
||||||
update_pusher_last_stream_ordering = __func__(
|
|
||||||
DataStore.update_pusher_last_stream_ordering
|
|
||||||
)
|
|
||||||
|
|
||||||
get_throttle_params_by_room = __func__(DataStore.get_throttle_params_by_room)
|
|
||||||
|
|
||||||
set_throttle_params = __func__(DataStore.set_throttle_params)
|
|
||||||
|
|
||||||
get_time_of_last_push_action_before = __func__(
|
|
||||||
DataStore.get_time_of_last_push_action_before
|
|
||||||
)
|
|
||||||
|
|
||||||
get_profile_displayname = __func__(DataStore.get_profile_displayname)
|
|
||||||
|
|
||||||
|
|
||||||
class PusherServer(HomeServer):
|
|
||||||
DATASTORE_CLASS = PusherSlaveStore
|
|
||||||
|
|
||||||
def remove_pusher(self, app_id, push_key, user_id):
|
|
||||||
self.get_tcp_replication().send_remove_pusher(app_id, push_key, user_id)
|
|
||||||
|
|
||||||
def _listen_http(self, listener_config):
|
|
||||||
port = listener_config["port"]
|
|
||||||
bind_addresses = listener_config["bind_addresses"]
|
|
||||||
site_tag = listener_config.get("tag", port)
|
|
||||||
resources = {}
|
|
||||||
for res in listener_config["resources"]:
|
|
||||||
for name in res["names"]:
|
|
||||||
if name == "metrics":
|
|
||||||
resources[METRICS_PREFIX] = MetricsResource(RegistryProxy)
|
|
||||||
|
|
||||||
root_resource = create_resource_tree(resources, NoResource())
|
|
||||||
|
|
||||||
_base.listen_tcp(
|
|
||||||
bind_addresses,
|
|
||||||
port,
|
|
||||||
SynapseSite(
|
|
||||||
"synapse.access.http.%s" % (site_tag,),
|
|
||||||
site_tag,
|
|
||||||
listener_config,
|
|
||||||
root_resource,
|
|
||||||
self.version_string,
|
|
||||||
),
|
|
||||||
)
|
|
||||||
|
|
||||||
logger.info("Synapse pusher now listening on port %d", port)
|
|
||||||
|
|
||||||
def start_listening(self, listeners):
|
|
||||||
for listener in listeners:
|
|
||||||
if listener["type"] == "http":
|
|
||||||
self._listen_http(listener)
|
|
||||||
elif listener["type"] == "manhole":
|
|
||||||
_base.listen_tcp(
|
|
||||||
listener["bind_addresses"],
|
|
||||||
listener["port"],
|
|
||||||
manhole(
|
|
||||||
username="matrix", password="rabbithole", globals={"hs": self}
|
|
||||||
),
|
|
||||||
)
|
|
||||||
elif listener["type"] == "metrics":
|
|
||||||
if not self.get_config().enable_metrics:
|
|
||||||
logger.warning(
|
|
||||||
(
|
|
||||||
"Metrics listener configured, but "
|
|
||||||
"enable_metrics is not True!"
|
|
||||||
)
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
_base.listen_metrics(listener["bind_addresses"], listener["port"])
|
|
||||||
else:
|
|
||||||
logger.warning("Unrecognized listener type: %s", listener["type"])
|
|
||||||
|
|
||||||
self.get_tcp_replication().start_replication(self)
|
|
||||||
|
|
||||||
def build_tcp_replication(self):
|
|
||||||
return PusherReplicationHandler(self)
|
|
||||||
|
|
||||||
|
|
||||||
class PusherReplicationHandler(ReplicationClientHandler):
|
|
||||||
def __init__(self, hs):
|
|
||||||
super(PusherReplicationHandler, self).__init__(hs.get_datastore())
|
|
||||||
|
|
||||||
self.pusher_pool = hs.get_pusherpool()
|
|
||||||
|
|
||||||
async def on_rdata(self, stream_name, token, rows):
|
|
||||||
await super(PusherReplicationHandler, self).on_rdata(stream_name, token, rows)
|
|
||||||
run_in_background(self.poke_pushers, stream_name, token, rows)
|
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
|
||||||
def poke_pushers(self, stream_name, token, rows):
|
|
||||||
try:
|
|
||||||
if stream_name == "pushers":
|
|
||||||
for row in rows:
|
|
||||||
if row.deleted:
|
|
||||||
yield self.stop_pusher(row.user_id, row.app_id, row.pushkey)
|
|
||||||
else:
|
|
||||||
yield self.start_pusher(row.user_id, row.app_id, row.pushkey)
|
|
||||||
elif stream_name == "events":
|
|
||||||
yield self.pusher_pool.on_new_notifications(token, token)
|
|
||||||
elif stream_name == "receipts":
|
|
||||||
yield self.pusher_pool.on_new_receipts(
|
|
||||||
token, token, {row.room_id for row in rows}
|
|
||||||
)
|
|
||||||
except Exception:
|
|
||||||
logger.exception("Error poking pushers")
|
|
||||||
|
|
||||||
def stop_pusher(self, user_id, app_id, pushkey):
|
|
||||||
key = "%s:%s" % (app_id, pushkey)
|
|
||||||
pushers_for_user = self.pusher_pool.pushers.get(user_id, {})
|
|
||||||
pusher = pushers_for_user.pop(key, None)
|
|
||||||
if pusher is None:
|
|
||||||
return
|
|
||||||
logger.info("Stopping pusher %r / %r", user_id, key)
|
|
||||||
pusher.on_stop()
|
|
||||||
|
|
||||||
def start_pusher(self, user_id, app_id, pushkey):
|
|
||||||
key = "%s:%s" % (app_id, pushkey)
|
|
||||||
logger.info("Starting pusher %r / %r", user_id, key)
|
|
||||||
return self.pusher_pool.start_pusher_by_id(app_id, pushkey, user_id)
|
|
||||||
|
|
||||||
|
|
||||||
def start(config_options):
|
|
||||||
try:
|
|
||||||
config = HomeServerConfig.load_config("Synapse pusher", config_options)
|
|
||||||
except ConfigError as e:
|
|
||||||
sys.stderr.write("\n" + str(e) + "\n")
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
assert config.worker_app == "synapse.app.pusher"
|
|
||||||
|
|
||||||
events.USE_FROZEN_DICTS = config.use_frozen_dicts
|
|
||||||
|
|
||||||
if config.start_pushers:
|
|
||||||
sys.stderr.write(
|
|
||||||
"\nThe pushers must be disabled in the main synapse process"
|
|
||||||
"\nbefore they can be run in a separate worker."
|
|
||||||
"\nPlease add ``start_pushers: false`` to the main config"
|
|
||||||
"\n"
|
|
||||||
)
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
# Force the pushers to start since they will be disabled in the main config
|
|
||||||
config.start_pushers = True
|
|
||||||
|
|
||||||
ps = PusherServer(
|
|
||||||
config.server_name,
|
|
||||||
config=config,
|
|
||||||
version_string="Synapse/" + get_version_string(synapse),
|
|
||||||
)
|
|
||||||
|
|
||||||
setup_logging(ps, config, use_worker_options=True)
|
|
||||||
|
|
||||||
ps.setup()
|
|
||||||
|
|
||||||
def start():
|
|
||||||
_base.start(ps, config.worker_listeners)
|
|
||||||
ps.get_pusherpool().start()
|
|
||||||
|
|
||||||
reactor.addSystemEventTrigger("before", "startup", start)
|
|
||||||
|
|
||||||
_base.start_worker_reactor("synapse-pusher", config)
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
with LoggingContext("main"):
|
with LoggingContext("main"):
|
||||||
ps = start(sys.argv[1:])
|
start(sys.argv[1:])
|
||||||
|
@ -13,454 +13,11 @@
|
|||||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
# See the License for the specific language governing permissions and
|
# See the License for the specific language governing permissions and
|
||||||
# limitations under the License.
|
# limitations under the License.
|
||||||
import contextlib
|
|
||||||
import logging
|
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
from six import iteritems
|
from synapse.app.generic_worker import start
|
||||||
|
from synapse.util.logcontext import LoggingContext
|
||||||
from twisted.internet import defer, reactor
|
|
||||||
from twisted.web.resource import NoResource
|
|
||||||
|
|
||||||
import synapse
|
|
||||||
from synapse.api.constants import EventTypes
|
|
||||||
from synapse.app import _base
|
|
||||||
from synapse.config._base import ConfigError
|
|
||||||
from synapse.config.homeserver import HomeServerConfig
|
|
||||||
from synapse.config.logger import setup_logging
|
|
||||||
from synapse.handlers.presence import PresenceHandler, get_interested_parties
|
|
||||||
from synapse.http.server import JsonResource
|
|
||||||
from synapse.http.site import SynapseSite
|
|
||||||
from synapse.logging.context import LoggingContext, run_in_background
|
|
||||||
from synapse.metrics import METRICS_PREFIX, MetricsResource, RegistryProxy
|
|
||||||
from synapse.replication.slave.storage._base import BaseSlavedStore, __func__
|
|
||||||
from synapse.replication.slave.storage.account_data import SlavedAccountDataStore
|
|
||||||
from synapse.replication.slave.storage.appservice import SlavedApplicationServiceStore
|
|
||||||
from synapse.replication.slave.storage.client_ips import SlavedClientIpStore
|
|
||||||
from synapse.replication.slave.storage.deviceinbox import SlavedDeviceInboxStore
|
|
||||||
from synapse.replication.slave.storage.devices import SlavedDeviceStore
|
|
||||||
from synapse.replication.slave.storage.events import SlavedEventStore
|
|
||||||
from synapse.replication.slave.storage.filtering import SlavedFilteringStore
|
|
||||||
from synapse.replication.slave.storage.groups import SlavedGroupServerStore
|
|
||||||
from synapse.replication.slave.storage.presence import SlavedPresenceStore
|
|
||||||
from synapse.replication.slave.storage.push_rule import SlavedPushRuleStore
|
|
||||||
from synapse.replication.slave.storage.receipts import SlavedReceiptsStore
|
|
||||||
from synapse.replication.slave.storage.registration import SlavedRegistrationStore
|
|
||||||
from synapse.replication.slave.storage.room import RoomStore
|
|
||||||
from synapse.replication.tcp.client import ReplicationClientHandler
|
|
||||||
from synapse.replication.tcp.streams.events import EventsStreamEventRow, EventsStreamRow
|
|
||||||
from synapse.rest.client.v1 import events
|
|
||||||
from synapse.rest.client.v1.initial_sync import InitialSyncRestServlet
|
|
||||||
from synapse.rest.client.v1.room import RoomInitialSyncRestServlet
|
|
||||||
from synapse.rest.client.v2_alpha import sync
|
|
||||||
from synapse.server import HomeServer
|
|
||||||
from synapse.storage.data_stores.main.monthly_active_users import (
|
|
||||||
MonthlyActiveUsersWorkerStore,
|
|
||||||
)
|
|
||||||
from synapse.storage.data_stores.main.presence import UserPresenceState
|
|
||||||
from synapse.util.httpresourcetree import create_resource_tree
|
|
||||||
from synapse.util.manhole import manhole
|
|
||||||
from synapse.util.stringutils import random_string
|
|
||||||
from synapse.util.versionstring import get_version_string
|
|
||||||
|
|
||||||
logger = logging.getLogger("synapse.app.synchrotron")
|
|
||||||
|
|
||||||
|
|
||||||
class SynchrotronSlavedStore(
|
|
||||||
SlavedReceiptsStore,
|
|
||||||
SlavedAccountDataStore,
|
|
||||||
SlavedApplicationServiceStore,
|
|
||||||
SlavedRegistrationStore,
|
|
||||||
SlavedFilteringStore,
|
|
||||||
SlavedPresenceStore,
|
|
||||||
SlavedGroupServerStore,
|
|
||||||
SlavedDeviceInboxStore,
|
|
||||||
SlavedDeviceStore,
|
|
||||||
SlavedPushRuleStore,
|
|
||||||
SlavedEventStore,
|
|
||||||
SlavedClientIpStore,
|
|
||||||
RoomStore,
|
|
||||||
MonthlyActiveUsersWorkerStore,
|
|
||||||
BaseSlavedStore,
|
|
||||||
):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
UPDATE_SYNCING_USERS_MS = 10 * 1000
|
|
||||||
|
|
||||||
|
|
||||||
class SynchrotronPresence(object):
|
|
||||||
def __init__(self, hs):
|
|
||||||
self.hs = hs
|
|
||||||
self.is_mine_id = hs.is_mine_id
|
|
||||||
self.http_client = hs.get_simple_http_client()
|
|
||||||
self.store = hs.get_datastore()
|
|
||||||
self.user_to_num_current_syncs = {}
|
|
||||||
self.clock = hs.get_clock()
|
|
||||||
self.notifier = hs.get_notifier()
|
|
||||||
|
|
||||||
active_presence = self.store.take_presence_startup_info()
|
|
||||||
self.user_to_current_state = {state.user_id: state for state in active_presence}
|
|
||||||
|
|
||||||
# user_id -> last_sync_ms. Lists the users that have stopped syncing
|
|
||||||
# but we haven't notified the master of that yet
|
|
||||||
self.users_going_offline = {}
|
|
||||||
|
|
||||||
self._send_stop_syncing_loop = self.clock.looping_call(
|
|
||||||
self.send_stop_syncing, 10 * 1000
|
|
||||||
)
|
|
||||||
|
|
||||||
self.process_id = random_string(16)
|
|
||||||
logger.info("Presence process_id is %r", self.process_id)
|
|
||||||
|
|
||||||
def send_user_sync(self, user_id, is_syncing, last_sync_ms):
|
|
||||||
if self.hs.config.use_presence:
|
|
||||||
self.hs.get_tcp_replication().send_user_sync(
|
|
||||||
user_id, is_syncing, last_sync_ms
|
|
||||||
)
|
|
||||||
|
|
||||||
def mark_as_coming_online(self, user_id):
|
|
||||||
"""A user has started syncing. Send a UserSync to the master, unless they
|
|
||||||
had recently stopped syncing.
|
|
||||||
|
|
||||||
Args:
|
|
||||||
user_id (str)
|
|
||||||
"""
|
|
||||||
going_offline = self.users_going_offline.pop(user_id, None)
|
|
||||||
if not going_offline:
|
|
||||||
# Safe to skip because we haven't yet told the master they were offline
|
|
||||||
self.send_user_sync(user_id, True, self.clock.time_msec())
|
|
||||||
|
|
||||||
def mark_as_going_offline(self, user_id):
|
|
||||||
"""A user has stopped syncing. We wait before notifying the master as
|
|
||||||
its likely they'll come back soon. This allows us to avoid sending
|
|
||||||
a stopped syncing immediately followed by a started syncing notification
|
|
||||||
to the master
|
|
||||||
|
|
||||||
Args:
|
|
||||||
user_id (str)
|
|
||||||
"""
|
|
||||||
self.users_going_offline[user_id] = self.clock.time_msec()
|
|
||||||
|
|
||||||
def send_stop_syncing(self):
|
|
||||||
"""Check if there are any users who have stopped syncing a while ago
|
|
||||||
and haven't come back yet. If there are poke the master about them.
|
|
||||||
"""
|
|
||||||
now = self.clock.time_msec()
|
|
||||||
for user_id, last_sync_ms in list(self.users_going_offline.items()):
|
|
||||||
if now - last_sync_ms > 10 * 1000:
|
|
||||||
self.users_going_offline.pop(user_id, None)
|
|
||||||
self.send_user_sync(user_id, False, last_sync_ms)
|
|
||||||
|
|
||||||
def set_state(self, user, state, ignore_status_msg=False):
|
|
||||||
# TODO Hows this supposed to work?
|
|
||||||
return defer.succeed(None)
|
|
||||||
|
|
||||||
get_states = __func__(PresenceHandler.get_states)
|
|
||||||
get_state = __func__(PresenceHandler.get_state)
|
|
||||||
current_state_for_users = __func__(PresenceHandler.current_state_for_users)
|
|
||||||
|
|
||||||
def user_syncing(self, user_id, affect_presence):
|
|
||||||
if affect_presence:
|
|
||||||
curr_sync = self.user_to_num_current_syncs.get(user_id, 0)
|
|
||||||
self.user_to_num_current_syncs[user_id] = curr_sync + 1
|
|
||||||
|
|
||||||
# If we went from no in flight sync to some, notify replication
|
|
||||||
if self.user_to_num_current_syncs[user_id] == 1:
|
|
||||||
self.mark_as_coming_online(user_id)
|
|
||||||
|
|
||||||
def _end():
|
|
||||||
# We check that the user_id is in user_to_num_current_syncs because
|
|
||||||
# user_to_num_current_syncs may have been cleared if we are
|
|
||||||
# shutting down.
|
|
||||||
if affect_presence and user_id in self.user_to_num_current_syncs:
|
|
||||||
self.user_to_num_current_syncs[user_id] -= 1
|
|
||||||
|
|
||||||
# If we went from one in flight sync to non, notify replication
|
|
||||||
if self.user_to_num_current_syncs[user_id] == 0:
|
|
||||||
self.mark_as_going_offline(user_id)
|
|
||||||
|
|
||||||
@contextlib.contextmanager
|
|
||||||
def _user_syncing():
|
|
||||||
try:
|
|
||||||
yield
|
|
||||||
finally:
|
|
||||||
_end()
|
|
||||||
|
|
||||||
return defer.succeed(_user_syncing())
|
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
|
||||||
def notify_from_replication(self, states, stream_id):
|
|
||||||
parties = yield get_interested_parties(self.store, states)
|
|
||||||
room_ids_to_states, users_to_states = parties
|
|
||||||
|
|
||||||
self.notifier.on_new_event(
|
|
||||||
"presence_key",
|
|
||||||
stream_id,
|
|
||||||
rooms=room_ids_to_states.keys(),
|
|
||||||
users=users_to_states.keys(),
|
|
||||||
)
|
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
|
||||||
def process_replication_rows(self, token, rows):
|
|
||||||
states = [
|
|
||||||
UserPresenceState(
|
|
||||||
row.user_id,
|
|
||||||
row.state,
|
|
||||||
row.last_active_ts,
|
|
||||||
row.last_federation_update_ts,
|
|
||||||
row.last_user_sync_ts,
|
|
||||||
row.status_msg,
|
|
||||||
row.currently_active,
|
|
||||||
)
|
|
||||||
for row in rows
|
|
||||||
]
|
|
||||||
|
|
||||||
for state in states:
|
|
||||||
self.user_to_current_state[state.user_id] = state
|
|
||||||
|
|
||||||
stream_id = token
|
|
||||||
yield self.notify_from_replication(states, stream_id)
|
|
||||||
|
|
||||||
def get_currently_syncing_users(self):
|
|
||||||
if self.hs.config.use_presence:
|
|
||||||
return [
|
|
||||||
user_id
|
|
||||||
for user_id, count in iteritems(self.user_to_num_current_syncs)
|
|
||||||
if count > 0
|
|
||||||
]
|
|
||||||
else:
|
|
||||||
return set()
|
|
||||||
|
|
||||||
|
|
||||||
class SynchrotronTyping(object):
|
|
||||||
def __init__(self, hs):
|
|
||||||
self._latest_room_serial = 0
|
|
||||||
self._reset()
|
|
||||||
|
|
||||||
def _reset(self):
|
|
||||||
"""
|
|
||||||
Reset the typing handler's data caches.
|
|
||||||
"""
|
|
||||||
# map room IDs to serial numbers
|
|
||||||
self._room_serials = {}
|
|
||||||
# map room IDs to sets of users currently typing
|
|
||||||
self._room_typing = {}
|
|
||||||
|
|
||||||
def stream_positions(self):
|
|
||||||
# We must update this typing token from the response of the previous
|
|
||||||
# sync. In particular, the stream id may "reset" back to zero/a low
|
|
||||||
# value which we *must* use for the next replication request.
|
|
||||||
return {"typing": self._latest_room_serial}
|
|
||||||
|
|
||||||
def process_replication_rows(self, token, rows):
|
|
||||||
if self._latest_room_serial > token:
|
|
||||||
# The master has gone backwards. To prevent inconsistent data, just
|
|
||||||
# clear everything.
|
|
||||||
self._reset()
|
|
||||||
|
|
||||||
# Set the latest serial token to whatever the server gave us.
|
|
||||||
self._latest_room_serial = token
|
|
||||||
|
|
||||||
for row in rows:
|
|
||||||
self._room_serials[row.room_id] = token
|
|
||||||
self._room_typing[row.room_id] = row.user_ids
|
|
||||||
|
|
||||||
|
|
||||||
class SynchrotronApplicationService(object):
|
|
||||||
def notify_interested_services(self, event):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
class SynchrotronServer(HomeServer):
|
|
||||||
DATASTORE_CLASS = SynchrotronSlavedStore
|
|
||||||
|
|
||||||
def _listen_http(self, listener_config):
|
|
||||||
port = listener_config["port"]
|
|
||||||
bind_addresses = listener_config["bind_addresses"]
|
|
||||||
site_tag = listener_config.get("tag", port)
|
|
||||||
resources = {}
|
|
||||||
for res in listener_config["resources"]:
|
|
||||||
for name in res["names"]:
|
|
||||||
if name == "metrics":
|
|
||||||
resources[METRICS_PREFIX] = MetricsResource(RegistryProxy)
|
|
||||||
elif name == "client":
|
|
||||||
resource = JsonResource(self, canonical_json=False)
|
|
||||||
sync.register_servlets(self, resource)
|
|
||||||
events.register_servlets(self, resource)
|
|
||||||
InitialSyncRestServlet(self).register(resource)
|
|
||||||
RoomInitialSyncRestServlet(self).register(resource)
|
|
||||||
resources.update(
|
|
||||||
{
|
|
||||||
"/_matrix/client/r0": resource,
|
|
||||||
"/_matrix/client/unstable": resource,
|
|
||||||
"/_matrix/client/v2_alpha": resource,
|
|
||||||
"/_matrix/client/api/v1": resource,
|
|
||||||
}
|
|
||||||
)
|
|
||||||
|
|
||||||
root_resource = create_resource_tree(resources, NoResource())
|
|
||||||
|
|
||||||
_base.listen_tcp(
|
|
||||||
bind_addresses,
|
|
||||||
port,
|
|
||||||
SynapseSite(
|
|
||||||
"synapse.access.http.%s" % (site_tag,),
|
|
||||||
site_tag,
|
|
||||||
listener_config,
|
|
||||||
root_resource,
|
|
||||||
self.version_string,
|
|
||||||
),
|
|
||||||
)
|
|
||||||
|
|
||||||
logger.info("Synapse synchrotron now listening on port %d", port)
|
|
||||||
|
|
||||||
def start_listening(self, listeners):
|
|
||||||
for listener in listeners:
|
|
||||||
if listener["type"] == "http":
|
|
||||||
self._listen_http(listener)
|
|
||||||
elif listener["type"] == "manhole":
|
|
||||||
_base.listen_tcp(
|
|
||||||
listener["bind_addresses"],
|
|
||||||
listener["port"],
|
|
||||||
manhole(
|
|
||||||
username="matrix", password="rabbithole", globals={"hs": self}
|
|
||||||
),
|
|
||||||
)
|
|
||||||
elif listener["type"] == "metrics":
|
|
||||||
if not self.get_config().enable_metrics:
|
|
||||||
logger.warning(
|
|
||||||
(
|
|
||||||
"Metrics listener configured, but "
|
|
||||||
"enable_metrics is not True!"
|
|
||||||
)
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
_base.listen_metrics(listener["bind_addresses"], listener["port"])
|
|
||||||
else:
|
|
||||||
logger.warning("Unrecognized listener type: %s", listener["type"])
|
|
||||||
|
|
||||||
self.get_tcp_replication().start_replication(self)
|
|
||||||
|
|
||||||
def build_tcp_replication(self):
|
|
||||||
return SyncReplicationHandler(self)
|
|
||||||
|
|
||||||
def build_presence_handler(self):
|
|
||||||
return SynchrotronPresence(self)
|
|
||||||
|
|
||||||
def build_typing_handler(self):
|
|
||||||
return SynchrotronTyping(self)
|
|
||||||
|
|
||||||
|
|
||||||
class SyncReplicationHandler(ReplicationClientHandler):
|
|
||||||
def __init__(self, hs):
|
|
||||||
super(SyncReplicationHandler, self).__init__(hs.get_datastore())
|
|
||||||
|
|
||||||
self.store = hs.get_datastore()
|
|
||||||
self.typing_handler = hs.get_typing_handler()
|
|
||||||
# NB this is a SynchrotronPresence, not a normal PresenceHandler
|
|
||||||
self.presence_handler = hs.get_presence_handler()
|
|
||||||
self.notifier = hs.get_notifier()
|
|
||||||
|
|
||||||
async def on_rdata(self, stream_name, token, rows):
|
|
||||||
await super(SyncReplicationHandler, self).on_rdata(stream_name, token, rows)
|
|
||||||
run_in_background(self.process_and_notify, stream_name, token, rows)
|
|
||||||
|
|
||||||
def get_streams_to_replicate(self):
|
|
||||||
args = super(SyncReplicationHandler, self).get_streams_to_replicate()
|
|
||||||
args.update(self.typing_handler.stream_positions())
|
|
||||||
return args
|
|
||||||
|
|
||||||
def get_currently_syncing_users(self):
|
|
||||||
return self.presence_handler.get_currently_syncing_users()
|
|
||||||
|
|
||||||
async def process_and_notify(self, stream_name, token, rows):
|
|
||||||
try:
|
|
||||||
if stream_name == "events":
|
|
||||||
# We shouldn't get multiple rows per token for events stream, so
|
|
||||||
# we don't need to optimise this for multiple rows.
|
|
||||||
for row in rows:
|
|
||||||
if row.type != EventsStreamEventRow.TypeId:
|
|
||||||
continue
|
|
||||||
assert isinstance(row, EventsStreamRow)
|
|
||||||
|
|
||||||
event = await self.store.get_event(
|
|
||||||
row.data.event_id, allow_rejected=True
|
|
||||||
)
|
|
||||||
if event.rejected_reason:
|
|
||||||
continue
|
|
||||||
|
|
||||||
extra_users = ()
|
|
||||||
if event.type == EventTypes.Member:
|
|
||||||
extra_users = (event.state_key,)
|
|
||||||
max_token = self.store.get_room_max_stream_ordering()
|
|
||||||
self.notifier.on_new_room_event(
|
|
||||||
event, token, max_token, extra_users
|
|
||||||
)
|
|
||||||
elif stream_name == "push_rules":
|
|
||||||
self.notifier.on_new_event(
|
|
||||||
"push_rules_key", token, users=[row.user_id for row in rows]
|
|
||||||
)
|
|
||||||
elif stream_name in ("account_data", "tag_account_data"):
|
|
||||||
self.notifier.on_new_event(
|
|
||||||
"account_data_key", token, users=[row.user_id for row in rows]
|
|
||||||
)
|
|
||||||
elif stream_name == "receipts":
|
|
||||||
self.notifier.on_new_event(
|
|
||||||
"receipt_key", token, rooms=[row.room_id for row in rows]
|
|
||||||
)
|
|
||||||
elif stream_name == "typing":
|
|
||||||
self.typing_handler.process_replication_rows(token, rows)
|
|
||||||
self.notifier.on_new_event(
|
|
||||||
"typing_key", token, rooms=[row.room_id for row in rows]
|
|
||||||
)
|
|
||||||
elif stream_name == "to_device":
|
|
||||||
entities = [row.entity for row in rows if row.entity.startswith("@")]
|
|
||||||
if entities:
|
|
||||||
self.notifier.on_new_event("to_device_key", token, users=entities)
|
|
||||||
elif stream_name == "device_lists":
|
|
||||||
all_room_ids = set()
|
|
||||||
for row in rows:
|
|
||||||
room_ids = await self.store.get_rooms_for_user(row.user_id)
|
|
||||||
all_room_ids.update(room_ids)
|
|
||||||
self.notifier.on_new_event("device_list_key", token, rooms=all_room_ids)
|
|
||||||
elif stream_name == "presence":
|
|
||||||
await self.presence_handler.process_replication_rows(token, rows)
|
|
||||||
elif stream_name == "receipts":
|
|
||||||
self.notifier.on_new_event(
|
|
||||||
"groups_key", token, users=[row.user_id for row in rows]
|
|
||||||
)
|
|
||||||
except Exception:
|
|
||||||
logger.exception("Error processing replication")
|
|
||||||
|
|
||||||
|
|
||||||
def start(config_options):
|
|
||||||
try:
|
|
||||||
config = HomeServerConfig.load_config("Synapse synchrotron", config_options)
|
|
||||||
except ConfigError as e:
|
|
||||||
sys.stderr.write("\n" + str(e) + "\n")
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
assert config.worker_app == "synapse.app.synchrotron"
|
|
||||||
|
|
||||||
synapse.events.USE_FROZEN_DICTS = config.use_frozen_dicts
|
|
||||||
|
|
||||||
ss = SynchrotronServer(
|
|
||||||
config.server_name,
|
|
||||||
config=config,
|
|
||||||
version_string="Synapse/" + get_version_string(synapse),
|
|
||||||
application_service_handler=SynchrotronApplicationService(),
|
|
||||||
)
|
|
||||||
|
|
||||||
setup_logging(ss, config, use_worker_options=True)
|
|
||||||
|
|
||||||
ss.setup()
|
|
||||||
reactor.addSystemEventTrigger(
|
|
||||||
"before", "startup", _base.start, ss, config.worker_listeners
|
|
||||||
)
|
|
||||||
|
|
||||||
_base.start_worker_reactor("synapse-synchrotron", config)
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
with LoggingContext("main"):
|
with LoggingContext("main"):
|
||||||
|
@ -14,217 +14,10 @@
|
|||||||
# See the License for the specific language governing permissions and
|
# See the License for the specific language governing permissions and
|
||||||
# limitations under the License.
|
# limitations under the License.
|
||||||
|
|
||||||
import logging
|
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
from twisted.internet import defer, reactor
|
from synapse.app.generic_worker import start
|
||||||
from twisted.web.resource import NoResource
|
from synapse.util.logcontext import LoggingContext
|
||||||
|
|
||||||
import synapse
|
|
||||||
from synapse import events
|
|
||||||
from synapse.app import _base
|
|
||||||
from synapse.config._base import ConfigError
|
|
||||||
from synapse.config.homeserver import HomeServerConfig
|
|
||||||
from synapse.config.logger import setup_logging
|
|
||||||
from synapse.http.server import JsonResource
|
|
||||||
from synapse.http.site import SynapseSite
|
|
||||||
from synapse.logging.context import LoggingContext, run_in_background
|
|
||||||
from synapse.metrics import METRICS_PREFIX, MetricsResource, RegistryProxy
|
|
||||||
from synapse.replication.slave.storage._base import BaseSlavedStore
|
|
||||||
from synapse.replication.slave.storage.appservice import SlavedApplicationServiceStore
|
|
||||||
from synapse.replication.slave.storage.client_ips import SlavedClientIpStore
|
|
||||||
from synapse.replication.slave.storage.events import SlavedEventStore
|
|
||||||
from synapse.replication.slave.storage.registration import SlavedRegistrationStore
|
|
||||||
from synapse.replication.tcp.client import ReplicationClientHandler
|
|
||||||
from synapse.replication.tcp.streams.events import (
|
|
||||||
EventsStream,
|
|
||||||
EventsStreamCurrentStateRow,
|
|
||||||
)
|
|
||||||
from synapse.rest.client.v2_alpha import user_directory
|
|
||||||
from synapse.server import HomeServer
|
|
||||||
from synapse.storage.data_stores.main.user_directory import UserDirectoryStore
|
|
||||||
from synapse.storage.database import Database
|
|
||||||
from synapse.util.caches.stream_change_cache import StreamChangeCache
|
|
||||||
from synapse.util.httpresourcetree import create_resource_tree
|
|
||||||
from synapse.util.manhole import manhole
|
|
||||||
from synapse.util.versionstring import get_version_string
|
|
||||||
|
|
||||||
logger = logging.getLogger("synapse.app.user_dir")
|
|
||||||
|
|
||||||
|
|
||||||
class UserDirectorySlaveStore(
|
|
||||||
SlavedEventStore,
|
|
||||||
SlavedApplicationServiceStore,
|
|
||||||
SlavedRegistrationStore,
|
|
||||||
SlavedClientIpStore,
|
|
||||||
UserDirectoryStore,
|
|
||||||
BaseSlavedStore,
|
|
||||||
):
|
|
||||||
def __init__(self, database: Database, db_conn, hs):
|
|
||||||
super(UserDirectorySlaveStore, self).__init__(database, db_conn, hs)
|
|
||||||
|
|
||||||
events_max = self._stream_id_gen.get_current_token()
|
|
||||||
curr_state_delta_prefill, min_curr_state_delta_id = self.db.get_cache_dict(
|
|
||||||
db_conn,
|
|
||||||
"current_state_delta_stream",
|
|
||||||
entity_column="room_id",
|
|
||||||
stream_column="stream_id",
|
|
||||||
max_value=events_max, # As we share the stream id with events token
|
|
||||||
limit=1000,
|
|
||||||
)
|
|
||||||
self._curr_state_delta_stream_cache = StreamChangeCache(
|
|
||||||
"_curr_state_delta_stream_cache",
|
|
||||||
min_curr_state_delta_id,
|
|
||||||
prefilled_cache=curr_state_delta_prefill,
|
|
||||||
)
|
|
||||||
|
|
||||||
def stream_positions(self):
|
|
||||||
result = super(UserDirectorySlaveStore, self).stream_positions()
|
|
||||||
return result
|
|
||||||
|
|
||||||
def process_replication_rows(self, stream_name, token, rows):
|
|
||||||
if stream_name == EventsStream.NAME:
|
|
||||||
self._stream_id_gen.advance(token)
|
|
||||||
for row in rows:
|
|
||||||
if row.type != EventsStreamCurrentStateRow.TypeId:
|
|
||||||
continue
|
|
||||||
self._curr_state_delta_stream_cache.entity_has_changed(
|
|
||||||
row.data.room_id, token
|
|
||||||
)
|
|
||||||
return super(UserDirectorySlaveStore, self).process_replication_rows(
|
|
||||||
stream_name, token, rows
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
class UserDirectoryServer(HomeServer):
|
|
||||||
DATASTORE_CLASS = UserDirectorySlaveStore
|
|
||||||
|
|
||||||
def _listen_http(self, listener_config):
|
|
||||||
port = listener_config["port"]
|
|
||||||
bind_addresses = listener_config["bind_addresses"]
|
|
||||||
site_tag = listener_config.get("tag", port)
|
|
||||||
resources = {}
|
|
||||||
for res in listener_config["resources"]:
|
|
||||||
for name in res["names"]:
|
|
||||||
if name == "metrics":
|
|
||||||
resources[METRICS_PREFIX] = MetricsResource(RegistryProxy)
|
|
||||||
elif name == "client":
|
|
||||||
resource = JsonResource(self, canonical_json=False)
|
|
||||||
user_directory.register_servlets(self, resource)
|
|
||||||
resources.update(
|
|
||||||
{
|
|
||||||
"/_matrix/client/r0": resource,
|
|
||||||
"/_matrix/client/unstable": resource,
|
|
||||||
"/_matrix/client/v2_alpha": resource,
|
|
||||||
"/_matrix/client/api/v1": resource,
|
|
||||||
}
|
|
||||||
)
|
|
||||||
|
|
||||||
root_resource = create_resource_tree(resources, NoResource())
|
|
||||||
|
|
||||||
_base.listen_tcp(
|
|
||||||
bind_addresses,
|
|
||||||
port,
|
|
||||||
SynapseSite(
|
|
||||||
"synapse.access.http.%s" % (site_tag,),
|
|
||||||
site_tag,
|
|
||||||
listener_config,
|
|
||||||
root_resource,
|
|
||||||
self.version_string,
|
|
||||||
),
|
|
||||||
)
|
|
||||||
|
|
||||||
logger.info("Synapse user_dir now listening on port %d", port)
|
|
||||||
|
|
||||||
def start_listening(self, listeners):
|
|
||||||
for listener in listeners:
|
|
||||||
if listener["type"] == "http":
|
|
||||||
self._listen_http(listener)
|
|
||||||
elif listener["type"] == "manhole":
|
|
||||||
_base.listen_tcp(
|
|
||||||
listener["bind_addresses"],
|
|
||||||
listener["port"],
|
|
||||||
manhole(
|
|
||||||
username="matrix", password="rabbithole", globals={"hs": self}
|
|
||||||
),
|
|
||||||
)
|
|
||||||
elif listener["type"] == "metrics":
|
|
||||||
if not self.get_config().enable_metrics:
|
|
||||||
logger.warning(
|
|
||||||
(
|
|
||||||
"Metrics listener configured, but "
|
|
||||||
"enable_metrics is not True!"
|
|
||||||
)
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
_base.listen_metrics(listener["bind_addresses"], listener["port"])
|
|
||||||
else:
|
|
||||||
logger.warning("Unrecognized listener type: %s", listener["type"])
|
|
||||||
|
|
||||||
self.get_tcp_replication().start_replication(self)
|
|
||||||
|
|
||||||
def build_tcp_replication(self):
|
|
||||||
return UserDirectoryReplicationHandler(self)
|
|
||||||
|
|
||||||
|
|
||||||
class UserDirectoryReplicationHandler(ReplicationClientHandler):
|
|
||||||
def __init__(self, hs):
|
|
||||||
super(UserDirectoryReplicationHandler, self).__init__(hs.get_datastore())
|
|
||||||
self.user_directory = hs.get_user_directory_handler()
|
|
||||||
|
|
||||||
async def on_rdata(self, stream_name, token, rows):
|
|
||||||
await super(UserDirectoryReplicationHandler, self).on_rdata(
|
|
||||||
stream_name, token, rows
|
|
||||||
)
|
|
||||||
if stream_name == EventsStream.NAME:
|
|
||||||
run_in_background(self._notify_directory)
|
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
|
||||||
def _notify_directory(self):
|
|
||||||
try:
|
|
||||||
yield self.user_directory.notify_new_event()
|
|
||||||
except Exception:
|
|
||||||
logger.exception("Error notifiying user directory of state update")
|
|
||||||
|
|
||||||
|
|
||||||
def start(config_options):
|
|
||||||
try:
|
|
||||||
config = HomeServerConfig.load_config("Synapse user directory", config_options)
|
|
||||||
except ConfigError as e:
|
|
||||||
sys.stderr.write("\n" + str(e) + "\n")
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
assert config.worker_app == "synapse.app.user_dir"
|
|
||||||
|
|
||||||
events.USE_FROZEN_DICTS = config.use_frozen_dicts
|
|
||||||
|
|
||||||
if config.update_user_directory:
|
|
||||||
sys.stderr.write(
|
|
||||||
"\nThe update_user_directory must be disabled in the main synapse process"
|
|
||||||
"\nbefore they can be run in a separate worker."
|
|
||||||
"\nPlease add ``update_user_directory: false`` to the main config"
|
|
||||||
"\n"
|
|
||||||
)
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
# Force the pushers to start since they will be disabled in the main config
|
|
||||||
config.update_user_directory = True
|
|
||||||
|
|
||||||
ss = UserDirectoryServer(
|
|
||||||
config.server_name,
|
|
||||||
config=config,
|
|
||||||
version_string="Synapse/" + get_version_string(synapse),
|
|
||||||
)
|
|
||||||
|
|
||||||
setup_logging(ss, config, use_worker_options=True)
|
|
||||||
|
|
||||||
ss.setup()
|
|
||||||
reactor.addSystemEventTrigger(
|
|
||||||
"before", "startup", _base.start, ss, config.worker_listeners
|
|
||||||
)
|
|
||||||
|
|
||||||
_base.start_worker_reactor("synapse-user-dir", config)
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
with LoggingContext("main"):
|
with LoggingContext("main"):
|
||||||
|
@ -32,6 +32,7 @@ from synapse.storage.data_stores.main.state import StateGroupWorkerStore
|
|||||||
from synapse.storage.data_stores.main.stream import StreamWorkerStore
|
from synapse.storage.data_stores.main.stream import StreamWorkerStore
|
||||||
from synapse.storage.data_stores.main.user_erasure_store import UserErasureWorkerStore
|
from synapse.storage.data_stores.main.user_erasure_store import UserErasureWorkerStore
|
||||||
from synapse.storage.database import Database
|
from synapse.storage.database import Database
|
||||||
|
from synapse.util.caches.stream_change_cache import StreamChangeCache
|
||||||
|
|
||||||
from ._base import BaseSlavedStore
|
from ._base import BaseSlavedStore
|
||||||
from ._slaved_id_tracker import SlavedIdTracker
|
from ._slaved_id_tracker import SlavedIdTracker
|
||||||
@ -68,6 +69,21 @@ class SlavedEventStore(
|
|||||||
|
|
||||||
super(SlavedEventStore, self).__init__(database, db_conn, hs)
|
super(SlavedEventStore, self).__init__(database, db_conn, hs)
|
||||||
|
|
||||||
|
events_max = self._stream_id_gen.get_current_token()
|
||||||
|
curr_state_delta_prefill, min_curr_state_delta_id = self.db.get_cache_dict(
|
||||||
|
db_conn,
|
||||||
|
"current_state_delta_stream",
|
||||||
|
entity_column="room_id",
|
||||||
|
stream_column="stream_id",
|
||||||
|
max_value=events_max, # As we share the stream id with events token
|
||||||
|
limit=1000,
|
||||||
|
)
|
||||||
|
self._curr_state_delta_stream_cache = StreamChangeCache(
|
||||||
|
"_curr_state_delta_stream_cache",
|
||||||
|
min_curr_state_delta_id,
|
||||||
|
prefilled_cache=curr_state_delta_prefill,
|
||||||
|
)
|
||||||
|
|
||||||
# Cached functions can't be accessed through a class instance so we need
|
# Cached functions can't be accessed through a class instance so we need
|
||||||
# to reach inside the __dict__ to extract them.
|
# to reach inside the __dict__ to extract them.
|
||||||
|
|
||||||
@ -120,6 +136,10 @@ class SlavedEventStore(
|
|||||||
backfilled=False,
|
backfilled=False,
|
||||||
)
|
)
|
||||||
elif row.type == EventsStreamCurrentStateRow.TypeId:
|
elif row.type == EventsStreamCurrentStateRow.TypeId:
|
||||||
|
self._curr_state_delta_stream_cache.entity_has_changed(
|
||||||
|
row.data.room_id, token
|
||||||
|
)
|
||||||
|
|
||||||
if data.type == EventTypes.Member:
|
if data.type == EventTypes.Member:
|
||||||
self.get_rooms_for_user_with_stream_ordering.invalidate(
|
self.get_rooms_for_user_with_stream_ordering.invalidate(
|
||||||
(data.state_key,)
|
(data.state_key,)
|
||||||
|
@ -197,6 +197,84 @@ class PusherWorkerStore(SQLBaseStore):
|
|||||||
|
|
||||||
return result
|
return result
|
||||||
|
|
||||||
|
@defer.inlineCallbacks
|
||||||
|
def update_pusher_last_stream_ordering(
|
||||||
|
self, app_id, pushkey, user_id, last_stream_ordering
|
||||||
|
):
|
||||||
|
yield self.db.simple_update_one(
|
||||||
|
"pushers",
|
||||||
|
{"app_id": app_id, "pushkey": pushkey, "user_name": user_id},
|
||||||
|
{"last_stream_ordering": last_stream_ordering},
|
||||||
|
desc="update_pusher_last_stream_ordering",
|
||||||
|
)
|
||||||
|
|
||||||
|
@defer.inlineCallbacks
|
||||||
|
def update_pusher_last_stream_ordering_and_success(
|
||||||
|
self, app_id, pushkey, user_id, last_stream_ordering, last_success
|
||||||
|
):
|
||||||
|
"""Update the last stream ordering position we've processed up to for
|
||||||
|
the given pusher.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
app_id (str)
|
||||||
|
pushkey (str)
|
||||||
|
last_stream_ordering (int)
|
||||||
|
last_success (int)
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Deferred[bool]: True if the pusher still exists; False if it has been deleted.
|
||||||
|
"""
|
||||||
|
updated = yield self.db.simple_update(
|
||||||
|
table="pushers",
|
||||||
|
keyvalues={"app_id": app_id, "pushkey": pushkey, "user_name": user_id},
|
||||||
|
updatevalues={
|
||||||
|
"last_stream_ordering": last_stream_ordering,
|
||||||
|
"last_success": last_success,
|
||||||
|
},
|
||||||
|
desc="update_pusher_last_stream_ordering_and_success",
|
||||||
|
)
|
||||||
|
|
||||||
|
return bool(updated)
|
||||||
|
|
||||||
|
@defer.inlineCallbacks
|
||||||
|
def update_pusher_failing_since(self, app_id, pushkey, user_id, failing_since):
|
||||||
|
yield self.db.simple_update(
|
||||||
|
table="pushers",
|
||||||
|
keyvalues={"app_id": app_id, "pushkey": pushkey, "user_name": user_id},
|
||||||
|
updatevalues={"failing_since": failing_since},
|
||||||
|
desc="update_pusher_failing_since",
|
||||||
|
)
|
||||||
|
|
||||||
|
@defer.inlineCallbacks
|
||||||
|
def get_throttle_params_by_room(self, pusher_id):
|
||||||
|
res = yield self.db.simple_select_list(
|
||||||
|
"pusher_throttle",
|
||||||
|
{"pusher": pusher_id},
|
||||||
|
["room_id", "last_sent_ts", "throttle_ms"],
|
||||||
|
desc="get_throttle_params_by_room",
|
||||||
|
)
|
||||||
|
|
||||||
|
params_by_room = {}
|
||||||
|
for row in res:
|
||||||
|
params_by_room[row["room_id"]] = {
|
||||||
|
"last_sent_ts": row["last_sent_ts"],
|
||||||
|
"throttle_ms": row["throttle_ms"],
|
||||||
|
}
|
||||||
|
|
||||||
|
return params_by_room
|
||||||
|
|
||||||
|
@defer.inlineCallbacks
|
||||||
|
def set_throttle_params(self, pusher_id, room_id, params):
|
||||||
|
# no need to lock because `pusher_throttle` has a primary key on
|
||||||
|
# (pusher, room_id) so simple_upsert will retry
|
||||||
|
yield self.db.simple_upsert(
|
||||||
|
"pusher_throttle",
|
||||||
|
{"pusher": pusher_id, "room_id": room_id},
|
||||||
|
params,
|
||||||
|
desc="set_throttle_params",
|
||||||
|
lock=False,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
class PusherStore(PusherWorkerStore):
|
class PusherStore(PusherWorkerStore):
|
||||||
def get_pushers_stream_token(self):
|
def get_pushers_stream_token(self):
|
||||||
@ -282,81 +360,3 @@ class PusherStore(PusherWorkerStore):
|
|||||||
|
|
||||||
with self._pushers_id_gen.get_next() as stream_id:
|
with self._pushers_id_gen.get_next() as stream_id:
|
||||||
yield self.db.runInteraction("delete_pusher", delete_pusher_txn, stream_id)
|
yield self.db.runInteraction("delete_pusher", delete_pusher_txn, stream_id)
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
|
||||||
def update_pusher_last_stream_ordering(
|
|
||||||
self, app_id, pushkey, user_id, last_stream_ordering
|
|
||||||
):
|
|
||||||
yield self.db.simple_update_one(
|
|
||||||
"pushers",
|
|
||||||
{"app_id": app_id, "pushkey": pushkey, "user_name": user_id},
|
|
||||||
{"last_stream_ordering": last_stream_ordering},
|
|
||||||
desc="update_pusher_last_stream_ordering",
|
|
||||||
)
|
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
|
||||||
def update_pusher_last_stream_ordering_and_success(
|
|
||||||
self, app_id, pushkey, user_id, last_stream_ordering, last_success
|
|
||||||
):
|
|
||||||
"""Update the last stream ordering position we've processed up to for
|
|
||||||
the given pusher.
|
|
||||||
|
|
||||||
Args:
|
|
||||||
app_id (str)
|
|
||||||
pushkey (str)
|
|
||||||
last_stream_ordering (int)
|
|
||||||
last_success (int)
|
|
||||||
|
|
||||||
Returns:
|
|
||||||
Deferred[bool]: True if the pusher still exists; False if it has been deleted.
|
|
||||||
"""
|
|
||||||
updated = yield self.db.simple_update(
|
|
||||||
table="pushers",
|
|
||||||
keyvalues={"app_id": app_id, "pushkey": pushkey, "user_name": user_id},
|
|
||||||
updatevalues={
|
|
||||||
"last_stream_ordering": last_stream_ordering,
|
|
||||||
"last_success": last_success,
|
|
||||||
},
|
|
||||||
desc="update_pusher_last_stream_ordering_and_success",
|
|
||||||
)
|
|
||||||
|
|
||||||
return bool(updated)
|
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
|
||||||
def update_pusher_failing_since(self, app_id, pushkey, user_id, failing_since):
|
|
||||||
yield self.db.simple_update(
|
|
||||||
table="pushers",
|
|
||||||
keyvalues={"app_id": app_id, "pushkey": pushkey, "user_name": user_id},
|
|
||||||
updatevalues={"failing_since": failing_since},
|
|
||||||
desc="update_pusher_failing_since",
|
|
||||||
)
|
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
|
||||||
def get_throttle_params_by_room(self, pusher_id):
|
|
||||||
res = yield self.db.simple_select_list(
|
|
||||||
"pusher_throttle",
|
|
||||||
{"pusher": pusher_id},
|
|
||||||
["room_id", "last_sent_ts", "throttle_ms"],
|
|
||||||
desc="get_throttle_params_by_room",
|
|
||||||
)
|
|
||||||
|
|
||||||
params_by_room = {}
|
|
||||||
for row in res:
|
|
||||||
params_by_room[row["room_id"]] = {
|
|
||||||
"last_sent_ts": row["last_sent_ts"],
|
|
||||||
"throttle_ms": row["throttle_ms"],
|
|
||||||
}
|
|
||||||
|
|
||||||
return params_by_room
|
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
|
||||||
def set_throttle_params(self, pusher_id, room_id, params):
|
|
||||||
# no need to lock because `pusher_throttle` has a primary key on
|
|
||||||
# (pusher, room_id) so simple_upsert will retry
|
|
||||||
yield self.db.simple_upsert(
|
|
||||||
"pusher_throttle",
|
|
||||||
{"pusher": pusher_id, "room_id": room_id},
|
|
||||||
params,
|
|
||||||
desc="set_throttle_params",
|
|
||||||
lock=False,
|
|
||||||
)
|
|
||||||
|
@ -13,7 +13,7 @@
|
|||||||
# See the License for the specific language governing permissions and
|
# See the License for the specific language governing permissions and
|
||||||
# limitations under the License.
|
# limitations under the License.
|
||||||
|
|
||||||
from synapse.app.frontend_proxy import FrontendProxyServer
|
from synapse.app.generic_worker import GenericWorkerServer
|
||||||
|
|
||||||
from tests.unittest import HomeserverTestCase
|
from tests.unittest import HomeserverTestCase
|
||||||
|
|
||||||
@ -22,7 +22,7 @@ class FrontendProxyTests(HomeserverTestCase):
|
|||||||
def make_homeserver(self, reactor, clock):
|
def make_homeserver(self, reactor, clock):
|
||||||
|
|
||||||
hs = self.setup_test_homeserver(
|
hs = self.setup_test_homeserver(
|
||||||
http_client=None, homeserverToUse=FrontendProxyServer
|
http_client=None, homeserverToUse=GenericWorkerServer
|
||||||
)
|
)
|
||||||
|
|
||||||
return hs
|
return hs
|
||||||
@ -46,9 +46,7 @@ class FrontendProxyTests(HomeserverTestCase):
|
|||||||
# Grab the resource from the site that was told to listen
|
# Grab the resource from the site that was told to listen
|
||||||
self.assertEqual(len(self.reactor.tcpServers), 1)
|
self.assertEqual(len(self.reactor.tcpServers), 1)
|
||||||
site = self.reactor.tcpServers[0][1]
|
site = self.reactor.tcpServers[0][1]
|
||||||
self.resource = (
|
self.resource = site.resource.children[b"_matrix"].children[b"client"]
|
||||||
site.resource.children[b"_matrix"].children[b"client"].children[b"r0"]
|
|
||||||
)
|
|
||||||
|
|
||||||
request, channel = self.make_request("PUT", "presence/a/status")
|
request, channel = self.make_request("PUT", "presence/a/status")
|
||||||
self.render(request)
|
self.render(request)
|
||||||
@ -76,9 +74,7 @@ class FrontendProxyTests(HomeserverTestCase):
|
|||||||
# Grab the resource from the site that was told to listen
|
# Grab the resource from the site that was told to listen
|
||||||
self.assertEqual(len(self.reactor.tcpServers), 1)
|
self.assertEqual(len(self.reactor.tcpServers), 1)
|
||||||
site = self.reactor.tcpServers[0][1]
|
site = self.reactor.tcpServers[0][1]
|
||||||
self.resource = (
|
self.resource = site.resource.children[b"_matrix"].children[b"client"]
|
||||||
site.resource.children[b"_matrix"].children[b"client"].children[b"r0"]
|
|
||||||
)
|
|
||||||
|
|
||||||
request, channel = self.make_request("PUT", "presence/a/status")
|
request, channel = self.make_request("PUT", "presence/a/status")
|
||||||
self.render(request)
|
self.render(request)
|
||||||
|
@ -16,7 +16,7 @@ from mock import Mock, patch
|
|||||||
|
|
||||||
from parameterized import parameterized
|
from parameterized import parameterized
|
||||||
|
|
||||||
from synapse.app.federation_reader import FederationReaderServer
|
from synapse.app.generic_worker import GenericWorkerServer
|
||||||
from synapse.app.homeserver import SynapseHomeServer
|
from synapse.app.homeserver import SynapseHomeServer
|
||||||
|
|
||||||
from tests.unittest import HomeserverTestCase
|
from tests.unittest import HomeserverTestCase
|
||||||
@ -25,7 +25,7 @@ from tests.unittest import HomeserverTestCase
|
|||||||
class FederationReaderOpenIDListenerTests(HomeserverTestCase):
|
class FederationReaderOpenIDListenerTests(HomeserverTestCase):
|
||||||
def make_homeserver(self, reactor, clock):
|
def make_homeserver(self, reactor, clock):
|
||||||
hs = self.setup_test_homeserver(
|
hs = self.setup_test_homeserver(
|
||||||
http_client=None, homeserverToUse=FederationReaderServer
|
http_client=None, homeserverToUse=GenericWorkerServer
|
||||||
)
|
)
|
||||||
return hs
|
return hs
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user