使用golang打印进度百分比
在golang中,我们经常需要处理一些需要耗费时间的任务,例如文件上传、数据处理等等。在这些任务中,我们经常会需要显示任务的进度百分比,以便给用户一个直观的反馈。本文将介绍如何使用golang来实现打印进度百分比的功能。
使用goroutine和channel进行任务分发
首先,我们可以将任务拆分成多个子任务,并使用goroutine来并发执行这些子任务。我们可以使用一个channel来接收每个子任务完成的消息。
```go package main import ( "fmt" "time" ) func worker(id int, jobs <-chan int,="" results="">-chan><- int)="" {="" for="" j="" :="range" jobs="" {="" fmt.println("worker",="" id,="" "started="" job",="" j)="" time.sleep(time.second)="" 模拟任务耗时="" fmt.println("worker",="" id,="" "finished="" job",="" j)="" results="">-><- j="" *="" 2="" }="" }="" func="" main()="" {="" jobs="" :="make(chan" int,="" 100)="" results="" :="make(chan" int,="" 100)="" 启动多个worker并发执行任务="" for="" w="" :="1;" w="">-><= 3;="" w++="" {="" go="" worker(w,="" jobs,="" results)="" }="" 分发任务="" for="" j="" :="1;" j="">=><= 9;="" j++="" {="" jobs="">=><- j="" }="" close(jobs)="" 收集结果="" for="" a="" :="1;" a="">-><= 9;="" a++="" {="">=><-results }="" }="" ```="">-results>在上面的代码中,我们定义了一个`worker`函数来执行子任务。每个子任务的执行过程中,我们打印出任务的开始和结束信息,并将结果发送到`results`通道中。
计算任务进度百分比
接下来,我们可以使用一个`progress`变量来统计任务的进度。我们可以通过`atomic`包提供的`AddInt64`函数来原子操作地增加`progress`变量的值。
```go package main import ( "fmt" "sync" "sync/atomic" "time" ) func worker(id int, jobs <-chan int,="" results="">-chan><- int,="" progress="" *int64,="" total="" int)="" {="" for="" j="" :="range" jobs="" {="" fmt.println("worker",="" id,="" "started="" job",="" j)="" time.sleep(time.second)="" 模拟任务耗时="" fmt.println("worker",="" id,="" "finished="" job",="" j)="" results="">-><- j="" *="" 2="" atomic.addint64(progress,="" 1)="" 增加进度="" }="" }="" func="" main()="" {="" jobs="" :="make(chan" int,="" 100)="" results="" :="make(chan" int,="" 100)="" progress="" :="new(int64)" total="" :="9" 总任务数="" 启动多个worker并发执行任务="" for="" w="" :="1;" w="">-><= 3;="" w++="" {="" go="" worker(w,="" jobs,="" results,="" progress,="" total)="" }="" 分发任务="" for="" j="" :="1;" j="">=><= total;="" j++="" {="" jobs="">=><- j="" }="" close(jobs)="" 收集结果="" for="" a="" :="1;" a="">-><= total;="" a++="" {="">=><-results }="" 计算进度百分比="" p="" :="atomic.LoadInt64(progress)" fmt.println("progress:",="" p*100/int64(total),="" "%")="" }="" ```="">-results>在上面的代码中,我们使用`sync/atomic`包提供的原子操作函数来增加`progress`变量的值。最后,我们通过除以总任务数,计算出了任务的进度百分比。
打印进度百分比
为了实时显示任务的进度百分比,我们可以使用一个定时器和`fmt.Printf`函数来打印进度信息。
```go package main import ( "fmt" "sync" "sync/atomic" "time" ) func worker(id int, jobs <-chan int,="" results="">-chan><- int,="" progress="" *int64,="" total="" int)="" {="" for="" j="" :="range" jobs="" {="" fmt.println("worker",="" id,="" "started="" job",="" j)="" time.sleep(time.second)="" 模拟任务耗时="" fmt.println("worker",="" id,="" "finished="" job",="" j)="" results="">-><- j="" *="" 2="" atomic.addint64(progress,="" 1)="" 增加进度="" }="" }="" func="" main()="" {="" jobs="" :="make(chan" int,="" 100)="" results="" :="make(chan" int,="" 100)="" progress="" :="new(int64)" total="" :="9" 总任务数="" 启动多个worker并发执行任务="" for="" w="" :="1;" w="">-><= 3;="" w++="" {="" go="" worker(w,="" jobs,="" results,="" progress,="" total)="" }="" 分发任务="" for="" j="" :="1;" j="">=><= total;="" j++="" {="" jobs="">=><- j="" }="" close(jobs)="" 打印进度百分比="" go="" func()="" {="" start="" :="time.Now()" var="" ticker="time.Tick(time.Second)" for="" {="">-><-ticker p="" :="atomic.LoadInt64(progress)" fmt.printf("\rprogress:="" %d%%="" (%d/%d)="" elapsed="" time:="" %v",="" p*100/int64(total),="" p,="" total,="" time.since(start))="" if="" p="=" int64(total)="" {="" break="" }="" }="" }()="" 收集结果="" for="" a="" :="1;" a="">-ticker><= total;="" a++="" {="">=><-results }="" }="" ```="">-results>在上面的代码中,我们使用了一个定时器和`fmt.Printf`实时打印进度信息。我们使用了`\r`字符来实现将光标移动到行首的效果,这样可以实现覆盖之前打印的信息,最终显示最新的进度。
总结
通过使用goroutine、channel、原子操作和定时器,我们实现了在golang中打印进度百分比的功能。这个功能对于需要处理耗时任务的应用程序来说非常有用,可以给用户提供一个直观的任务进度反馈。

评论