diff --git a/synapse/notifier.py b/synapse/notifier.py
index 01a50b9d62..ba23257f54 100644
--- a/synapse/notifier.py
+++ b/synapse/notifier.py
@@ -46,6 +46,7 @@ from synapse.types import (
JsonDict,
PersistedEventPosition,
RoomStreamToken,
+ StreamKeyType,
StreamToken,
UserID,
)
@@ -370,7 +371,7 @@ class Notifier:
if users or rooms:
self.on_new_event(
- "room_key",
+ StreamKeyType.ROOM,
max_room_stream_token,
users=users,
rooms=rooms,
@@ -440,7 +441,7 @@ class Notifier:
for room in rooms:
user_streams |= self.room_to_user_streams.get(room, set())
- if stream_key == "to_device_key":
+ if stream_key == StreamKeyType.TO_DEVICE:
issue9533_logger.debug(
"to-device messages stream id %s, awaking streams for %s",
new_token,
|