Python之进程锁

背景:多个进程操作同一个资源,会引发数据安全问题

# 模拟买票过程
from multiprocessing import Process
import time
import json

# 首先创建一个余票文件
# dic = {'ticket':1}
# f = open('ticketDB','w')
# json.dump(dic,f)
# f.close()

# 定义一个买票类
class BuyTicket:
    def __init__(self,ticketDB):
        self.__ticketDB=ticketDB
    # 定义一个查看余票的方法
    def showTicket(self,name):
        # 查看票据有网络延迟
        time.sleep(0.2)
        tickets = self.getTicket()
        print("%s查看余票数:"%(name),tickets['ticket'])
    # 定义一个获取余票的方法
    def getTicket(self):
        ticketNum = json.load(open(self.__ticketDB))
        return ticketNum
    # 定义一个买票的方法
    def buyTicket(self,name):
        # 买票要先获取余票数,也存在网络延迟
        time.sleep(0.2)
        tickets = self.getTicket()
        if tickets['ticket'] > 0:
            tickets['ticket'] = tickets['ticket'] - 1
            # 将买完后的余票数写回到数据库中,存在网络延迟
            time.sleep(0.2)
            f = open(self.__ticketDB,'w')
            json.dump(tickets,f)
            f.close()
            print("%s买到票了"%(name))
        else:
            print("%s没有买到票"%(name))
# 开始模拟10个人买票
bt = BuyTicket('ticketDB')
if __name__ == '__main__':
    # 10个人来查看余票
    for i in range(1,11):
        p = Process(target=bt.showTicket,args=(i,))
        p.start()
    # 10个人抢票
    for i in range(1,11):
        p = Process(target=bt.buyTicket,args=(i,))
        p.start()
'''
输出结果:
1查看余票数: 1
2查看余票数: 1
3查看余票数: 1
4查看余票数: 1
5查看余票数: 1
6查看余票数: 1
7查看余票数: 1
8查看余票数: 1
9查看余票数: 1
10查看余票数: 1
1买到票了
7没有买到票
8没有买到票
2买到票了
3买到票了
9没有买到票
4买到票了
10没有买到票
5买到票了
6买到票了
'''

由以上可以看出:余票只有一张,但确有多个人买到了票,即出现了数据安全问题

引入进程锁,可以解决这个问题

from multiprocessing import Process
from multiprocessing import Lock
import time
import json

# 首先创建一个余票文件
dic = {'ticket':1}
f = open('ticketDB','w')
json.dump(dic,f)
f.close()

# 定义一个买票类
class BuyTicket:
    def __init__(self,ticketDB):
        # 定义一个锁
        self.__lock = Lock()
        self.__ticketDB=ticketDB
    # 定义一个查看余票的方法
    def showTicket(self,name):
        # 查看票据有网络延迟
        time.sleep(0.2)
        tickets = self.getTicket()
        print("%s查看余票数:"%(name),tickets['ticket'])
    # 定义一个获取余票的方法
    def getTicket(self):
        ticketNum = json.load(open(self.__ticketDB))
        return ticketNum
    # 定义一个买票的方法
    def buyTicket(self,name):
        # 买票要先获取余票数,也存在网络延迟
        time.sleep(0.2)
        # 在网络延迟后,获取到票,到买完回写到票据中心的过程加锁,其它进程在这中间不可访问票据中心的数据
        self.__lock.acquire()
        tickets = self.getTicket()
        if tickets['ticket'] > 0:
            tickets['ticket'] = tickets['ticket'] - 1
            time.sleep(0.2)
            f = open(self.__ticketDB,'w')
            json.dump(tickets,f)
            f.close()
            print("%s买到票了"%(name))
        else:
            print("%s没有买到票"%(name))
        self.__lock.release()
# 开始模拟10个人买票
bt = BuyTicket('ticketDB')
if __name__ == '__main__':
    # 10个人来查看余票
    for i in range(1,11):
        p = Process(target=bt.showTicket,args=(i,))
        p.start()
    # 10个人抢票
    for i in range(1,11):
        p = Process(target=bt.buyTicket,args=(i,))
        p.start()
'''
2查看余票数: 1
1查看余票数: 1
3查看余票数: 1
4查看余票数: 1
5查看余票数: 1
6查看余票数: 1
7查看余票数: 1
8查看余票数: 1
9查看余票数: 1
10查看余票数: 1
1买到票了
2没有买到票
3没有买到票
4没有买到票
5没有买到票
6没有买到票
7没有买到票
8没有买到票
9没有买到票
10没有买到票
'''

加锁要加在合适的地方,否则回造成死锁问题,锁加错导致死锁的过程

from multiprocessing import Process
from multiprocessing import Lock
import time
import json

# 首先创建一个余票文件
dic = {'ticket':1}
f = open('ticketDB','w')
json.dump(dic,f)
f.close()

# 定义一个买票类
class BuyTicket:
    def __init__(self,ticketDB):
        # 定义一个锁
        self.__lock = Lock()
        self.__ticketDB=ticketDB
    # 定义一个查看余票的方法
    def showTicket(self,name):
        # 查看票据有网络延迟
        time.sleep(0.2)
        tickets = self.getTicket()
        print("%s查看余票数:"%(name),tickets['ticket'])
    # 定义一个获取余票的方法
    def getTicket(self):
        ticketNum = json.load(open(self.__ticketDB))
        return ticketNum
    # 定义一个买票的方法
    def buyTicket(self,name):
        # 买票要先获取余票数,也存在网络延迟
        time.sleep(0.2)
        # 在网络延迟后,获取到票,到买完回写到票据中心的过程加锁,其它进程在这中间不可访问票据中心的数据
        self.__lock.acquire()
        tickets = self.getTicket()
        if tickets['ticket'] > 0:
            tickets['ticket'] = tickets['ticket'] - 1
            time.sleep(0.2)
            f = open(self.__ticketDB,'w')
            json.dump(tickets,f)
            f.close()
            self.__lock.release() # 最初锁加在了此处
            print("%s买到票了"%(name))
        else:
            print("%s没有买到票"%(name))
# 开始模拟10个人买票
bt = BuyTicket('ticketDB')
if __name__ == '__main__':
    # 10个人来查看余票
    for i in range(1,11):
        p = Process(target=bt.showTicket,args=(i,))
        p.start()
    # 10个人抢票
    for i in range(1,11):
        p = Process(target=bt.buyTicket,args=(i,))
        p.start()
'''
2查看余票数: 1
1查看余票数: 1
3查看余票数: 1
4查看余票数: 1
5查看余票数: 1
6查看余票数: 1
7查看余票数: 1
8查看余票数: 1
9查看余票数: 1
10查看余票数: 1
1买到票了
2没有买到票
程序到此卡住了
'''

当第二个人进来先申请了一个锁,但是发现票据余量为0,直接打印没有买到票则推出了,锁没有释放,因此造成了后续进程无法进程

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值