Add SetWriter to specify io.Writer

schollz · Jan 29, 2018 · 974cb68 · 974cb68
1 parent 702bf7c
commit 974cb68
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/examples/main.go b/examples/main.go
@@ -1,6 +1,7 @@
 package main
 
 import (
+	"os"
 	"time"
 
 	"github.com/schollz/progressbar"
@@ -18,6 +19,14 @@ func main() {
 	bar.Reset()
 	for i := 0; i < 1000; i++ {
 		bar.Add(1)
-		time.Sleep(10 * time.Millisecond)
+		time.Sleep(5 * time.Millisecond)
 	}
+
+	bar.Reset()
+	bar.SetWriter(os.Stderr)
+	for i := 0; i < 1000; i++ {
+		bar.Add(1)
+		time.Sleep(5 * time.Millisecond)
+	}
+
 }
diff --git a/progressbar.go b/progressbar.go
@@ -73,6 +73,13 @@ func (p *ProgressBar) SetSize(size int) {
 	p.size = size
 }
 
+// SetWriter will specify a different writer than os.Stdout
+func (p *ProgressBar) SetWriter(w io.Writer) {
+	p.Lock()
+	defer p.Unlock()
+	p.w = w
+}
+
 // Add with increase the current count on the progress bar
 func (p *ProgressBar) Add(num int) error {
 	p.RLock()