Channel在Golang中的有趣用法(1)-channel实现非阻塞队列

版权声明 本站原创文章 由 萌叔 发表
转载请注明 萌叔 | https://vearne.cc

前言

Channel是Golang中非常重要的数据结构, 默认它是阻塞的。那么如何实现一个非阻塞队列呢，你可以参考我的实现。

实现1

package main

import (
    "errors"
    "fmt"
)

func push(q chan int, item int) error {
    select {
    case q <- item:
        return nil
    default:
        return errors.New("queue full")
    }

}

func get(q chan int) (int, error) {
    var item int
    select {
    case item = <-q:
        return item, nil
    default:
        return 0, errors.New("queue empty")
    }

}

func main() {
    q := make(chan int, 5)
    x := []int{1, 2, 3, 4, 5, 6}
    for _, value := range x {
        err := push(q, value)
        fmt.Printf("error:%v\n", err)
    }

    for _, value := range x {
        fmt.Println(value)
        v, err := get(q)
        fmt.Printf("v:%v, error:%v\n", v, err)
    }
}

实现2

我们还可以把channel变成一个带超时的阻塞队列

package main

import (
    "errors"
    "fmt"
    "time"
)

func push(q chan int, item int, timeoutSecs int) error {

    select {
    case q <- item:
        return nil
    case <-time.After(time.Duration(timeoutSecs) * time.Second):
        return errors.New("queue full, wait timeout")
    }
}

func get(q chan int, timeoutSecs int) (int, error) {
    var item int
    select {
    case item = <-q:
        return item, nil
    case <-time.After(time.Duration(timeoutSecs) * time.Second):
        return 0, errors.New("queue empty, wait timeout")
    }

}

func main() {
    q := make(chan int, 5)
    x := []int{1, 2, 3, 4, 5, 6}
    for _, value := range x {
        err := push(q, value, 3)
        fmt.Printf("error:%v\n", err)
    }

    for _, value := range x {
        fmt.Println(value)
        v, err := get(q, 3)
        fmt.Printf("v:%v, error:%v\n", v, err)
    }
}

请我喝瓶饮料