Etcd教程 — 第九章 Etcd之实现分布式锁_etcd实现分布式锁

• • 1.1 思路分析
• 2 具体实现
• • 2.1 流程图

• 2.2 实现代码

1 通过 Etcd txn 实现分布式锁

通过Etcd实现分布式锁，同样需要满足一致性、互斥性和可靠性等要求。Etcd中的事务 txn、lease租约以及 watch 监听特性，能够使得基于Etcd实现上述要求的分布式锁。

1.1 思路分析

1. 通过 Etcd 的事务特性可以帮助我们实现一致性和互斥性。Etcd 的事务特性，使用的 IF-Then-Else 语句，IF 语言判断 Etcd 服务端是否存在指定的 key，即该 key 创建版本号 create_revision 是否为 0 来检查 key 是否已存在，因为该 key 已存在的话，它的 create_revision 版本号就不是 0。
2. 满足 IF 条件的情况下则使用 then 执行 put 操作，否则 else 语句返回抢锁失败的结果。当然，除了使用 key 是否创建成功作为 IF 的判断依据，还可以创建前缀相同的 key，比较这些 key 的 revision 来判断分布式锁应该属于哪个请求。
3. 客户端请求在获取到分布式锁之后，如果发生异常，需要及时将锁给释放掉。因此需要租约，当我们申请分布式锁的时候需要指定租约时间。超过 lease 租期时间将会自动释放锁，保证了业务的可用性。
4. 是不是这样就够了呢？在执行业务逻辑时，如果客户端发起的是一个耗时的操作，操作未完成的请情况下，租约时间过期，导致其他请求获取到分布式锁，造成不一致。这种情况下则需要续租，即刷新租约，使得客户端能够和 etcd 服务端保持心跳。

2 具体实现

2.1 流程图

我们基于如上分析的思路，绘制出实现 etcd 分布式锁的流程图，如下所示：

2.2 实现代码

基于 Go 语言实现的 etcd 分布式锁，测试代码如下所示：

func TestLock(t *testing.T) {
// 客户端配置
config = clientv3.Config{
Endpoints: []string{“localhost:2379”},
DialTimeout: 5 * time.Second,
}
// 建立连接
if client, err = clientv3.New(config); err != nil {
fmt.Println(err)
return
}

// 1. 上锁并创建租约
lease = clientv3.NewLease(client)

if leaseGrantResp, err = lease.Grant(context.TODO(), 5); err != nil {
panic(err)
}
leaseId = leaseGrantResp.ID

// 2 自动续约
// 创建一个可取消的租约，主要是为了退出的时候能够释放
ctx, cancelFunc = context.WithCancel(context.TODO())

// 3. 释放租约
defer cancelFunc()
defer lease.Revoke(context.TODO(), leaseId)

if keepRespChan, err = lease.KeepAlive(ctx, leaseId); err != nil {
panic(err)
}
// 续约应答
go func() {
for {
select {
case keepResp = <-keepRespChan:
if keepRespChan == nil {
fmt.Println(“租约已经失效了”)
goto END
} else { // 每秒会续租一次, 所以就会受到一次应答
fmt.Println(“收到自动续租应答:”, keepResp.ID)
}
}
}
END:
}()

// 1.3 在租约时间内去抢锁（etcd 里面的锁就是一个 key）
kv = clientv3.NewKV(client)

// 创建事务
txn = kv.Txn(context.TODO())

//if 不存在 key，then 设置它，else 抢锁失败
txn.If(clientv3.Compare(clientv3.CreateRevision(“lock”), “=”, 0)).
Then(clientv3.OpPut(“lock”, “g”, clientv3.WithLease(leaseId))).
Else(clientv3.OpGet(“lock”))

// 提交事务
if txnResp, err = txn.Commit(); err != nil {
panic(err)
}

if !txnResp.Succeeded {
fmt.Println(“锁被占用:”, string(txnResp.Responses[0].GetResponseRange().Kvs[0].Value))
return
}

// 抢到锁后执行业务逻辑，没有抢到退出
fmt.Println(“处理任务”)
time.Sleep(5 * time.Second)

自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数Go语言工程师，想要提升技能，往往是自己摸索成长或者是报班学习，但对于培训机构动则几千的学费，着实压力不小。自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年Go语言全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上Golang知识点，真正体系化！

由于文件比较大，这里只是将部分目录大纲截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且后续会持续更新

如果你觉得这些内容对你有帮助，可以添加V获取：vip1024b （备注Go）

一个人可以走的很快，但一群人才能走的更远。不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎扫码加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！

-cbQ4Y2Fv-1712868492960)]

一个人可以走的很快，但一群人才能走的更远。不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎扫码加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！