Python强制关闭线程的一种办法

开源中国首席鉴定大湿 发布于 2016/03/23 22:04
阅读 13K+
收藏 2

由于经常被Python非Daemon线程阻塞,导致程序无法结束。所以到处找办法解决,但是经常没有找到点上。导致无功而返。

今天突发奇想来搜了一下相关的解决方案,竟然被我找到了。

首先是百度了一下(懒得开VPN)

然后找到了一个网友分享的解决方案:

http://www.cnblogs.com/rainduck/archive/2013/03/29/2989810.html

但是试验之后并没有什么卵用(┑( ̄Д  ̄)┍),我是在我的MAC上面试验的。python 2.7.10

然后再次谷歌了一下使用到的API,在最佳回答的评论区找到了答案。

http://stackoverflow.com/questions/323972/is-there-any-way-to-kill-a-thread-in-python
第六条评论

于是最终解决方案如下:

import threading
import time
import inspect
import ctypes

def _async_raise(tid, exctype):
    """raises the exception, performs cleanup if needed"""
    tid = ctypes.c_long(tid)
    if not inspect.isclass(exctype):
        exctype = type(exctype)
    res = ctypes.pythonapi.PyThreadState_SetAsyncExc(tid, ctypes.py_object(exctype))
    if res == 0:
        raise ValueError("invalid thread id")
    elif res != 1:
        # """if it returns a number greater than one, you're in trouble,
        # and you should call it again with exc=NULL to revert the effect"""
        ctypes.pythonapi.PyThreadState_SetAsyncExc(tid, None)
        raise SystemError("PyThreadState_SetAsyncExc failed")

def stop_thread(thread):
    _async_raise(thread.ident, SystemExit)

class TestThread(threading.Thread):
    def run(self):
        print "begin"
        while True:
            time.sleep(0.1)
        print "end"
if __name__ == "__main__":
    t = TestThread()
    t.start()
    time.sleep(1)
    stop_thread(t)
    print "stoped"

改造后的方案,只是在 _async_raise 函数最前面,将tid转换成了c_long类型。因为传到API中的类型需要是C的长整形,不然会越界。因为在我的环境中,PID是一个较大的值。

解决方案利用的是python内置API,通过ctypes模块调用,在线程中丢出异常,使线程退出。

希望我的分享能给各位python程序猿一些帮助。


加载中
0
乌龟壳
乌龟壳
sys.exit()就行了
乌龟壳
乌龟壳
回复 @开源中国首席鉴定大湿 : 这个问题以前碰到过,肯定没那么麻烦的
开源中国首席鉴定大湿
开源中国首席鉴定大湿
sys.exit() 也退出不了阻塞的线程,你可以去试试
0
乌龟壳
乌龟壳
import threading
import time

class TestThread(threading.Thread):
    def run(self):
        print("begin")
        while True:
            time.sleep(0.1)
        print("end")
        
if __name__ == "__main__":
    t = TestThread()
    t.setDaemon(True)
    t.start()
    time.sleep(1)



开源中国首席鉴定大湿
开源中国首席鉴定大湿
我说的就是强制关闭非daemon线程。 daemon线程也会因为某些原因阻塞主线程,这是我实际遇到过的。
0
乌龟壳
乌龟壳

如果要关闭含有非Daemon的线程,可以给自己发送SIGTERM命令,建议不要发送SIGKILL,SIGKILL不会释放资源

import threading
import time
import os
import signal
 
class TestThread(threading.Thread):
    def run(self):
        print("begin")
        while True:
            time.sleep(0.1)
        print("end")
         
if __name__ == "__main__":
    t = TestThread()
    t.start()
    time.sleep(1)
    os.kill(os.getpid(), signal.SIGTERM)



乌龟壳
乌龟壳
回复 @乌龟壳 : 所以也算第一次在这里遇到这类问题吧
乌龟壳
乌龟壳
回复 @开源中国首席鉴定大湿 : 可能我做的程序都和IO有关,主进程直接关了fd,子进程自己报错走了
开源中国首席鉴定大湿
开源中国首席鉴定大湿
我要做的并不是关闭这个进程。 我只是想关闭进程里面的某一个线程而已。 如果是一个服务里面的线程执行超时了,难道我也要去把这个服务重启一次吗?
0
mihello
mihello

666, 今天用django + channels(websocket) 模拟 tail -f  ,浏览器实时打印日志。用的是threading,但是有个bug每次connect就新开一个线程(导致刷新页面一次就多打印一行),我想在disconnect的时候,kill掉这个子线程。但是找了很多资料都没成功。这次多些博主了。

返回顶部
顶部