#!/usr/local/bin/python3
"""
多線程程序如果沒有競爭資源的場景那么通常會比較簡單
臨界資源 - 被多個線程競爭的資源
當多個線程競爭臨界資源的時候如果缺乏必要的保護措施就會導致數據錯亂
"""
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
        # 自定義線程的初始化方法中必須調用父類的初始化方法
        super().__init__()

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

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


if __name__ == '__main__':
    main()