1 分布式鎖概述

談到分布式鎖，必然是因?yàn)閱螜C(jī)鎖無(wú)法滿足要求，在現(xiàn)階段微服務(wù)多實(shí)例部署的情況下，單機(jī)語(yǔ)言級(jí)別的鎖，無(wú)法滿足并發(fā)互斥資源的安全訪問(wèn)。常見(jiàn)的單機(jī)鎖如Java的jvm鎖Lock、synchronized，golang的Mutex等 對(duì)于分布式鎖有很多種實(shí)現(xiàn)方式，常見(jiàn)的有以下幾種：

• 基于數(shù)據(jù)庫(kù)：通過(guò)數(shù)據(jù)庫(kù)事務(wù)鎖例如for update操作
• 基于緩存中間件：redis分布式鎖、etcd分布式鎖等
• 基于ZK臨時(shí)節(jié)點(diǎn)：zookeeper 臨時(shí)節(jié)點(diǎn)實(shí)現(xiàn)分布式鎖

每種方式實(shí)現(xiàn)的分布式鎖各有優(yōu)缺點(diǎn)簡(jiǎn)單介紹一下：

• 數(shù)據(jù)庫(kù)實(shí)現(xiàn)不用額外引入新的中間件，減少系統(tǒng)的依賴性和不穩(wěn)定性，但性能不會(huì)太高，且并發(fā)量大時(shí)，對(duì)數(shù)據(jù)庫(kù)壓力比較大。
• ZK實(shí)現(xiàn)分布式，因?yàn)閦k滿足了CP，能夠保證其數(shù)據(jù)一致性，不會(huì)出現(xiàn)加鎖成功后又丟失的問(wèn)題，但相反性能會(huì)降低，并且可用性降低 CAP A不是滿足的，詳細(xì)可以自行了解zk細(xì)節(jié)
• redis 實(shí)現(xiàn)：最大的優(yōu)點(diǎn)性能高，能保證AP，保證其高可用。但無(wú)法保證一致性，因?yàn)閞edis滿足的是AP，可能存在某一個(gè)時(shí)間節(jié)點(diǎn)集群數(shù)據(jù)S-M同步不一致。

2 分布式鎖要點(diǎn)

實(shí)現(xiàn)分布式鎖需要滿足一下幾點(diǎn)：

• 鎖載體：redis 受用 K-V 鍵值作為鎖載體，ZK使用臨時(shí)節(jié)點(diǎn)作為載體
• 鎖租期：進(jìn)程持有分布式鎖后不能一直占用，如果因?yàn)殄礄C(jī)情況造成鎖釋放失敗，就會(huì)一直占用，reds 可以設(shè)置過(guò)期時(shí)間，zk臨時(shí)節(jié)點(diǎn)也會(huì)自動(dòng)刪除。
• 其他要求：比如減少驚群效應(yīng)、可重入機(jī)制、公平鎖機(jī)制，不同的實(shí)現(xiàn)方式有的不能完全滿足。

分布式鎖選擇：

• qps不大的情況下，那種方式都可以
• 結(jié)合目前技術(shù)體系，在不引入新的技術(shù)中間件情況下解決問(wèn)題
• qps并發(fā)極高，但容忍極少的數(shù)據(jù)丟失或者不一致，建議使用redis實(shí)現(xiàn)分布式鎖
• 如果業(yè)務(wù)要求任何情況下都不允許數(shù)據(jù)丟失，可以使用zk或者etcd實(shí)現(xiàn)

3 Etcd 實(shí)現(xiàn)機(jī)制

• 鎖載體： 使用 k-v 結(jié)構(gòu)實(shí)現(xiàn)
• 鎖租期： Etcd 通過(guò)lease可以對(duì) kv 設(shè)置租約，當(dāng)租約到期，kv 將失效刪除；避免長(zhǎng)時(shí)間占用鎖不釋放放。
• 自動(dòng)續(xù)期： Etcd 可以對(duì)租約進(jìn)行自動(dòng)續(xù)期，通過(guò)KeepAlive實(shí)現(xiàn)
• 公平鎖： 多個(gè)程序同時(shí)搶鎖時(shí)，會(huì)根據(jù) Revision 值大小依次獲得鎖，可以有效避免 “驚群效應(yīng)”，公平獲取。
• Watch 機(jī)制： 監(jiān)聽(tīng)機(jī)制，Watch 機(jī)制支持 Watch 某個(gè)固定的 key或者目錄， key 或目錄發(fā)生變化，客戶端可以收到通知。

4 代碼實(shí)現(xiàn)

操作步驟：

• 初始化客戶端
• 創(chuàng)建一個(gè)session并設(shè)置默認(rèn)租期30s
• 獲取指定前綴的鎖對(duì)象
• 加鎖
• 執(zhí)行業(yè)務(wù)
• 釋放鎖

代碼：

package main
import (
	"context"
	clientv3 "go.etcd.io/etcd/client/v3"
	"go.etcd.io/etcd/client/v3/concurrency"
	"log"
	"time"
)
func main() {
	// 初始化客戶端
	log.Println("客戶端初始化")
	client, err := clientv3.New(clientv3.Config{Endpoints: []string{"localhost:2379"}, DialTimeout: time.Second * 3})
	if err != nil {
		log.Fatalf("客戶端初始化失敗:%v\n", err)
	}
	// 創(chuàng)建一個(gè)session并設(shè)置默認(rèn)租期30s,即鎖默認(rèn)超過(guò)30s會(huì)自動(dòng)釋放(內(nèi)部會(huì)自動(dòng)續(xù)期Etcd KeepAlive)
	log.Println("Session初始化")
	session, err := concurrency.NewSession(client, concurrency.WithTTL(30))
	if err != nil {
		log.Fatalf("Session初始化失敗:%v\n", err)
		return
	}
	defer func(session *concurrency.Session) {
		err := session.Close()
		if err != nil {
			log.Fatalf("Session關(guān)閉失敗:%v\n", err)
		}
	}(session)
	// 獲取指定前綴的鎖對(duì)象
	mutex := concurrency.NewMutex(session, "my-lock")
	// 加鎖默認(rèn)等待3s
	log.Println("TryLock加鎖失敗不會(huì)等待")
	ctx, cancel := context.WithTimeout(context.Background(), time.Second*3)
	defer cancel()
	err = mutex.TryLock(ctx)
	if err != nil {
		log.Fatalf("加鎖失敗立即返回:%v\n", err)
		return
	}
	//log.Println("加鎖最多等待3s")
	//ctx, cancel := context.WithTimeout(context.Background(), time.Second*3)
	//defer cancel()
	//err = mutex.Lock(ctx)
	//if err != nil {
	//	log.Fatalf("加鎖失敗:%v\n", err)
	//	return
	//}
	// Exe biz
	log.Println("加鎖成功開(kāi)始執(zhí)行業(yè)務(wù)")
	for i := 1; i <= 10; i++ {
		time.Sleep(time.Second)
		log.Printf("執(zhí)行 %%%d ...", i*10)
	}
	// 釋放鎖
	err = mutex.Unlock(context.TODO())
	if err != nil {
		log.Fatalf("釋放鎖失敗:%v\n", err)
		return
	}
	log.Println("釋放鎖完成")
}

測(cè)試結(jié)果 

到此這篇關(guān)于Go語(yǔ)言使用Etcd實(shí)現(xiàn)分布式鎖的文章就介紹到這了,更多相關(guān)Go Etcd分布式鎖內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論