Use the account handler in additional places. (#9166)

This commit is contained in:
Patrick Cloke 2021-01-20 05:44:52 -05:00 committed by GitHub
parent 47d48a5853
commit fa842a9866
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 5 additions and 2 deletions

1
changelog.d/9166.feature Normal file
View File

@ -0,0 +1 @@
Add experimental support for moving off receipts and account data persistence off master.

View File

@ -42,6 +42,7 @@ class ResourceLimitsServerNotices:
self._auth = hs.get_auth() self._auth = hs.get_auth()
self._config = hs.config self._config = hs.config
self._resouce_limited = False self._resouce_limited = False
self._account_data_handler = hs.get_account_data_handler()
self._message_handler = hs.get_message_handler() self._message_handler = hs.get_message_handler()
self._state = hs.get_state_handler() self._state = hs.get_state_handler()
@ -177,7 +178,7 @@ class ResourceLimitsServerNotices:
# tag already present, nothing to do here # tag already present, nothing to do here
need_to_set_tag = False need_to_set_tag = False
if need_to_set_tag: if need_to_set_tag:
max_id = await self._store.add_tag_to_room( max_id = await self._account_data_handler.add_tag_to_room(
user_id, room_id, SERVER_NOTICE_ROOM_TAG, {} user_id, room_id, SERVER_NOTICE_ROOM_TAG, {}
) )
self._notifier.on_new_event("account_data_key", max_id, users=[user_id]) self._notifier.on_new_event("account_data_key", max_id, users=[user_id])

View File

@ -35,6 +35,7 @@ class ServerNoticesManager:
self._store = hs.get_datastore() self._store = hs.get_datastore()
self._config = hs.config self._config = hs.config
self._account_data_handler = hs.get_account_data_handler()
self._room_creation_handler = hs.get_room_creation_handler() self._room_creation_handler = hs.get_room_creation_handler()
self._room_member_handler = hs.get_room_member_handler() self._room_member_handler = hs.get_room_member_handler()
self._event_creation_handler = hs.get_event_creation_handler() self._event_creation_handler = hs.get_event_creation_handler()
@ -163,7 +164,7 @@ class ServerNoticesManager:
) )
room_id = info["room_id"] room_id = info["room_id"]
max_id = await self._store.add_tag_to_room( max_id = await self._account_data_handler.add_tag_to_room(
user_id, room_id, SERVER_NOTICE_ROOM_TAG, {} user_id, room_id, SERVER_NOTICE_ROOM_TAG, {}
) )
self._notifier.on_new_event("account_data_key", max_id, users=[user_id]) self._notifier.on_new_event("account_data_key", max_id, users=[user_id])