Source file `parallel_parany.ml`

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
open Configuration
open Command_input

let process_interactive ~f paths number_of_workers =
  let reduce (acc, c) (path, result) =
    match result with
    | Some rewritten_source, c' ->
      Interactive.{path; rewritten_source}::acc, c+c'
    | None, c' ->
      acc, c+c'
  in
  let init = ([], 0) in
  let map path = path, f ~input:(Path path) ~path:(Some path) in
  Parany.Parmap.parfold ~csize:16 number_of_workers map reduce init paths

let process ~f number_of_workers _bound_count sources =
  match sources with
  | `Paths paths ->
    Parany.Parmap.parfold ~csize:16 number_of_workers (fun path -> f ~input:(Path path) ~output_path:(Some path)) (+) 0 paths
  | `Zip _ -> failwith "Not supported"