我正在嘗試在我的程式中處理 KeyboardInterrupt 例外，但我不知道如何使用多處理池來處理它。即使我將池操作放在 try-exception 塊中并處理例外，我也會收到 4 個 KeyboardInterrupt 例外。
import time
import multiprocessing as mp
def calc(i):
return i*i
def main():
try:
with mp.Pool(4) as p:
while True:
print(p.map(calc, range(10)))
time.sleep(1)
except KeyboardInterrupt:
print("\nShutting down.")
except Exception as e:
print(e)
if __name__ == '__main__':
main()
我知道這些行程在一個隔離的環境中運行，但我也想以某種方式處理例外。
編輯：運行代碼時得到的輸出：
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
^CProcess ForkPoolWorker-3:
Process ForkPoolWorker-5:
Process ForkPoolWorker-4:
Traceback (most recent call last):
File "/usr/lib/python3.6/multiprocessing/process.py", line 258, in _bootstrap
self.run()
Traceback (most recent call last):
File "/usr/lib/python3.6/multiprocessing/process.py", line 93, in run
self._target(*self._args, **self._kwargs)
File "/usr/lib/python3.6/multiprocessing/pool.py", line 108, in worker
task = get()
File "/usr/lib/python3.6/multiprocessing/queues.py", line 334, in get
with self._rlock:
File "/usr/lib/python3.6/multiprocessing/synchronize.py", line 95, in __enter__
return self._semlock.__enter__()
File "/usr/lib/python3.6/multiprocessing/process.py", line 258, in _bootstrap
self.run()
KeyboardInterrupt
File "/usr/lib/python3.6/multiprocessing/process.py", line 93, in run
self._target(*self._args, **self._kwargs)
File "/usr/lib/python3.6/multiprocessing/pool.py", line 108, in worker
task = get()
File "/usr/lib/python3.6/multiprocessing/queues.py", line 335, in get
res = self._reader.recv_bytes()
File "/usr/lib/python3.6/multiprocessing/connection.py", line 216, in recv_bytes
buf = self._recv_bytes(maxlength)
File "/usr/lib/python3.6/multiprocessing/connection.py", line 407, in _recv_bytes
buf = self._recv(4)
File "/usr/lib/python3.6/multiprocessing/connection.py", line 379, in _recv
chunk = read(handle, remaining)
KeyboardInterrupt
Traceback (most recent call last):
File "/usr/lib/python3.6/multiprocessing/process.py", line 258, in _bootstrap
self.run()
File "/usr/lib/python3.6/multiprocessing/process.py", line 93, in run
self._target(*self._args, **self._kwargs)
File "/usr/lib/python3.6/multiprocessing/pool.py", line 108, in worker
task = get()
File "/usr/lib/python3.6/multiprocessing/queues.py", line 334, in get
with self._rlock:
File "/usr/lib/python3.6/multiprocessing/synchronize.py", line 95, in __enter__
return self._semlock.__enter__()
KeyboardInterrupt
Process ForkPoolWorker-6:
Traceback (most recent call last):
File "/usr/lib/python3.6/multiprocessing/process.py", line 258, in _bootstrap
self.run()
File "/usr/lib/python3.6/multiprocessing/process.py", line 93, in run
self._target(*self._args, **self._kwargs)
File "/usr/lib/python3.6/multiprocessing/pool.py", line 108, in worker
task = get()
File "/usr/lib/python3.6/multiprocessing/queues.py", line 334, in get
with self._rlock:
File "/usr/lib/python3.6/multiprocessing/synchronize.py", line 95, in __enter__
return self._semlock.__enter__()
KeyboardInterrupt
Shutting down.
uj5u.com熱心網友回復：
由于您似乎在 Linux 型別的平臺下運行（您確實應該使用該平臺標記您的多處理問題），您需要忽略池行程中的 CTRL-C。最簡單的方法是在創建池時使用初始化引數：
import time
import multiprocessing as mp
def init_pool_processes():
"""
Each pool process will execute this as part of its
initialization.
"""
import signal
signal.signal(signal.SIGINT, signal.SIG_IGN)
def calc(i):
return i*i
def main():
try:
with mp.Pool(4, initializer=init_pool_processes) as p:
while True:
print(p.map(calc, range(10)))
time.sleep(1)
except KeyboardInterrupt:
print("\nShutting down.")
except Exception as e:
print(e)
if __name__ == '__main__':
main()
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/438409.html