gitlab-com-seaweedfs/seaweedfs
1
0
Fork 0
mirror of https://github.com/seaweedfs/seaweedfs.git synced 2024-01-19 02:48:24 +00:00
Code Issues Projects Releases Wiki Activity

refactoring

Browse source
This commit is contained in:
chrislu 2022-08-22 14:11:13 -07:00
parent 7ea2c24c5d
commit 57e7582c36
2 changed files with 13 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
weed/shell/commands.go
View file

@ -73,7 +73,7 @@ func (ce *CommandEnv) isDirectory(path string) bool {
func (ce *CommandEnv) confirmIsLocked(args []string) error {
if ce.locker.IsLocking() {
if ce.locker.IsLocked() {
return nil
}
ce.locker.SetMessage(fmt.Sprintf("%v", args))
@ -82,6 +82,10 @@ func (ce *CommandEnv) confirmIsLocked(args []string) error {
}
func (ce *CommandEnv) isLocked() bool {
return ce.locker.IsLocked()
}
func (ce *CommandEnv) checkDirectory(path string) error {
dir, name := util.FullPath(path).DirAndName()

15
weed/wdclient/exclusive_locks/exclusive_locker.go
View file

@ -19,7 +19,7 @@ const (
type ExclusiveLocker struct {
token int64
lockTsNs int64
isLocking bool
isLocked bool
masterClient *wdclient.MasterClient
lockName string
message string
@ -32,8 +32,8 @@ func NewExclusiveLocker(masterClient *wdclient.MasterClient, lockName string) *E
}
}
func (l *ExclusiveLocker) IsLocking() bool {
return l.isLocking
func (l *ExclusiveLocker) IsLocked() bool {
return l.isLocked
}
func (l *ExclusiveLocker) GetToken() (token int64, lockTsNs int64) {
@ -45,7 +45,7 @@ func (l *ExclusiveLocker) GetToken() (token int64, lockTsNs int64) {
}
func (l *ExclusiveLocker) RequestLock(clientName string) {
if l.isLocking {
if l.isLocked {
return
}
@ -74,14 +74,14 @@ func (l *ExclusiveLocker) RequestLock(clientName string) {
}
}
l.isLocking = true
l.isLocked = true
// start a goroutine to renew the lease
go func() {
ctx2, cancel2 := context.WithCancel(context.Background())
defer cancel2()
for l.isLocking {
for l.isLocked {
if err := l.masterClient.WithClient(false, func(client master_pb.SeaweedClient) error {
resp, err := client.LeaseAdminToken(ctx2, &master_pb.LeaseAdminTokenRequest{
PreviousToken: atomic.LoadInt64(&l.token),
@ -98,6 +98,7 @@ func (l *ExclusiveLocker) RequestLock(clientName string) {
return err
}); err != nil {
glog.Errorf("failed to renew lock: %v", err)
l.isLocked = false
return
} else {
time.Sleep(RenewInteval)
@ -109,7 +110,7 @@ func (l *ExclusiveLocker) RequestLock(clientName string) {
}
func (l *ExclusiveLocker) ReleaseLock() {
l.isLocking = false
l.isLocked = false
ctx, cancel := context.WithCancel(context.Background())
defer cancel()