大数据存储秒杀

import redis
import time

# 连接Redis服务器
r = redis.Redis(host='localhost', port=6379, db=0)

def seckill_product(product_id, user_id):
    # 使用Redis的setnx命令实现分布式锁
    lock_key = f"lock:product:{product_id}"
    if r.setnx(lock_key, "locked"):
        r.expire(lock_key, 10)  # 设置锁的过期时间，防止死锁
        try:
            # 检查库存
            stock_key = f"stock:product:{product_id}"
            stock = int(r.get(stock_key) or 0)
            if stock > 0:
                # 减少库存
                r.decr(stock_key)
                # 记录用户购买信息
                r.sadd(f"purchased:{product_id}", user_id)
                return True
            else:
                return False
        finally:
            r.delete(lock_key)  # 释放锁
    else:
        return False

# 模拟秒杀请求
start_time = time.time()
for i in range(1000):
    seckill_product("product_123", f"user_{i}")
end_time = time.time()

print(f"处理 {1000} 次秒杀请求耗时: {end_time - start_time} 秒")