parent
fd8099fb68
commit
57995fa8cf
|
@ -1,36 +0,0 @@
|
||||||
defmodule Mix.Tasks.Pleroma.Marker do
|
|
||||||
use Mix.Task
|
|
||||||
import Mix.Pleroma
|
|
||||||
import Ecto.Query
|
|
||||||
|
|
||||||
alias Pleroma.Notification
|
|
||||||
alias Pleroma.Repo
|
|
||||||
|
|
||||||
def run(["update_markers"]) do
|
|
||||||
start_pleroma()
|
|
||||||
|
|
||||||
from(q in Notification,
|
|
||||||
select: %{
|
|
||||||
timeline: "notifications",
|
|
||||||
user_id: q.user_id,
|
|
||||||
unread_count: fragment("SUM( CASE WHEN seen = false THEN 1 ELSE 0 END )"),
|
|
||||||
last_read_id:
|
|
||||||
type(fragment("MAX( CASE WHEN seen = true THEN id ELSE null END )"), :string)
|
|
||||||
},
|
|
||||||
group_by: [q.user_id]
|
|
||||||
)
|
|
||||||
|> Repo.all()
|
|
||||||
|> Enum.each(fn attrs ->
|
|
||||||
Pleroma.Marker
|
|
||||||
|> struct(attrs)
|
|
||||||
|> Ecto.Changeset.change()
|
|
||||||
|> Pleroma.Repo.insert(
|
|
||||||
returning: true,
|
|
||||||
on_conflict: {:replace, [:last_read_id, :unread_count]},
|
|
||||||
conflict_target: [:user_id, :timeline]
|
|
||||||
)
|
|
||||||
end)
|
|
||||||
|
|
||||||
shell_info("Done")
|
|
||||||
end
|
|
||||||
end
|
|
|
@ -2,12 +2,15 @@ defmodule Pleroma.Repo.Migrations.AddUnreadToMarker do
|
||||||
use Ecto.Migration
|
use Ecto.Migration
|
||||||
import Ecto.Query
|
import Ecto.Query
|
||||||
alias Pleroma.Repo
|
alias Pleroma.Repo
|
||||||
alias Pleroma.Notification
|
|
||||||
|
|
||||||
def up do
|
def up do
|
||||||
alter table(:markers) do
|
alter table(:markers) do
|
||||||
add_if_not_exists(:unread_count, :integer, default: 0)
|
add_if_not_exists(:unread_count, :integer, default: 0)
|
||||||
end
|
end
|
||||||
|
|
||||||
|
flush()
|
||||||
|
|
||||||
|
update_markers()
|
||||||
end
|
end
|
||||||
|
|
||||||
def down do
|
def down do
|
||||||
|
@ -15,4 +18,31 @@ def down do
|
||||||
remove_if_exists(:unread_count, :integer)
|
remove_if_exists(:unread_count, :integer)
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
|
def update_markers do
|
||||||
|
now = NaiveDateTime.utc_now()
|
||||||
|
|
||||||
|
markers_attrs =
|
||||||
|
from(q in "notifications",
|
||||||
|
select: %{
|
||||||
|
timeline: "notifications",
|
||||||
|
user_id: q.user_id,
|
||||||
|
unread_count: fragment("SUM( CASE WHEN seen = false THEN 1 ELSE 0 END )"),
|
||||||
|
last_read_id:
|
||||||
|
type(fragment("MAX( CASE WHEN seen = true THEN id ELSE null END )"), :string)
|
||||||
|
},
|
||||||
|
group_by: [q.user_id]
|
||||||
|
)
|
||||||
|
|> Repo.all()
|
||||||
|
|> Enum.map(fn attrs ->
|
||||||
|
attrs
|
||||||
|
|> Map.put_new(:inserted_at, now)
|
||||||
|
|> Map.put_new(:updated_at, now)
|
||||||
|
end)
|
||||||
|
|
||||||
|
Repo.insert_all("markers", markers_attrs,
|
||||||
|
on_conflict: {:replace, [:last_read_id, :unread_count]},
|
||||||
|
conflict_target: [:user_id, :timeline]
|
||||||
|
)
|
||||||
|
end
|
||||||
end
|
end
|
||||||
|
|
Loading…
Reference in New Issue