feat(prints): Add a WithProgressBar method. (#92)

* chore(prints): Make some adjustments to color configurations Signed-off-by: Flc゛ <[email protected]> * feat(prints): Add a `WithProgressBar` method. Signed-off-by: Flc゛ <[email protected]> * feat(prints): Add a `WithProgressBar` method. Signed-off-by: Flc゛ <[email protected]> --------- Signed-off-by: Flc゛ <[email protected]>
go-fries · Feb 7, 2024 · 316ffdf · 316ffdf
1 parent ea462d1
commit 316ffdf
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/prints/progressbar.go b/prints/progressbar.go
@@ -41,3 +41,9 @@ func NewProgressBar(total int, opts ...ProgressBarOption) *ProgressBar {
 
 	return p
 }
+
+func WithProgressBar(total int, fc func(bar *ProgressBar), opts ...ProgressBarOption) {
+	bar := NewProgressBar(total, opts...)
+	fc(bar)
+	bar.Finish()
+}
diff --git a/prints/progressbar_test.go b/prints/progressbar_test.go
@@ -24,3 +24,11 @@ func TestNewProgressBar(t *testing.T) {
 	}
 	p2.Finish()
 }
+
+func TestWithProgressBar(*testing.T) {
+	WithProgressBar(100, func(p *ProgressBar) {
+		for i := 1; i <= 100; i++ {
+			p.Increment()
+		}
+	})
+}