【python小技巧】并发中的'lock'---模拟买票的数据延迟问题

场景：我们在买票过程中，通常第一步先查找还有多少余票，然后点击购买，购买后，余票的数量-1,实际情况中是会有多人同时买票，其实就是一个并发的过程，但这个过程中，因为并发的子进程在读取数据，修改回填的过程中，可能会因为出现的延迟，造成数据的错误，比如这个例子，我们可以模拟20个人同时买票，设计1张余票，代码如下：

from multiprocessing import Process
from multiprocessing import Lock
import time
import json
def buy(i,lock):
    with open('code') as f:
        dic = json.load(f)
        time.sleep(0.1)
        if dic['dic']>0:
            print('\33[32m%s买到票了\33[0m'%i)
            dic['dic']-=1
        else:
            print('没票了')
        time.sleep(0.1)
        with open('code', 'w', encoding='utf-8') as f:
            json.dump(dic, f)
if __name__=='__main__':
    lock = Lock()
    for i in range(20):
        p=Process(target=buy,args=(i,lock))
        p.start()

结果如下图所示：

实际我们在文件中只写了1张余票，但出现16个人同时买到票的情况，为了避免这种情况，我们引入了‘锁’的概念，当某个子进程进行时，其他子进程会发生阻塞(lock.acquire),相当于锁的过程，当运行到lock.release()，才会释放新的钥匙，相当于还钥匙的过程，这样就解决了延迟的问题，具体的代码如下：

from multiprocessing import Process
from multiprocessing import Lock
import time
import json
def buy(i,lock):
    with open('code') as f:
        lock.acquire()
        dic = json.load(f)
        time.sleep(0.1)
        if dic['dic']>0:
            print('\33[32m%s买到票了\33[0m'%i)
            dic['dic']-=1
        else:
            print('没票了')
        time.sleep(0.1)
        with open('code', 'w', encoding='utf-8') as f:
            json.dump(dic, f)
        lock.release()
if __name__=='__main__':
    lock = Lock()
    for i in range(20):
        p=Process(target=buy,args=(i,lock))
        p.start()

具体的代码有3步
1.实例化对象lock = Lock()
2.锁的过程lock.acquire()
3.还钥匙的过程lock.release()
这样就可以完美解决上面的问题