MySQL数据库秒杀

import redis
import pymysql

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

def seckill(product_id):
    stock = r.get(f'stock:{product_id}')
    if stock and int(stock) > 0:
        r.decr(f'stock:{product_id}')
        # 更新MySQL中的库存
        conn = pymysql.connect(host='localhost', user='root', password='password', db='test')
        cursor = conn.cursor()
        cursor.execute("UPDATE products SET stock = stock - 1 WHERE id = %s", (product_id,))
        conn.commit()
        cursor.close()
        conn.close()
        return True
    return False

import pika

connection = pika.BlockingConnection(pika.ConnectionParameters('localhost'))
channel = connection.channel()

channel.queue_declare(queue='seckill_queue')

def callback(ch, method, properties, body):
    product_id = body.decode()
    # 处理秒杀逻辑
    seckill(product_id)

channel.basic_consume(queue='seckill_queue', on_message_callback=callback, auto_ack=True)
channel.start_consuming()

START TRANSACTION;
UPDATE products SET stock = stock - 1 WHERE id = 1 AND stock > 0;
SELECT ROW_COUNT() INTO @result;
IF @result = 0 THEN
    ROLLBACK;
ELSE
    COMMIT;
END IF;