67 lines
1.7 KiB
Python
67 lines
1.7 KiB
Python
"""
|
||
多个线程竞争一个资源 - 保护临界资源 - 锁（Lock/RLock）
|
||
多个线程竞争多个资源（线程数>资源数） - 信号量（Semaphore）
|
||
多个线程的调度 - 暂停线程执行/唤醒等待中的线程 - Condition
|
||
"""
|
||
from concurrent.futures import ThreadPoolExecutor
|
||
from random import randint
|
||
from time import sleep
|
||
|
||
import threading
|
||
|
||
|
||
class Account():
|
||
"""银行账户"""
|
||
|
||
def __init__(self, balance=0):
|
||
self.balance = balance
|
||
lock = threading.Lock()
|
||
self.condition = threading.Condition(lock)
|
||
|
||
def withdraw(self, money):
|
||
"""取钱"""
|
||
with self.condition:
|
||
while money > self.balance:
|
||
self.condition.wait()
|
||
new_balance = self.balance - money
|
||
sleep(0.001)
|
||
self.balance = new_balance
|
||
|
||
def deposit(self, money):
|
||
"""存钱"""
|
||
with self.condition:
|
||
new_balance = self.balance + money
|
||
sleep(0.001)
|
||
self.balance = new_balance
|
||
self.condition.notify_all()
|
||
|
||
|
||
def add_money(account):
|
||
while True:
|
||
money = randint(5, 10)
|
||
account.deposit(money)
|
||
print(threading.current_thread().name,
|
||
':', money, '====>', account.balance)
|
||
sleep(0.5)
|
||
|
||
|
||
def sub_money(account):
|
||
while True:
|
||
money = randint(10, 30)
|
||
account.withdraw(money)
|
||
print(threading.current_thread().name,
|
||
':', money, '<====', account.balance)
|
||
sleep(1)
|
||
|
||
|
||
def main():
|
||
account = Account()
|
||
with ThreadPoolExecutor(max_workers=10) as pool:
|
||
for _ in range(5):
|
||
pool.submit(add_money, account)
|
||
pool.submit(sub_money, account)
|
||
|
||
|
||
if __name__ == '__main__':
|
||
main()
|