common
/
machinery-plugins
Private
1
Fork 0
Code Pull Requests Releases Activity

feat: add lock

Browse Source
Operations
Operations
Create branch
Create tag
Revert
Cherry-pick
main
git 2024-05-25 00:32:23 +08:00
parent a095e14a71
commit 074d740817
Signed by: git
GPG Key ID: 3F65EFFA44207ADD
1 changed files with 12 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
locks/etcd/etcd.go
View File

@ -51,7 +51,7 @@ func (r etcdLock) LockWithRetries(key string, unixTsToExpireNs int64) error {
return nil return nil
} }
log.INFO.Printf("acquired lock=%s failed, retries=%d", key, i) log.INFO.Printf("acquired lock=%s failed, retries=%d, err=%s", key, i, err)
time.Sleep(time.Millisecond * 100) time.Sleep(time.Millisecond * 100)
} }
return ErrRedisLockFailed return ErrRedisLockFailed
@ -61,26 +61,21 @@ func (r etcdLock) LockWithRetries(key string, unixTsToExpireNs int64) error {
func (r etcdLock) Lock(key string, unixTsToExpireNs int64) error { func (r etcdLock) Lock(key string, unixTsToExpireNs int64) error {
now := time.Now().UnixNano() now := time.Now().UnixNano()
ttl := time.Duration(unixTsToExpireNs + 1 - now) ttl := time.Duration(unixTsToExpireNs + 1 - now)
lease := clientv3.NewLease(r.cli)
// 创建一个新的session
s, err := concurrency.NewSession(r.cli, concurrency.WithTTL(int(ttl.Seconds())))
if err != nil {
return err
}
defer s.Orphan()
lockKey := fmt.Sprintf("/machinery/v2/lock/%s", strings.TrimRight(key, "/"))
m := concurrency.NewMutex(s, lockKey)
ctx, cancel := context.WithTimeout(context.Background(), time.Second*2) ctx, cancel := context.WithTimeout(context.Background(), time.Second*2)
defer cancel() defer cancel()
resp, err := lease.Grant(ctx, int64(ttl.Seconds()))
if err != nil {
return err
}
// 创建一个新的session
s, err := concurrency.NewSession(r.cli, concurrency.WithLease(resp.ID))
if err != nil {
return err
}
key = fmt.Sprintf("/machinery/v2/lock/%s", strings.TrimRight(key, "/"))
m := concurrency.NewMutex(s, key)
if err := m.Lock(ctx); err != nil { if err := m.Lock(ctx); err != nil {
_ = s.Close()
return err return err
} }