open Types_
include Runner

let k_ls = Task_local_storage.Private_.Storage.k_storage

let run_async_ ~name ~ls f =
  let cur_ls = ref ls in
  TLS.set k_ls (Some cur_ls);
  let sp = Tracing_.enter_span name in
  try
    let x = f () in
    Tracing_.exit_span sp;
    TLS.set k_ls None;
    x
  with e ->
    let bt = Printexc.get_raw_backtrace () in
    Tracing_.exit_span sp;
    TLS.set k_ls None;
    Printexc.raise_with_backtrace e bt

let runner : t =
  Runner.For_runner_implementors.create
    ~size:(fun () -> 0)
    ~num_tasks:(fun () -> 0)
    ~shutdown:(fun ~wait:_ () -> ())
    ~run_async:run_async_ ()