Squash duplicate errors in hammer (#250)

transparency-dev · Sep 12, 2024 · a984f59 · a984f59
1 parent d86e84c
commit a984f59
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/hammer/hammer.go b/hammer/hammer.go
@@ -315,6 +315,8 @@ func (a *HammerAnalyser) updateStatsLoop(ctx context.Context) {
 func (a *HammerAnalyser) errorLoop(ctx context.Context) {
 	tick := time.NewTicker(time.Second)
 	pbCount := 0
+	lastErr := ""
+	lastErrCount := 0
 	for {
 		select {
 		case <-ctx.Done(): //context cancelled
@@ -324,12 +326,24 @@ func (a *HammerAnalyser) errorLoop(ctx context.Context) {
 				klog.Warningf("%d requests received pushback from log", pbCount)
 				pbCount = 0
 			}
+			if lastErrCount > 0 {
+				klog.Warningf("(%d x) %s", lastErrCount, lastErr)
+				lastErrCount = 0
+
+			}
 		case err := <-a.errChan:
 			if errors.Is(err, ErrRetry) {
 				pbCount++
 				continue
 			}
-			klog.Warning(err)
+			es := err.Error()
+			if es != lastErr && lastErrCount > 0 {
+				klog.Warningf("(%d x) %s", lastErrCount, lastErr)
+				lastErr = es
+				lastErrCount = 0
+				continue
+			}
+			lastErrCount++
 		}
 	}
 }