Implement `only=highlight` on `/notifications`

This commit is contained in:
David Baker 2016-09-08 13:43:35 +01:00
parent 61cd9af09b
commit 4ef222ab61
2 changed files with 10 additions and 2 deletions

View File

@ -45,11 +45,12 @@ class NotificationsServlet(RestServlet):
from_token = parse_string(request, "from", required=False)
limit = parse_integer(request, "limit", default=50)
only = parse_string(request, "only", required=False)
limit = min(limit, 500)
push_actions = yield self.store.get_push_actions_for_user(
user_id, from_token, limit
user_id, from_token, limit, only_highlight=(only == "highlight")
)
receipts_by_room = yield self.store.get_receipts_for_user_with_orderings(

View File

@ -338,7 +338,8 @@ class EventPushActionsStore(SQLBaseStore):
defer.returnValue(notifs[:limit])
@defer.inlineCallbacks
def get_push_actions_for_user(self, user_id, before=None, limit=50):
def get_push_actions_for_user(self, user_id, before=None, limit=50,
only_highlight=False):
def f(txn):
before_clause = ""
if before:
@ -346,6 +347,12 @@ class EventPushActionsStore(SQLBaseStore):
args = [user_id, before, limit]
else:
args = [user_id, limit]
if only_highlight:
if len(before_clause) > 0:
before_clause += " "
before_clause += "AND epa.highlight = 1"
sql = (
"SELECT epa.event_id, epa.room_id,"
" epa.stream_ordering, epa.topological_ordering,"