Golang 实现的线程安全的队列

Dec 29, 2015  
本文系原创作品,转载请与 henson.lu(AT)gmail.com 联系!

Golang 的 channel 可以作为线程安全的队列来使用,但是只能是固定大小的,如果填满的 channel,就只能阻塞的等待。

goqueue 就是来解决这个问题的,参考了 Python 里面的 Queue(Py3 queue)模块。

详见 https://github.com/Damnever/goqueue

package main

import (
"github.com/Damnever/goqueue"
"fmt"
)

func main() {
    queue := goqueue.NewQueue(0)

    worker := func(queue *goqueue.Queue) {
        for {
            e, err := queue.Get(true, 0)
            if err != nil {
                fmt.Println("Unexpect Error: %v\n", err)
            }
            num := e.Value.(int)
            fmt.Printf("-> %v\n", num)
            queue.TaskDone()
            if num % 3 == 0 {
                for i := num + 1; i < num + 3; i ++ {
                     queue.PutNoWait(i)
                }
            }
        }
    }

    for i := 0; i <= 27; i += 3 {
        queue.PutNoWait(i)
    }

    for i := 0; i < 5; i++ {
        go worker(queue)
    }

    queue.WaitAllComplete()
    fmt.Println("All task done!!!")
}