萬盛學電腦網

 萬盛學電腦網 >> 網絡編程 >> 編程語言綜合 >> python實現線程池的方法

python實現線程池的方法

   本文實例講述了python實現線程池的方法。分享給大家供大家參考。具體如下:

  原理:建立一個任務隊列,然多個線程都從這個任務隊列中取出任務然後執行,當然任務隊列要加鎖,詳細請看代碼

  文件名:thrd_pool.py 系統環境:ubuntu linux & python2.6

  ?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 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

  執行結果:

  ?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 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程序設計有所幫助。

copyright © 萬盛學電腦網 all rights reserved