package project.data;

import java.util.ArrayList;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.TimeUnit;

/**
 * @author JORGE
 * @description 通用队列
 * @param <R>
 */
@SuppressWarnings("serial")
public class CommonQueue<R> extends LinkedBlockingQueue<R>{
	/**
	 * 构造方法
	 */
	public CommonQueue(){}
	
	/**
	 * 构造方法
	 * @param capacity 队列尺寸
	 */
	public CommonQueue(Integer capacity){
		super(capacity);
	}
	
	/**
	 * 返回队列中元素数量
	 * @return 元素数量
	 */
	public int length(){
		return super.size();
	}
	
	/**
	 * 阻塞直到队列中有数据时返回并移除队首元素
	 * @return 泛化类型
	 */
	public R get(){
		try {
			return super.take();
		} catch (InterruptedException e) {
			e.printStackTrace();
		}
		return null;
	}
	
	/**
	 * 从队首开始返回并移除队列中给定数量的元素,若队列中没有指定数量的元素则返回队列中所有元素,若队列为空则返回NULL
	 * @param maxNumber 需要获取的元素数量(可选),若未给定batchSize参数则返回并移除队列中所有元素
	 * @return 元素集合,本方法快速返回、无阻塞
	 */
	public ArrayList<R> batchGet(Integer... batchSize){
		ArrayList<R> resultList=new ArrayList<R>();
		if(null==batchSize || 0==batchSize.length) {
			super.drainTo(resultList);
		}else{
			super.drainTo(resultList,batchSize[0]);
		}
		return 0==resultList.size()?null:resultList;
	}
	
	/**
	 * 等待给定的最大时间内返回并移除队列中队首元素,如果超时则返回NULL,
	 * 若未给定timeout参数则快速无阻塞返回队首元素(若队列为空则返回NULL)
	 * @param timeout 超时时间(单位:毫秒),即等待队列中有元素到来前的最大等待时间
	 * @return 泛化类型
	 */
	public R get(Long... timeout){
		try {
			if(null==timeout || 0==timeout.length) return super.poll();
			return super.poll(timeout[0], TimeUnit.MILLISECONDS);
		} catch (InterruptedException e) {
			e.printStackTrace();
		}
		return null;
	}
}