最近需要使用到一個Queue隊列, 并發(fā)寫入的.
作為gopher, 第一個馬上想到的是channel, channel用起來是很爽, 并發(fā)安全, 直接一端取, 一端塞, 非常和諧友好. 但是, channel不支持動態(tài)擴容啊, 當一個channel滿了之后, 想要擴容, 只能make一個新的, 然后做一個遷移, 這會卡住好長時間, 業(yè)務需求不滿足.
然后找到了container/list這個包, 有個list結(jié)構(gòu)體, 使用簡單, 接口清晰, 代碼實現(xiàn)也簡單, 用著也挺舒服的. 就是不是并發(fā)安全的, 只能自己動手包了一層并發(fā)的皮:

import (
    "container/list"
    "sync"
)

type Queue struct {
    l *list.List
    m sync.Mutex
}

func NewQueue() *Queue {
    return &Queue{l: list.New()}
}

func (q *Queue) PushBack(v interface{}) {
    if v == nil {
        return
    }
    q.m.Lock()
    defer q.m.Unlock()
    q.l.PushBack(v)
}

func (q *Queue) Front() *list.Element {
    q.m.Lock()
    defer q.m.Unlock()
    return q.l.Front()
}

func (q *Queue) Remove(e *list.Element) {
    if e == nil {
        return
    }
    q.m.Lock()
    defer q.m.Unlock()
    q.l.Remove(e)
}

func (q *Queue) Len() int {
    q.m.Lock()
    defer q.m.Unlock()
    return q.l.Len()
}

這樣就基本滿足了.

使用過程中, 發(fā)現(xiàn)list有個小坑: 遍歷的時候不能Remove

for e := l.Front(); e != nil; e = e.Next {
    l.Remove(e)
}

按照設想, 這應該會移除list里所有的元素, 但是, 結(jié)果是只移除了第一個. 原因是: Remove時候, e.next = nil, 使得for判斷中, e != nil不成立了, 所以退出了循環(huán).
這時候有兩種解決辦法:

var next *list.Element
for e := l.Front(); e != nil; e = next {
    next = e.Next()
    l.Remove(e)
}

for {
    e := l.Front()
    if e == nil {
            break
    }
    l.Remove(e)
}

that's all