GO中使用select实现带优先级的channel队列

一般Go语言同时使用多个channel的方法是使用select/case语句配合<-操作符，比如

1
2
3
4
5
6


select {
case <- chan1:
    // do something
case <- chan2:
    // do something
}

但是这种实现方式下chan1和chan2是同等优先级的。如果要实现带优先级的channel则需要用到defalut语句。

在go语言中，如果select/case中没有default子句，则程序会阻塞在select中，直到其中一个case语句接收到了数据。如果有default语句，则不会阻塞，如果case接收到数据，就执行case中的语句，如果case未收到信号，则会执行defalut中的语句，随后跳出select块。

使用这个特性可以实现带优先级的channel队列。以2个优先级的channel举例，实现方法是使用多层select，将高优先级channel放在最外层select语句的case后，并跟上一个default语句以免当高优先级的channel没有数据时阻塞。 defalut内依然是一个select语句，在这个select语句中，将高优先级和低优先级的case都放入，并且没有default语句。这样内层select就会阻塞直到其中一个case收到数据。

这种实现方式相当于高优先级的channel比低优先级的多了一次被处理的机会，即外层select，只有高优先级没有数据时，才会执行内层select，此时先产生数据的channel先被执行。

也就是说，当高优先级和低优先级都有数据时，高优先级先被处理，也就是实现了优先级。示例如下：

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13


for {
    select {
    case data := <- highChan:
        handleHigh(data)
    default:
        select {
        case data := <- highChan:
            handleHigh(data)
        case data := <- lowChan:
            handleLow(data)
        }
    }
}

文章目录