realloc/frostfs-node
1
0
Fork 0
forked from TrueCloudLab/frostfs-node
Code Releases Packages Activity

[#281] service/audit: Run each SG check in separate routing at PoR

Browse source 
Signed-off-by: Alex Vanin <alexey@nspcc.ru>
This commit is contained in:
Alex Vanin 2020-12-24 14:26:07 +03:00 • committed by Alex Vanin
parent 77cf97c236
commit e5108cf135
3 changed files with 55 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

29
pkg/services/audit/auditor/context.go
View file

@ -12,6 +12,7 @@ import (
"github.com/nspcc-dev/neofs-node/pkg/services/object_manager/placement"
"github.com/nspcc-dev/neofs-node/pkg/util"
"github.com/nspcc-dev/neofs-node/pkg/util/logger"
"go.uber.org/atomic"
"go.uber.org/zap"
)
@ -23,13 +24,13 @@ type Context struct {
report *audit.Report
// consider adding mutex to access caches
sgMembersMtx sync.RWMutex
sgMembersCache map[int][]*object.ID
placementMtx sync.Mutex
placementCache map[string][]netmap.Nodes
porRequests, porRetries uint32
porRequests, porRetries atomic.Uint32
pairs []gamePair
@ -42,6 +43,7 @@ type Context struct {
cnrNodesNum int
headMtx sync.RWMutex
headResponses map[string]shortHeader
}
@ -130,7 +132,7 @@ func (c *Context) WithTask(t *audit.Task) *Context {
return c
}
// WithPDPWorkerPool sets worker pool for PDP pairs processing..
// WithPDPWorkerPool sets worker pool for PDP pairs processing.
func (c *Context) WithPDPWorkerPool(pool util.WorkerPool) *Context {
if c != nil {
c.pdpWorkerPool = pool
@ -198,6 +200,9 @@ func (c *Context) writeReport() {
}
func (c *Context) buildPlacement(id *object.ID) ([]netmap.Nodes, error) {
c.placementMtx.Lock()
defer c.placementMtx.Unlock()
strID := id.String()
if nn, ok := c.placementCache[strID]; ok {
@ -219,6 +224,9 @@ func (c *Context) buildPlacement(id *object.ID) ([]netmap.Nodes, error) {
}
func (c *Context) objectSize(id *object.ID) uint64 {
c.headMtx.RLock()
defer c.headMtx.RUnlock()
strID := id.String()
if hdr, ok := c.headResponses[strID]; ok {
@ -229,6 +237,9 @@ func (c *Context) objectSize(id *object.ID) uint64 {
}
func (c *Context) objectHomoHash(id *object.ID) []byte {
c.headMtx.RLock()
defer c.headMtx.RUnlock()
strID := id.String()
if hdr, ok := c.headResponses[strID]; ok {
@ -239,6 +250,9 @@ func (c *Context) objectHomoHash(id *object.ID) []byte {
}
func (c *Context) updateHeadResponses(hdr *object.Object) {
c.headMtx.Lock()
defer c.headMtx.Unlock()
strID := hdr.ID().String()
if _, ok := c.headResponses[strID]; !ok {
@ -248,3 +262,10 @@ func (c *Context) updateHeadResponses(hdr *object.Object) {
}
}
}
func (c *Context) updateSGInfo(ind int, members []*object.ID) {
c.sgMembersMtx.Lock()
defer c.sgMembersMtx.Unlock()
c.sgMembersCache[ind] = members
}