如何将此代码转换为非阻塞&无锁?

时间:2014-02-21 10:21:26

标签: synchronization go queue nonblocking

我有队列,女巫必须有不断增长的缓冲区,这不包括缓冲频道的使用。

经过一些谷歌搜索后,我想出了这段代码:

import (
    "sync"
)

type Queue struct {
    nodes []interface{}
    head, tail, count int
    lck sync.RWMutex
}

func (q *Queue) Enqueue(v interface{}) {
    q.lck.Lock()
    defer q.lck.Unlock()

    if q.nodes == nil {
        q.nodes = make([]interface{}, 2)
    }

    if q.head == q.tail && q.count > 0 {
        nodes := make([]interface{}, len(q.nodes) * 2)
        copy(nodes, q.nodes[q.head:])
        copy(nodes[len(q.nodes) - q.head:], q.nodes[:q.head])
        q.head = 0
        q.tail = len(q.nodes)
        q.nodes = nodes
    }

    q.nodes[q.tail] = v
    q.tail = (q.tail + 1) % len(q.nodes)
    q.count++
}

func (q *Queue) Dequeue() interface{} {
    q.lck.Lock()
    defer q.lck.Unlock()

    if len(q.nodes) == 0 {
        return nil
    }

    node := q.nodes[q.head]
    q.head = (q.head + 1) % len(q.nodes)
    q.count--

    return node
}

func (q *Queue) Len() int {
    q.lck.RLock()
    defer q.lck.RUnlock()

    return q.count
}

我有什么方法可以将其转换为非阻塞& amp;无锁队列?

1 个答案:

答案 0 :(得分:2)

Evan Huus's channels package提供了ResizableChannel类型,它似乎提供了您所追求的内容。

  
    

ResizableChannel实现Channel接口,在输入和输出之间有一个可调整大小的缓冲区。通道最初的缓冲区大小为1,但可以通过调用Resize()来调整大小。