forked from AkkomaGang/akkoma
move sql (update_markers) from migrate to mix task
This commit is contained in:
parent
d719078699
commit
1b82eb6d41
3 changed files with 54 additions and 46 deletions
36
lib/mix/tasks/pleroma/marker.ex
Normal file
36
lib/mix/tasks/pleroma/marker.ex
Normal 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
|
|
@ -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
|
18
priv/repo/migrations/20191030202008_add_unread_to_marker.exs
Normal file
18
priv/repo/migrations/20191030202008_add_unread_to_marker.exs
Normal 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
|
Loading…
Reference in a new issue