You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
77 lines
1.8 KiB
77 lines
1.8 KiB
package golock
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"go.etcd.io/etcd/client/v3"
|
|
)
|
|
|
|
type LockEtcd struct {
|
|
etcdClient *clientv3.Client // 驱动
|
|
}
|
|
|
|
func NewLockEtcd(etcdClient *clientv3.Client) *LockEtcd {
|
|
return &LockEtcd{etcdClient: etcdClient}
|
|
}
|
|
|
|
// Lock 上锁
|
|
// key 锁名
|
|
// val 锁内容
|
|
// ttl 锁过期时间
|
|
func (e *LockEtcd) Lock(key string, val string, ttl int64) (string, error) {
|
|
if ttl <= 0 {
|
|
return "", errors.New("长期请使用 LockForever 方法")
|
|
}
|
|
// 1、获取
|
|
get, err := e.etcdClient.Get(context.Background(), key)
|
|
if err != nil {
|
|
return "", errors.New("获取异常")
|
|
}
|
|
if len(get.Kvs) > 0 {
|
|
return "", errors.New("上锁失败，已存在")
|
|
}
|
|
// 2、申请一个lease(租约)
|
|
lease := clientv3.NewLease(e.etcdClient)
|
|
// 3、申请一个*秒的租约
|
|
leaseGrantResp, err := lease.Grant(context.TODO(), ttl)
|
|
if err != nil {
|
|
return "", errors.New(fmt.Sprintf("申请租约失败 %s", err))
|
|
}
|
|
// 5、获得kv api子集
|
|
kv := clientv3.NewKV(e.etcdClient)
|
|
// 4、设置
|
|
kv.Put(context.TODO(), key, val, clientv3.WithLease(leaseGrantResp.ID))
|
|
if err != nil {
|
|
return "", errors.New("上锁失败")
|
|
}
|
|
return val, nil
|
|
}
|
|
|
|
// Unlock 解锁
|
|
// key 锁名
|
|
func (e *LockEtcd) Unlock(key string) error {
|
|
_, err := e.etcdClient.Delete(context.Background(), key)
|
|
return err
|
|
}
|
|
|
|
// LockForever 永远上锁
|
|
// key 锁名
|
|
// val 锁内容
|
|
func (e *LockEtcd) LockForever(key string, val string) (string, error) {
|
|
// 1、获取
|
|
get, err := e.etcdClient.Get(context.Background(), key)
|
|
if err != nil {
|
|
return "", errors.New("获取异常")
|
|
}
|
|
if len(get.Kvs) > 0 {
|
|
return "", errors.New("上锁失败，已存在")
|
|
}
|
|
// 2、设置
|
|
e.etcdClient.Put(context.TODO(), key, val)
|
|
if err != nil {
|
|
return "", errors.New("上锁失败")
|
|
}
|
|
return val, nil
|
|
}
|