ThreadPoolExecutor线程池介绍

ThreadPoolExecutor

from concurrent.futures import ThreadPoolExecutor
import time

def get_html(times):
    time.sleep(times)
    print('get page {} sucess'.format(times))
    return 'get done'

exector = ThreadPoolExecutor(max_workers=2)
#submit方法提交执行的函数到线程池中
task1 = exector.submit(get_html,(1))
task2 = exector.submit(get_html,(2))

#判断某个任务是否完成
print(task1.done())
#可以取消线程池中的某一个线程，线程submit之后会立即执行，已经执行的线程会取消失败（线程池大小为2，第三个submit的线程就可以取消）
print(task2.cancel())

time.sleep(4)
print(task1.done())

#判断某个任务的执行结果
print(task1.result())

------------------------------------------------------------------
False
True
get page 1 sucess
True
get done

获取已经成功的task的返回

as_completed将线程池中已经完成的task获取。

from concurrent.futures import ThreadPoolExecutor ,as_completed
import time
def get_html(times):
    time.sleep(times)
    print('get page {} sucess'.format(times))
    return times
exector = ThreadPoolExecutor(max_workers=2)
urls = [2,3,4,5]
all_task = [exector.submit(get_html,(url)) for url in urls]
for future in as_completed(all_task):
    data = future.result()
    print("get {} page".format(data))

---

from concurrent.futures import wait
可以设置线程等待