子线程中的I / O函数调用超时

时间:2018-09-06 17:41:50

标签: python flask-restful

我有一个Flask-RESTful API(Python 3),并且正在尝试使运行终端命令的函数调用超时,该命令执行C ++程序可能需要很长时间才能完成。此函数如下所示:

def func(args):
    ...
    result = subprocess.getoutput('./c_program')
    ...

我搜索了如何执行此操作,发现了以下内容: Timeout on a function call 我尝试了两种建议的方法,但不幸的是,这两种方法均无效。 Signal仅在主线程中起作用(并且我在线程API中),并且多处理不会停止I / O工作。我一直在寻找,但只发现提到这两种方法。有人知道对此有什么解决办法吗?

1 个答案:

答案 0 :(得分:2)

最好使用Popen模块中的subprocess。我还加入了How to terminate a python subprocess launched with shell=True中的策略,以解决有关终止进程的问题:

import os
import signal
import subprocess
import time
from subprocess import PIPE, Popen

p = Popen("sleep 1; echo 1", shell=True, stdout=PIPE, preexec_fn=os.setsid)

WAIT = 0.5

started = time.time()

# do some other work for WAIT seconds:
while time.time() < started + WAIT:
    print("do something else")
    time.sleep(.1)

print("kill")

try:
    os.killpg(
        os.getpgid(p.pid), signal.SIGTERM
    )
except ProcessLookupError:
    # process possibly died already
    pass

print(p.stdout.read())

您可以将WAIT设置为1.5以使过程完成,在这种情况下,您将看到echo的输出。