秒杀系统
作者:互联网
秒杀系统(多进程模拟)
'''
秒杀程序
'''
import os
import arrow
import redis
from multiprocessing import Pool
from redlock.lock import RedLock
HOTKEY = 'num' # 设置秒杀数量
r = redis.Redis(host='localhost', port=6379)
# 秒杀函数
def miaosha():
name = os.getpid() # 获取pid 主要是知道哪个pid执行任务
v = r.get(HOTKEY)
if int(v) > 0:
print (name, '库存',int(v))
r.decr(HOTKEY) # 自减1
else:
print(name,'商品库存为0')
# 不加锁,每5秒中执行一次秒杀函数
def run_without_lock(name):
while True:
if arrow.now().second % 5 == 0:
miaosha() # 秒杀函数
return
# 使用redlock 模块
rlock = RedLock('miaosha',[{'host': 'localhost','port': 6379,'db': 8,}])
# 使用 锁
def run_with_redis_lock(name):
while True:
if arrow.now().second % 5 == 0:
# 获取锁
if rlock.acquire():
# 秒杀函数
miaosha()
# 释放锁
rlock.release()
return
if __name__ == '__main__':
p = Pool(80) # 进程池
r.set(HOTKEY, 10) # 设置10件商品
for i in range(80):
# p.apply_async(run_without_lock, args=(i,)) # 异步非阻塞
p.apply_async(run_with_redis_lock, args=(i,))
p.close()
p.join()
print('执行完毕')
标签:__,name,lock,系统,秒杀,HOTKEY,import 来源: https://www.cnblogs.com/wuxiaoshi/p/12179241.html