task_supervisor.ex

defmodule MyApp.TaskSupervisor do
  use DynamicSupervisor

  alias MyApp.Management
  alias MyApp.WorkerStateStorage

  def start_link(arg) do
    DynamicSupervisor.start_link(__MODULE__, arg, name: __MODULE__)
  end

  def init(arg) do
    DynamicSupervisor.init(arg)
  end

  def start_task_worker(task_id) do
    spec = {MyApp.TaskWorker, %{task_id: task_id}}
    DynamicSupervisor.start_child(__MODULE__, spec)
  end

  def start_all_task_workers() do
    Enum.each(Management.list_all_tasks(), fn %{id: task_id} ->
      start_task_worker(task_id)
    end)
  end

  def terminate_task_worker(pid, task_id) do
    WorkerStateStorage.delete(task_id)
    Supervisor.terminate_child(__MODULE__, pid)
  end
end