forked from TrueCloudLab/frostfs-node
[#2164] network/cache: Separate mutex for addr
Signed-off-by: Evgenii Stratonikov <e.stratonikov@yadro.com>
This commit is contained in:
parent
0ace28e43d
commit
f3caf6acfe
1 changed files with 26 additions and 2 deletions
26
pkg/network/cache/multi.go
vendored
26
pkg/network/cache/multi.go
vendored
|
@ -17,6 +17,8 @@ type multiClient struct {
|
|||
|
||||
clients map[string]clientcore.Client
|
||||
|
||||
// addrMtx protects addr field. Should not be taken before the mtx.
|
||||
addrMtx sync.RWMutex
|
||||
addr network.AddressGroup
|
||||
|
||||
opts ClientCacheOpts
|
||||
|
@ -74,6 +76,22 @@ func (x *multiClient) updateGroup(group network.AddressGroup) {
|
|||
return false
|
||||
})
|
||||
|
||||
x.addrMtx.RLock()
|
||||
oldGroup := x.addr
|
||||
x.addrMtx.RUnlock()
|
||||
if len(oldGroup) == len(cache) {
|
||||
needUpdate := false
|
||||
for i := range oldGroup {
|
||||
if cache[i] != oldGroup[i].String() {
|
||||
needUpdate = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if !needUpdate {
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
x.mtx.Lock()
|
||||
defer x.mtx.Unlock()
|
||||
loop:
|
||||
|
@ -87,13 +105,19 @@ loop:
|
|||
}
|
||||
|
||||
// Then add new clients.
|
||||
x.addrMtx.Lock()
|
||||
x.addr = group
|
||||
x.addrMtx.Unlock()
|
||||
}
|
||||
|
||||
func (x *multiClient) iterateClients(ctx context.Context, f func(clientcore.Client) error) error {
|
||||
var firstErr error
|
||||
|
||||
x.addr.IterateAddresses(func(addr network.Address) bool {
|
||||
x.addrMtx.RLock()
|
||||
group := x.addr
|
||||
x.addrMtx.RUnlock()
|
||||
|
||||
group.IterateAddresses(func(addr network.Address) bool {
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
firstErr = context.Canceled
|
||||
|
|
Loading…
Reference in a new issue