deal with case when label < labelLength

mutantmonkey · Oct 16, 2015 · e33b607 · e33b607
1 parent a4a177b
commit e33b607
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/progress/progress.go b/progress/progress.go
@@ -9,8 +9,8 @@ import (
 )
 
 const (
-	labelLength = 40
-	termWidth = 80
+	labelLength    = 40
+	termWidth      = 80
 	updateInterval = time.Second / 10
 )
 
@@ -61,7 +61,11 @@ func (pr *ProgressReader) clearProgress() {
 func (pr *ProgressReader) printProgress(n int64) {
 	percent := float64(n*100) / float64(pr.total)
 	formatStr := fmt.Sprintf("\r%%-%dv %%7.2f%%%%", labelLength)
-	fmt.Fprintf(pr.output, formatStr, pr.label[:labelLength], percent)
+	if len(pr.label) > labelLength {
+		fmt.Fprintf(pr.output, formatStr, pr.label[:labelLength], percent)
+	} else {
+		fmt.Fprintf(pr.output, formatStr, pr.label, percent)
+	}
 }
 
 func (pr *ProgressReader) update() {