企业身份数据连接限时秒杀

from flask import Flask, request, jsonify
import redis
import time

app = Flask(__name__)
redis_client = redis.StrictRedis(host='localhost', port=6379, db=0)

PRODUCT_ID = 'product_123'
SECKILL_STOCK_KEY = f'seckill_stock:{PRODUCT_ID}'
SECKILL_USERS_KEY = f'seckill_users:{PRODUCT_ID}'

@app.route('/seckill', methods=['POST'])
def seckill():
    user_id = request.json.get('user_id')
    if not user_id:
        return jsonify({'error': 'User ID is required'}), 400

    # 使用Redis原子操作减少库存
    stock = redis_client.decr(SECKILL_STOCK_KEY)
    if stock < 0:
        redis_client.incr(SECKILL_STOCK_KEY)  # 库存回滚
        return jsonify({'error': 'Out of stock'}), 400

    # 记录成功秒杀的用户
    redis_client.sadd(SECKILL_USERS_KEY, user_id)

    return jsonify({'message': 'Seckill successful'}), 200

if __name__ == '__main__':
    app.run(debug=True)