forked from AkkomaGang/akkoma
Add a migration to remove embeded objects
This commit is contained in:
parent
83589ca6a5
commit
a11ca87f40
2 changed files with 10 additions and 57 deletions
|
@ -1,57 +0,0 @@
|
|||
defmodule Mix.Tasks.CompactDatabase do
|
||||
@moduledoc """
|
||||
Compact the database by flattening the object graph.
|
||||
"""
|
||||
|
||||
require Logger
|
||||
|
||||
use Mix.Task
|
||||
import Ecto.Query
|
||||
alias Pleroma.Activity
|
||||
alias Pleroma.Repo
|
||||
|
||||
defp maybe_compact(%Activity{data: %{"object" => %{"id" => object_id}}} = activity) do
|
||||
data =
|
||||
activity.data
|
||||
|> Map.put("object", object_id)
|
||||
|
||||
{:ok, activity} =
|
||||
Activity.change(activity, %{data: data})
|
||||
|> Repo.update()
|
||||
|
||||
{:ok, activity}
|
||||
end
|
||||
|
||||
defp maybe_compact(%Activity{} = activity), do: {:ok, activity}
|
||||
|
||||
defp activity_query(min_id, max_id) do
|
||||
from(
|
||||
a in Activity,
|
||||
where: fragment("?->>'type' = 'Create'", a.data),
|
||||
where: a.id >= ^min_id,
|
||||
where: a.id < ^max_id
|
||||
)
|
||||
end
|
||||
|
||||
def run(_args) do
|
||||
Application.ensure_all_started(:pleroma)
|
||||
|
||||
max = Repo.aggregate(Activity, :max, :id)
|
||||
Logger.info("Considering #{max} activities")
|
||||
|
||||
chunks = 0..round(max / 100)
|
||||
|
||||
Enum.each(chunks, fn i ->
|
||||
min = i * 100
|
||||
max = min + 100
|
||||
|
||||
activity_query(min, max)
|
||||
|> Repo.all()
|
||||
|> Enum.each(&maybe_compact/1)
|
||||
|
||||
IO.write(".")
|
||||
end)
|
||||
|
||||
Logger.info("Finished.")
|
||||
end
|
||||
end
|
|
@ -0,0 +1,10 @@
|
|||
defmodule Pleroma.Repo.Migrations.RemoveEmbededObjects do
|
||||
use Ecto.Migration
|
||||
|
||||
# TODO: bench on a real DB and add clippy if it takes too long
|
||||
def change do
|
||||
execute """
|
||||
update activities set data = jsonb_set(data, '{object}'::text[], data->'object'->'id') where data->>'type' = 'Create' and data->'object'->>'id' is not null;
|
||||
"""
|
||||
end
|
||||
end
|
Loading…
Reference in a new issue