快捷導(dǎo)航

Python中用Ctrl+C終止多線程程序的問題解決

更新時(shí)間：2013年03月30日 14:23:14 投稿：shangke

花了一天時(shí)間用python為服務(wù)寫了個(gè)壓力測試。很簡單，多線程向服務(wù)器發(fā)請求。但寫完之后發(fā)現(xiàn)如果中途想停下來，按Ctrl+C達(dá)不到效果，自然想到要用信號處理函數(shù)捕捉信號，使線程都停下來，問題解決的方法請往下看：

復(fù)制代碼代碼如下:

#!/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()

上面是一個(gè)模擬程序，并不真正向服務(wù)發(fā)送請求，而代之以在一千萬以內(nèi)，每個(gè)線程每隔并發(fā)數(shù)個(gè)（cc個(gè)）打印一個(gè)整數(shù)。很明顯，當(dāng)所有線程都完成自己的任務(wù)后，進(jìn)程會正常退出。但如果我們中途想退出（試想一個(gè)壓力測試程序，在中途已經(jīng)發(fā)現(xiàn)了問題，需要停止測試），該腫么辦？你當(dāng)然可以用ps查找到進(jìn)程號，然后kill -9殺掉，但這樣太繁瑣了，捕捉Ctrl+C是最自然的想法。上面示例程序中已經(jīng)捕捉了這個(gè)信號，并修改全局變量is_exit，線程中會檢測這個(gè)變量，及時(shí)退出。

但事實(shí)上這個(gè)程序并不work，當(dāng)你按下Ctrl+C時(shí)，程序照常運(yùn)行，并無任何響應(yīng)。網(wǎng)上搜了一些資料，明白是python的子線程如果不是daemon的話，主線程是不能響應(yīng)任何中斷的。但設(shè)為daemon后主線程會隨之退出，接著整個(gè)進(jìn)程很快就退出了，所以還需要在主線程中檢測各個(gè)子線程的狀態(tài)，直到所有子線程退出后自己才退出，因此上例29行之后的代碼可以修改為：

復(fù)制代碼代碼如下:

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()

重新試一下，問題依然沒有解決，進(jìn)程還是沒有響應(yīng)Ctrl+C，這是因?yàn)閖oin()函數(shù)同樣會waiting在一個(gè)鎖上，使主線程無法捕獲信號。因此繼續(xù)修改，調(diào)用線程的isAlive()函數(shù)判斷線程是否完成：

復(fù)制代碼代碼如下:

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

這樣修改后，程序完全按照預(yù)想運(yùn)行了：可以順利的打印每個(gè)線程應(yīng)該打印的所有數(shù)字，也可以中途用Ctrl+C終結(jié)整個(gè)進(jìn)程。完整的代碼如下：

復(fù)制代碼代碼如下:

#!/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