服务器最近压测的时候发现了问题。发现是往channel写数据,但是channel满了,导致服务器部分功能阻塞。golang的channel读或写是会造成阻塞的,但是可以用select的多路复用解决这个问题。
不阻塞读channel(也可以加上超时)
func readChan(c chan int) (int, error) {
select {
case num := <-c:
return num, nil
default:
return 0, errors.New("chan do not have data")
}
}
// 加上超时时间
func readChanWithTimeout(c chan int) (int, error) {
timeout := time.NewTimer(time.Microsecond * 100)
select {
case num := <-c:
return num , nil
case <-timeout.C:
return 0, errors.New("read chan time out")
}
}
非阻塞写入channel(也可以加上超时)
func writeChan(num int, c chan int) error {
select {
case ch <- num:
return nil
default:
return errors.New("chan is full")
}
}
// 加上超时时间
func writeChanWithTimeout(num int, c chan int) error {
timeout := time.NewTimer(time.Microsecond * 100)
select {
case c <- num:
return nil
case <-timeout.C:
return errors.New("write chan time out")
}
}
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)