Browse Source

refactor

pull/4608/head
chrislu 2 years ago
parent
commit
26eff062c8
  1. 8
      weed/cluster/lock_manager/distributed_lock_manager.go
  2. 2
      weed/cluster/lock_manager/lock_ring.go

8
weed/cluster/lock_manager/distributed_lock_manager.go

@ -15,7 +15,7 @@ func NewDistributedLockManager() *DistributedLockManager {
} }
func (dlm *DistributedLockManager) Lock(host pb.ServerAddress, key string, expiredAtNs int64, token string, servers []pb.ServerAddress) (renewToken string, movedTo pb.ServerAddress, err error) { func (dlm *DistributedLockManager) Lock(host pb.ServerAddress, key string, expiredAtNs int64, token string, servers []pb.ServerAddress) (renewToken string, movedTo pb.ServerAddress, err error) {
server := HashKeyToServer(key, servers)
server := hashKeyToServer(key, servers)
if server != host { if server != host {
movedTo = server movedTo = server
return return
@ -25,7 +25,7 @@ func (dlm *DistributedLockManager) Lock(host pb.ServerAddress, key string, expir
} }
func (dlm *DistributedLockManager) Unlock(host pb.ServerAddress, key string, token string, servers []pb.ServerAddress) (movedTo pb.ServerAddress, err error) { func (dlm *DistributedLockManager) Unlock(host pb.ServerAddress, key string, token string, servers []pb.ServerAddress) (movedTo pb.ServerAddress, err error) {
server := HashKeyToServer(key, servers)
server := hashKeyToServer(key, servers)
if server != host { if server != host {
movedTo = server movedTo = server
return return
@ -41,10 +41,10 @@ func (dlm *DistributedLockManager) InsertLock(key string, expiredAtNs int64, tok
} }
func (dlm *DistributedLockManager) SelectNotOwnedLocks(host pb.ServerAddress, servers []pb.ServerAddress) (locks []*Lock) { func (dlm *DistributedLockManager) SelectNotOwnedLocks(host pb.ServerAddress, servers []pb.ServerAddress) (locks []*Lock) {
return dlm.lockManager.SelectLocks(func(key string) bool { return dlm.lockManager.SelectLocks(func(key string) bool {
server := HashKeyToServer(key, servers)
server := hashKeyToServer(key, servers)
return server != host return server != host
}) })
} }
func (dlm *DistributedLockManager) CalculateTargetServer(key string, servers []pb.ServerAddress) pb.ServerAddress { func (dlm *DistributedLockManager) CalculateTargetServer(key string, servers []pb.ServerAddress) pb.ServerAddress {
return HashKeyToServer(key, servers)
return hashKeyToServer(key, servers)
} }

2
weed/cluster/lock_manager/lock_ring.go

@ -161,7 +161,7 @@ func (r *LockRing) GetSnapshot() (servers []pb.ServerAddress) {
return r.snapshots[0].servers return r.snapshots[0].servers
} }
func HashKeyToServer(key string, servers []pb.ServerAddress) pb.ServerAddress {
func hashKeyToServer(key string, servers []pb.ServerAddress) pb.ServerAddress {
if len(servers) == 0 { if len(servers) == 0 {
return "" return ""
} }

Loading…
Cancel
Save