这是一个解决方案。我不确定您是否喜欢它,但我想不到其他任何东西。我已经修改了您的代码以使其正常运行。
from concurrent.futures import ThreadPoolExecutor
import time
quit = False
def pri():
print("Hello World!!!")
def start():
while quit is not True:
time.sleep(1)
pri()
try:
pool = ThreadPoolExecutor(max_workers=3)
pool.submit(start)
while quit is not True:
print("hei")
time.sleep(1)
except KeyboardInterrupt:
quit = True
要点如下:
如果wait
为True,则该方法将在所有未完成的期货执行完毕并且与执行者关联的资源被释放之前不会返回。如果等待,False
则此方法将立即返回,并且当所有未决的期货执行完毕时,与执行程序关联的资源将被释放。不管等待值是多少,整个Python程序都不会退出,直到所有未完成的期货执行完毕。
如果使用该with
语句,则可以避免显式调用此方法,这将关闭Executor
(等待,就像Executor.shutdown()
在调用wait时将其设置为True
)
就像join()
在线程上调用一样。 这就是为什么我将其替换为:
pool = ThreadPoolExecutor(max_workers=3)
pool.submit(start)
主线程必须执行“工作”才能捕获Ctrl + C。因此,您不能仅将主线程留在那里并退出,最简单的方法是运行无限循环
现在您已经在主线程中运行了一个循环,当您点击时CTRL+C
,程序将进入该except KeyboardInterrupt
块并进行设置quit=True
。然后,您的工作线程可以退出。
严格来说,这只是一种解决方法。在我看来,没有其他办法可以做到。
我不确定是什么困扰着您,但是您可以在另一个线程中捕获异常而不会出现问题:
import socket
import time
from concurrent.futures import ThreadPoolExecutor
s = socket.socket(socket.AF_INET,socket.soCK_STREAM)
def con():
try:
raise socket.gaierror
main()
except socket.gaierror:
print("gaierror occurred")
err()
def err():
print("err invoked")
time.sleep(1)
con()
def main():
s.send("[+] Hello")
with ThreadPoolExecutor(3) as exe:
exe.submit(con)
输出量
gaierror occurred
err invoked
gaierror occurred
err invoked
gaierror occurred
err invoked
gaierror occurred
...