func randomChain() chan interface{} { data := Data{0} nc := make(chan interface{},4096) count := int64(0) go func() { ticker := time.NewTicker(1 * time.Second) for { select { case <-ticker.C: fmt.Println("Per second: ", count, len(nc)) count = 0 default: nc <- &data count++ } } }() go func() { ticker := time.NewTicker(1 * time.Second) for { select { case <-ticker.C: fmt.Println("Per second: ", count, len(nc)) count = 0 default: nc <- &data count++ } } }() // 在程序退出时打印管道中的剩余数据数量 c := make(chan os.Signal, 1) signal.Notify(c, os.Interrupt, syscall.SIGTERM) go func() { <-c fmt.Println("Remaining data count:", len(nc)) os.Exit(1) }() return nc } 优化
时间: 2024-03-02 17:51:28 浏览: 18
这段代码中存在两个重复的 goroutine,它们的代码实现是完全一样的,可以将它们合并成一个。另外,可以将计数器和 ticker 的定义提到外部函数中,避免重复定义。最后,在程序退出时,可以直接使用 defer 关键字来打印管道中的剩余数据数量,避免使用 signal 包。
下面是优化后的代码:
```
func randomChain() chan interface{} {
data := Data{0}
nc := make(chan interface{}, 4096)
count := int64(0)
ticker := time.NewTicker(1 * time.Second)
go func() {
for {
select {
case <-ticker.C:
fmt.Println("Per second: ", count, len(nc))
count = 0
default:
nc <- &data
count++
}
}
}()
// 在程序退出时打印管道中的剩余数据数量
defer func() {
fmt.Println("Remaining data count:", len(nc))
}()
return nc
}
```