У меня есть требование создавать дочерние процессы, получать результаты с помощью Future, а затем при необходимости убивать некоторые из них.
Для этого я создал подкласс класса multiprocessing.Process и возвращаю объект Future из метода start().
Проблема в том, что я не могу получить результат в функции cb(), так как она никогда не вызывается.
Пожалуйста, помогите/предложите, можно ли это сделать каким-то другим способом или что-то, чего мне не хватает в моей текущей реализации?
Ниже приведен мой текущий подход
from multiprocessing import Process, Queue
from concurrent.futures import _base
import threading
from time import sleep
def foo(x,q):
print('result {}'.format(x*x))
result = x*x
sleep(5)
q.put(result)
class MyProcess(Process):
def __init__(self, target, args):
super().__init__()
self.target = target
self.args = args
self.f = _base.Future()
def run(self):
q = Queue()
worker_thread = threading.Thread(target=self.target, args=(self.args+ (q,)))
worker_thread.start()
r = q.get(block=True)
print('setting result {}'.format(r))
self.f.set_result(result=r)
print('done setting result')
def start(self):
f = _base.Future()
run_thread = threading.Thread(target=self.run)
run_thread.start()
return f
def cb(future):
print('received result in callback {}'.format(future))
def main():
p1 = MyProcess(target=foo, args=(2,))
f = p1.start()
f.add_done_callback(fn=cb)
sleep(10)
if __name__ == '__main__':
main()
print('Main thread dying')