From 44d9f9513761fb07b1d34e94dbc1422891676dea Mon Sep 17 00:00:00 2001
From: Alex Vanin <alexey@nspcc.ru>
Date: Thu, 24 Dec 2020 13:22:14 +0300
Subject: [PATCH] [#276] service/audit: Count Head requests and retries in PoR

Signed-off-by: Alex Vanin <alexey@nspcc.ru>
---
 go.sum                                | Bin 60804 -> 60119 bytes
 pkg/services/audit/auditor/context.go |   2 ++
 pkg/services/audit/auditor/por.go     |  14 +++++++++-----
 pkg/services/audit/report.go          |   6 ++++++
 4 files changed, 17 insertions(+), 5 deletions(-)

diff --git a/go.sum b/go.sum
index 5ba29c188eaed0c49fc15c04ede37d5d8f2fd1eb..60b2b66b973d1d7d8b313c72e8be4a1094155f97 100644
GIT binary patch
delta 49
zcmV-10M7q}+XL6t1F#&Kvt*e!0kh<qnHjU;v>5`kVYkr@vzEn)5VLL8F$}Zx;`sry
HHswwTY>ybv

delta 448
zcmXBPIgZm%002-yilG3Bf(8kR^eMc?3!U?sub9~J7Jn~Po*4UknZy%M{A1UgU{Vbo
zH$WI#8cu*TFa-_JQBZILBy)gQyxXs%*WX9)o?KI7%sj&xbLP)=v@0zvc;uk8lZy}_
z$_do+VmonbOyVlyFO@~wkFL!-0|GL3fZ<6Ty~Zjk;ymJb5($fTqIO<c7H+n&C$mdC
zjQL)E(?1?iAyzNgRxvd*J*$fX#@Gh6B8ml@v`)x$3EP3Zu|THk$Di+?_TPHy?f@M`
zR;m)2fORRQX|!d?Mxuf0^H~KXZ8|H-f+2X$;dA=<_rtjV^YQnC0j3@l1u3VrDBP!&
z6t@oVlDaMf5OlJi!b4N0^@iTabE$v(`P0JzS)7}3h@$|?Hqe8a$H^+83w@7vb{F_G
zSqg>;E4o}M$1k^!u5M2Hmp9Y<gUxl=33{TLG{q-w8<wvmX6L)xgE4~{BO@0s&~w8|
yw%xkVevJS3=tf&H-08ltXuNEqf@pgfQhBh`sww1(wKw-;J56(ND)rl6_})ML)|?yw

diff --git a/pkg/services/audit/auditor/context.go b/pkg/services/audit/auditor/context.go
index 282c17f9..2ac379c8 100644
--- a/pkg/services/audit/auditor/context.go
+++ b/pkg/services/audit/auditor/context.go
@@ -29,6 +29,8 @@ type Context struct {
 
 	placementCache map[string][]netmap.Nodes
 
+	porRequests, porRetries uint32
+
 	pairs []gamePair
 
 	pairedMtx   sync.Mutex
diff --git a/pkg/services/audit/auditor/por.go b/pkg/services/audit/auditor/por.go
index 59d29f64..71599313 100644
--- a/pkg/services/audit/auditor/por.go
+++ b/pkg/services/audit/auditor/por.go
@@ -14,6 +14,7 @@ func (c *Context) executePoR() {
 	for i, sg := range c.task.StorageGroupList() {
 		c.checkStorageGroupPoR(i, sg) // consider parallel it
 	}
+	c.report.SetPoRCounters(c.porRequests, c.porRetries)
 }
 
 func (c *Context) checkStorageGroupPoR(ind int, sg *object.ID) {
@@ -52,15 +53,18 @@ func (c *Context) checkStorageGroupPoR(ind int, sg *object.ID) {
 			flat[i], flat[j] = flat[j], flat[i]
 		})
 
-		for _, node := range flat {
-			hdr, err := c.cnrCom.GetHeader(c.task, node, members[i], true)
+		for i := range flat {
+			c.porRequests++
+			if i > 0 { // in best case audit get object header on first iteration
+				c.porRetries++
+			}
+
+			hdr, err := c.cnrCom.GetHeader(c.task, flat[i], members[i], true)
 			if err != nil {
 				c.log.Debug("can't head object",
-					zap.String("remote_node", node.Address()),
+					zap.String("remote_node", flat[i].Address()),
 					zap.Stringer("oid", members[i]))
 
-				// todo: count all fails and successes for audit report
-
 				continue
 			}
 
diff --git a/pkg/services/audit/report.go b/pkg/services/audit/report.go
index 2a508383..c41d0c01 100644
--- a/pkg/services/audit/report.go
+++ b/pkg/services/audit/report.go
@@ -60,3 +60,9 @@ func (r *Report) SetPDPResults(passed, failed [][]byte) {
 	r.res.SetPassNodes(passed)
 	r.res.SetFailNodes(failed)
 }
+
+// SetPoRCounters sets amounts of head requests and retries at PoR audit stage.
+func (r *Report) SetPoRCounters(requests, retries uint32) {
+	r.res.SetRequests(requests)
+	r.res.SetRetries(retries)
+}