python多处理执行之间的睡眠

时间:2014-12-09 23:59:18

标签: python python-multiprocessing

我有一个python脚本,应该并行运行多个作业。我将最大进程设置为20但我需要脚本在发送作业之间休眠5秒。 所以这是我的示例代码:

#!/usr/bin/env python

import multiprocessing
import subprocess


def prcss(cmd):
    sbc = subprocess.call
    com = sbc(cmd, shell='True')
    return (com)


if __name__=='__main__':

    s = 'sleep 5'
    cmd= []
    for j in range(1,21):
        for i in range(10):
            sis = "nohup ~/mycodes/code > str(j)"+"/"+"out"+str(i)+".dat"
            cmd.append(sis)
            cmd.append(s)

    pool=multiprocessing.Pool(processes=20)
    pool.map(prcss,cmd)

虽然我在'sis'命令之间睡了5,但是当我运行我的脚本时,所有工作都会立即启动。我需要在'sis'命令之间休眠,因为每个作业的输出取决于计算机时钟。因此,如果我运行20个作业,它们都以相同的系统时钟开始,因此它们都将具有相同的输出。

知道如何让我的脚本在'sis'命令之间睡觉吗?

Abedin

1 个答案:

答案 0 :(得分:2)

看看docs for pool.map()。创建项目列表然后使用map将它们提交到池时,所有作业将一起提交到池中。由于您有20个工作流程,因此您的20个工作将同时启动(有效)。这包括您的sis命令和睡眠命令。甚至没有保证它们将以相同的顺序执行和完成,只是您将以相同的顺序收到结果。 apply_async()函数可能对您更好,因为您可以控制何时将作业提交到池中。

在我看来,你希望你的Python脚本在发出sis命令之前等待5秒钟,所以你没有理由需要在工作进程中执行sleep命令。尝试重构成这样的东西:

import multiprocessing
import subprocess
import time

def prcss(cmd):
  # renaming the subprocess call is silly - remove the rename
  com = subprocess.call(cmd, shell='True') 
  return (com)

if __name__=='__main__':

  pool = multiprocessing.Pool(processes=20)
  results_objects = []

  for j in range(1,21):
    for i in range(10):
      sis = 'nohup ~/mycodes/code >'+str(j)+'/'+'out'+str(i)+'.dat'

      # make an asynchronous that will execute our target function with the
      # sis command
      results_objects.append(pool.apply_async(prcss, args=(sis,))
      # don't forget the extra comma in the args - Process must receive a tuple

      # now we pause for five sections before submitting the next job
      time.sleep(5)

  # close the pool and wait for everything to finish
  pool.close()
  pool.join() 

  # retrieve all of the results
  result = [result.get() for result in results_objects]

另一个注意事项:由于语法高亮显示已应用,因此很容易看到您在sis字符串中缺少结束引号,也可能是'+'。请考虑使用string.format()

,而不是手动构建字符串
sis = 'nohup ~/mycodes/code > {}/out{}.dat'.format(j, i)

如果反斜杠用于分隔路径层次结构,则应使用os.path.join()

import os
sis = os.path.join('nohup ~/mycodes/code > {}'.format(j), 'out{}.dat'.format(i))

生成的第一个字符串(在任何一种情况下)都是:

  

nohup~ / mycodes / code> 1 / out0.dat