限制分叉进程的数量

时间:2016-02-14 00:31:23

标签: ruby process

我正在尝试处理一个数组,每个元素需要一个合理的时间来处理(≈30分钟)。为了加快我想要使用fork的过程,所以元素可以在不同的进程中并行处理。但是我的数组有大约1000个元素,我不想一次制作1000个分支。

我可以分批 n 进行fork次,然后等到所有这些都以Process.waitall结束,但有没有办法让我可以开始新进程很快,上一个结束了?

1 个答案:

答案 0 :(得分:6)

可以通过以下方式实现解决方案:

# assign maximum number of processes
counter = @max_processes

# when a child process ends increase the counter of available processes
Signal.trap("CLD")  { counter += 1 }    

your_array.each do |element|

  # wait for a child process to end if there are no more "available processes"
  Process.wait if counter <= 0

  # before launching a process decrease the "available processes" counter
  counter -= 1

  fork {
    # do something with an element
  }

end

更简单的选择是将数组拆分为子数组并并行运行:

your_array.each_slice(@max_processes) do |elements|

  elements.each do |element|
    fork {
      # do something with an element
    }
  end

  Process.waitall

end