Graham McIntire
65e3159a85
Mix tasks that call app.start were also booting Oban's cron scheduler, causing PropagationGridWorker and other cron jobs to fire during backfills. Add Oban.pause_all_queues(Oban) immediately after app.start in every mix task that only needs Repo access.
70 lines
2.5 KiB
Elixir
70 lines
2.5 KiB
Elixir
defmodule Mix.Tasks.HrrrNativeBackfill do
|
|
@shortdoc "Enqueue HrrrNativeGridWorker jobs for the top-N hours by contact count"
|
|
@moduledoc """
|
|
Backfills the `hrrr_native_profiles` table by enqueueing one
|
|
`HrrrNativeGridWorker` job per distinct `(year, month, day, hour)`
|
|
where we have contacts, prioritized by contact count so the most
|
|
data-dense hours land first.
|
|
|
|
Native HRRR files are ~566 MB each, so think about the total
|
|
bandwidth before running with a large limit. Typical plan:
|
|
|
|
mix hrrr_native_backfill --limit 50 # one-time smoke backfill
|
|
mix hrrr_native_backfill --limit 500 # full Phase 2 backfill (~280 GB)
|
|
|
|
Jobs are deduplicated by Oban's unique constraint on
|
|
`{year, month, day, hour}` so running the task twice is safe.
|
|
"""
|
|
use Mix.Task
|
|
|
|
import Ecto.Query
|
|
|
|
alias Microwaveprop.Radio.Contact
|
|
alias Microwaveprop.Repo
|
|
alias Microwaveprop.Workers.HrrrNativeGridWorker
|
|
|
|
@impl Mix.Task
|
|
def run(argv) do
|
|
Mix.Task.run("app.start")
|
|
Oban.pause_all_queues(Oban)
|
|
|
|
{opts, _, _} = OptionParser.parse(argv, switches: [limit: :integer])
|
|
limit = Keyword.get(opts, :limit, 50)
|
|
|
|
hours = top_hours_by_contact_count(limit)
|
|
Mix.shell().info("Enqueueing #{length(hours)} HrrrNativeGridWorker jobs")
|
|
|
|
Enum.each(hours, fn %{year: y, month: m, day: d, hour: h, contacts: n} ->
|
|
args = %{"year" => y, "month" => m, "day" => d, "hour" => h}
|
|
|
|
case Oban.insert(HrrrNativeGridWorker.new(args)) do
|
|
{:ok, _job} -> Mix.shell().info(" #{y}-#{pad(m)}-#{pad(d)} #{pad(h)}Z (#{n} contacts)")
|
|
{:error, reason} -> Mix.shell().error(" #{y}-#{pad(m)}-#{pad(d)} #{pad(h)}Z failed: #{inspect(reason)}")
|
|
end
|
|
end)
|
|
end
|
|
|
|
defp pad(n), do: n |> Integer.to_string() |> String.pad_leading(2, "0")
|
|
|
|
defp top_hours_by_contact_count(limit) do
|
|
from(c in Contact,
|
|
where: not is_nil(c.pos1),
|
|
select: %{
|
|
year: fragment("EXTRACT(YEAR FROM ?)::int", c.qso_timestamp),
|
|
month: fragment("EXTRACT(MONTH FROM ?)::int", c.qso_timestamp),
|
|
day: fragment("EXTRACT(DAY FROM ?)::int", c.qso_timestamp),
|
|
hour: fragment("EXTRACT(HOUR FROM ?)::int", c.qso_timestamp),
|
|
contacts: count(c.id)
|
|
},
|
|
group_by: [
|
|
fragment("EXTRACT(YEAR FROM ?)::int", c.qso_timestamp),
|
|
fragment("EXTRACT(MONTH FROM ?)::int", c.qso_timestamp),
|
|
fragment("EXTRACT(DAY FROM ?)::int", c.qso_timestamp),
|
|
fragment("EXTRACT(HOUR FROM ?)::int", c.qso_timestamp)
|
|
],
|
|
order_by: [desc: count(c.id)],
|
|
limit: ^limit
|
|
)
|
|
|> Repo.all()
|
|
end
|
|
end
|