本文实例讲述了python实现线程池的方法。分享给大家供大家参考。具体如下:
原理:建立一个任务队列,然多个线程都从这个任务队列中取出任务然后执行,当然任务队列要加锁,详细请看代码
文件名:thrd_pool.py 系统环境:ubuntu linux & python2.6
import threading
import time
import signal
import 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
执行方式:python thrd_pool.py
执行结果:
count:0, thrd_name:thread-1
count:1, thrd_name:thread-2
count:2, thrd_name:thread-3
count:3, thrd_name:thread-4
count:4, thrd_name:thread-5
count:5, thrd_name:thread-1
count:6, thrd_name:thread-6
count:7, thrd_name:thread-2
count:8, thrd_name:thread-3
count:9, thrd_name:thread-4
count:10, thrd_name:thread-5
count:11, thrd_name:thread-1
count:12, thrd_name:thread-6
count:13, thrd_name:thread-2
count:14, thrd_name:thread-3
(‘oops! got signal %s’, 15)
希望本文所述对大家的python程序设计有所帮助。