gin框架用go-redis+redsync實現分布式鎖

一，安裝需要用到的庫

1,go-redis的地址:

https://github.com/go-redis/redis

2,安裝go-redis

liuhongdi@ku:~$ go get -u github.com/go-redis/redis/v8

3,redsync的地址

https://github.com/go-redsync/redsync

4，安裝redsync

liuhongdi@ku:~$ go get -u github.com/go-redsync/redsync/v4

5,gorm的地址

https://gorm.io/

6,安裝gorm

liuhongdi@ku:~$ go get -u gorm.io/gorm

說明：劉宏締的go森林是一個專注golang的博客，
          地址：https://blog.csdn.net/weixin_43881017

說明：作者:劉宏締 郵箱: [e-mail]

 

二，演示項目的相關信息

1,  地址：

https://github.com/liuhongdi/digv23

2，功能說明：演示了使用分布式鎖避免高並發下單減庫存時多扣庫存

3,  項目結構；如圖:

三，數據庫的sql

1，建表sql

1.  
   CREATE TABLE `goods` (
2.  
   `goodsId` int NOT NULL AUTO_INCREMENT COMMENT 'id',
3.  
   `goodsName` varchar(500) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci NOT NULL DEFAULT '' COMMENT '商品名稱',
4.  
   `subject` varchar(200) NOT NULL DEFAULT '' COMMENT '標題',
5.  
   `price` decimal(15,2) NOT NULL DEFAULT '0.00' COMMENT '價格',
6.  
   `stock` int NOT NULL DEFAULT '0' COMMENT '庫存數量',
7.  
   PRIMARY KEY (`goodsId`)
8.  
   ) ENGINE=InnoDB AUTO_INCREMENT=0 DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci COMMENT='商品表'

2,插入演示數據：

1.  
   INSERT INTO `goods` (`goodsId`, `goodsName`, `subject`, `price`, `stock`) VALUES
2.  
   ( 1, '蜂蜜牛奶手工皂', '深入滋養，肌膚細膩嫩滑', '70.00', 3),
3.  
   ( 2, '紫光筷子筒', '紫光智護，干爽防潮更健康', '189.00', 40),
4.  
   ( 3, '野性mini便攜式藍牙音箱', '強悍機能，品味豪邁', '499.00', 100),
5.  
   ( 4, '樂穿梭茶具', '茶具+茶葉精美端午禮盒', '200.00', 40);

四，go代碼說明

1,controller/goodsController.go

1.  
   package controller
2.  
    
3.  
   import (
4.  
   "github.com/gin-gonic/gin"
5.  
   "github.com/liuhongdi/digv23/global"
6.  
   "github.com/liuhongdi/digv23/service"
7.  
   )
8.  
    
9.  
   type GoodsController struct{}
10.  
    func NewGoodsController() GoodsController {
11.  
    return GoodsController{}
12.  
    }
13.  
    //購買一件商品
14.  
    func (g *GoodsController) BuyOne(c *gin.Context) {
15.  
    result := global.NewResult(c)
16.  
     
17.  
    var goodsId int64 = 1
18.  
    buyNum := 1
19.  
    err := service.BuyOneGoods(goodsId,buyNum);
20.  
    if err != nil {
21.  
    result.Error( 404,"數據查詢錯誤")
22.  
    } else {
23.  
    result.Success( "減庫存成功");
24.  
    }
25.  
    return
26.  
    }
27.  
    //購買一件商品,by lock
28.  
    func (g *GoodsController) LockBuyOne(c *gin.Context) {
29.  
    result := global.NewResult(c)
30.  
     
31.  
    var goodsId int64 = 1
32.  
    buyNum := 1
33.  
    err := service.LockBuyOneGoods(goodsId,buyNum);
34.  
    if err != nil {
35.  
    result.Error( 404,"數據查詢錯誤")
36.  
    } else {
37.  
    result.Success( "減庫存成功");
38.  
    }
39.  
    return
40.  
    }

2,dao/goods.go

1.  
   package dao
2.  
    
3.  
   import (
4.  
   "errors"
5.  
   "fmt"
6.  
   "github.com/liuhongdi/digv23/global"
7.  
   "github.com/liuhongdi/digv23/model"
8.  
   "gorm.io/gorm"
9.  
   )
10.  
     
11.  
    //decrease stock
12.  
    func DecreaseOneGoodsStock(goodsId int64,buyNum int) error {
13.  
    //查詢商品信息
14.  
    goodsOne:=&model.Goods{}
15.  
    err := global.DBLink.Where( "goodsId=?",goodsId).First(&goodsOne).Error
16.  
    //fmt.Println(goodsOne)
17.  
    if (err != nil) {
18.  
    return err
19.  
    }
20.  
    //得到庫存
21.  
    stock := goodsOne.Stock
22.  
    fmt.Println( "當前庫存:",stock)
23.  
    //fmt.Println(stock)
24.  
    if (stock < buyNum || stock <= 0) {
25.  
    return errors.New("庫存不足")
26.  
    }
27.  
     
28.  
    //減庫存
29.  
    result := global.DBLink.Debug().Table( "goods").Where("goodsId = ? ", goodsId,buyNum).Update("stock", gorm.Expr("stock - ?", buyNum))
30.  
    if (result.Error != nil) {
31.  
    return result.Error
32.  
    } else {
33.  
    fmt.Println( "成功減庫存一次")
34.  
    return nil
35.  
    }
36.  
    }

3,service/goods.go

1.  
   package service
2.  
    
3.  
   import (
4.  
   "github.com/liuhongdi/digv23/dao"
5.  
   "github.com/liuhongdi/digv23/global"
6.  
   "strconv"
7.  
   "github.com/go-redsync/redsync/v4"
8.  
   "github.com/go-redsync/redsync/v4/redis/goredis/v8"
9.  
   )
10.  
     
11.  
    //購買一件商品
12.  
    func BuyOneGoods(goodsId int64,buyNum int) error {
13.  
    return dao.DecreaseOneGoodsStock(goodsId,buyNum);
14.  
    }
15.  
     
16.  
    //購買一件商品,by lock
17.  
    func LockBuyOneGoods(goodsId int64,buyNum int) error {
18.  
     
19.  
    pool := goredis.NewPool(global.RedisDb) // or, pool := redigo.NewPool(...)
20.  
    // Create an instance of redisync to be used to obtain a mutual exclusion
21.  
    // lock.
22.  
    rs := redsync.New(pool)
23.  
    // Obtain a new mutex by using the same name for all instances wanting the
24.  
    // same lock.
25.  
    mutexname := "goods_"+strconv.FormatInt(goodsId,10)
26.  
    mutex := rs.NewMutex(mutexname)
27.  
    // Obtain a lock for our given mutex. After this is successful, no one else
28.  
    // can obtain the same lock (the same mutex name) until we unlock it.
29.  
    if err := mutex.Lock(); err != nil {
30.  
    return err
31.  
    }
32.  
    // Do your work that requires the lock.
33.  
    errdecre := dao.DecreaseOneGoodsStock(goodsId,buyNum);
34.  
    //fmt.Println(errdecre)
35.  
     
36.  
    // Release the lock so other processes or threads can obtain a lock.
37.  
    if ok, err := mutex.Unlock(); !ok || err != nil {
38.  
    return err
39.  
    }
40.  
     
41.  
    if (errdecre!=nil){
42.  
    return errdecre
43.  
    }
44.  
     
45.  
    return nil
46.  
    }

4,其他相關代碼可訪問github

五，測試效果

1,設置id為1的商品庫存為3

2,測試不加鎖的訪問:

liuhongdi@ku:~$ ab -c 100 -n 100 http://127.0.0.1:8080/goods/buyone

查看控制台的輸出:

1.  
   當前庫存: 3
2.  
    
3.  
   2021/01/21 12:22:17 /data/liuhongdi/digv23/dao/goods.go:29
4.  
   [1.681ms] [rows:1] UPDATE `goods` SET `stock`=stock - 1 WHERE goodsId = 1
5.  
   成功減庫存一次
6.  
   當前庫存: 2
7.  
   當前庫存: 2
8.  
   當前庫存: 2
9.  
   當前庫存: 2
10.  
    當前庫存: 2
11.  
    當前庫存: 2
12.  
    當前庫存: 2
13.  
    當前庫存: 2
14.  
    當前庫存: 2
15.  
    當前庫存: 2
16.  
    當前庫存: 2
17.  
    當前庫存: 2
18.  
    當前庫存: 2
19.  
    當前庫存: 2
20.  
    當前庫存: 2
21.  
    當前庫存: 2
22.  
    當前庫存: 2
23.  
    當前庫存: 2
24.  
    當前庫存: 2
25.  
    當前庫存: 2
26.  
    當前庫存: 2
27.  
    當前庫存: 2
28.  
    當前庫存: 2
29.  
    當前庫存: 2
30.  
    當前庫存: 2
31.  
    當前庫存: 2
32.  
    當前庫存: 2
33.  
    當前庫存: 2
34.  
    當前庫存: 2
35.  
    當前庫存: 2
36.  
    當前庫存: 2
37.  
    當前庫存: 2
38.  
    當前庫存: 2
39.  
    當前庫存: 2
40.  
    當前庫存: 2
41.  
    當前庫存: 2
42.  
    當前庫存: 2
43.  
    當前庫存: 2
44.  
    當前庫存: 2
45.  
    當前庫存: 2
46.  
    當前庫存: 2
47.  
    當前庫存: 2
48.  
    當前庫存: 2
49.  
    當前庫存: 2
50.  
    當前庫存: 2
51.  
    當前庫存: 2
52.  
    當前庫存: 2
53.  
    當前庫存: 2
54.  
    當前庫存: 2
55.  
    當前庫存: 2
56.  
    當前庫存: 2
57.  
    當前庫存: 2
58.  
    當前庫存: 2
59.  
    當前庫存: 2
60.  
    當前庫存: 2
61.  
    當前庫存: 2
62.  
    當前庫存: 2
63.  
    當前庫存: 2
64.  
     
65.  
    2021/01/21 12:22:18 /data/liuhongdi/digv23/dao/goods.go:29
66.  
    [17.357ms] [rows:1] UPDATE `goods` SET `stock`=stock - 1 WHERE goodsId = 1
67.  
    成功減庫存一次
68.  
    當前庫存: 2
69.  
    當前庫存: 1
70.  
    當前庫存: 1
71.  
    當前庫存: 1
72.  
    當前庫存: 1
73.  
    當前庫存: 1
74.  
    當前庫存: 1
75.  
     
76.  
    2021/01/21 12:22:18 /data/liuhongdi/digv23/dao/goods.go:29
77.  
    [39.838ms] [rows:1] UPDATE `goods` SET `stock`=stock - 1 WHERE goodsId = 1
78.  
    成功減庫存一次
79.  
    當前庫存: 0
80.  
    當前庫存: 0
81.  
     
82.  
    2021/01/21 12:22:18 /data/liuhongdi/digv23/dao/goods.go:29
83.  
    [85.284ms] [rows:1] UPDATE `goods` SET `stock`=stock - 1 WHERE goodsId = 1
84.  
    成功減庫存一次
85.  
    當前庫存: 0
86.  
    當前庫存: -1
87.  
    當前庫存: -1
88.  
    當前庫存: -1
89.  
    當前庫存: 0
90.  
    當前庫存: -1
91.  
    當前庫存: -1
92.  
     
93.  
    2021/01/21 12:22:18 /data/liuhongdi/digv23/dao/goods.go:29
94.  
    [95.104ms] [rows:1] UPDATE `goods` SET `stock`=stock - 1 WHERE goodsId = 1
95.  
    成功減庫存一次
96.  
    當前庫存: -2
97.  
    ....

注意因為是並發的訪問，數據庫同時返回了多個結果：庫存是2，
導致后面的多個並發執行減庫存，使庫存數出現負數

3,把庫存數重置為3，

   測試加鎖的減庫存:

liuhongdi@ku:~$ ab -c 100 -n 100 http://127.0.0.1:8080/goods/lockbuyone

執行會比較慢，因為每個訪問都需要先獲得鎖之后再執行sql

1.  
   DecreaseOneGoodsStock begin
2.  
   當前庫存: 3
3.  
    
4.  
   2021/01/21 12:44:16 /data/liuhongdi/digv23/dao/goods.go:30
5.  
   [2.115ms] [rows:1] UPDATE `goods` SET `stock`=stock - 1 WHERE goodsId = 1
6.  
   成功減庫存一次
7.  
   DecreaseOneGoodsStock begin
8.  
   當前庫存: 2
9.  
    
10.  
    2021/01/21 12:44:16 /data/liuhongdi/digv23/dao/goods.go:30
11.  
    [18.724ms] [rows:1] UPDATE `goods` SET `stock`=stock - 1 WHERE goodsId = 1
12.  
    成功減庫存一次
13.  
    DecreaseOneGoodsStock begin
14.  
    當前庫存: 1
15.  
     
16.  
    2021/01/21 12:44:16 /data/liuhongdi/digv23/dao/goods.go:30
17.  
    [2.782ms] [rows:1] UPDATE `goods` SET `stock`=stock - 1 WHERE goodsId = 1
18.  
    成功減庫存一次
19.  
    DecreaseOneGoodsStock begin
20.  
    當前庫存: 0
21.  
    DecreaseOneGoodsStock begin
22.  
    當前庫存: 0
23.  
    DecreaseOneGoodsStock begin
24.  
    當前庫存: 0
25.  
    DecreaseOneGoodsStock begin
26.  
    .....

注意庫存數的返回，

因為獲取鎖之后才查詢，所以沒有同時返回多個相同數字以致減庫存成負數的情況

 

4,查看redis中的key,注意因為redis的切換很快，不一定可以看到:

1.  
   root@ku :/data/liuhongdi/digv23# /usr/local/soft/redis6/bin/redis-cli
2.  
   127.0.0.1:6379> keys *
3.  
   1) "goods_1"

 

六，查看庫的版本:

1.  
   module github.com/liuhongdi/digv23
2.  
    
3.  
   go 1.15
4.  
    
5.  
   require (
6.  
   github.com/gin-gonic/gin v1.6.3
7.  
   github.com/go-redis/redis/v8 v8.3.3
8.  
   gorm.io/driver/mysql v1.0.1
9.  
   gorm.io/gorm v1.20.6
10.  
    github.com/go-redsync/redsync/v4 v4.0.3
11.  
    )

12.