golang 进度条功能实现

最近在作一个需求，功能很简单，就是开发一个轻量级客户端，将一个指定文件中的内容经过TCP发送到服务器。因为该文件特别大，有可能到达100G的数量级，所以处理起来会比较慢，为了给用户提供比较友好的展现界面，所以，在其中加入了进度条显示功能。

在这里，说一下我在实现该进度条功能时的一些思路。

成果演示

先看一下最终的成品效果展现：

该进度条一共分三部分组成，第一部分是主体进度条，第二部分是百分比，第三部分是当前完成的数据和总数据的一个动态展现。

源码分析

因为是要在终端上打印出进度条的效果，所以，主要仍是利用fmt.Printf函数中的\r格式控制符。有了这个基础，咱们就能够先设计一下结构，以下所示：

type Bar struct {
    percent int64  //百分比
    cur     int64  //当前进度位置
    total   int64  //总进度
    rate    string //进度条
    graph   string //显示符号
}

其中，百分比没什么说的，cur和total是一组，表示的就是第三部分动态展现的当前完成数据和总数据。rate就是第一部分不断变化的进度条，它是一个string类型的字符串。
这个进度条显示工具还提供了一个叫graph的属性，有了它，用户就能够自定义进度条显示的图案，好比能够把进度条中的方块换成#、=、@等你能够想获得的图案。

初始化

为了可以方便的调用该进度条工具，所以，为该结构提供了两个初始化的方法，分别为NewOption和NewOptionWithGraph,第二个初始化的方法便可以本身指定显示图案。
NewOption使用的是默认的显示图案，也就是上图展现的方框。其实现代码以下所示：

func (bar *Bar) NewOption(start, total int64) {
    bar.cur = start
    bar.total = total
    if bar.graph == "" {
        bar.graph = "█"
    }
    bar.percent = bar.getPercent()
    for i := 0; i < int(bar.percent); i += 2 {
        bar.rate += bar.graph //初始化进度条位置
    }
}

该函数提供了两个参数，分别为start和total，total不用说，它表明的是总的任务量，还提供了一个start参数，说明能够不从0开始，这也就意味着， 若是你的程序要支持断点续传功能，这个进度条工具依然能够完美支持，只须要将start值设置在断点处便可。固然了，若是你不须要断点续传，每次都从0开始，只须要将start值设置为0便可。
若是你注意到我在初始化进度条位置的时候，我使用了i += 2的步长，这就是我接下来要说的。由于百分比老是从0到100，而个人进度条长度最长为50个字符，这也就意味着，每增加2%，进度条就要涨一格，所以，这里的步长为2。
getPercent是一个根据cur和total获取当前进度完成百分比的一个函数，其实现比较简单：

func (bar *Bar) getPercent() int64 {
    return int64(float32(bar.cur) / float32(bar.total) * 100)
}

第二个初始化函数就比较容易实现了，只须要把graph从新覆盖以后，直接调用上面的初始化函数便可。

func (bar *Bar) NewOptionWithGraph(start, total int64, graph string) {
    bar.graph = graph
    bar.NewOption(start, total)
}

进度条展现

那么，如何实现显示功能呢？
通常调用显示进度条时，都是放在循环中执行的，所以，咱们只须要在循环中可以展现出每轮循环当前的进度状态便可。

func (bar *Bar) Play(cur int64) {
    bar.cur = cur
    last := bar.percent
    bar.percent = bar.getPercent()
    if bar.percent != last && bar.percent%2 == 0 {
        bar.rate += bar.graph
    }
    fmt.Printf("\r[%-50s]%3d%%  %8d/%d", bar.rate, bar.percent, bar.cur, bar.total)
}

这段代码中，最重要的就是最后的使用fmt.Printf打印的那一句，经过\r控制打印效果。
固然了，在构建rate进度条时，我须要保存上一次完成的百分比，只有当百分比发生了变化，且步长变化了2时，才须要改变进度条的长度。若是你的屏幕足够大，你也可让你的进度条长度为100个字符，这样，你就不须要控制进度条的步长为2了，每增加1%，进度条前进1格，也是没有问题的。

结束

因为上面的打印没有打印换行符，所以，在进度所有结束以后（也就是跳出循环以外时），须要打印一个换行符，所以，封装了一个Finish函数，该函数纯粹的打印一个换行，表示进度条已经完成。

func (bar *Bar) Finish(){
    fmt.Println()
}

如何调用

调用该进度条功能，首先，确定要构建一个Bar对象，使用该对象进行初始化后，便可完成进度条的调用了，一个完整的调用程序以下所示：

func main(){
    var bar progressbar.Bar    
    bar.NewOption(0, 100)
    for i:= 0; i<=100; i++{
        time.Sleep(100*time.Millisecond)
        bar.Play(int64(i))
    }
    bar.Finish()
}

以上是一个最简单的调用，其运行效果以下所示：

固然了，你也可使用另外一个初始化函数指定显示的图标，以下所示：

bar.NewOptionWithGraph(0, 100, "#")

展现效果则以下所示：

固然，实际使用中，你太可能只利用睡眠，而是须要实现本身的函数功能，只须要将time.Sleep(100*time.Millisecond)换成本身的代码逻辑便可。