一站式企业社区秒杀

from flask import Flask, request, jsonify
import redis
import pymysql
import pika

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

# 模拟数据库连接
db = pymysql.connect(host='localhost', user='root', password='password', database='test')

@app.route('/seckill', methods=['POST'])
def seckill():
    product_id = request.json.get('product_id')
    user_id = request.json.get('user_id')

    # 使用Redis检查库存并锁定
    stock_key = f'stock:{product_id}'
    stock = redis_client.decr(stock_key)
    if stock < 0:
        redis_client.incr(stock_key)  # 库存不足，回滚
        return jsonify({'status': 'fail', 'message': 'Out of stock'})

    # 发送消息到队列处理订单
    connection = pika.BlockingConnection(pika.ConnectionParameters('localhost'))
    channel = connection.channel()
    channel.queue_declare(queue='order_queue')
    channel.basic_publish(exchange='', routing_key='order_queue', body=f'{product_id},{user_id}')
    connection.close()

    return jsonify({'status': 'success', 'message': 'Order placed successfully'})

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