Fork me on GitHub

版权声明 本站原创文章 由 萌叔 发表
转载请注明 萌叔 | http://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)
    }
}


如果我的文章对你有帮助,你可以给我打赏以促使我拿出更多的时间和精力来分享我的经验和思考总结。

微信支付码

发表评论

电子邮件地址不会被公开。

此站点使用Akismet来减少垃圾评论。了解我们如何处理您的评论数据