#!/usr/local/bin/python3
"""
多線程程序如果沒有競爭資源的場景那么通常會比較簡單
臨界資源 - 被多個線程競爭的資源
當多個線程競爭臨界資源的時候如果缺乏必要的保護措施就會導致數(shù)據(jù)錯亂
"""
import time
import threading

from concurrent.futures import ThreadPoolExecutor


class Account(object):
    """銀行賬戶"""

    def __init__(self):
        self.balance = 0.0
        self.lock = threading.Lock()

    def deposit(self, money):
        # 通過鎖保護臨界資源
        # 可以寫try-finally也可以使用上下文語法
        # self.lock.acquire()
        # try:
        #     pass
        # finally:
        #     self.lock.release()
        with self.lock:
            new_balance = self.balance + money
            time.sleep(0.001)
            self.balance = new_balance


def add_money(account, money):
    account.deposit(money)


# 自定義線程類
class AddMoneyThread(threading.Thread):

    def __init__(self, account, money):
        self.account = account
        self.money = money
        # 自定義線程的初始化方法中必須調(diào)用父類的初始化方法
        super().__init__()

    # run方法是線程啟動之后要執(zhí)行的回調(diào)方法（鉤子函數(shù)）
    # 所以啟動線程不能夠直接調(diào)用run方法而是通過start方法啟動線程
    # 什么時候需要使用回調(diào)式編程？
    # 你知道要做什么但不知道什么時候會做這件事情
    def run(self):
        # 線程啟動之后要執(zhí)行的操作
        pass

def main():
    account = Account()
    # 創(chuàng)建線程池
    pool = ThreadPoolExecutor(max_workers=10)
    futures = []
    for _ in range(100):
        # 創(chuàng)建線程的第1種方式
        # threading.Thread(
        #     target=add_money, args=(account, 1)
        # ).start()
        # 創(chuàng)建線程的第2種方式
        # AddMoneyThread(account, 1).start()
        # 創(chuàng)建線程的第3種方式
        # 調(diào)用線程池中的線程來執(zhí)行特定的任務(wù)
        future = pool.submit(add_money, account, 1)
        futures.append(future)
    # 關(guān)閉線程池
    pool.shutdown()
    for future in futures:
        future.result()
    print(account.balance)


if __name__ == '__main__':
    main()