脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → Ctrl+C终止多线程程序

Python中用Ctrl+C终止多线程程序的问题解决

 更新时间：2013年03月30日 14:23:14   投稿：shangke  
花了一天时间用python为服务写了个压力测试。很简单，多线程向服务器发请求。但写完之后发现如果中途想停下来，按Ctrl+C达不到效果，自然想到要用信号处理函数捕捉信号，使线程都停下来，问题解决的方法请往下看：

复制代码 代码如下:

#!/bin/env python
 # -*- coding: utf-8 -*-
 #filename: peartest.py
 
 import threading, signal
 
 is_exit = False
 
 def doStress(i, cc):
     global is_exit
     idx = i
     while not is_exit:
         if (idx < 10000000):
             print "thread[%d]: idx=%d"%(i, idx)
             idx = idx + cc
         else:
             break
     print "thread[%d] complete."%i
 
 def handler(signum, frame):
     global is_exit
     is_exit = True
     print "receive a signal %d, is_exit = %d"%(signum, is_exit)
 
 if __name__ == "__main__":
     signal.signal(signal.SIGINT, handler)
     signal.signal(signal.SIGTERM, handler)
     cc = 5
     for i in range(cc):
         t = threading.Thread(target=doStress, args=(i,cc))
         t.start()

上面是一个模拟程序，并不真正向服务发送请求，而代之以在一千万以内，每个线程每隔并发数个（cc个）打印一个整数。很明显，当所有线程都完成自己的任务后，进程会正常退出。但如果我们中途想退出（试想一个压力测试程序，在中途已经发现了问题，需要停止测试），该肿么办？你当然可以用ps查找到进程号，然后kill -9杀掉，但这样太繁琐了，捕捉Ctrl+C是最自然的想法。上面示例程序中已经捕捉了这个信号，并修改全局变量is_exit，线程中会检测这个变量，及时退出。

但事实上这个程序并不work，当你按下Ctrl+C时，程序照常运行，并无任何响应。网上搜了一些资料，明白是python的子线程如果不是daemon的话，主线程是不能响应任何中断的。但设为daemon后主线程会随之退出，接着整个进程很快就退出了，所以还需要在主线程中检测各个子线程的状态，直到所有子线程退出后自己才退出，因此上例29行之后的代码可以修改为：

复制代码 代码如下:

threads=[]
     for i in range(cc):
         t = threading.Thread(target=doStress, args=(i, cc))
         t.setDaemon(True)
         threads.append(t)
         t.start()
     for i in range(cc):
         threads[i].join()

重新试一下，问题依然没有解决，进程还是没有响应Ctrl+C，这是因为join()函数同样会waiting在一个锁上，使主线程无法捕获信号。因此继续修改，调用线程的isAlive()函数判断线程是否完成：

复制代码 代码如下:

while 1:
         alive = False
         for i in range(cc):
             alive = alive or threads[i].isAlive()
         if not alive:
             break

这样修改后，程序完全按照预想运行了：可以顺利的打印每个线程应该打印的所有数字，也可以中途用Ctrl+C终结整个进程。完整的代码如下：

复制代码 代码如下:

#!/bin/env python
 # -*- coding: utf-8 -*-
 #filename: peartest.py
 
 import threading, signal
 
 is_exit = False
 
 def doStress(i, cc):
     global is_exit
     idx = i
     while not is_exit:
         if (idx < 10000000):
             print "thread[%d]: idx=%d"%(i, idx)
             idx = idx + cc
         else:
             break
     if is_exit:
         print "receive a signal to exit, thread[%d] stop."%i
     else:
         print "thread[%d] complete."%i
 
 def handler(signum, frame):
     global is_exit
     is_exit = True
     print "receive a signal %d, is_exit = %d"%(signum, is_exit)
 
 if __name__ == "__main__":
     signal.signal(signal.SIGINT, handler)
     signal.signal(signal.SIGTERM, handler)
     cc = 5
     threads = []
     for i in range(cc):
         t = threading.Thread(target=doStress, args=(i,cc))
         t.setDaemon(True)
         threads.append(t)
         t.start()
     while 1:
         alive = False
         for i in range(cc):
             alive = alive or threads[i].isAlive()
         if not alive:
             break

1.把所有子线程设为Daemon；
2.使用isAlive()函数判断所有子线程是否完成，而不是在主线程中用join()函数等待完成；
3.写一个响应Ctrl+C信号的函数，修改全局变量，使得各子线程能够检测到，并正常退出。

相关文章

  • Python3中urlencode和urldecode的用法详解

    Python3中urlencode和urldecode的用法详解

    今天小编就为大家分享一篇Python3中urlencode和urldecode的用法详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2019-07-07
  • python文件读取read及readlines两种方法使用详解

    python文件读取read及readlines两种方法使用详解

    这篇文章主要为大家介绍了python文件读取read及readlines两种方法的使用示例及区别详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
    2022-07-07
  • Python与Node.js之间实现通信的JSON数据接收发送

    Python与Node.js之间实现通信的JSON数据接收发送

    Python和Node.js是两个流行且功能强大的编程语言，它们之间使用JSON格式进行数据交换是一种高效和灵活的方式,本文将详细介绍如何在Python和Node.js之间通过JSON进行数据通信，包括发送和接收JSON数据以及一些常见的交互示例代码
    2024-01-01
  • Python实现WGS84火星百度及web墨卡托四种坐标系相互转换

    Python实现WGS84火星百度及web墨卡托四种坐标系相互转换

    主流被使用的地理坐标系并不统一,常用的有WGS84、GCJ02（火星坐标系）、BD09（百度坐标系）以及百度地图中保存矢量信息的web墨卡托,本文利用Python编写相关类以实现4种坐标系统之间的互相转换
    2023-08-08
  • Python程序控制语句用法实例分析

    Python程序控制语句用法实例分析

    这篇文章主要介绍了Python程序控制语句用法,结合实例形式分析了Python流程控制语句的条件、循环以及函数、类、文件、异常处理等相关操作使用技巧,需要的朋友可以参考下
    2020-01-01
  • 利用Python绘制酷炫的3D地图

    利用Python绘制酷炫的3D地图

    pyecharts是一款将python与echarts结合的强大的数据可视化工具。本文将为大家介绍如何利用pyecharts绘制酷炫的3D地图，感兴趣的小伙伴可以试一试
    2022-03-03
  • 分享一下Python数据分析常用的8款工具

    分享一下Python数据分析常用的8款工具

    Python是数据处理常用工具，可以处理数量级从几K至几T不等的数据，具有较高的开发效率和可维护性，还具有较强的通用性和跨平台性，这里就为大家分享几个不错的数据分析工具,需要的朋友可以参考下
    2018-04-04
  • 小白教你PyCharm从下载到安装再到科学使用PyCharm2020最新激活码

    小白教你PyCharm从下载到安装再到科学使用PyCharm2020最新激活码

    这篇文章主要介绍了PyCharm最新版从下载到安装再到科学使用PyCharm2020最新激活码,需要的朋友可以参考下
    2020-09-09
  • Python实现调用摄像头拍摄照片

    Python实现调用摄像头拍摄照片

    这篇文章主要为大家详细介绍了如何利用Python实现调用摄像头拍摄照片以及邮箱收集功能，文中的示例代码讲解详细，感兴趣的可以动手尝试一下
    2022-06-06
  • 深入Python解释器理解Python中的字节码

    深入Python解释器理解Python中的字节码

    这篇文章深入Python解释器、从其对字节码的处理过程来详解Python中的字节码,需要的朋友可以参考下
    2015-04-04

最新评论