Go 言語の同時データ構造: キューとスタックのパフォーマンスの最適化

type ConcurrentQueue struct {
    m     sync.Mutex
    items []interface{}
    conds sync.Cond
}

func (q *ConcurrentQueue) Enqueue(v interface{}) {
    q.m.Lock()
    defer q.m.Unlock()
    q.items = append(q.items, v)
    q.conds.Signal()
}

func (q *ConcurrentQueue) Dequeue() interface{} {
    q.m.Lock()
    defer q.m.Unlock()
    var v interface{}
    if len(q.items) > 0 {
        v = q.items[0]
        q.items = q.items[1:]
    }
    return v
}

type ConcurrentStack struct {
    m sync.Mutex
    top *node
}

type node struct {
    data interface{}
    next *node
}

func (s *ConcurrentStack) Push(v interface{}) {
    s.m.Lock()
    defer s.m.Unlock()
    n := &node{data: v}
    n.next = s.top
    s.top = n
}

func (s *ConcurrentStack) Pop() interface{} {
    s.m.Lock()
    defer s.m.Unlock()
    if s.top == nil {
        return nil
    }
    v := s.top.data
    s.top = s.top.next
    return v
}

func main() {
    q := ConcurrentQueue{}
    s := ConcurrentStack{}

    for i := 0; i < 1000; i++ {
        // 向队列中并发添加任务
        go func(i int) {
            q.Enqueue(i)
        }(i)
    }

    for i := 0; i < 1000; i++ {
        // 从队列中并发获取任务并推入栈中
        go func() {
            if v := q.Dequeue(); v != nil {
                s.Push(v)
            }
        }()
    }

    for i := 0; i < 1000; i++ {
        // 从栈中弹出任务并处理
        go func() {
            if v := s.Pop(); v != nil {
                // 处理任务 v
            }
        }()
    }
}