需求:
最近要实现一个自动派单功能,就是如果有任务来了,那么任务平均分配给现有的员工,如果之前取了任务的员工下次来任务时就分配给下一个,如果有新员工则将员工添加到队列的最后,如果有员工离职就将员工信息从队列里移除。
实现:
1.写一个定时任务,定时将任务放入任务队列中TaskQueue,并将员工放入User队列中,如果员工已存在则不添加,如果新员工则加入队列中,如果员工不存在了就从队列中移除。
2.写一个线程,从任务队列中取任务,如果任务不存在则阻塞,取到任务后,将该任务派给下一个员工,并将该员工添加到员工队列末尾,这样达到按员工次序分配任务。
3.实现一个任务队列,如果在取任务时任务不存在则阻塞。
4.实现一个员工队列,在取下一个员工后将该员工从列头移到列尾,并且可以将新员工加入到队列未尾,可以将不存在的员工从队列中移除。
核心代码:
import java.util.Collection;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.LinkedBlockingDeque;
/**
* 任务队列
* 如果在取任务时没有任务,则阻塞
* @author dg
*
*/
public class TaskQueue {
private static TaskQueue instance;
private static BlockingQueue queue = null;
private TaskQueue(){
queue = new LinkedBlockingDeque();
}
public static TaskQueue getInstance(){
if(instance == null){
instance = new TaskQueue();
}
return instance;
}
/**
* 获取队列大小
* @return
*/
public static int getSize(){
return queue.size();
}
/**
* 放入队列中
* @param msg
*/
public void put(Object obj){
if(obj != null){
try {
queue.put(obj);
} ca