基于数组阻塞队列 ArrayBlockingQueue 的一个队列工具类
java语言基于ArrayBlockingQueue 开发的一个根据特定前缀和后缀的队列。每天自动循环生成。
1.定义队列基类 Cookie
package com.bytter.util.queue;
import java.util.Date;
import java.util.concurrent.ArrayBlockingQueue;
public class Cookie {
private ArrayBlockingQueue<String> queue;
private int cursor = 1;
private Date date = new Date();
private int maxPoolQueue = 500;
private int mixPoolQueue = 100;
public Cookie(int maxPoolQueue,int mixPoolQueue,Date date){
this.maxPoolQueue=maxPoolQueue;
this.mixPoolQueue=mixPoolQueue;
this.date=date;
this.queue = new ArrayBlockingQueue<String>(this.maxPoolQueue,true);
}
public Cookie(int maxPoolQueue,int mixPoolQueue,Date date, int cursor){
this.maxPoolQueue = maxPoolQueue;
this.mixPoolQueue = mixPoolQueue;
this.date=date;
this.cursor=cursor;
this.queue = new ArrayBlockingQueue<String>(this.maxPoolQueue,true);
}
public ArrayBlockingQueue<String> getQueue() {
return queue;
}
public Cookie setQueue(ArrayBlockingQueue<String> queue) {
this.queue = queue;
return this;
}
public int getCursor() {
return cursor;
}
public Cookie setCursor(int cursor) {
this.cursor = cursor;
return this;
}
public Date getDate() {
return date;
}
public Cookie setDate(Date date) {
this.date = date;
return this;
}
public int getMaxPoolQueue() {
return maxPoolQueue;
}
public Cookie setMaxPoolQueue(int maxPoolQueue) {
this.maxPoolQueue = maxPoolQueue;
return this;
}
public int getMixPoolQueue() {
return mixPoolQueue;
}
public Cookie setMixPoolQueue(int mixPoolQueue) {
this.mixPoolQueue = mixPoolQueue;
return this;
}
}
2.QueueUtil 队列工具类，用于获取队列中的值(主要是获取付款的单号==)
package com.bytter.util.queue;
import java.util.Date;
import java.util.List;
import java.util.concurrent.ConcurrentHashMap;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import com.bytter.framework.persistence.dao.IBaseDAO;
import com.bytter.util.AppContextUtil;
import com.bytter.util.DateUtil;
/**
*
* @author
* May 6, 2019 10:35:39 AM
* @version
*/
public class QueueUtil{
private static Logger log = LogManager.getLogger(QueueUtil.class);
protected static IBaseDAO baseDao = null;
private static ConcurrentHashMap<String, Cookie> queueMap = new ConcurrentHashMap<String, Cookie>();
private static class SingletonClassInstance{
private static final QueueUtil instance=new QueueUtil();
}
private QueueUtil(){}
public static QueueUtil getInstance(){
return SingletonClassInstance.instance;
}
/**
* 调用事例： QueueUtil.getInstance().getSequence(SequenceEnum.BIS_EXC)
* @param sequenceEnum
* @return
* @throws InterruptedException
*/
@SuppressWarnings("unchecked")
public String getSequence(SequenceEnum sequenceEnum) throws InterruptedException{
String value = "";
String key = sequenceEnum.getTableName()+"_"+sequenceEnum.getFieldName();
Cookie cookie = queueMap.get(key);
value = cookie.getQueue().take();
String returnValue = sequenceEnum.getPrefix()+DateUtil.dateToStr(sequenceEnum.getDateFormat(), new Date());
for (int i = 0; i < sequenceEnum.getSequenceLength()-value.length(); i++) {
returnValue += "0";
}
returnValue += value;
log.info("取出的单号："+returnValue);
return returnValue;
}
public void initQueue() {
putAllMap();
Thread thread = new Produce(queueMap);
thread.setName("QueueUtil_Produce");
thread.start();
}
private static void putAllMap() {
queueMap.clear();
int curor = 1;
log.info("-----开始加载队列 -------");
for (SequenceEnum sequenceEnum : SequenceEnum.values()){
String key = sequenceEnum.getTableName()+"_"+sequenceEnum.getFieldName();
//同一天但是系统重启或者宕机的情况时，游标重置为1.此时需要查询表中最大的值并且置游标值+1
String sql="SELECT MAX("+sequenceEnum.getFieldName()+") FROM "+sequenceEnum.getTableName()+" WHERE "+sequenceEnum.getFieldName()+" LIKE '"+sequenceEnum.getPrefix()+"%'";
List list = baseDao.search_sql(null, sql, "", "", null);
if(list != null && list.size() >0 && list.get(0) != null){
String no = list.get(0).toString();
//单号，自增。
String count = no.substring(no.length()-sequenceEnum.getSequenceLength());
String date = no.replaceAll(count, "").replaceAll(sequenceEnum.getPrefix(), "");
/*if(date.equals(DateUtil.dateToStr(sequenceEnum.getDateFormat(), new Date()))){
curor = Integer.parseInt(count) + 1;
}*/
}
log.info("加载队列 :table=="+sequenceEnum.getTableName()+" field=="+sequenceEnum.getFieldName()+" 游标=="+curor);
queueMap.put(key, new Cookie(sequenceEnum.getMaxSeqSize(),sequenceEnum.getMinSeqSize(),new Date(),curor));
}
log.info("-----队列加载完毕 -------");
}
static{
baseDao = AppContextUtil.getBean("baseDao");
//第一次调用时开启队列
//initQueue();
}
}
3.生产者线程类
package com.bytter.util.queue;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Iterator;
import java.util.Map;
import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.ConcurrentHashMap;
import com.bytter.util.DateUtil;
public class Produce extends Thread{
private ConcurrentHashMap<String, Cookie> queueMap;
public Produce(ConcurrentHashMap<String, Cookie> queueMap){
this.queueMap=queueMap;
}
public void run(){
ArrayBlockingQueue<String> queue;
int mixSize;
int curor;
Cookie cookie;
Date date;
Map.Entry<String,Cookie> entry;
Iterator<Map.Entry<String,Cookie>> iteratorMap;
while (true) {
iteratorMap = queueMap.entrySet().iterator();
while (iteratorMap.hasNext()){
entry = iteratorMap.next();
cookie = entry.getValue();
queue = cookie.getQueue();
mixSize = cookie.getMixPoolQueue();
curor = cookie.getCursor();
date = cookie.getDate();
//新的一天，游标开始重置
if(!DateUtil.dateToStr("yyyy-MM-dd", new Date()).equals(DateUtil.dateToStr("yyyy-MM-dd", date))){
curor = 1;
queue.clear();
date = new Date();
System.out.println("重置游标时间===="+new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(new Date()));
}
//offer 在填满后返回false
if(queue.size() < mixSize){
while (queue.offer(curor + "")) {
curor++;
}
}
cookie.setQueue(queue).setCursor(curor).setDate(date);
queueMap.put(entry.getKey(),cookie);
}
}
}
}
4.自定义的生成单号的规则枚举类
package com.bytter.util.queue;
public enum SequenceEnum{
/**
* 付款指令编码 yyMMdd000001
* @return
*/
BIS_EXC("",6, "BIS_EXC", "VOUCHER_NO", "yyMMdd",500,100),
/**
* 付款单号
* @return
*/
PAYMENT_BILL_NO("P",7, "T_BIS_PAYMENT_INFO", "BILL_CODE", "yyMMddHHmmss",500,100),
;
/**
* 流水号前缀
*/
private String prefix;
/**
* 流水号数字部分长度
*/
private Integer sequenceLength;
/**
* 业务表名
*/
private String tableName;
/**
* 业务表中的字段名
*/
private String fieldName;
/**
* 日期格式化模板
*/
private String dateFormat;
/**
* 队列最大的长度
*/
private int maxSeqSize;
/**
* 队列最小长度
*/
private int minSeqSize;
/**
* @param prefix 流水号前缀
* @param sequenceLength 流水号自增长长度
* @param tableName 表名
* @param fieldName 字段名
* @param dateFormat 日期格式化模板
*/
SequenceEnum(String prefix, Integer sequenceLength, String tableName, String fieldName,String dateFormat,int maxSeqSize,int minSeqSize) {
this.prefix = prefix;
this.sequenceLength = sequenceLength;
this.tableName = tableName;
this.fieldName = fieldName;
this.dateFormat = dateFormat;
this.maxSeqSize = maxSeqSize;
this.minSeqSize = minSeqSize;
}
/**
* @param prefix 流水号前缀
* @param sequenceLength 流水号自增长长度
* @param tableName 表名
* @param fieldName 字段名
* @param dateFormat 日期格式化模板
*/
SequenceEnum(Integer sequenceLength, String tableName, String fieldName,String dateFormat) {
this.sequenceLength = sequenceLength;
this.tableName = tableName;
this.fieldName = fieldName;
this.dateFormat = dateFormat;
}
public String getPrefix() {
return prefix;
}
public void setPrefix(String prefix) {
this.prefix = prefix;
}
public Integer getSequenceLength() {
return sequenceLength;
}
public void setSequenceLength(Integer sequenceLength) {
this.sequenceLength = sequenceLength;
}
public String getTableName() {
return tableName;
}
public void setTableName(String tableName) {
this.tableName = tableName;
}
public String getFieldName() {
return fieldName;
}
public void setFieldName(String fieldName) {
this.fieldName = fieldName;
}
public String getDateFormat() {
return dateFormat;
}
public void setDateFormat(String dateFormat) {
this.dateFormat = dateFormat;
}
public int getMaxSeqSize() {
return maxSeqSize;
}
public void setMaxSeqSize(int maxSeqSize) {
this.maxSeqSize = maxSeqSize;
}
public int getMinSeqSize() {
return minSeqSize;
}
public void setMinSeqSize(int minSeqSize) {
this.minSeqSize = minSeqSize;
}
}
5.测试相关类
1> 消费者线程
package com.bytter.util.queue;
public class implements Runnable{
public void run(){
try {
for (int i = 0; i < 200; i++) {
Thread.sleep(Long.valueOf((long) (Math.random()*10*300)));
// System.out.println(Thread.currentThread().getName());
System.out.println(Thread.currentThread().getName()+"==="+SequenceUtilTest.getInstance().getSequence(SequenceEnum.BIS_EXC));
}
} catch (Exception e) {
// TODO 自动生成的 catch 块
e.printStackTrace();
}
}
}
2>测试main 方法
package com.bytter.util.queue;
public class tests {
public static void main(String[] args) throws InterruptedException {
new Thread(new Consume()).start();
new Thread(new Consume()).start();
new Thread(new Consume()).start();
new Thread(new Consume()).start();
new Thread(new Consume()).start();
// new Thread(new Consume()).start();
// new Thread(new Consume()).start();
// new Thread(new Consume()).start();
// new Thread(new Consume()).start();
// new Thread(new Consume()).start();
}
}
