squeegily/pleroma
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add pagination to notifications.

Browse Source
This commit is contained in:
Roger Braun 2017-09-12 09:11:36 +02:00
parent f1d9f2f6cd
commit cda01285f4
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
lib/pleroma/notification.ex
View File

@ -11,12 +11,28 @@ defmodule Pleroma.Notification do
timestamps() timestamps()
end end
# TODO: Make generic and unify (see activity_pub.ex)
defp restrict_max(query, %{"max_id" => max_id}) do
from activity in query, where: activity.id < ^max_id
end
defp restrict_max(query, _), do: query
defp restrict_since(query, %{"since_id" => since_id}) do
from activity in query, where: activity.id > ^since_id
end
defp restrict_since(query, _), do: query
def for_user(user, opts \\ %{}) do def for_user(user, opts \\ %{}) do
query = from n in Notification, query = from n in Notification,
where: n.user_id == ^user.id, where: n.user_id == ^user.id,
order_by: [desc: n.id], order_by: [desc: n.id],
preload: [:activity], preload: [:activity],
limit: 20 limit: 20
query = query
|> restrict_since(opts)
|> restrict_max(opts)
Repo.all(query) Repo.all(query)
end end