python中socket修改ttl_用Python编写TTL装饰器

我正在尝试在python中编写TTL装饰器.

基本上我给它引发一个异常,如果函数不

在选定的时间回答.

from thread2 import Thread

""" A TTL decorator. """

class Worker(Thread):

def __init__(self, q, f, args, kvargs):

Thread.__init__(self)

self.q = q

self.f = f

self.args = args

self.kvargs = kvargs

def run(self,):

try:

res = (True, self.f(*self.args, **self.kvargs))

self.q.put(res)

except Exception, e:

self.q.put((False, e))

class Referee(Thread):

def __init__(self,q, ttl,exception_factory):

Thread.__init__(self)

self.exception_factory=exception_factory

self.q=q

self.ttl=ttl

def run(self):

time.sleep(self.ttl)

res = (False, self.exception_factory())

self.q.put(res)

def raise_if_too_long(ttl, exception_factory=lambda :RuntimeError("Timeout")):

def raise_if_too_long_aux(f):

def ritl(*args,**kvargs):

q = Queue.Queue(2)

referee = Referee(q, ttl, exception_factory)

worker = Worker(q,f,args,kvargs)

worker.start()

referee.start()

(valid, res)= q.get(1)

q.task_done()

referee.terminate()

worker.terminate()

if valid:

return res

else:

raise res

return ritl

return raise_if_too_long_aux

但是,我得到了一些非常糟糕的结果.

似乎有时函数还可以返回

在达到TTL且出现错误之前,装饰器不会返回

加薪.

您在此代码中看到错误吗？

是否有在Python中使用TTL编写函数的通用方法/库？