FastAPI秒杀库存总变负数？Redis分布式锁能帮你守住底线吗

最新推荐文章于 2025-10-21 17:57:46 发布

原创

最新推荐文章于 2025-10-21 17:57:46 发布 · 644 阅读

22 ·

CC 4.0 BY-SA版权

文章标签：

#fastapi #redis #分布式

url: /posts/65ce343cc5df9faf3a8e2eeaab42ae45/
title: FastAPI秒杀库存总变负数？Redis分布式锁能帮你守住底线吗
date: 2025-09-17T03:43:34+08:00
lastmod: 2025-09-17T03:43:34+08:00
author: cmdragon

summary:
分布式锁在FastAPI中用于解决多实例并发访问共享资源时的数据一致性问题。其核心原理包括互斥性、安全性、可用性和容错性，常用Redis分布式锁实现。Redlock算法通过多节点投票确保锁的可靠性。FastAPI中通过aioredis实现异步分布式锁，支持锁的获取、释放和续约。测试策略覆盖单实例、多实例并发及锁超时等场景，确保锁的正确性和稳定性。

categories:

fastapi

tags:

FastAPI
分布式锁
Redis
异步编程
Redlock算法
并发控制
测试策略

扫描二维码关注或者微信搜一搜：编程智域前端至全栈交流与成长

发现1000+提升效率与开发的AI工具和实用程序：https://tools.cmdragon.cn/

一、分布式锁在FastAPI中的作用与原理

1.1 为什么需要分布式锁？

想象一个场景：你做了个FastAPI秒杀接口，商品库存只有1件。如果同时有100个请求打进来，单实例FastAPI能用asyncio.Lock（本地锁）保证同一时间只有一个请求处理库存。但如果部署了3个FastAPI实例（多进程/多机器），本地锁就失效了——每个实例都有自己的锁，100个请求会同时冲进3个实例，导致库存变成-99，彻底乱套。

分布式锁的本质：给跨进程、跨机器的资源竞争“上全局锁”，不管多少个FastAPI实例，同一时间只有一个请求能拿到锁，确保数据一致。

1.2 分布式锁的核心原理

分布式锁要满足4个核心要求：

互斥性：同一时间只有一个请求能拿到锁；
安全性：不能让A的锁被B释放；
可用性：Redis挂了一个节点，还能正常用；
容错性：持有锁的进程崩溃，锁要能自动释放。

FastAPI里最常用的是Redis分布式锁（轻量、性能高），底层用Redlock算法（解决Redis单点故障问题）。测试环境可以简化成单Redis节点，生产环境建议用3-5个节点。

1.3 Redlock算法简化理解

Redlock是“多节点投票制”：

向5个Redis节点发“锁请求”；
超过3个节点同意（半数以上），就算拿到锁；
计算总耗时，如果比锁超时时间短，锁有效；
否则，把所有节点的锁都删了，重新来。

测试环境不用这么复杂——先拿单Redis节点练手，生产再扩展。

二、FastAPI中分布式锁的实现

2.1 依赖准备与配置

首先装依赖：

pip install fastapi==0.109 aioredis==2.0.1 pydantic==2.5.3 pytest-asyncio==0.23.2

用pydantic写个配置类（统一管理Redis连接参数）：

# lock_config.py
from pydantic import BaseModel, Field

class RedisLockConfig(BaseModel):
    redis_url: str = Field(default="redis://localhost:6379", description="Redis连接地址")
    lock_prefix: str = Field(default="dist_lock:", description="锁键前缀，避免key冲突")
    timeout: int = Field(default=10, description="锁超时时间（秒），防止死锁")
    renew_interval: int = Field(default=3, description="锁续约间隔（秒），防止业务超时")

2.2 异步分布式锁实现（aioredis）

因为FastAPI是异步的，必须用aioredis（异步Redis客户端）。写个RedisDistributedLock类，封装锁的获取、释放、续约：

# distributed_lock.py
from aioredis import Redis, RedisError
from pydantic import BaseModel
import uuid
import asyncio

class RedisDistributedLock:
    def __init__(self, config: RedisLockConfig):
        self.config = config
        self.redis: Redis | None = None  # Redis客户端实例
        self.lock_key: str | None = None  # 当前锁的key
        self.lock_value: str | None = None  # 唯一标识（防误删别人的锁）
        self.renew_task: asyncio.Task | None = None  # 锁续约任务

    # 异步上下文管理器：自动连接/断开Redis
    async def __aenter__(self) -> "RedisDistributedLock":
        await self._connect()
        return self

    async def __aexit__(self, exc_type, exc_val, exc_tb):
        await self.release()
        await self._disconnect()

    # 连接Redis
    async def _connect(self):
        if not self.redis:
            self.redis = await Redis.from_url(self.config.redis_url)

    # 断开Redis连接
    async def _disconnect(self):
        if self.redis:
            await self.redis.close