mirror of
https://mau.dev/maunium/synapse.git
synced 2024-08-24 03:05:48 +00:00
782e4e64df
The aim here is to get to a stage where we have a `PersistEventStore` that holds all the write methods used during event persistence, so that we can take that class out of the `DataStore` mixin and instansiate it separately. This will allow us to instansiate it on processes other than master, while also ensuring it is only available on processes that are configured to write to events stream. This is a bit of an architectural change, where we end up with multiple classes per data store (rather than one per data store we have now). We end up having: 1. Storage classes that provide high level APIs that can talk to multiple data stores. 2. Data store modules that consist of classes that must point at the same database instance. 3. Classes in a data store that can be instantiated on processes depending on config.
72 lines
2.3 KiB
Python
72 lines
2.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright 2014-2016 OpenMarket Ltd
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from unpaddedbase64 import encode_base64
|
|
|
|
from twisted.internet import defer
|
|
|
|
from synapse.storage._base import SQLBaseStore
|
|
from synapse.util.caches.descriptors import cached, cachedList
|
|
|
|
|
|
class SignatureWorkerStore(SQLBaseStore):
|
|
@cached()
|
|
def get_event_reference_hash(self, event_id):
|
|
# This is a dummy function to allow get_event_reference_hashes
|
|
# to use its cache
|
|
raise NotImplementedError()
|
|
|
|
@cachedList(
|
|
cached_method_name="get_event_reference_hash", list_name="event_ids", num_args=1
|
|
)
|
|
def get_event_reference_hashes(self, event_ids):
|
|
def f(txn):
|
|
return {
|
|
event_id: self._get_event_reference_hashes_txn(txn, event_id)
|
|
for event_id in event_ids
|
|
}
|
|
|
|
return self.db.runInteraction("get_event_reference_hashes", f)
|
|
|
|
@defer.inlineCallbacks
|
|
def add_event_hashes(self, event_ids):
|
|
hashes = yield self.get_event_reference_hashes(event_ids)
|
|
hashes = {
|
|
e_id: {k: encode_base64(v) for k, v in h.items() if k == "sha256"}
|
|
for e_id, h in hashes.items()
|
|
}
|
|
|
|
return list(hashes.items())
|
|
|
|
def _get_event_reference_hashes_txn(self, txn, event_id):
|
|
"""Get all the hashes for a given PDU.
|
|
Args:
|
|
txn (cursor):
|
|
event_id (str): Id for the Event.
|
|
Returns:
|
|
A dict[unicode, bytes] of algorithm -> hash.
|
|
"""
|
|
query = (
|
|
"SELECT algorithm, hash"
|
|
" FROM event_reference_hashes"
|
|
" WHERE event_id = ?"
|
|
)
|
|
txn.execute(query, (event_id,))
|
|
return {k: v for k, v in txn}
|
|
|
|
|
|
class SignatureStore(SignatureWorkerStore):
|
|
"""Persistence for event signatures and hashes"""
|