hj/pleroma
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Merge branch 'tusooa/assign-app-user-oom' into 'develop'

Browse source 
Fix AssignAppUser migration OOM

Closes #3358

See merge request pleroma/pleroma!4326
This commit is contained in:
feld 2025-06-07 19:39:57 +00:00
commit f38e9228ef
2 changed files with 12 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

1
changelog.d/assign-app-user-oom.fix Normal file
View file

@ -0,0 +1 @@
Fix AssignAppUser migration OOM

18
priv/repo/migrations/20240904142434_assign_app_user.exs
View file

@ -1,20 +1,24 @@
defmodule Pleroma.Repo.Migrations.AssignAppUser do defmodule Pleroma.Repo.Migrations.AssignAppUser do
use Ecto.Migration use Ecto.Migration
import Ecto.Query
alias Pleroma.Repo alias Pleroma.Repo
alias Pleroma.Web.OAuth.App alias Pleroma.Web.OAuth.App
alias Pleroma.Web.OAuth.Token alias Pleroma.Web.OAuth.Token
def up do def up do
Repo.all(Token) Token
|> Enum.group_by(fn x -> Map.get(x, :app_id) end) |> where([t], not is_nil(t.user_id))
|> Enum.each(fn {_app_id, tokens} -> |> group_by([t], t.app_id)
token = |> select([t], %{app_id: t.app_id, id: min(t.id)})
Enum.filter(tokens, fn x -> not is_nil(x.user_id) end) |> order_by(asc: :app_id)
|> List.first() |> Repo.stream()
|> Stream.each(fn %{id: id} ->
token = Token.Query.get_by_id(id) |> Repo.one()
App.maybe_update_owner(token) App.maybe_update_owner(token)
end) end)
|> Stream.run()
end end
def down, do: :ok def down, do: :ok