Refactor _set_push_actions_for_event_and_users_txn to use events_and_contexts

This commit is contained in:
Erik Johnston 2018-02-20 12:29:50 +00:00
parent 324c3e9399
commit ad0ccf15ea
2 changed files with 42 additions and 34 deletions

View File

@ -88,11 +88,13 @@ class EventPushActionsStore(SQLBaseStore):
self._rotate_notifs, 30 * 60 * 1000 self._rotate_notifs, 30 * 60 * 1000
) )
def _set_push_actions_for_event_and_users_txn(self, txn, event): def _set_push_actions_for_event_and_users_txn(self, txn, events_and_contexts):
""" """Handles moving push actions from staging table to main
event_push_actions table for all events in `events_and_contexts`.
Args: Args:
event: the event set actions for events_and_contexts (list[(EventBase, EventContext)]): events
tuples: list of tuples of (user_id, actions) we are persisting
""" """
sql = """ sql = """
@ -105,34 +107,39 @@ class EventPushActionsStore(SQLBaseStore):
WHERE event_id = ? WHERE event_id = ?
""" """
txn.execute(sql, ( if events_and_contexts:
event.room_id, event.internal_metadata.stream_ordering, txn.executemany(sql, (
event.depth, event.event_id, (
)) event.room_id, event.internal_metadata.stream_ordering,
event.depth, event.event_id,
)
for event, _ in events_and_contexts
))
user_ids = self._simple_select_onecol_txn( for event, _ in events_and_contexts:
txn, user_ids = self._simple_select_onecol_txn(
table="event_push_actions_staging", txn,
keyvalues={ table="event_push_actions_staging",
"event_id": event.event_id, keyvalues={
}, "event_id": event.event_id,
retcol="user_id", },
) retcol="user_id",
self._simple_delete_txn(
txn,
table="event_push_actions_staging",
keyvalues={
"event_id": event.event_id,
},
)
for uid in user_ids:
txn.call_after(
self.get_unread_event_push_actions_by_room_for_user.invalidate_many,
(event.room_id, uid,)
) )
for uid in user_ids:
txn.call_after(
self.get_unread_event_push_actions_by_room_for_user.invalidate_many,
(event.room_id, uid,)
)
txn.executemany(
"DELETE FROM event_push_actions_staging WHERE event_id = ?",
(
(event.event_id,)
for event, _ in events_and_contexts
)
)
@cachedInlineCallbacks(num_args=3, tree=True, max_entries=5000) @cachedInlineCallbacks(num_args=3, tree=True, max_entries=5000)
def get_unread_event_push_actions_by_room_for_user( def get_unread_event_push_actions_by_room_for_user(
self, room_id, user_id, last_read_event_id self, room_id, user_id, last_read_event_id

View File

@ -1162,16 +1162,17 @@ class EventsStore(SQLBaseStore):
backfilled (bool): True if the events were backfilled backfilled (bool): True if the events were backfilled
""" """
# Insert all the push actions into the event_push_actions table.
self._set_push_actions_for_event_and_users_txn(
txn,
events_and_contexts=events_and_contexts,
)
if not events_and_contexts: if not events_and_contexts:
# nothing to do here # nothing to do here
return return
for event, context in events_and_contexts: for event, context in events_and_contexts:
# Insert all the push actions into the event_push_actions table.
self._set_push_actions_for_event_and_users_txn(
txn, event,
)
if event.type == EventTypes.Redaction and event.redacts is not None: if event.type == EventTypes.Redaction and event.redacts is not None:
# Remove the entries in the event_push_actions table for the # Remove the entries in the event_push_actions table for the
# redacted event. # redacted event.