def close(self):
pass
+ @classmethod
+ def parseargs(cls, **args):
+ if len(args) > 0:
+ raise ValueError("unknown handler argument: " + next(iter(args)))
+ return {}
+
class freethread(handler):
- def __init__(self, **kw):
+ def __init__(self, *, max=None, timeout=None, **kw):
super().__init__(**kw)
self.current = set()
self.lk = threading.Lock()
+ self.tcond = threading.Condition(self.lk)
+ self.max = max
+ self.timeout = timeout
+
+ @classmethod
+ def parseargs(cls, *, max=None, abort=None, **args):
+ ret = super().parseargs(**args)
+ if max:
+ ret["max"] = int(max)
+ if abort:
+ ret["timeout"] = int(abort)
+ return ret
def handle(self, req):
- reqthread(target=self.run, args=[req]).start()
+ with self.lk:
+ if self.max is not None:
+ if self.timeout is not None:
+ now = start = time.time()
+ while len(self.current) >= self.max:
+ self.tcond.wait(start + self.timeout - now)
+ now = time.time()
+ if now - start > self.timeout:
+ os.abort()
+ else:
+ while len(self.current) >= self.max:
+ self.tcond.wait()
+ th = reqthread(target=self.run, args=[req])
+ th.start()
+ while th.is_alive() and th not in self.current:
+ self.tcond.wait()
def ckflush(self, req):
while len(req.buffer) > 0:
th = threading.current_thread()
with self.lk:
self.current.add(th)
+ self.tcond.notify_all()
try:
env = req.mkenv()
with perf.request(env) as reqevent:
finally:
with self.lk:
self.current.remove(th)
+ self.tcond.notify_all()
finally:
req.close()
if len(self.current) > 0:
th = next(iter(self.current))
else:
- th = None
+ return
th.join()
class threadpool(handler):
- def __init__(self, *, min=0, max=20, live=10, **kw):
+ def __init__(self, *, min=0, max=20, live=300, **kw):
super().__init__(**kw)
self.current = set()
self.free = set()
for i in range(self.min):
self.newthread()
+ @classmethod
+ def parseargs(cls, *, min=None, max=None, live=None, **args):
+ ret = super().parseargs(**args)
+ if min:
+ ret["min"] = int(min)
+ if max:
+ ret["max"] = int(max)
+ if live:
+ ret["live"] = int(live)
+ return ret
+
def newthread(self):
with self.lk:
th = reqthread(target=self.loop)
names = {"free": freethread,
"pool": threadpool}
+
+def parsehspec(spec):
+ if ":" not in spec:
+ return spec, {}
+ nm, spec = spec.split(":", 1)
+ args = {}
+ while spec:
+ if "," in spec:
+ part, spec = spec.split(",", 1)
+ else:
+ part, spec = spec, None
+ if "=" in part:
+ key, val = part.split("=", 1)
+ else:
+ key, val = part, ""
+ args[key] = val
+ return nm, args