go?分布式鎖簡單實現(xiàn)實例詳解

更新時間：2022年09月15日 15:07:40 作者：水紋

這篇文章主要為大家介紹了go?分布式鎖簡單實現(xiàn)實例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪

正文

其實鎖這種東西，都能能不加就不加，鎖會導(dǎo)致程序一定程度上退回到串行化，進(jìn)而降低效率。

案例

首先，看一個案例，如果要實現(xiàn)一個計數(shù)器，并且是多個協(xié)程共同進(jìn)行的，就會出現(xiàn)以下的情況：

package main
import (
   "fmt"
   "sync"
)
func main() {
   numberFlag := 0
   wg := new(sync.WaitGroup)
   for i := 0; i < 200; i++ {
      wg.Add(1)
      go func() {
         defer wg.Done()
         numberFlag++
      }()
   }
   fmt.Println(numberFlag)
   wg.Wait()
}

每次執(zhí)行后的計數(shù)器結(jié)果都是不同的，這是因為計數(shù)器本身是被不同的協(xié)程搶著+1，會產(chǎn)生多個協(xié)程同時拿到numberFlag=N的情況。為了避免這種資源競爭，要對資源進(jìn)行加鎖，使得同一時刻只有一個協(xié)程能對資源進(jìn)行操控。

資源加鎖

package main
import (
   "fmt"
   "sync"
)
func main() {
   numberFlag := 0
   myLock := make(chan struct{}, 1)
   wg := new(sync.WaitGroup)
   for i := 0; i < 200; i++ {
      wg.Add(1)
      go func() {
         defer func() {
            // 釋放鎖
            <-myLock
         }()
         defer wg.Done()
         // 搶鎖
         myLock <- struct{}{}
         numberFlag++
      }()
   }
   wg.Wait()
   fmt.Println(numberFlag)
}

但是這種鎖只能用于你自己的本地服務(wù)，一旦出現(xiàn)多服務(wù)，比如分布式，微服務(wù)，這樣的場景，這個鎖就沒啥用了，這就需要分布式鎖。

關(guān)于分布式鎖，一般的實現(xiàn)就是用redis或者zookeeper實現(xiàn)。redis比較方便的就是大部分的服務(wù)都會使用redis，無需額外安裝依賴，而zookeeper普通服務(wù)用的并不多，即使是kafka也在新版放棄了zookeeper。

zookeeper最大的好處就是可以通過心跳檢測客戶端的情況，進(jìn)而避免重復(fù)得鎖的問題。
但是同時也產(chǎn)生了一些問題，這個心跳檢測多久一次，在心跳檢測的間隔如果出現(xiàn)了鎖超時的問題怎么辦，等等。

使用redis來實現(xiàn)分布式鎖

所以一些服務(wù)還是傾向于使用redis來實現(xiàn)分布式鎖。

package main
import (
   "fmt"
   "github.com/gomodule/redigo/redis"
   "go-face-test/redisTest/redisOne/redisConn"
   "sync"
   "time"
)
func main() {
   // 分布式鎖
   var LockName = "lockLock"
   // 十秒過期時間
   var ExpirationTime = 10
   wg := new(sync.WaitGroup)
   wg.Add(2)
   // 起兩個協(xié)程來模擬分布式服務(wù)的搶占
   go handleBusiness(LockName, ExpirationTime, "A", wg)
   go handleBusiness(LockName, ExpirationTime, "B", wg)
   wg.Wait()
}
func handleBusiness(lockName string, ExpTime int, nowGroName string, wg *sync.WaitGroup) {
   // One服務(wù)獲取鎖是否存在
   c := redisConn.Get()
   defer c.Close()
   for {
      isKeyExists, err := redis.Bool(c.Do("EXISTS", lockName))
      if err != nil {
         fmt.Println("err while checking keys:", err)
      } else {
         fmt.Println(isKeyExists)
      }
      if isKeyExists {
         // 存在這把鎖，開始自旋等待
         fmt.Println("當(dāng)前協(xié)程為: " + nowGroName + " 沒搶到鎖……")
         //休息1s
         time.Sleep(time.Second)
      } else {
         // 設(shè)置一把鎖
         // 值為1，過期時間為10秒
         reply, err := c.Do("SET", lockName, 2, "EX", ExpTime, "NX")
         // 搶占失敗
         if reply == nil {
            fmt.Println("當(dāng)前協(xié)程為: " + nowGroName + " 搶占鎖失敗")
            continue
         }
         // 開始業(yè)務(wù)處理
         fmt.Println("當(dāng)前協(xié)程為: " + nowGroName + " 啊啊啊啊。這是一個業(yè)務(wù)處理，預(yù)計處理時間為 3s 。處理開始........")
         fmt.Println("當(dāng)前協(xié)程為: " + nowGroName + " 距離處理完成還有---3s" + time.Now().Format("2006-01-02 15:04:05"))
         time.Sleep(time.Second)
         fmt.Println("當(dāng)前協(xié)程為: " + nowGroName + " 距離處理完成還有---2s" + time.Now().Format("2006-01-02 15:04:05"))
         time.Sleep(time.Second)
         fmt.Println("當(dāng)前協(xié)程為: " + nowGroName + " 距離處理完成還有---1s" + time.Now().Format("2006-01-02 15:04:05"))
         time.Sleep(time.Second)
         //業(yè)務(wù)結(jié)束,釋放鎖
         _, err = c.Do("DEL", lockName)
         if err != nil {
            fmt.Println("err while deleting:", err)
         }
         break
      }
   }
   wg.Done()
}

但是這個鎖明顯有問題:

第一，當(dāng)A服務(wù)（本案例中其實是協(xié)程模擬的）拿到鎖之后，處理超時了，鎖還沒有釋放，就已經(jīng)過期，過期后B服務(wù)就搶到了鎖，此時AB均認(rèn)為自己拿到了鎖

第二，A服務(wù)按理說只能去掉自己的服務(wù)加上的鎖，如果不止是有AB兩個服務(wù)，有更多的服務(wù)，那么A如果出現(xiàn)處理較慢，鎖超時后，B服務(wù)搶到鎖，A又處理完成所有的事釋放了鎖，那其實是釋放掉了B的鎖。

也就是說，釋放鎖的時候也必須判斷是否是自己的鎖

那么就得用redis的lua來保證原子性

redis lua保證原子性

package main
import (
   "fmt"
   "github.com/gomodule/redigo/redis"
   "go-face-test/redisTest/redisTwo/redisConn"
   "log"
   "math/rand"
   "strconv"
   "sync"
   "time"
)
var letterRunes = []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ")
var lockCommand = `if redis.call("GET", KEYS[1]) == ARGV[1] then
    redis.call("SET", KEYS[1], ARGV[1], "PX", ARGV[2] , "NX")
    return "OK"
else
    return redis.call("SET", KEYS[1], ARGV[1], "NX", "PX", ARGV[2])
end`
var delCommand = `if redis.call("GET", KEYS[1]) == ARGV[1] then
    return redis.call("DEL", KEYS[1])
else
    return 0
end`
func main() {
   // 分布式鎖
   var LockName = "lockLock"
   // 十秒過期時間
   var ExpirationTime = 3
   wg := new(sync.WaitGroup)
   wg.Add(2)
   // 起兩個協(xié)程來模擬分布式服務(wù)的搶占
   go handleBusiness(LockName, ExpirationTime, "A", wg)
   go handleBusiness(LockName, ExpirationTime, "B", wg)
   wg.Wait()
}
func init() {
   rand.Seed(time.Now().UnixNano())
}
func handleBusiness(lockName string, ExpTime int, nowGroName string, wg *sync.WaitGroup) {
   // One服務(wù)獲取鎖是否存在
   c := redisConn.Get()
   defer c.Close()
   for {
      isKeyExists, err := redis.Bool(c.Do("EXISTS", lockName))
      if err != nil {
         fmt.Println("err while checking keys:", err)
      } else {
         fmt.Println(isKeyExists)
      }
      if isKeyExists {
         // 存在這把鎖，開始自旋等待
         fmt.Println("當(dāng)前協(xié)程為: " + nowGroName + " 沒搶到鎖……")
         //休息1s
         time.Sleep(time.Second)
      } else {
         // 設(shè)置一把鎖
         // 鎖的值是根據(jù)當(dāng)前服務(wù)名稱和時間來的
         lockFlag, lockValue, _ := getLock(lockName, nowGroName, ExpTime, c)
         // 搶占失敗
         if !lockFlag {
            fmt.Println("當(dāng)前協(xié)程為: " + nowGroName + " 搶占鎖失敗")
            continue
         }
         // 開始業(yè)務(wù)處理
         fmt.Println("當(dāng)前協(xié)程為: " + nowGroName + " 啊啊啊啊。這是一個業(yè)務(wù)處理，預(yù)計處理時間為 " + strconv.Itoa(ExpTime) + "s 。處理開始........")
         for i := ExpTime - 1; i > 0; i-- {
            fmt.Println("當(dāng)前協(xié)程為: " + nowGroName + " 距離處理完成還有---" + strconv.Itoa(i) + "s " + time.Now().Format("2006-01-02 15:04:05"))
            time.Sleep(time.Second)
         }
         //業(yè)務(wù)結(jié)束,釋放鎖
         lockDelFlag, _ := delLock(lockName, lockValue, c)
         //獲取當(dāng)前鎖的值
         if lockDelFlag {
            fmt.Println("釋放鎖成功")
         } else {
            fmt.Println("這個鎖不是你的，或者這個鎖已經(jīng)超時")
         }
         break
      }
   }
   wg.Done()
}
// 獲得唯一鎖的值
func getLockOnlyValue(nowGroName string) string {
   nano := strconv.FormatInt(time.Now().UnixNano(), 10)
   return nowGroName + "_" + nano + "_" + RandStringRunes(6)
}
// 獲得一個鎖
func getLock(LockName, nowGroName string, timeOut int, conn redis.Conn) (bool, string, error) {
   myLockValue := getLockOnlyValue(nowGroName)
   lua := redis.NewScript(1, lockCommand)
   resp, err := lua.Do(conn, LockName, myLockValue, strconv.Itoa(timeOut*1000))
   if err != nil {
      log.Fatal(LockName, err)
      return false, "", err
   } else if resp == nil {
      return false, "", nil
   }
   s, ok := resp.(string)
   if !ok {
      return false, "", nil
   }
   if s != "OK" {
      return false, "", nil
   }
   return true, myLockValue, nil
}
// 刪除一個鎖
func delLock(LockName, LockeValue string, conn redis.Conn) (bool, error) {
   lua := redis.NewScript(1, delCommand)
   resp, err := lua.Do(conn, LockName, LockeValue)
   if err != nil {
      return false, err
   }
   reply, ok := resp.(int64)
   if !ok {
      return false, nil
   }
   return reply == 1, nil
}
func RandStringRunes(n int) string {
   b := make([]rune, n)
   for i := range b {
      b[i] = letterRunes[rand.Intn(len(letterRunes))]
   }
   return string(b)
}

以上就是go 分布式鎖簡單實現(xiàn)實例詳解的詳細(xì)內(nèi)容，更多關(guān)于go 分布式鎖的資料請關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: