最近在項目中遇到這一需求：
我需要一個函數(shù)工作，比如遠程連接一個端口，遠程讀取文件等，但是我給的時間有限，比如，4秒鐘如果你還沒有讀取完成或者連接成功，我就不等了，很可能對方已經(jīng)宕機或者拒絕了。這樣可以批量做一些事情而不需要一直等，浪費時間。
結合我的需求，我想到這種辦法：
1、在主進程執(zhí)行，調用一個進程執(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ù)式用來等待一個線程結束的，如果這個函數(shù)沒有結束的話，那么，就會阻塞當前運行的程序。關鍵是，這個參數(shù)有一個可選參數(shù)：join([timeout]):? 阻塞當前上下文環(huán)境的線程，直到調用此方法的線程終止或到達指定的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:
#//添加你要做的事情，如果成功了就設置一下
self.thread_stop變量。
[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)是很強大了。哈哈。
更多文章、技術交流、商務合作、聯(lián)系博主
微信掃碼或搜索：z360901061
微信掃一掃加我為好友
QQ號聯(lián)系： 360901061
您的支持是博主寫作最大的動力，如果您喜歡我的文章，感覺我的文章對您有幫助，請用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧，狠狠點擊下面給點支持吧，站長非常感激您！手機微信長按不能支付解決辦法：請將微信支付二維碼保存到相冊，切換到微信，然后點擊微信右上角掃一掃功能，選擇支付二維碼完成支付。
【本文對您有幫助就好】元
