托管数据库秒杀

import redis
import pymysql

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

# 连接MySQL数据库
db = pymysql.connect(host='localhost', user='root', password='password', database='test')

def seckill_product(product_id):
    # 先尝试从Redis缓存获取库存
    stock = redis_client.get(f'stock:{product_id}')
    if stock is None:
        # 如果缓存中没有，则查询数据库
        cursor = db.cursor()
        cursor.execute(f"SELECT stock FROM products WHERE id={product_id}")
        stock = cursor.fetchone()[0]
        redis_client.set(f'stock:{product_id}', stock)  # 将库存存入Redis缓存
    
    if int(stock) > 0:
        # 执行秒杀逻辑（此处简化处理）
        redis_client.decr(f'stock:{product_id}')  # 减少库存
        # 更新数据库中的库存（异步处理或事务中处理）
        return True
    else:
        return False

# 示例调用
if seckill_product(123):
    print("秒杀成功！")
else:
    print("库存不足，秒杀失败。")