时序数据存储秒杀

import redis
import time

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

def seckill(product_id, user_id):
    key = f"seckill:{product_id}"
    current_time = int(time.time())
    
    # 使用Lua脚本保证原子性操作
    script = """
    local stock = tonumber(redis.call('get', KEYS[1]))
    if stock and stock > 0 then
        redis.call('decr', KEYS[1])
        redis.call('zadd', 'users:' .. KEYS[1], ARGV[1], ARGV[2])
        return 1
    else
        return 0
    end
    """
    
    result = r.eval(script, 1, key, current_time, user_id)
    return result == 1

# 模拟秒杀请求
if seckill('product123', 'user456'):
    print("秒杀成功！")
else:
    print("秒杀失败，商品已售罄。")