from concurrent.futures import ProcessPoolExecutor
import time
# CPU密集型計算函數(shù)
def calculate_square(n):
    print(f"計算 {n} 的平方...")
    time.sleep(1)  # 模擬耗時計算
    return n * n
# 使用進(jìn)程池
with ProcessPoolExecutor(max_workers=4) as executor:
    # 提交任務(wù)到進(jìn)程池
    future1 = executor.submit(calculate_square, 5)
    future2 = executor.submit(calculate_square, 8)
    # 獲取任務(wù)結(jié)果
    print(f"5的平方 = {future1.result()}")
    print(f"8的平方 = {future2.result()}")

future = executor.submit(func, *args, **kwargs)

results = executor.map(func, iterable, timeout=None)

result = future.result()  # 阻塞直到結(jié)果返回

from concurrent.futures import as_completed
futures = [executor.submit(calculate_square, i) for i in range(1, 6)]
for future in as_completed(futures):
    print(f"結(jié)果: {future.result()}")

# 最多同時運(yùn)行2個進(jìn)程
with ProcessPoolExecutor(max_workers=2) as executor:
    results = list(executor.map(calculate_square, range(1, 5)))
    print(results)

future = executor.submit(calculate_square, 10)
if future.running():
    print("任務(wù)正在運(yùn)行...")
elif future.done():
    print("任務(wù)已完成!")

def result_callback(future):
    print(f"收到結(jié)果: {future.result()}")
with ProcessPoolExecutor() as executor:
    future = executor.submit(calculate_square, 15)
    future.add_done_callback(result_callback)

def divide(a, b):
    return a / b
try:
    future = executor.submit(divide, 10, 0)
    result = future.result()
except ZeroDivisionError as e:
    print(f"出現(xiàn)錯誤: {e}")

from PIL import Image
import os
from concurrent.futures import ProcessPoolExecutor
# 圖片處理函數(shù)
def process_image(image_path):
    try:
        img = Image.open(image_path)
        # 圖片處理操作
        img = img.resize((800, 600))
        img = img.convert('L')  # 轉(zhuǎn)為灰度圖
        # 保存處理后的圖片
        new_path = os.path.splitext(image_path)[0] + "_processed.jpg"
        img.save(new_path)
        return f"已處理: {image_path}"
    except Exception as e:
        return f"處理失敗: {image_path} - {str(e)}"
# 獲取圖片目錄中的所有圖片
image_dir = "images"
image_files = [os.path.join(image_dir, f) for f in os.listdir(image_dir) 
              if f.endswith(('.jpg', '.png'))]
# 使用進(jìn)程池處理
with ProcessPoolExecutor(max_workers=os.cpu_count()) as executor:
    # 提交所有任務(wù)
    futures = {executor.submit(process_image, img): img for img in image_files}
    # 獲取結(jié)果
    for future in as_completed(futures):
        result = future.result()
        print(result)

# 使用initializer預(yù)加載共享數(shù)據(jù)
def init_worker():
    global shared_data
    shared_data = load_big_data()
def process_item(item):
    return process(shared_data, item)
with ProcessPoolExecutor(initializer=init_worker) as executor:
    ...

# 減少小任務(wù)的數(shù)量
def process_chunk(chunk):
    return [calculate_square(n) for n in chunk]
chunks = [range(i, i+1000) for i in range(0, 10000, 1000)]
results = executor.map(process_chunk, chunks)

# 設(shè)置超時時間
try:
    result = future.result(timeout=60)  # 最多等待60秒
except TimeoutError:
    print("任務(wù)超時")

# 使用上下文管理器確保資源回收
with ProcessPoolExecutor() as executor:
    # 執(zhí)行代碼
# 離開with塊后自動關(guān)閉進(jìn)程池

from multiprocessing import Manager
def worker(shared_list, data):
    shared_list.append(process(data))
with Manager() as manager:
    shared_list = manager.list()
    with ProcessPoolExecutor() as executor:
        executor.map(worker, [shared_list]*len(data), data)
    print(list(shared_list))