Add back the guard against the user directory stream position not existing. (#9428)

As the comment says, this guard was there for when the
initial user directory update has yet to happen.
This commit is contained in:
Patrick Cloke 2021-02-18 08:44:19 -05:00 committed by GitHub
parent 626afd7e89
commit 43f1c82457
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 12 additions and 1 deletions

View file

@ -707,7 +707,13 @@ class UserDirectoryStore(UserDirectoryBackgroundUpdateStore):
return {row["room_id"] for row in rows}
async def get_user_directory_stream_pos(self) -> int:
async def get_user_directory_stream_pos(self) -> Optional[int]:
"""
Get the stream ID of the user directory stream.
Returns:
The stream token or None if the initial background update hasn't happened yet.
"""
return await self.db_pool.simple_select_one_onecol(
table="user_directory_stream_pos",
keyvalues={},