关系型数据库秒杀

import redis
import pymysql

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

# 连接MySQL
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:
        # 如果Redis中没有库存，从MySQL中获取并存入Redis
        with mysql_conn.cursor() as cursor:
            cursor.execute("SELECT stock FROM products WHERE id=%s FOR UPDATE", (product_id,))
            result = cursor.fetchone()
            stock = result[0]
            redis_client.set(f'stock:{product_id}', stock)
    
    stock = int(stock)
    if stock > 0:
        # 减少库存
        redis_client.decr(f'stock:{product_id}')
        
        # 记录秒杀成功
        with mysql_conn.cursor() as cursor:
            cursor.execute("INSERT INTO seckill_records (product_id, user_id) VALUES (%s, %s)", (product_id, user_id))
        mysql_conn.commit()
        return True
    else:
        return False

# 示例调用
if seckill(1, 1001):
    print("秒杀成功")
else:
    print("秒杀失败")