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

其實，如果用python寫一個服務，也需要這樣，因為負責服務的那個線程是永遠在那裡接收請求的，不會退出，而如果你想用Ctrl+C殺死整個服務，跟上面的壓力測試程式是一個道理。總結一下，python多線程中要響應Ctrl+C的訊號以殺死整個進程，需要：

1.把所有子線程設為Daemon；
2.使用isAlive()函數判斷所有子線程是否完成，而不是在主線程中用join()函數等待完成；
3.寫一個響應Ctrl+C訊號的函數，修改全域變數，使得各子線程能夠檢測到，並正常退出。