mirror of
https://gitee.com/johng/gf
synced 2026-06-06 02:25:47 +08:00
114 lines
2.6 KiB
Go
114 lines
2.6 KiB
Go
// Copyright 2017 gf Author(https://gitee.com/johng/gf). All Rights Reserved.
|
||
//
|
||
// This Source Code Form is subject to the terms of the MIT License.
|
||
// If a copy of the MIT was not distributed with this file,
|
||
// You can obtain one at https://gitee.com/johng/gf.
|
||
|
||
// 并发安全的动态队列.
|
||
// 优点：
|
||
// 1、队列初始化速度快；
|
||
// 2、可以向队头/队尾进行Push/Pop操作；
|
||
package gqueue
|
||
|
||
import (
|
||
"math"
|
||
"sync"
|
||
"errors"
|
||
"container/list"
|
||
)
|
||
|
||
type Queue struct {
|
||
mu sync.RWMutex
|
||
list *list.List // 数据队列
|
||
limit int // 队列限制大小
|
||
limits chan struct{} // 用于队列大小限制
|
||
events chan struct{} // 用于内部数据写入事件通知
|
||
closed bool // 队列是否关闭
|
||
}
|
||
|
||
// 队列大小为非必须参数，默认不限制
|
||
func New(limit...int) *Queue {
|
||
size := 0
|
||
if len(limit) > 0 {
|
||
size = limit[0]
|
||
}
|
||
return &Queue {
|
||
list : list.New(),
|
||
limits : make(chan struct{}, size),
|
||
events : make(chan struct{}, math.MaxInt64),
|
||
}
|
||
}
|
||
|
||
// 将数据压入队列, 队尾
|
||
func (q *Queue) PushBack(v interface{}) error {
|
||
q.mu.RLock()
|
||
if q.closed {
|
||
q.mu.RUnlock()
|
||
return errors.New("closed")
|
||
}
|
||
if q.limit > 0 {
|
||
q.limits <- struct{}{}
|
||
}
|
||
q.list.PushBack(v)
|
||
q.events <- struct{}{}
|
||
q.mu.RUnlock()
|
||
return nil
|
||
}
|
||
|
||
// 将数据压入队列, 队头
|
||
func (q *Queue) PushFront(v interface{}) error {
|
||
q.mu.RLock()
|
||
if q.closed {
|
||
q.mu.RUnlock()
|
||
return errors.New("closed")
|
||
}
|
||
if q.limit > 0 {
|
||
q.limits <- struct{}{}
|
||
}
|
||
q.list.PushFront(v)
|
||
q.events <- struct{}{}
|
||
q.mu.RUnlock()
|
||
return nil
|
||
}
|
||
|
||
// 从队头先进先出地从队列取出一项数据，当没有数据可获取时，阻塞等待
|
||
func (q *Queue) PopFront() interface{} {
|
||
<- q.events
|
||
if q.limit > 0 {
|
||
<- q.limits
|
||
}
|
||
if elem := q.list.Front(); elem != nil {
|
||
item := q.list.Remove(elem)
|
||
return item
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// 从队尾先进先出地从队列取出一项数据，当没有数据可获取时，阻塞等待
|
||
func (q *Queue) PopBack() interface{} {
|
||
<- q.events
|
||
if q.limit > 0 {
|
||
<- q.limits
|
||
}
|
||
if elem := q.list.Front(); elem != nil {
|
||
item := q.list.Remove(elem)
|
||
return item
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// 关闭队列(通知所有通过Pop*阻塞的协程退出)
|
||
func (q *Queue) Close() {
|
||
q.mu.Lock()
|
||
if !q.closed {
|
||
q.closed = true
|
||
close(q.limits)
|
||
close(q.events)
|
||
}
|
||
q.mu.Unlock()
|
||
}
|
||
|
||
// 获取当前队列大小
|
||
func (q *Queue) Size() int {
|
||
return len(q.events)
|
||
} |