move sql (update_markers) from migrate to mix task

This commit is contained in:
Maksim Pechnikov 2019-10-30 23:22:38 +03:00
parent d719078699
commit 1b82eb6d41
3 changed files with 54 additions and 46 deletions

View File

@ -0,0 +1,36 @@
defmodule Mix.Tasks.Pleroma.Marker do
use Mix.Task
import Mix.Pleroma
import Ecto.Query
alias Pleroma.Repo
alias Pleroma.Notification
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

View File

@ -1,46 +0,0 @@
defmodule Pleroma.Repo.Migrations.AddUnreadToMarker do
use Ecto.Migration
import Ecto.Query
alias Pleroma.Repo
alias Pleroma.Notification
def up do
alter table(:markers) do
add_if_not_exists(:unread_count, :integer, default: 0)
end
flush()
update_markers()
end
def down do
alter table(:markers) do
remove_if_exists(:unread_count, :integer)
end
end
def update_markers do
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)
end
end

View File

@ -0,0 +1,18 @@
defmodule Pleroma.Repo.Migrations.AddUnreadToMarker do
use Ecto.Migration
import Ecto.Query
alias Pleroma.Repo
alias Pleroma.Notification
def up do
alter table(:markers) do
add_if_not_exists(:unread_count, :integer, default: 0)
end
end
def down do
alter table(:markers) do
remove_if_exists(:unread_count, :integer)
end
end
end