Допустим, у меня есть функция:
def f():
while True:
x = generate_something()
if x == condition:
return x
if __name__ == '__main__':
p=Pool(4)
Я хочу запустить эту функцию в мультипроцессе, и когда один из процессов соответствует условию моей функции, я хочу, чтобы все остальные процессы остановились.
Вы можете использовать event и terminate в multiprocessing, так как вы хотите остановить все процессы после выполнения условия в одном из дочерних процессов.
Проверьте приведенный ниже рабочий пример, в котором я создаю два процесса, которые будут проверять, что значение переменной x равно 5 или нет.
Как только один из процессов устанавливает значение x в 5, событие устанавливается.
Event равно set или unset постоянно проверяется в коде main.
Код:
import random
import multiprocessing
import sys
import time
def generate_something():
return random.choice(range(10))
def f(event):
while True:
x = generate_something()
print "Value is ",x
if x == 5:
print "Got what I am searching for."
event.set()
time.sleep(2)
if __name__ == '__main__':
jobs = []
#Create Event
event = multiprocessing.Event()
#Create two processes
for i in range(2):
p = multiprocessing.Process(target=f,args=(event,))
p.start()
jobs.append(p)
#Check whether event is set or not
#When set close all child processes
while True:
if event.is_set():
print "Exiting all child processess.."
for i in jobs:
#Terminate each process
i.terminate()
#Terminating main process
sys.exit(1)
time.sleep(2)
Вывод:
"C:\Program Files (x86)\Python27\python.exe" C:/Users/punddin/PycharmProjects/demo/a.py
Value is 4
Value is 2
Value is 7
Value is 6
Value is 3
Value is 4
Value is 8
Value is 3
Value is 8
Value is 1
Value is 9
Value is 9
Value is 1
Value is 6
Value is 5
Got what I am searching for.
Value is 6
Exiting all processess..
