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

select的基本用法

select会监听IO操作,当IO操作发生时会触发相应的动作,如果没有发生符合条件的IO操作则执行default分支,如没有default分支则一直阻塞。

ch1 := make(chan int,1)
ch2 := make(chan int,1)

select {
    case <- ch1:
        fmt.Println("recv from ch1")
    case <- ch2:
       fmt.Println("recv from ch2")
    default:
       fmt.Println("recv from default")
 }

实现优先级队列

我们可以使用select中的default特性实现优先级队列。
例如我们有a,b,c 这3个channel,我们要实现的功能是先执行从a队列收到的消息,其次执行b队列收到的消息,最后执行c队列收到的消息,那么执行顺序就是a>b>c

a := make(chan int, 1)
b := make(chan int, 1)
c := make(chan int, 1)

select {
    case <-a:
        fmt.Println("recv from a")
    default:
        select {
            case <-b:
                fmt.Println("recv from b")
            default:
                select {
                    case <-c:
                    fmt.Println("recv from c")
                }
        }
}

你可能感兴趣的:(go)