Fix AssignAppUser migration OOM

changelog.d/assign-app-user-oom.fix

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

priv/repo/migrations/20240904142434_assign_app_user.exs

@@ -1,20 +1,24 @@
 defmodule Pleroma.Repo.Migrations.AssignAppUser do
   use Ecto.Migration
 
+  import Ecto.Query
+
   alias Pleroma.Repo
   alias Pleroma.Web.OAuth.App
   alias Pleroma.Web.OAuth.Token
 
   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)
     end)
+    |> Stream.run()
   end
 
   def down, do: :ok