Python的執行緒池

昀溪發表於2018-12-15
#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
concurrent 用於執行緒池和程式池程式設計而且更加容易,在Python3.2中才有。
"""

import sys
from concurrent.futures import ThreadPoolExecutor, as_completed, wait
from multiprocessing import Manager

Manager().dict()

"""
執行緒池
主執行緒可以獲取某一個執行緒的狀態或者某一個任務的狀態以及返回值
當一個執行緒完成的時候讓主執行緒立即知道
"""

import time
def task(num):
    time.sleep(num)
    print("Thread", num)
    return num


def NormalUsage():
    # 例項化執行緒池並設定執行緒池大小
    executor = ThreadPoolExecutor(max_workers=5)
    # 透過submit提交執行的函式到執行緒池,並返回futures的類物件,這裡是立即返回非阻塞
    task1 = executor.submit(task, (1))
    task2 = executor.submit(task, (2))
    # done()方法用於判定某個任務十分完成
    print(task1.done())
    # result() 方法可以獲取返回值,阻塞的方法
    print(task1.result())
    # 關閉執行緒池
    executor.shutdown(wait=True)


def BBB():
    executor = ThreadPoolExecutor(max_workers=5)
    # 如何已經成功的任務的返回
    tList = []
    for num in range(10):
        tList.append(executor.submit(task, (num)))
    # 主要就是 as_completed()方法,這是一個生成器,傳遞列表進去,這個列表就是包含執行緒的列表
    for future in as_completed(tList):
        data = future.result()
        print(data)
    executor.shutdown(wait=True)


def CCC():
    # 例項化執行緒池並設定執行緒池大小
    executor = ThreadPoolExecutor(max_workers=5)
    # 透過submit提交執行的函式到執行緒池,並返回futures的類物件,這裡是立即返回非阻塞
    task1 = executor.submit(task, (1))
    task2 = executor.submit(task, (2))
    # 等待執行緒完成後再執行。return_when 是什麼時候返回,預設是所有執行緒執行完畢,你可以設定 FIRST_COMPLETED 第一個執行緒完成就不阻塞了
    wait([task1, task2], return_when="ALL_COMPLETED")
    print("main")
    executor.shutdown(wait=True)


if __name__ == '__main__':
    CCC()

  

相關文章