python 并发编程 多进程 模拟抢票
抢票是并发执行
多个进程可以访问同一个文件
多个进程共享同一文件,我们可以把文件当数据库,用多个进程模拟多个人执行抢票任务
db.txt
{"count": 1}
并发运行,效率高,但竞争写同一文件,数据写入错乱,只有一张票,都卖成功给了10个人
#文件db.txt的内容为:{"count":1} #注意一定要用双引号,不然json无法识别 from multiprocessing import Process import time import json class Foo(object): def search(self, name): with open("db.txt", "r") as f_read: dic = json.load(f_read) time.sleep(1) # 模拟读数据的网络延迟 print("<%s>用户 查看剩余票数为 [%s]" % (name, dic["count"])) def get(self, name): with open("db.txt", "r") as f_read: dic = json.load(f_read) if dic["count"] > 0: dic["count"] -= 1 time.sleep(1) # 模拟写数据的网络延迟 with open("db.txt", "w") as f_write: json.dump(dic, f_write) print("<%s> 购票成功" % name) print("剩余票数为 [%s]" % dic["count"]) else: print("没票了,抢光了") def task(self, name): self.search(name) self.get(name) if __name__ == "__main__": obj = Foo() for i in range(1,11): # 模拟并发10个客户端抢票 p = Process(target=obj.task, args=("路人%s" % i,)) p.start()
总结:程序出现数据写入错乱
大家都查到票为1,都购票成功
<路人1>用户 查看剩余票数为 [1] <路人2>用户 查看剩余票数为 [1] <路人3>用户 查看剩余票数为 [1] <路人4>用户 查看剩余票数为 [1] <路人5>用户 查看剩余票数为 [1] <路人6>用户 查看剩余票数为 [1] <路人7>用户 查看剩余票数为 [1] <路人8>用户 查看剩余票数为 [1] <路人9>用户 查看剩余票数为 [1] <路人10>用户 查看剩余票数为 [1] <路人1> 购票成功 剩余票数为 [0] <路人2> 购票成功 剩余票数为 [0] <路人3> 购票成功 剩余票数为 [0] <路人4> 购票成功 剩余票数为 [0] <路人5> 购票成功 剩余票数为 [0] <路人6> 购票成功 剩余票数为 [0] <路人7> 购票成功 剩余票数为 [0] <路人8> 购票成功 剩余票数为 [0] <路人9> 购票成功 剩余票数为 [0] <路人10> 购票成功 剩余票数为 [0] 总结程序出现数据写入错乱
加锁处理:购票行为由并发变成了串行,牺牲了运行效率,但保证了数据安全
购票功能不应该并发执行,查票应该是并发执行的
查票准不准确不重要,有可能这张票就被别人买走
一个人写完以后,让另外一个人基于上一个人写的结果,再做购票操作
#把文件db.txt的内容重置为:{"count":1} from multiprocessing import Process from multiprocessing import Lock import time import json class Foo(object): def search(self, name): with open("db.txt", "r") as f_read: dic = json.load(f_read) time.sleep(1) # 模拟读数据的网络延迟 print("<%s>用户 查看剩余票数为 [%s]" % (name, dic["count"])) def get(self, name): with open("db.txt", "r") as f_read: dic = json.load(f_read) if dic["count"] > 0: dic["count"] -= 1 time.sleep(1) # 模拟写数据的网络延迟 with open("db.txt", "w") as f_write: json.dump(dic, f_write) print("<%s> 购票成功" % name) print("剩余票数为 [%s]" % dic["count"]) else: print("没票了,抢光了") def task(self, name, mutex): self.search(name) mutex.acquire() self.get(name) mutex.release() if __name__ == "__main__": mutex = Lock() obj = Foo() for i in range(1,11): # 模拟并发10个客户端抢票 p = Process(target=obj.task, args=("路人%s" % i, mutex)) p.start()
执行结果
<路人2>用户 查看剩余票数为 [1] <路人3>用户 查看剩余票数为 [1] <路人1>用户 查看剩余票数为 [1] <路人4>用户 查看剩余票数为 [1] <路人5>用户 查看剩余票数为 [1] <路人7>用户 查看剩余票数为 [1] <路人6>用户 查看剩余票数为 [1] <路人8>用户 查看剩余票数为 [1] <路人9>用户 查看剩余票数为 [1] <路人10>用户 查看剩余票数为 [1] <路人2> 购票成功 剩余票数为 [0] 没票了,抢光了 没票了,抢光了 没票了,抢光了 没票了,抢光了 没票了,抢光了 没票了,抢光了 没票了,抢光了 没票了,抢光了 没票了,抢光了
with lock
相当于lock.acquire(),执行完自代码块自动执行lock.release()
from multiprocessing import Process from multiprocessing import Lock import time import json class Foo(object): def search(self, name): with open("db.txt", "r") as f_read: dic = json.load(f_read) time.sleep(1) # 模拟读数据的网络延迟 print("<%s>用户 查看剩余票数为 [%s]" % (name, dic["count"])) def get(self, name): with open("db.txt", "r") as f_read: dic = json.load(f_read) if dic["count"] > 0: dic["count"] -= 1 time.sleep(1) # 模拟写数据的网络延迟 with open("db.txt", "w") as f_write: json.dump(dic, f_write) print("<%s> 购票成功" % name) print("剩余票数为 [%s]" % dic["count"]) else: print("没票了,抢光了") def task(self, name, mutex): self.search(name) with mutex: # 相当于lock.acquire(),执行完自代码块自动执行lock.release() self.get(name) if __name__ == "__main__": mutex = Lock() obj = Foo() for i in range(1,11): # 模拟并发10个客户端抢票 p = Process(target=obj.task, args=("路人%s" % i, mutex)) p.start()