在 go 框架中，实现分布式锁有以下两种方法：etcd：利用 etcd 的 watch 特性，当锁被释放时触发事件通知，实现分布式锁。redis：使用 redis 的 setnx 和 expire 命令，实现简单可靠的分布式锁。
Go 框架实现分布式锁
在分布式系统中，需要确保不同服务或进程访问共享资源时的顺序访问，此时就需要使用分布式锁。在 Go 框架中，有几个流行的库可以实现分布式锁，如分布式协调服务 Etcd 和分布式 KV 数据库 Redis。
Etcd
Etcd 是一个分布式、高可用的键值存储，可用于实现分布式锁。Etcd 的锁机制通过利用其 Watch 特性，当锁被释放时触发事件通知。
// import "github.com/coreos/etcd/clientv3"
// EtcdLock实现了分布式锁
type EtcdLock struct {
client *clientv3.Client
key string
}
// NewEtcdLock 创建一个EtcdLock
func NewEtcdLock(client *clientv3.Client, key string) *EtcdLock {
return &EtcdLock{
client: client,
key: key,
}
}
// Lock 尝试获取锁
func (l *EtcdLock) Lock() error {
ctx := context.Background()
lease, err := l.client.Grant(ctx, 10) // 租约时间为10秒
if err != nil {
return err
}
txn := l.client.Txn(ctx).If(
clientv3.Compare(clientv3.Value(l.key), "=", ""),
).Then(
clientv3.OpPut(l.key, "locked", clientv3.WithLease(lease.ID)),
)
resp, err := txn.Commit()
if err != nil {
return err
}
if resp.Succeeded {
return nil
}
return errors.New("获取锁失败")
}
// Unlock 释放锁
func (l *EtcdLock) Unlock() error {
ctx := context.Background()
_, err := l.client.Delete(ctx, l.key)
return err
}Redis
Redis 是一种分布式 KV 数据库，也支持分布式锁的实现。Redis 提供了 SETNX 和 EXPIRE 命令，可以用来实现简单可靠的锁。
立即学习“go语言免费学习笔记（深入）”；
// import "github.com/go-redis/redis/v8"
// RedisLock实现了分布式锁
type RedisLock struct {
client *redis.Client
key string
}
// NewRedisLock 创建一个RedisLock
func NewRedisLock(client *redis.Client, key string) *RedisLock {
return &RedisLock{
client: client,
key: key,
}
}
// Lock 尝试获取锁
func (l *RedisLock) Lock() error {
ctx := context.Background()
res, err := l.client.SetNX(ctx, l.key, "locked", 10*time.Second).Result() // 锁10秒
if err != nil {
return err
}
if !res {
return errors.New("获取锁失败")
}
return nil
}
// Unlock 释放锁
func (l *RedisLock) Unlock() error {
ctx := context.Background()
_, err := l.client.Del(ctx, l.key).Result()
return err
}实战案例
考虑一个场景，多个微服务共享一个数据库连接池，为了防止并发访问导致数据库连接被耗尽，需要对连接池进行分布式锁控制。
lock := NewEtcdLock(client, "db-pool-lock")
if err := lock.Lock(); err != nil {
log.Fatalf("获取锁失败: %v", err)
}
// 访问数据库连接池...
if err := lock.Unlock(); err != nil {
log.Fatalf("释放锁失败: %v", err)
} 
