在 .NET Core WebApi 中使用 Redis 创建分布式锁可以通过 StackExchange.Redis
库来实现。分布式锁用于确保在分布式系统中,同一时间只有一个进程可以执行某段代码。
1. 场景描述
在支付系统中,可能会出现以下并发问题:
- 用户同时发起多次支付请求,导致重复扣款。
- 多个请求同时处理同一个订单,导致数据不一致。
通过分布式锁,可以确保同一时间只有一个请求能够执行关键操作(如扣款)。
2. 实现步骤
2.1 安装 StackExchange.Redis
包
首先,安装 Redis 客户端库:
dotnet add package StackExchange.Redis
2.2 配置 Redis 连接
在 appsettings.json
中添加 Redis 连接字符串:
{
"ConnectionStrings": {
"Redis": "localhost:6379"
}
}
2.3 创建分布式锁工具类
创建一个工具类来封装 Redis 分布式锁的逻辑:
using StackExchange.Redis;
using System;
using System.Threading.Tasks;
public class RedisDistributedLock
{
private readonly IDatabase _redisDatabase;
private readonly string _lockKey;
private readonly string _lockValue;
private readonly TimeSpan _expiry;
public RedisDistributedLock(IDatabase redisDatabase, string lockKey, string lockValue, TimeSpan expiry)
{
_redisDatabase = redisDatabase;
_lockKey = lockKey;
_lockValue = lockValue;
_expiry = expiry;
}
public async Task<bool> AcquireLockAsync()
{
// 尝试设置锁,仅当键不存在时才成功
return await _redisDatabase.StringSetAsync(_lockKey, _lockValue, _expiry, When.NotExists);
}
public async Task ReleaseLockAsync()
{
// 使用 Lua 脚本确保只有锁的持有者才能释放锁
var luaScript = @"
if redis.call('GET', KEYS[1]) == ARGV[1] then
return redis.call('DEL', KEYS[1])
else
return 0
end";
await _redisDatabase.ScriptEvaluateAsync(luaScript, new RedisKey[] { _lockKey }, new RedisValue[] { _lockValue });
}
}
2.4 在 Web API 中使用分布式锁
在 Web API 的控制器中使用分布式锁来确保支付操作的原子性。
2.4.1 注册 Redis 服务
在 Startup.cs
或 Program.cs
中注册 Redis 服务:
// 添加 Redis 服务
builder.Services.AddSingleton<IConnectionMultiplexer>(sp =>
ConnectionMultiplexer.Connect(builder.Configuration.GetConnectionString("Redis")));
2.4.2 创建支付控制器
在 Controllers
文件夹中创建一个 PaymentController
,并在其中使用分布式锁:
using Microsoft.AspNetCore.Mvc;
using StackExchange.Redis;
using System;
using System.Threading.Tasks;
[ApiController]
[Route("api/[controller]")]
public class PaymentController : ControllerBase
{
private readonly IDatabase _redisDatabase;
public PaymentController(IConnectionMultiplexer redis)
{
_redisDatabase = redis.GetDatabase();
}
[HttpPost("pay")]
public async Task<IActionResult> ProcessPayment([FromBody] PaymentRequest request)
{
// 创建分布式锁
var lockKey = $"PaymentLock:{request.OrderId}"; // 锁的键,基于订单 ID
var lockValue = Guid.NewGuid().ToString(); // 锁的值,确保唯一性
var expiry = TimeSpan.FromSeconds(10); // 锁的过期时间
var distributedLock = new RedisDistributedLock(_redisDatabase, lockKey, lockValue, expiry);
try
{
// 尝试获取锁
if (await distributedLock.AcquireLockAsync())
{
Console.WriteLine("已获取锁,正在处理付款...");
// 模拟支付处理
bool paymentSuccess = await ProcessPaymentAsync(request.UserId, request.OrderId, request.Amount);
if (paymentSuccess)
{
return Ok(new { Message = "付款成功!" });
}
else
{
return BadRequest(new { Message = "付款失败!" });
}
}
else
{
return Conflict(new { Message = "正在处理此订单的另一个付款请求..." });
}
}
finally
{
// 释放锁
await distributedLock.ReleaseLockAsync();
}
}
}
3. 代码说明
3.1 分布式锁的实现
AcquireLockAsync
: 使用Redis
的SET key value NX EX
命令尝试获取锁。NX
表示仅在键不存在时设置,`EX 设置键的过期时间。ReleaseLockAsync
: 使用Lua
脚本确保只有锁的持有者才能释放锁,避免误删其他请求的锁。
3.2 支付控制器的使用
- 锁的键: 使用订单 ID 作为锁的键(如
PaymentLock:202501061410455506968463210
),确保同一订单的支付请求串行化。 - 锁的值: 使用 GUID 作为锁的值,确保锁的唯一性。
- 锁的过期时间: 设置合理的过期时间(如 10 秒),防止锁被长时间占用。
3.3 支付处理逻辑
ProcessPaymentAsync
: 模拟支付处理逻辑,包括调用支付网关、扣减余额等操作。
4. 测试 API
4.1 启动 Web API
运行项目,启动 Web API。
4.2 发送支付请求
使用工具(如 Postman 或 curl)发送支付请求:
POST /api/payment/pay
Content-Type: application/json
{
"userId": "9527",
"orderId": "202501061410455506968463210"
}
4.3 测试并发场景
同时发送多个相同的支付请求,观察是否只有一个请求能够成功获取锁并处理支付。
5. 注意事项
-
锁的粒度:
- 锁的粒度要适中。如果锁的粒度过大(如全局锁),可能导致性能问题;如果粒度过小,可能增加复杂性。
- 在支付系统中,通常以订单 ID 或用户 ID 作为锁的粒度。
-
锁的过期时间:
- 设置合理的过期时间,避免锁被长时间占用导致死锁。
- 如果业务逻辑执行时间较长,可以动态延长锁的过期时间。
-
锁的可靠性:
- Redis 需要高可用,否则可能导致锁失效。可以使用 Redis 集群或 Redlock 算法提高可靠性。
-
异常处理:
- 确保锁的释放操作放在
finally
块中,避免因异常导致锁无法释放。
- 确保锁的释放操作放在
-
幂等性:
- 支付系统需要支持幂等性,即使多次请求,也只会产生一次扣款。
6. 总结
在 .NET Core Web API 中使用 Redis 创建分布式锁,可以带来以下好处:
- 解决并发问题,确保数据一致性。
- 提高系统的可靠性和性能。
- 简化代码逻辑,降低开发复杂度。
- 支持高并发、分布式环境和高可用需求。
通过合理使用 Redis 分布式锁,可以构建高可靠、高性能的分布式系统,满足复杂的业务需求。