Python：线程之定位与销毁

时间 2019-11-07

标签 python 线程定位销毁栏目 Python 繁體版

原文原文链接

背景

开工前我就以为有什么不太对劲，感受要背锅。这可不，上班第三天就捅锅了。html

咱们有个了不得的后台程序，能够动态加载模块，并以线程方式运行，经过这种形式实现插件的功能。而模块更新时候，后台程序自身不会退出，只会将模块对应的线程关闭、更新代码再启动，6 得不行。python

因而乎我就写了个模块准备大展身手，结果忘记写退出函数了，致使每次更新模块都新建立一个线程，除非重启那个程序，不然那些线程就一直苟活着。segmentfault

这可不行啊，得想个办法清理呀，要否则怕是要炸了。api

那么怎么清理呢？我能想到的就是两步走：多线程

找出须要清理的线程号 tid；
销毁它们；

找出线程ID

和平时的故障排查类似，先经过 ps 命令看看目标进程的线程状况，由于已是 setName 设置过线程名，因此正常来讲应该是看到对应的线程的。直接用下面代码来模拟这个线程：并发

Python 版本的多线程async

#coding: utf8
import threading
import os
import time

def tt():
    info = threading.currentThread()
    while True:
        print 'pid: ', os.getpid()
        print info.name, info.ident
        time.sleep(3)

t1 = threading.Thread(target=tt)
t1.setName('OOOOOPPPPP')
t1.setDaemon(True)
t1.start()

t2 = threading.Thread(target=tt)
t2.setName('EEEEEEEEE')
t2.setDaemon(True)
t2.start()


t1.join()
t2.join()

输出:ide

root@10-46-33-56:~# python t.py
pid:  5613
OOOOOPPPPP 139693508122368
pid:  5613
EEEEEEEEE 139693497632512
...

能够看到在 Python 里面输出的线程名就是咱们设置的那样，然而 Ps 的结果倒是令我怀疑人生：函数

root@10-46-33-56:~# ps -Tp 5613
  PID  SPID TTY          TIME CMD
 5613  5613 pts/2    00:00:00 python
 5613  5614 pts/2    00:00:00 python
 5613  5615 pts/2    00:00:00 python

正常来讲不应是这样呀，我有点迷了，难道我一直都是记错了？用别的语言版本的多线程来测试下：测试

C 版本的多线程

#include<stdio.h>
#include<sys/syscall.h>
#include<sys/prctl.h>
#include<pthread.h>

void *test(void *name)
{    
    pid_t pid, tid;
    pid = getpid();
    tid = syscall(__NR_gettid);
    char *tname = (char *)name;
    
    // 设置线程名字
    prctl(PR_SET_NAME, tname);
    
    while(1)
    {
        printf("pid: %d, thread_id: %u, t_name: %s\n", pid, tid, tname);
        sleep(3);
    }
}

int main()
{
    pthread_t t1, t2;
    void *ret;
    pthread_create(&t1, NULL, test,  (void *)"Love_test_1");
    pthread_create(&t2, NULL, test,  (void *)"Love_test_2");
    pthread_join(t1, &ret);
    pthread_join(t2, &ret);
}

输出：

root@10-46-33-56:~# gcc t.c -lpthread && ./a.out
pid: 5575, thread_id: 5577, t_name: Love_test_2
pid: 5575, thread_id: 5576, t_name: Love_test_1
pid: 5575, thread_id: 5577, t_name: Love_test_2
pid: 5575, thread_id: 5576, t_name: Love_test_1
...

用 PS 命令再次验证：

root@10-46-33-56:~# ps -Tp 5575
  PID  SPID TTY          TIME CMD
 5575  5575 pts/2    00:00:00 a.out
 5575  5576 pts/2    00:00:00 Love_test_1
 5575  5577 pts/2    00:00:00 Love_test_2

这个才是正确嘛，线程名确实是能够经过 Ps 看出来的嘛！

不过为啥 Python 那个看不到呢？既然是经过 setName 设置线程名的，那就看看定义咯：

[threading.py]
class Thread(_Verbose):
    ...
    @property
    def name(self):
        """A string used for identification purposes only.

        It has no semantics. Multiple threads may be given the same name. The
        initial name is set by the constructor.

        """
        assert self.__initialized, "Thread.__init__() not called"
        return self.__name

    @name.setter
    def name(self, name):
        assert self.__initialized, "Thread.__init__() not called"
        self.__name = str(name)
        
    def setName(self, name):
        self.name = name
    ...

看到这里其实只是在 Thread 对象的属性设置了而已，并无动到根本，那确定就是看不到咯~

这样看起来，咱们已经没办法经过 ps 或者 /proc/ 这类手段在外部搜索 python 线程名了，因此咱们只能在 Python 内部来解决。

因而问题就变成了，怎样在 Python 内部拿到全部正在运行的线程呢？

threading.enumerate 能够完美解决这个问题！Why?

Because 在下面这个函数的 doc 里面说得很清楚了，返回全部活跃的线程对象，不包括终止和未启动的。

[threading.py]

def enumerate():
    """Return a list of all Thread objects currently alive.

    The list includes daemonic threads, dummy thread objects created by
    current_thread(), and the main thread. It excludes terminated threads and
    threads that have not yet been started.

    """
    with _active_limbo_lock:
        return _active.values() + _limbo.values()

由于拿到的是 Thread 的对象，因此咱们经过这个能到该线程相关的信息！

请看完整代码示例：

#coding: utf8

import threading
import os
import time


def get_thread():
    pid = os.getpid()
    while True:
        ts = threading.enumerate()
        print '------- Running threads On Pid: %d -------' % pid
        for t in ts:
            print t.name, t.ident
        print
        time.sleep(1)
        
def tt():
    info = threading.currentThread()
    pid = os.getpid()
    while True:
        print 'pid: {}, tid: {}, tname: {}'.format(pid, info.name, info.ident)
        time.sleep(3)
        return

t1 = threading.Thread(target=tt)
t1.setName('Thread-test1')
t1.setDaemon(True)
t1.start()

t2 = threading.Thread(target=tt)
t2.setName('Thread-test2')
t2.setDaemon(True)
t2.start()

t3 = threading.Thread(target=get_thread)
t3.setName('Checker')
t3.setDaemon(True)
t3.start()

t1.join()
t2.join()
t3.join()

输出：

root@10-46-33-56:~# python t_show.py
pid: 6258, tid: Thread-test1, tname: 139907597162240
pid: 6258, tid: Thread-test2, tname: 139907586672384

------- Running threads On Pid: 6258 -------
MainThread 139907616806656
Thread-test1 139907597162240
Checker 139907576182528
Thread-test2 139907586672384

------- Running threads On Pid: 6258 -------
MainThread 139907616806656
Thread-test1 139907597162240
Checker 139907576182528
Thread-test2 139907586672384

------- Running threads On Pid: 6258 -------
MainThread 139907616806656
Thread-test1 139907597162240
Checker 139907576182528
Thread-test2 139907586672384

------- Running threads On Pid: 6258 -------
MainThread 139907616806656
Checker 139907576182528
...

代码看起来有点长，可是逻辑至关简单，Thread-test1 和 Thread-test2 都是打印出当前的 pid、线程 id 和线程名字，而后 3s 后退出，这个是想模拟线程正常退出。

而 Checker 线程则是每秒经过 threading.enumerate 输出当前进程内全部活跃的线程。

能够明显看到一开始是能够看到 Thread-test1 和 Thread-test2的信息，当它俩退出以后就只剩下 MainThread 和 Checker 自身而已了。

销毁指定线程

既然能拿到名字和线程 id，那咱们也就能干掉指定的线程了！

假设如今 Thread-test2 已经黑化，发疯了，咱们须要制止它，那咱们就能够经过这种方式解决了：

在上面的代码基础上，增长和补上下列代码：

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:
        ctypes.pythonapi.PyThreadState_SetAsyncExc(tid, None)
        raise SystemError("PyThreadState_SetAsyncExc failed")

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

def get_thread():
    pid = os.getpid()
    while True:
        ts = threading.enumerate()
        print '------- Running threads On Pid: %d -------' % pid
        for t in ts:
            print t.name, t.ident, t.is_alive()
            if t.name == 'Thread-test2':
                print 'I am go dying! Please take care of yourself and drink more hot water!'
                stop_thread(t)
        print
        time.sleep(1)

输出

root@10-46-33-56:~# python t_show.py
pid: 6362, tid: 139901682108160, tname: Thread-test1
pid: 6362, tid: 139901671618304, tname: Thread-test2
------- Running threads On Pid: 6362 -------
MainThread 139901706389248 True
Thread-test1 139901682108160 True
Checker 139901661128448 True
Thread-test2 139901671618304 True
Thread-test2: I am go dying. Please take care of yourself and drink more hot water!

------- Running threads On Pid: 6362 -------
MainThread 139901706389248 True
Thread-test1 139901682108160 True
Checker 139901661128448 True
Thread-test2 139901671618304 True
Thread-test2: I am go dying. Please take care of yourself and drink more hot water!

pid: 6362, tid: 139901682108160, tname: Thread-test1
------- Running threads On Pid: 6362 -------
MainThread 139901706389248 True
Thread-test1 139901682108160 True
Checker 139901661128448 True
// Thread-test2 已经不在了

一顿操做下来，虽然咱们这样对待 Thread-test2，但它仍是关心着咱们：多喝热水，

PS: 热水虽好，八杯足矣，请勿贪杯哦。

书回正传，上述的方法是极为粗暴的，为何这么说呢？

由于它的原理是：利用 Python 内置的 API，触发指定线程的异常，让其能够自动退出；

万不得已真不要用这种方法，有必定几率触发不可描述的问题。切记！别问我为何会知道...

为何中止线程这么难

多线程自己设计就是在进程下的协做并发，是调度的最小单元，线程间分食着进程的资源，因此会有许多锁机制和状态控制。

若是使用强制手段干掉线程，那么很大概率出现意想不到的bug。并且最重要的锁资源释放可能也会出现意想不到问题。

咱们甚至也没法经过信号杀死进程那样直接杀线程，由于 kill 只有对付进程才能达到咱们的预期，而对付线程明显不能够，无论杀哪一个线程，整个进程都会退出！

而由于有 GIL，使得不少童鞋都以为 Python 的线程是Python 自行实现出来的，并不是实际存在，Python 应该能够直接销毁吧？

然而事实上 Python 的线程都是货真价实的线程！

什么意思呢？Python 的线程是操做系统经过 pthread 建立的原生线程。Python 只是经过 GIL 来约束这些线程，来决定何时开始调度，比方说运行了多少个指令就交出 GIL，至于谁夺得花魁，得听操做系统的。

若是是单纯的线程，其实系统是有办法终止的，好比: pthread_exit,pthread_kill 或 pthread_cancel，详情可看：https://www.cnblogs.com/Creat...

很惋惜的是： Python 层面并无这些方法的封装！个人天，好气！可能人家以为，线程就该温柔对待吧。

如何温柔退出线程

想要温柔退出线程，其实差很少就是一句废话了~

要么运行完退出，要么设置标志位，时常检查标记位，该退出的就退出咯。

扩展

《如何正确的终止正在运行的子线程》：https://www.cnblogs.com/Creat...
《不要粗暴的销毁python线程》：http://xiaorui.cc/2017/02/22/...

欢迎各位大神指点交流, QQ讨论群: 258498217
转载请注明来源: https://segmentfault.com/a/11...