fix migrate

update migrate
This commit is contained in:
Maksim Pechnikov 2019-11-02 21:19:01 +03:00
parent fd8099fb68
commit 57995fa8cf
2 changed files with 31 additions and 37 deletions

View File

@ -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

View File

@ -2,12 +2,15 @@ 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
@ -15,4 +18,31 @@ def down do
remove_if_exists(:unread_count, :integer)
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