no divide by zero

4d677ee3 · Michael Yang · 7ea90587 · 4d677ee3
Commit 4d677ee3 authored Nov 15, 2023 by Michael Yang
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 1 deletion

progress/bar.go progress/bar.go +6 -1

No files found.
--- a/progress/bar.go
+++ b/progress/bar.go
@@ -105,7 +105,12 @@ func (b *Bar) percent() float64 {
 }

 func (b *Bar) rate() float64 {
-	return (float64(b.currentValue) - float64(b.initialValue)) / b.elapsed().Seconds()
+	elapsed := b.elapsed()
+	if elapsed.Seconds() > 0 {
+		return (float64(b.currentValue) - float64(b.initialValue)) / elapsed.Seconds()
+	}
+
+	return 0
 }

 func (b *Bar) elapsed() time.Duration {