Skip to content
This repository has been archived by the owner on Apr 26, 2024. It is now read-only.

Commit

Permalink
Add/update some docstrings.
Browse files Browse the repository at this point in the history
  • Loading branch information
clokep committed Apr 28, 2022
1 parent 629aa51 commit 5145bee
Show file tree
Hide file tree
Showing 2 changed files with 44 additions and 1 deletion.
1 change: 1 addition & 0 deletions changelog.d/12581.misc
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
Improve docstrings for the receipts store.
44 changes: 43 additions & 1 deletion synapse/storage/databases/main/receipts.py
Original file line number Diff line number Diff line change
Expand Up @@ -126,6 +126,16 @@ async def get_users_with_read_receipts_in_room(self, room_id: str) -> Set[str]:
async def get_receipts_for_room(
self, room_id: str, receipt_type: str
) -> List[Dict[str, Any]]:
"""
Fetch the event IDs for the latest receipt for all users in a room with the given receipt type.
Args:
room_id: The room ID to fetch the receipt for.
receipt_type: The receipt type to fetch.
Returns:
A list of dictionaries of the user ID and event ID of the latest receipt for each user.
"""
return await self.db_pool.simple_select_list(
table="receipts_linearized",
keyvalues={"room_id": room_id, "receipt_type": receipt_type},
Expand All @@ -137,6 +147,17 @@ async def get_receipts_for_room(
async def get_last_receipt_event_id_for_user(
self, user_id: str, room_id: str, receipt_type: str
) -> Optional[str]:
"""
Fetch the event ID for the latest receipt in a room with the given receipt type.
Args:
user_id: The user to fetch receipts for.
room_id: The room ID to fetch the receipt for.
receipt_type: The receipt type to fetch.
Returns:
The event ID of the latest receipt, if one exists.
"""
return await self.db_pool.simple_select_one_onecol(
table="receipts_linearized",
keyvalues={
Expand All @@ -153,6 +174,16 @@ async def get_last_receipt_event_id_for_user(
async def get_receipts_for_user(
self, user_id: str, receipt_type: str
) -> Dict[str, str]:
"""
Fetch the event IDs for the latest receipt in all rooms with the given receipt type.
Args:
user_id: The user to fetch receipts for.
receipt_type: The receipt types to fetch.
Returns:
A map of room ID to the event ID of the latest receipt for that room.
"""
rows = await self.db_pool.simple_select_list(
table="receipts_linearized",
keyvalues={"user_id": user_id, "receipt_type": receipt_type},
Expand All @@ -165,6 +196,17 @@ async def get_receipts_for_user(
async def get_receipts_for_user_with_orderings(
self, user_id: str, receipt_type: str
) -> JsonDict:
"""
Fetch receipts in all rooms for a user.
Args:
user_id: The user to fetch receipts for.
receipt_type: The receipt type to fetch.
Returns:
A map of room ID to the latest receipt information.
"""

def f(txn: LoggingTransaction) -> List[Tuple[str, str, int, int]]:
sql = (
"SELECT rl.room_id, rl.event_id,"
Expand Down Expand Up @@ -541,7 +583,7 @@ def insert_linearized_receipt_txn(
data: JsonDict,
stream_id: int,
) -> Optional[int]:
"""Inserts a read-receipt into the database if it's newer than the current RR
"""Inserts a receipt into the database if it's newer than the current one.
Returns:
None if the RR is older than the current RR
Expand Down

0 comments on commit 5145bee

Please sign in to comment.