golang 打印进度百分比

使用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 } } ```

在上面的代码中，我们定义了一个`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), "%") } ```

在上面的代码中，我们使用`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 <= total; a++ { <-results } } ```

在上面的代码中，我们使用了一个定时器和`fmt.Printf`实时打印进度信息。我们使用了`\r`字符来实现将光标移动到行首的效果，这样可以实现覆盖之前打印的信息，最终显示最新的进度。

总结

通过使用goroutine、channel、原子操作和定时器，我们实现了在golang中打印进度百分比的功能。这个功能对于需要处理耗时任务的应用程序来说非常有用，可以给用户提供一个直观的任务进度反馈。