在别的地方看到过,觉得挺有意思的,于是亲自实现一次。
原理很简单,就是利用通道,在主线程开始执行耗操作时,另起一个goroutine在同一个位置循环打印-\|/
形成等待动画,然后在耗时协程结束后给通道一个信号量,在动画函数中一旦接收到信号,便停止打印并退出。
package main
import (
"fmt"
"time"
)
func main() {
sig := make(chan bool)
go spinner(time.Duration(time.Millisecond*40), sig)
time.Sleep(4 * time.Second) // 耗时任务...
sig <- true
}
func spinner(delay time.Duration, done chan bool) {
for {
for _, i := range "-\\|/" {
select {
case <-done: //通道接收到信号退出
return
case <-time.After(delay): //在一定时间内没有收到信号
fmt.Printf("\r%c", i)
}
}
}
}
代码的意义大概是在一些比较耗时的操作(比如统计vcenter集群下的设备数量)让用户知道程序在运行,而不是没反应。