首页 > 技术文章 > python多进程并行代码

codeDevotee 2019-09-30 04:07 原文

from multiprocessing import Process
import sys, os
import time


def timetask(string):
  while True:
    print(string)


def works(func, arg, worknum):
  proc_record = []
  for i in range(worknum):
    p = Process(target = func, args = (i,))
    p.start()
    proc_record.append(p)
  for p in proc_record:
    p.join()


if __name__ == '__main__':
  arg = 5
  procs = 4
  works(timetask, arg, procs)

 

效果图【未加锁,会发现数字1没出现,进程间资源抢夺导致】:

 

 

 

优化之后效果:

 

 

代码:

from multiprocessing import Process
import sys, os
import time
from threading import Lock


def timetask(string):
  mutex.acquire()
  while True:
    print(string)
    if mutex.locked():
      mutex.release()


mutex = Lock()

def works(func, arg, worknum):
  proc_record = []
  for i in range(worknum):
    p = Process(target = func, args = (i,))
    p.start()
    proc_record.append(p)
  for p in proc_record:
    p.join()


if __name__ == '__main__':
  arg = 5
  procs = 4
  works(timetask, arg, procs)

 

推荐阅读