golang 线程安全队列 读取函数带超时功能

type Deque struct {
    sync.RWMutex
    notEmptyNotify        chan int 
    container *list.List
}
func (s *Deque) Put(item interface{}) {
    s.Lock()
    s.container.PushFront(item)
    s.Unlock()
    select {
    case s.notEmptyNotify <-1:
    default:
    }   
}
func (s *Deque) Get(timeout int) (interface{}, error) {
    s.Lock()
    var item interface{} = nil
    var lastContainerItem *list.Element = nil
    endTime := time.Now().Add(time.Duration(timeout) * time.Second)
    for {
        if s.container.Back() != nil {
            break
        }
        remaining := endTime.Sub(time.Now())
        s.Unlock()
        if remaining < 0 {
            return nil, errors.New("time out in Get")
        }
        select {
        case <-s.notEmptyNotify:
        case <-time.After(remaining):
        return nil, errors.New("time out in Get")
        }
        s.Lock()
    }
    lastContainerItem = s.container.Back()
    item = s.container.Remove(lastContainerItem)
    s.Unlock()
    return item, nil
}

 
queueFreeServer := Deque{ container: list.New(),  notEmptyNotify: make(chan int)}

最近提供人像分割服务给业务方时, 发现需要有一个服务分发请求, 开始用python flask框架发现问题挺多, 用golang重写时发现网上实现的golang版队列, 都没有像python那样的Get超时功能. 于是结合网上代码和python queue代码自己写了一个, 看来golang还没完全忘记.

队列不支持设定固定size, 队列满时Put等待和Put超时, 需要的朋友可以自己扩展一下.

github地址: https://github.com/bolun365/golang-queue.git