go etcd 分布式锁-Go服务器开发

etcd是一个分布式的键值存储系统，支持watch、ttl等特性，同时提供了分布式锁的实现。使用etcd分布式锁可以实现多个节点之间的协同工作，避免因为并发操作而导致的数据不一致问题。

实现分布式锁的基本思路如下：

在etcd中创建一个临时有序节点，代表当前请求正在竞争锁。
判断自己是否得到锁（是否是最小的节点）。
如果没有得到锁，则监听前面的节点，等待其释放锁。
如果得到锁，则开始执行任务，在任务完成后删除自己创建的节点，释放锁。

具体代码实现如下：

package main
import (
	"context"
	"fmt"
	"go.etcd.io/etcd/clientv3"
	"time"
)
type DistributedLock struct {
	cli  *clientv3.Client
	name string
}
func NewDistributedLock(endpoints []string, name string) (*DistributedLock, error) {
	cli, err := clientv3.New(clientv3.Config{
		Endpoints:   endpoints,
		DialTimeout: 5 * time.Second,
	})
	if err != nil {
		return nil, err
	}
	return &DistributedLock{
		cli:  cli,
		name: name,
	}, nil
}
func (dl *DistributedLock) Lock() error {
	resp, err := dl.cli.Grant(context.Background(), 5)
	if err != nil {
		return err
	}
	leaseID := resp.ID
	key := fmt.Sprintf("/locks/%s", dl.name)
	_, err = dl.cli.Txn(context.Background()).
		If(clientv3.Compare(clientv3.CreateRevision(key), "=", 0)).
		Then(clientv3.OpPut(key, "", clientv3.WithLease(leaseID))).
		Commit()
	if err != nil {
		return err
	}
	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()
	ch := dl.cli.Watch(ctx, key)
	for {
		select {
		case <-ch:
			resp, _ := dl.cli.Get(context.Background(), key)
			if len(resp.Kvs) == 0 {
				return nil
			}
			minKey := resp.Kvs[0]
			for _, kv := range resp.Kvs {
				if kv.Lease < minKey.Lease {
					minKey = kv
				}
			}
			if string(minKey.Key) == key {
				return nil
			}
		case <-time.After(2 * time.Second):
			return fmt.Errorf("Lock timeout")
		}
	}
}
func (dl *DistributedLock) Unlock() error {
	key := fmt.Sprintf("/locks/%s", dl.name)
	_, err := dl.cli.Delete(context.Background(), key)
	if err != nil {
		return err
	}
	return nil
}
func main() {
	dl1, err := NewDistributedLock([]string{"localhost:2379"}, "test")
	if err != nil {
		panic(err)
	}
	err = dl1.Lock()
	if err != nil {
		panic(err)
	}
	fmt.Println("Got lock 1")
	dl2, err := NewDistributedLock([]string{"localhost:2379"}, "test")
	if err != nil {
		panic(err)
	}
	err = dl2.Lock()
	if err == nil {
		panic("Should not get the lock")
	}
	fmt.Println("Can't get lock 2")
	err = dl1.Unlock()
	if err != nil {
		panic(err)
	}
	fmt.Println("Released lock 1")
	err = dl2.Lock()
	if err != nil {
		panic(err)
	}
	fmt.Println("Got lock 2")
	err = dl2.Unlock()
	if err != nil {
		panic(err)
	}
	fmt.Println("Released lock 2")
}