golang程序进度条实现示例详解

func main() {
	bar := progress.New(100)
	for i := 0; i < 100; i++ {
		time.Sleep(time.Second / 10)
		bar.Done(1)
	}
	bar.Finish()
}

type Bar struct {
	total         int64         // 总进度
	current       int64         // 当前进度
	filler        string        // 进度填充字符
	filler_length int64         // 进度条长度
	time_format   string        // 进度条时间格式
	interval      time.Duration // 打印时间间隔
	begin         time.Time     // 任务开始时间
}

// New 新建进度条实例
func New(total int64, opts ...func(*Bar)) *Bar {
	bar := &Bar{
		total:         total,
		filler:        "█",
		filler_length: 25,
		time_format:   "15:04:05", // 2006-01-02T15:04:05
		interval:      time.Second,
		begin:         time.Now(),
	}
	for _, opt := range opts {
		opt(bar)
	}
	// 定时打印
	ticker := time.NewTicker(bar.interval)
	go func() {
		for bar.current < bar.total {
			fmt.Print(bar.get_progress_string())// 打印进度
			<-ticker.C
		}
	}()
	return bar
}
// Done 更新完成进度
func (bar *Bar) Done(i int64) {
	bar.current += i
}
// Finish 完成最后进度条
func (bar *Bar) Finish() {
	fmt.Println(bar.get_progress_string())
}
// WithFiller 设置进度条填充字符
func WithFiller(filler string) func(*Bar) {
	return func(bar *Bar) {
		if len(bar.filler) != 0 {
			bar.filler = filler
		}
	}
}

//get_percent 获取进度百分比,区间0-100
func (bar *Bar) get_percent() int64 {
	return bar.current * 100 / bar.total
}

//get_eta 获取eta时间
func (bar *Bar) get_eta(now time.Time) string {
	eta := (now.Unix() - bar.begin.Unix()) * 100 / (bar.get_percent() + 1)
	return bar.begin.Add(time.Second * time.Duration(eta)).Format(bar.time_format)
}

//get_progress_string 获取打印控制台字符串
func (bar *Bar) get_progress_string() string {
	fills := bar.get_percent() * bar.filler_length / 100
	for i := int64(0); i < bar.filler_length; i++ {
		switch {
		case i < fills:
			chunks[i] = bar.filler
		default:
			chunks[i] = " "
		}
	}
	now := time.Now()
	eta := bar.get_eta(now)
	qps := bar.current / (now.Unix() - bar.begin.Unix() + 1)
	return fmt.Sprintf("\r[%s]%d/%d [eta]%s [qps]%d ", strings.Join(chunks, ""), bar.current, bar.total, eta, qps)
}