当前位置：首页 > news >正文

go 分布式redis锁的实现方式

news 2025/8/25 18:59:35

go 语言以高并发著称。那么在实际的项目中 经常会用到锁的情况。比如说秒杀抢购等等场景。下面主要介绍 redis 布式锁实现的两种高并发抢购场景。其实 高并发 和 分布式锁 是一个互斥的两个状态：

方式一 setNX：

使用 redis自带的API setNX 来实现。能解决高并发场景下的绝大多数场景，待优化点 锁的续命和等待锁的实现。实现流程：

redis setNX 设置键值。如果键存在则返回 false 反之则为 true
使用 setNX 来设置一个键值，值为当前协程设置的随机值。
当程序运行完成之后，删除该键值

这里只有当减库存成功：

抢购流程成功则返回 410。其余失败则返回 200. 这样就能通过返回码很容易看到成功抢购的数量 我么使用 postman 模拟 1600 用户点击十分钟。库存为一个亿。

// redis分布式锁  方式1：自己动手
// 该方案可以解决大多数场景中的 redis 锁的问题，
// 还剩余一个 锁续命的问题 极高并发下的微小概率事件
func redisLock_0(c *gin.Context) {// 实现逻辑// 1 先用商品ID为 key, uuid为值,  这一步是防止别人把自己的锁删除// 2 用SetNX 设置一个键值 锁住一个商品，并设置超时时间。 当 SetNX key 存在则 返回false, 反之为 truerdb := Rdb()lockKey := "product_001"newUUID := uuid.New()// 只能删除锁  并切判断是不是自己的锁，只有自己的锁才会删除defer func() {keyValue, err := rdb.Get(ctx, lockKey).Result()if err != nil {fmt.Println("keyValue error:", keyValue, err)c.JSON(http.StatusOK, gin.H{"message": "获取锁失败",})return}if keyValue == newUUID.String() {rdb.Del(ctx, lockKey)}}()//设置锁，30秒过期，只有当锁不存在时才会成功设置，//设置时间是为了 防止特殊情况所没有成功释放。success, err := rdb.SetNX(ctx, lockKey, newUUID.String(), time.Second*30).Result()if err != nil {fmt.Println("Error setting lock: %v", err)c.JSON(http.StatusOK, gin.H{"message": "设置锁单出错",})return}// 判断是否成功获得锁if success {fmt.Println("Successfully acquired lock:", newUUID)// 执行需要锁保护的操作 获取真实的 库存count, err := strconv.Atoi(rdb.Get(ctx, "product_count").Val())if err != nil {fmt.Println("Error getting product count: %v", err)c.JSON(http.StatusOK, gin.H{"message": "Error getting product count",})return}if count > 1 {stock := count - 1err := rdb.Set(ctx, "product_count", strconv.Itoa(stock), 0).Err()if err != nil {fmt.Println("Error setting product count: %v", err)c.JSON(http.StatusOK, gin.H{"message": "Error setting product count",})return} else {fmt.Println("减库存操作成功， 现在库存为: %v", stock)c.JSON(http.StatusGone, gin.H{"message": "Hello, World!",})return}} else {fmt.Println("库存为 0 ")c.JSON(http.StatusOK, gin.H{"message": "Hello, World!",})return}} else {///没有获得锁！  可以做延迟 轮询处理fmt.Println("Failed to acquire lock. The key already exists.")c.JSON(http.StatusOK, gin.H{"message": "Hello, World!",})return}
}

经过十分钟我们看下数据：

该方案整体数据：

一共请求了 534,979 次
并发 877
成功销售 280,367 个商品即返回值为 410的个数。

方式二 redisson:

使用 go-redisson 库，这个类似 java redisson:

go-redisson command - github.com/paceew/go-redisson - Go Packageshttps://pkg.go.dev/github.com/paceew/go-redisson

该方案使用起来就很简单了：

我们来测试一样的数据：

func redisLock_1(c *gin.Context) {//获取一个锁对象mutex := RedSon().NewMutex("godisson")//尝试加锁， 并且设置超时时间和等待时间，//如果加锁失败 会阻塞等待，或超时 或 加锁成功err := mutex.TryLock(20000, 20000)if err != nil {log.Println("can't obtained lock")c.JSON(http.StatusOK, gin.H{"message": "Error can't obtained lock",})return}defer func(mutex *godisson.Mutex) {_, err := mutex.Unlock()if err != nil {log.Println("can't obtained lock")c.JSON(http.StatusOK, gin.H{"message": "Error1 can't obtained lock",})}}(mutex)// 执行需要锁保护的操作 获取真实的 库存count, err := strconv.Atoi(rdb.Get(ctx, "product_count").Val())if err != nil {fmt.Println("Error getting product count: %v", err)c.JSON(http.StatusOK, gin.H{"message": "Error getting product count",})return}if count > 1 {stock := count - 1err := rdb.Set(ctx, "product_count", strconv.Itoa(stock), 0).Err()if err != nil {fmt.Println("Error setting product count: %v", err)c.JSON(http.StatusOK, gin.H{"message": "Error setting product count",})return} else {fmt.Println("减库存操作成功， 现在库存为: %v", stock)c.JSON(http.StatusGone, gin.H{"message": "Hello, World!",})return}} else {fmt.Println("库存为 0 ")c.JSON(http.StatusOK, gin.H{"message": "Hello, World!",})return}
}

该方案整体数据：

一共请求 528,686
并发 868
成功销售 343,381 个商品即返回值为 410的个数。应该是实现了锁等待。所有这个方案比自己实现的抢购要高。

如何提高吞吐优化性能问题

分段锁：

分段锁的核心思路就是：之前的方案都是一个锁，处理所有请求。这里呢开十把锁。那吞吐性能不就快了十倍了麽。那么我们就采用redisson 来做十把分段锁：

把一个亿的商品库存，分成1千万的十份。然后用十把锁。这样：

func redisLock_2(c *gin.Context) {rand.Seed(time.Now().UnixNano())// 生成包含0和9的随机数num := rand.Intn(10)mutexKey := "godisson_" + strconv.Itoa(num)product_key := "product_count_" + strconv.Itoa(num)//获取一个锁对象mutex := RedSon().NewMutex(mutexKey)//尝试加锁， 并且设置超时时间和等待时间，//如果加锁失败 会阻塞等待，或超时 或 加锁成功err := mutex.TryLock(20000, 20000)if err != nil {log.Println("can't obtained lock")c.JSON(http.StatusOK, gin.H{"message": "Error can't obtained lock",})return}defer func(mutex *godisson.Mutex) {_, err := mutex.Unlock()if err != nil {log.Println("can't obtained lock")c.JSON(http.StatusOK, gin.H{"message": "Error1 can't obtained lock",})}}(mutex)// 执行需要锁保护的操作 获取真实的 库存count, err := strconv.Atoi(rdb.Get(ctx, product_key).Val())if err != nil {fmt.Println("Error getting product count: %v", err)c.JSON(http.StatusOK, gin.H{"message": "Error getting product count",})return}if count > 1 {stock := count - 1err := rdb.Set(ctx, product_key, strconv.Itoa(stock), 0).Err()if err != nil {fmt.Println("Error setting product count: %v", err)c.JSON(http.StatusOK, gin.H{"message": "Error setting product count",})return} else {fmt.Println("减库存操作成功， 现在库存为: %v", stock)c.JSON(http.StatusGone, gin.H{"message": "Hello, World!",})return}} else {fmt.Println("库存为 0 ")c.JSON(http.StatusOK, gin.H{"message": "Hello, World!",})return}
}