这几天在实践的时候,有一个问题一直在困扰我,比如timer,定时器,python没有这么好的机制,需要自己实现。这个网上也是有一些。但是我自己的需求可能并不一样。
需求:
我需要一个函数工作,比如远程连接一个端口,远程读取文件等,但是我给的时间有限,比如,4秒钟如果你还没有读取完成或者连接成功,我就不等了,很可能对方已经宕机或者拒绝了。这样可以批量做一些事情而不需要一直等,浪费时间。
定时器是最好的,我觉得这个需求,一旦定时过了,就将这个读取函数直接kill。但是这出现了问题。
第一,怎么定时?
第二,时间到了,怎么kill函数
第三,定时函数和kill函数怎么判断该kill了。
写一个多线程,一个线程专门用来计时间,一个用来执行函数,一旦时间到了,就杀死执行函数的线程。我在多线程,python很弱,似乎没有这样的比较简单的一个接口就调用的。不像windows编程。那么,该怎办?
结合我的需求,我这么做:
1、在主进程执行,调用一个进程执行函数,然后主进程sleep,等时间到了,就kill 执行函数的进程。
测试一个例子:
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()
这样就定了4秒钟了。
问题还有一个,如果函数读取远程文件成功了,那么需要立即返回,这个功能,上面的程序做不到,只能等4秒,那该怎么做呢??
我们遗忘了一个函数,join函数,这个函数式用来等待一个线程结束的,如果这个函数没有结束的话,那么,就会阻塞当前运行的程序。
关键是,这个参数有一个可选参数:
join([timeout]): 阻塞当前上下文环境的线程,直到调用此方法的线程终止或到达指定的timeout(可选参数)。
那么我们的思路就是,在循环中一直调用一个线程做任务,然后join(3),就会等3秒,3秒之后,我们判断这个进程还存不存在,有函数isAlive()
如果存在,说明已经超时了。我们可以调用之前写的线程的结束函数主动结束线程。如果线程不存在,说明任务已经完成返回了。那么我们就可以对成功做了任务的进行下一步操作。
具体代码如下:
#!/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:
#//添加你要做的事情,如果成功了就设置一下<span style="font-family: Arial, Helvetica, sans-serif;">self.thread_stop变量。</span>
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 listif __name__ == '__main__': eachline = ['1.1.1.1','222.73.5.54'] test(eachline)
多线程编程:
http://blog.csdn.net/bagboy_taobao_com/article/details/8573478这个博文写的还是比较详细。
具体想要看join函数等,我想你能找到更多的资料