MySQL 数据库运维限时秒杀

import redis
import pymysql

# 连接 Redis 和 MySQL
redis_client = redis.StrictRedis(host='localhost', port=6379, db=0)
mysql_conn = pymysql.connect(host='localhost', user='root', password='password', db='test')

def seckill(product_id, user_id):
    # 尝试从 Redis 中获取库存
    stock = redis_client.get(f'stock:{product_id}')
    if stock is None or int(stock) <= 0:
        return "秒杀失败，库存不足"

    # 减少 Redis 中的库存
    redis_client.decr(f'stock:{product_id}')

    try:
        with mysql_conn.cursor() as cursor:
            # 使用事务确保数据一致性
            mysql_conn.begin()
            sql = "UPDATE products SET stock = stock - 1 WHERE id = %s AND stock > 0"
            cursor.execute(sql, (product_id,))
            if cursor.rowcount == 0:
                mysql_conn.rollback()
                redis_client.incr(f'stock:{product_id}')  # 回滚 Redis 库存
                return "秒杀失败，库存不足"

            sql = "INSERT INTO orders (product_id, user_id) VALUES (%s, %s)"
            cursor.execute(sql, (product_id, user_id))
            mysql_conn.commit()
    except Exception as e:
        mysql_conn.rollback()
        redis_client.incr(f'stock:{product_id}')  # 回滚 Redis 库存
        return f"秒杀失败，数据库错误: {str(e)}"

    return "秒杀成功"

# 示例调用
result = seckill(1, 1001)
print(result)