python2でmultiprocessing中にエラーが発生すると応答が帰ってこなくなる。

Question

概要

python2でmultiprocessing中にエラーが発生すると応答が帰ってこなくなる。

この原因と解決法が知りたいです。

環境

OS: CentOS7
言語：python2.7

やりたかった事

特定のコマンドを並列で実行し、結果を取得。

ソース

#!/bin/env python

import subprocess
from multiprocessing import Pool

def testSP (cmd):
  print(cmd)
  try:
    ret=subprocess.check_output(cmd)
    return {"rc":0,"stdout":ret}
  except subprocess.CalledProcessError as e:
    print("catch CalledProcessError")
    print(e)
    return {"rc":1,"stdout":e}
  except:
    return {"rc":255,"stdout":"-"}
########################

print("==true==================")
print(testSP("true"))

print("==false==================")
print(testSP("false"))

print("==subprocess==================")
p = Pool(2)
ret=p.map(testSP,["true","false"])
print(ret)

実行結果

# ./test.py
==true==================
true
{'stdout': '', 'rc': 0}
==false==================
false
catch CalledProcessError
Command 'false' returned non-zero exit status 1
{'stdout': CalledProcessError(), 'rc': 1}
==subprocess==================
true
false
catch CalledProcessError
Command 'false' returned non-zero exit status 1
Exception in thread Thread-3:
Traceback (most recent call last):
  File "/usr/lib64/python2.7/threading.py", line 811, in __bootstrap_inner
    self.run()
  File "/usr/lib64/python2.7/threading.py", line 764, in run
    self.__target(*self.__args, **self.__kwargs)
  File "/usr/lib64/python2.7/multiprocessing/pool.py", line 376, in _handle_results
    task = get()
TypeError: ('__init__() takes at least 3 arguments (1 given)', <class 'subprocess.CalledProcessError'>, ())

以降応答なし、ctrl+cを押しても下記エラーが出てプロンプトは戻らない

●"ctrl+c"を押下●
Process PoolWorker-2:
Traceback (most recent call last):
  File "/usr/lib64/python2.7/multiprocessing/process.py", line 258, in _bootstrap
    self.run()
  File "/usr/lib64/python2.7/multiprocessing/process.py", line 114, in run
    self._target(*self._args, **self._kwargs)
  File "/usr/lib64/python2.7/multiprocessing/pool.py", line 102, in worker
    task = get()
  File "/usr/lib64/python2.7/multiprocessing/queues.py", line 374, in get
    racquire()
KeyboardInterrupt

以上、よろしくおねがいします。

Answer 1

Poolの中で実行している処理にKeyboardInterruptの入力が伝達されずに、wait関数がいつまでも待ち続けている状態になることに起因すると思われます。

ret=p.map(testSP,["true","false"])

を

ret=p.map_async(testSP,["true","false"]).get(9999999)

に変更して、タイムアウトの設定をしてみてはどうでしょうか。

Comments

[katosei]

ありがとうございます。
反応遅くなってすみません。

そもそもCtrl+Cを押す前に、応答が帰ってこなくなる理由ってわかりますでしょうか？
なぜmultiprocessingがエラーになるのかわからなくて。