48 lines
967 B
Go
48 lines
967 B
Go
|
// Tested on etcd 3.1+
|
||
|
|
package etcd3locker
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/tus/tusd"
|
||
|
|
"go.etcd.io/etcd/clientv3/concurrency"
|
||
|
|
)
|
||
|
|
|
||
|
|
type etcd3Lock struct {
|
||
|
|
Id string
|
||
|
|
Mutex *concurrency.Mutex
|
||
|
|
Session *concurrency.Session
|
||
|
|
}
|
||
|
|
|
||
|
|
func newEtcd3Lock(session *concurrency.Session, id string) *etcd3Lock {
|
||
|
|
return &etcd3Lock{
|
||
|
|
Mutex: concurrency.NewMutex(session, id),
|
||
|
|
Session: session,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (lock *etcd3Lock) Acquire() error {
|
||
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||
|
|
defer cancel()
|
||
|
|
|
||
|
|
// this is a blocking call; if we receive DeadlineExceeded
|
||
|
|
// the lock is most likely already taken
|
||
|
|
if err := lock.Mutex.Lock(ctx); err != nil {
|
||
|
|
if err == context.DeadlineExceeded {
|
||
|
|
return tusd.ErrFileLocked
|
||
|
|
} else {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (lock *etcd3Lock) Release() error {
|
||
|
|
return lock.Mutex.Unlock(context.Background())
|
||
|
|
}
|
||
|
|
|
||
|
|
func (lock *etcd3Lock) CloseSession() error {
|
||
|
|
return lock.Session.Close()
|
||
|
|
}
|