游戏数据库存储秒杀

import redis
import pymysql

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

# 连接MySQL
mysql_conn = pymysql.connect(host='localhost', user='user', password='password', db='game_db')

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

    # 使用Redis事务确保原子性
    pipe = redis_client.pipeline()
    try:
        pipe.watch(f'stock:{item_id}')
        pipe.multi()
        pipe.decr(f'stock:{item_id}')
        pipe.execute()
    except redis.WatchError:
        return "秒杀失败，库存被其他用户修改"

    # 记录秒杀成功到MySQL
    try:
        with mysql_conn.cursor() as cursor:
            sql = "INSERT INTO seckill_records (item_id, user_id) VALUES (%s, %s)"
            cursor.execute(sql, (item_id, user_id))
        mysql_conn.commit()
    except Exception as e:
        # 回滚Redis操作
        redis_client.incr(f'stock:{item_id}')
        return f"秒杀失败，数据库错误: {str(e)}"

    return "秒杀成功"

# 示例调用
result = seckill_item(123, 456)
print(result)