Wait, notify实现的生产消费模型

最新推荐文章于 2025-09-06 09:33:29 发布

转载最新推荐文章于 2025-09-06 09:33:29 发布 · 384 阅读

文章标签：

#java

java 专栏收录该内容

35 篇文章

订阅专栏

本文介绍了一个基于Java的任务管理器的设计与实现细节。该任务管理器采用单例模式，并利用线程池来处理任务队列中的BaseTask实例。通过TaskExecutor类实现了任务的异步执行，同时使用了同步机制来确保线程安全。

public class TaskManager {

	private TaskManager(){
		
	}
	
	private final static TaskManager instance = new TaskManager();
	
	public static TaskManager getInstance(){
		return instance;
	}
	
	private ExecutorService executor = null;
	
	private ConcurrentLinkedQueue<BaseTask> taskQueue = new ConcurrentLinkedQueue<BaseTask>();
	
	private TaskExecutor taskExecutor;
	
	public void addTask(BaseTask task){
		taskQueue.add(task);
		taskExecutor.notifyTaskAdded();
	}
	
	public BaseTask popTask(){
		return taskQueue.poll();
	}
	
	public int getTaskSize(){
		return taskQueue.size();
	}
	
	public void startService(){
		executor = Executors.newFixedThreadPool(1);
		taskExecutor = new TaskExecutor();
		executor.execute(taskExecutor);
	}
	
	public void stopService(){
		
	}
}

public class TaskExecutor implements Runnable{

	private boolean isStarted = true;
	
	private Object obj = new Object();
	
	@Override
	public void run() {
		while(isStarted){
			int taskSize = TaskManager.getInstance().getTaskSize();
			synchronized(obj){
				if(taskSize == 0){
					try {
						LogUtil.logDebug("There is no task to execute now, waiting...");
						obj.wait();
					} catch (InterruptedException e) {
					}
				}
			}
			BaseTask task = TaskManager.getInstance().popTask();
			if(task != null){
				LogUtil.logDebug("execute one task...");
				task.run();
			}
		}
	}
	
	public void notifyTaskAdded(){
		synchronized(obj){
			obj.notify();
		}
	}

	public void stop(){
		this.isStarted = false;
	}
}