Redsync分布式锁
Redsync是一个专为Go语言设计的基于Redis实现的分布式互斥锁库,它遵循了分布式锁算法,并提供了高性能、高可用、防死锁和防误删的特性。
使用场景
- 扣减库存:在电商系统中,多个用户可能同时购买同一件商品,此时可以使用Redsync锁来控制库存的扣减,确保不会出现超卖的情况。
- 缓存击穿/缓存雪崩:在缓存系统中,当大量请求同时访问一个不存在的缓存项时,会导致缓存击穿;而当大量缓存项同时失效时,会导致缓存雪崩。此时可以使用Redsync锁来控制缓存的访问和更新,从而减轻数据库的压力。
- 任务调度:在分布式任务调度系统中,可以使用Redsync锁来确保同一时间只有一个任务执行器在执行某个任务。
**需要引入的包**
github.com/go-redsync/redsync/v4
// 此函数首先创建一个Redis客户端,并使用该客户端来初始化一个连接池和一个基于Redis的分布式锁管理器(redsync)。
// 分布式锁用于确保在高并发环境下,对共享资源(如库存)的访问是互斥的,避免数据竞争和不一致。
func (c *Server) Sell(ctx context.Context, in *proto.SellInfo) (*proto.Empty, error) {
// 因为'goredis.NewPool'通常需要一个符合其特定连接工厂接口的实现。
// 这里假设'goredis'是一个假设的包,实际使用中可能需要替换为如'redis.Pool'等标准实现。
pool := goredis.NewPool(global.RedisClient)
// 创建一个新的redsync实例,该实例将用于管理分布式锁。
rs := redsync.New(pool)
// 定义一个锁的名称,这里假设它是全局唯一的。
mutexName := "my-global-mutex"
// 创建一个新的分布式锁实例,设置锁的过期时间和尝试加锁的次数。
mutex := rs.NewMutex(mutexName, redsync.WithExpiry(10*time.Second), redsync.WithTries(5))
// 尝试在上下文中加锁。如果加锁失败(如因超时、竞争等),则返回错误。
if err := mutex.LockContext(ctx); err != nil {
// 加锁失败,返回内部服务器错误。
return nil, status.Errorf(codes.Internal, "获取Redis锁失败: %v", err)
}
// 开始一个数据库事务。
tx := global.Db.Begin()
// 遍历传入的售卖信息中的每个商品信息。
for _, info := range in.GoodsInfo {
// 查询当前商品的库存信息。
var inv model.Inventory
result := tx.Where(&model.Inventory{GoodsId: info.GoodsId}).First(&inv)
// 如果查询结果为空或发生错误,则回滚事务并返回错误。
if result.RowsAffected == 0 || result.Error != nil {
tx.Rollback()
return nil, status.Errorf(codes.Internal, "库存详情查询失败")
}
// 检查库存是否足够。
if inv.Number < info.Num {
tx.Rollback()
return nil, status.Errorf(codes.Internal, "库存不足")
}
// 减少库存数量。
inv.Number -= info.Num
// 更新库存信息。
res := tx.Save(&inv)
// 如果更新失败,则回滚事务并返回错误。
if res.Error != nil {
tx.Rollback()
return nil, status.Errorf(codes.Internal, "设置库存失败")
}
}
// 如果所有操作都成功,则提交事务。
tx.Commit()
// 释放锁。如果释放失败(如锁已过期或已被其他进程释放),则抛出panic。
// 在实际生产环境中,应该避免使用panic,而是应该优雅地处理错误。
if ok, err := mutex.UnlockContext(ctx); !ok || err != nil {
return nil, status.Errorf(codes.Internal, "释放Redis锁失败: %v", err)
}
// 返回成功消息。
return &proto.Empty{}, nil
}
Redsync的注意事项
- 确保Redis节点的一致性:在使用Redsync时,需要确保Redis节点之间的一致性,以避免出现锁状态不一致的情况。
- 合理设置超时时间:在加锁时,需要合理设置超时时间,以确保在锁被持有者意外释放或系统崩溃时,锁能够自动释放。
- 避免频繁加锁和解锁:频繁加锁和解锁会增加Redis的负担,并可能影响系统的性能。因此,在不需要时应及时释放锁,并避免不必要的加锁操作。