首页 > 编程 > Python > 正文

Python实现线程池代码分享

2019-11-25 17:16:27
字体:
来源:转载
供稿:网友

原理:建立一个任务队列,然多个线程都从这个任务队列中取出任务然后执行,当然任务队列要加锁,详细请看代码

import threadingimport timeimport signalimport os class task_info(object):  def __init__(self):    self.func = None    self.parm0 = None    self.parm1 = None    self.parm2 = None   class task_list(object):  def __init__(self):    self.tl = []    self.mutex = threading.Lock()    self.sem = threading.Semaphore(0)     def append(self, ti):    self.mutex.acquire()    self.tl.append(ti)    self.mutex.release()    self.sem.release()     def fetch(self):    self.sem.acquire()    self.mutex.acquire()    ti = self.tl.pop(0)        self.mutex.release()    return ti   class thrd(threading.Thread):  def __init__(self, tl):    threading.Thread.__init__(self)    self.tl = tl     def run(self):    while True:      tsk = self.tl.fetch()      tsk.func(tsk.parm0, tsk.parm1, tsk.parm2)   class thrd_pool(object):  def __init__(self, thd_count, tl):    self.thds = []         for i in range(thd_count):      self.thds.append(thrd(tl))     def run(self):    for thd in self.thds:      thd.start()              def func(parm0=None, parm1=None, parm2=None):  print 'count:%s, thrd_name:%s'%(str(parm0), threading.currentThread().getName())   def cleanup(signo, stkframe):  print ('Oops! Got signal %s', signo)      os._exit(0)   if __name__ == '__main__':     signal.signal(signal.SIGINT, cleanup)  signal.signal(signal.SIGQUIT, cleanup)  signal.signal(signal.SIGTERM, cleanup)     tl = task_list()  tp = thrd_pool(6, tl)  tp.run()     count = 0  while True:         ti = task_info()    ti.parm0 = count    ti.func = func    tl.append(ti)    count += 1         time.sleep(2)  pass

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表