Python的线程池

#!/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()