defmodule Pleroma.Repo.Migrations.MigrateOldBookmarks do
use Ecto.Migration
+ import Ecto.Query
alias Pleroma.Activity
alias Pleroma.Bookmark
alias Pleroma.User
alias Pleroma.Repo
def up do
- Repo.all(User)
- |> Enum.each(fn user ->
- Enum.each(user.old_bookmarks, fn id ->
- activity = Activity.get_create_by_object_ap_id(id)
- {:ok, _} = Bookmark.create(user.id, activity.id)
+ query =
+ from(u in User,
+ where: u.local == true,
+ where: fragment("array_length(bookmarks, 1)") > 0,
+ select: %{id: u.id, bookmarks: fragment("bookmarks")}
+ )
+
+ Repo.stream(query)
+ |> Enum.each(fn %{id: user_id, bookmarks: bookmarks} ->
+ Enum.each(bookmarks, fn ap_id ->
+ activity = Activity.get_create_by_object_ap_id(ap_id)
+ unless is_nil(activity), do: {:ok, _} = Bookmark.create(user_id, activity.id)
end)
end)
+
+ alter table(:users) do
+ remove(:bookmarks)
+ end
end
def down do
- execute("TRUNCATE TABLE bookmarks")
+ alter table(:users) do
+ add :bookmarks, {:array, :string}, null: false, default: []
+ end
end
end