git.squeep.com Git - akkoma/blob - priv/repo/migrations/20200825061316_move_activity_expirations_to_oban.exs

   1 defmodule Pleroma.Repo.Migrations.MoveActivityExpirationsToOban do
   2   use Ecto.Migration
   3
   4   import Ecto.Query, only: [from: 2]
   5
   6   def change do
   7     Pleroma.Config.Oban.warn()
   8
   9     Supervisor.start_link([{Oban, Pleroma.Config.get(Oban)}],
  10       strategy: :one_for_one,
  11       name: Pleroma.Supervisor
  12     )
  13
  14     from(e in "activity_expirations",
  15       select: %{id: e.id, activity_id: e.activity_id, scheduled_at: e.scheduled_at}
  16     )
  17     |> Pleroma.Repo.stream()
  18     |> Stream.each(fn expiration ->
  19       with {:ok, expires_at} <- DateTime.from_naive(expiration.scheduled_at, "Etc/UTC") do
  20         Pleroma.Workers.PurgeExpiredActivity.enqueue(%{
  21           activity_id: FlakeId.to_string(expiration.activity_id),
  22           expires_at: expires_at
  23         })
  24       end
  25     end)
  26     |> Stream.run()
  27   end
  28 end