mirror of
https://github.com/zeromicro/go-zero.git
synced 2025-01-24 01:30:25 +08:00
29 lines
457 B
Go
29 lines
457 B
Go
package filex
|
|
|
|
import "gopkg.in/cheggaaa/pb.v1"
|
|
|
|
type (
|
|
Scanner interface {
|
|
Scan() bool
|
|
Text() string
|
|
}
|
|
|
|
progressScanner struct {
|
|
Scanner
|
|
bar *pb.ProgressBar
|
|
}
|
|
)
|
|
|
|
func NewProgressScanner(scanner Scanner, bar *pb.ProgressBar) Scanner {
|
|
return &progressScanner{
|
|
Scanner: scanner,
|
|
bar: bar,
|
|
}
|
|
}
|
|
|
|
func (ps *progressScanner) Text() string {
|
|
s := ps.Scanner.Text()
|
|
ps.bar.Add64(int64(len(s)) + 1) // take newlines into account
|
|
return s
|
|
}
|