From cda01285f4f36ffaac0034d6f0a5da64b4a26a58 Mon Sep 17 00:00:00 2001 From: Roger Braun Date: Tue, 12 Sep 2017 09:11:36 +0200 Subject: [PATCH] Add pagination to notifications. --- lib/pleroma/notification.ex | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/lib/pleroma/notification.ex b/lib/pleroma/notification.ex index 4a9e835bf..35f817d1d 100644 --- a/lib/pleroma/notification.ex +++ b/lib/pleroma/notification.ex @@ -11,12 +11,28 @@ defmodule Pleroma.Notification do timestamps() 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 query = from n in Notification, where: n.user_id == ^user.id, order_by: [desc: n.id], preload: [:activity], limit: 20 + + query = query + |> restrict_since(opts) + |> restrict_max(opts) + Repo.all(query) end