Insert push actions in a single db query rather than one per user/profile_tag

This commit is contained in:
David Baker 2015-12-22 17:04:31 +00:00
parent 77f06856b6
commit 4c8f6a7e42
2 changed files with 24 additions and 17 deletions

View File

@ -43,7 +43,9 @@ class ActionGenerator:
actions_by_user = bulk_evaluator.action_for_event_by_user(event)
for uid,actions in actions_by_user.items():
self.store.set_actions_for_event(
event, uid, None, actions
)
yield self.store.set_actions_for_event_and_users(
event,
[
(uid, None, actions) for uid, actions in actions_by_user.items()
]
)

View File

@ -24,22 +24,27 @@ logger = logging.getLogger(__name__)
class EventActionsStore(SQLBaseStore):
@defer.inlineCallbacks
def set_actions_for_event(self, event, user_id, profile_tag, actions):
actionsJson = json.dumps(actions)
ret = yield self.runInteraction(
"_set_actions_for_event",
self._simple_upsert_txn,
EventActionsTable.table_name,
{
def set_actions_for_event_and_users(self, event, tuples):
"""
:param event: the event set actions for
:param tuples: list of tuples of (user_id, profile_tag, actions)
"""
values = []
for uid, profile_tag, actions in tuples:
values.append({
'room_id': event['room_id'],
'event_id': event['event_id'],
'user_id': user_id,
'profile_tag': profile_tag
},
{'actions': actionsJson}
'user_id': uid,
'profile_tag': profile_tag,
'actions': json.dumps(actions)
})
yield self.runInteraction(
"set_actions_for_event_and_users",
self._simple_insert_many_txn,
EventActionsTable.table_name,
values
)
defer.returnValue(ret)
@defer.inlineCallbacks
def get_unread_event_actions_by_room_for_user(