q.go

package utee

import (
	"errors"
	"log"
)

var ErrFull = errors.New("queue is full")

// MemQueue memory queue
type MemQueue chan interface{}

// NewMemQueue create memory queue
func NewMemQueue(cap int) MemQueue {
	return make(chan interface{}, cap)
}

// EnqBlocking enqueue, block if queue is full
func (p MemQueue) EnqBlocking(data interface{}) {
	p <- data
}

// Enq enqueue, return error if queue is full
func (p MemQueue) Enq(data interface{}) error {
	select {
	case p <- data:
	default:
		return ErrFull
	}

	return nil
}

// Deq
func (p MemQueue) Deq() interface{} {
	return <-p
}

// DeqN dequeue less than n in a batch
func (p MemQueue) DeqN(n int) []interface{} {
	if n <= 0 {
		log.Println("[MemQueue] deqn err, n must > 0")
		return nil
	}

	var l []interface{}

	for {
		select {
		case data := <-p:
			l = append(l, data)
			if len(l) == n {
				return l
			}
		default:
			return l
		}
	}
}

// Len queue length
func (p MemQueue) Len() int {
	return len(p)
}

// Cap queue capacity
func (p MemQueue) Cap() int {
	return cap(p)
}