fix fault recovery issue

harmony-one · Aug 22, 2023 · 1ae0bd0 · 1ae0bd0
1 parent 78d71da
commit 1ae0bd0
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/p2p/stream/types/stream.go b/p2p/stream/types/stream.go
@@ -85,11 +85,14 @@ func (st *BaseStream) FailedTimes() int {
 }
 
 func (st *BaseStream) AddFailedTimes(faultRecoveryThreshold time.Duration) {
-	durationSinceLastFailure := time.Now().Sub(st.lastFailureTime)
-	if durationSinceLastFailure >= faultRecoveryThreshold {
-		st.ResetFailedTimes()
+	if st.failedTimes > 0 {
+		durationSinceLastFailure := time.Now().Sub(st.lastFailureTime)
+		if durationSinceLastFailure >= faultRecoveryThreshold {
+			st.ResetFailedTimes()
+		}
 	}
 	st.failedTimes++
+	st.lastFailureTime = time.Now()
 }
 
 func (st *BaseStream) ResetFailedTimes() {