Add ResetFailCount to subscriberAPI

Signed-off-by: Jack Ding <[email protected]>
redhat-cne · Aug 22, 2024 · a0b03bc · a0b03bc
1 parent 51aca34
commit a0b03bc
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/pkg/protocol/http/http.go b/pkg/protocol/http/http.go
@@ -587,6 +587,7 @@ func (h *Server) SendTo(wg *sync.WaitGroup, clientID uuid.UUID, clientAddress, r
 				}
 				log.Errorf("connection lost addressing %s", clientAddress)
 			} else {
+				h.subscriberAPI.ResetFailCount(clientID)
 				localmetrics.UpdateEventCreatedCount(resourceAddress, localmetrics.SUCCESS, 1)
 				h.DataOut <- &channel.DataChan{
 					Address: resourceAddress,

diff --git a/v1/subscriber/subscriber.go b/v1/subscriber/subscriber.go
@@ -395,6 +395,13 @@ func (p *API) IncFailCountToFail(clientID uuid.UUID) bool {
 	return false
 }
 
+// ResetFailCount ..reset fail count
+func (p *API) ResetFailCount(clientID uuid.UUID) {
+	if subStore, ok := p.SubscriberStore.Get(clientID); ok {
+		subStore.ResetFailCount()
+	}
+}
+
 // FailCountThreshold .. get threshold
 func (p *API) FailCountThreshold() int {
 	return subscriber.SetConnectionToFailAfter