数据库部署限时秒杀

import redis
import pymysql
from pymysql.cursors import DictCursor

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

# 连接MySQL
mysql_conn = pymysql.connect(host='localhost', user='root', password='password', db='秒杀', cursorclass=DictCursor)

def seckill_product(product_id, user_id):
    # 尝试从Redis获取库存
    stock = redis_client.get(f'stock:{product_id}')
    if not stock or int(stock) <= 0:
        return "库存不足"

    # 使用Redis事务确保原子性
    pipe = redis_client.pipeline()
    try:
        pipe.watch(f'stock:{product_id}')
        pipe.multi()
        pipe.decr(f'stock:{product_id}')
        pipe.execute()
    except redis.WatchError:
        return "秒杀失败，请重试"

    # 更新MySQL数据库
    with mysql_conn.cursor() as cursor:
        sql = "INSERT INTO orders (product_id, user_id) VALUES (%s, %s)"
        cursor.execute(sql, (product_id, user_id))
    mysql_conn.commit()

    return "秒杀成功"

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