python殺死一個線程的方法
最近在項目中遇到這一需求:
我需要一個函數(shù)工作,比如遠程連接一個端口,遠程讀取文件等,但是我給的時間有限,比如,4秒鐘如果你還沒有讀取完成或者連接成功,我就不等了,很可能對方已經(jīng)宕機或者拒絕了。這樣可以批量做一些事情而不需要一直等,浪費時間。
結(jié)合我的需求,我想到這種辦法:
1、在主進程執(zhí)行,調(diào)用一個進程執(zhí)行函數(shù),然后主進程sleep,等時間到了,就kill 執(zhí)行函數(shù)的進程。
測試一個例子:
import time import threading def p(i): print i class task(threading.Thread): def __init__(self,fun,i): threading.Thread.__init__(self) self.fun = fun self.i = i self.thread_stop = False def run(self): while not self.thread_stop: self.fun(self.i) def stop(self): self.thread_stop = True def test(): thread1 = task(p,2) thread1.start() time.sleep(4) thread1.stop() return if __name__ == '__main__': test()
經(jīng)過測試只定了4秒鐘。
經(jīng)過我的一番折騰,想到了join函數(shù),這個函數(shù)式用來等待一個線程結(jié)束的,如果這個函數(shù)沒有結(jié)束的話,那么,就會阻塞當前運行的程序。關(guān)鍵是,這個參數(shù)有一個可選參數(shù):join([timeout]): 阻塞當前上下文環(huán)境的線程,直到調(diào)用此方法的線程終止或到達指定的timeout(可選參數(shù))。
不多說了貼下面代碼大家看下:
#!/usr/bin/env python #-*-coding:utf-8-*- ''''' author:cogbee time:2014-6-13 function:readme ''' import pdb import time import threading import os #pdb.set_trace() class task(threading.Thread): def __init__(self,ip): threading.Thread.__init__(self) self.ip = ip self.thread_stop = False def run(self): while not self.thread_stop: #//添加你要做的事情,如果成功了就設(shè)置一下<span style="font-family: Arial, Helvetica, sans-serif;">self.thread_stop變量。</span> [python] view plaincopy在CODE上查看代碼片派生到我的代碼片 if file != '': self.thread_stop = True def stop(self): self.thread_stop = True def test(eachline): global file list = [] for ip in eachline: thread1 = task(ip) thread1.start() thread1.join(3) if thread1.isAlive(): thread1.stop() continue #將可以讀取的都存起來 if file != '': list.append(ip) print list if __name__ == '__main__': eachline = ['1.1.1.1','222.73.5.54'] test(eachline)
下面給大家分享我寫的一段殺死線程的代碼。
由于python線程沒有提供abort方法,分享下面一段代碼殺死線程:
import threading import inspect import ctypes def _async_raise(tid, exctype): """raises the exception, performs cleanup if needed""" if not inspect.isclass(exctype): raise TypeError("Only types can be raised (not instances)") 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, 0) raise SystemError("PyThreadState_SetAsyncExc failed") class Thread(threading.Thread): def _get_my_tid(self): """determines this (self's) thread id""" if not self.isAlive(): raise threading.ThreadError("the thread is not active") # do we have it cached? if hasattr(self, "_thread_id"): return self._thread_id # no, look for it in the _active dict for tid, tobj in threading._active.items(): if tobj is self: self._thread_id = tid return tid raise AssertionError("could not determine the thread's id") def raise_exc(self, exctype): """raises the given exception type in the context of this thread""" _async_raise(self._get_my_tid(), exctype) def terminate(self): """raises SystemExit in the context of the given thread, which should cause the thread to exit silently (unless caught)""" self.raise_exc(SystemExit)
使用例子:
>>> import time >>> from thread2 import Thread >>> >>> def f(): ... try: ... while True: ... time.sleep(0.1) ... finally: ... print "outta here" ... >>> t = Thread(target = f) >>> t.start() >>> t.isAlive() True >>> t.terminate() >>> t.join() outta here >>> t.isAlive() False
試了一下,很不錯,只是在要kill的線程中如果有time.sleep()時,好像工作不正常,沒有找出真正的原因是什么。已經(jīng)是很強大了。哈哈。
相關(guān)文章
Python的反射函數(shù)與內(nèi)省工具深入解析
這篇文章主要為大家介紹了Python的反射函數(shù)與內(nèi)省工具深入解析,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2023-06-06由Python運算π的值深入Python中科學計算的實現(xiàn)
這篇文章主要介紹了由Python運算π的值深入Python中科學計算的實現(xiàn),由簡單的計算發(fā)散出各種算法的講解,需要的朋友可以參考下2015-04-04在Python的Flask框架中驗證注冊用戶的Email的方法
這篇文章主要介紹了在Python的Flask框架中驗證注冊用戶的Email的方法,包括非常詳細的測試過程,極力推薦!需要的朋友可以參考下2015-09-09Python連接SQLite數(shù)據(jù)庫操作實戰(zhàn)指南從入門到精通
在Python中使用SQLite進行數(shù)據(jù)庫操作時,我們將深入研究SQLite數(shù)據(jù)庫的創(chuàng)建、表格管理、數(shù)據(jù)插入、查詢、更新和刪除等關(guān)鍵主題,幫助你全面了解如何使用SQLite進行數(shù)據(jù)庫操作2023-11-11