public class Message {
private String data;
public Message(String data) {
this.data = data;
}
public String getData() {
return data;
}
}
//====================================
public class MessageQueue {
//任务队列
private final LinkedList<Message> queue;
//默认上限值
private final static int DEFAULT_MAX_LIMIT = 100;
//实际上限值
private final int limit;
public MessageQueue() {
this(DEFAULT_MAX_LIMIT);
}
public MessageQueue(final int limit) {
this.limit = limit;
this.queue = new LinkedList<>();
}
public void put(final Message message) throws InterruptedException {
synchronized (queue) {
while (queue.size() > limit) {
//如果队列的大小超过上限值,阻塞生产者线程不让继续往队列put
queue.wait();
}
//添加任务到队尾
queue.addLast(message);
queue.notifyAll(); //唤醒阻塞的消费者线程,可以从队列take了
}
}
public Message take() throws InterruptedException {
synchronized (queue) {
while (queue.isEmpty()) {
//如果队列为空 阻塞消费者take
queue.wait();
}
//从队首拿出元素
Message message = queue.removeFirst();
queue.notifyAll(); //唤醒等待的生产者可以生产了
return message;
}
}
public int getMaxLimit() {
return this.limit;
}
public int getMessageSize() {
synchronized (queue) {
return queue.size();
}
}
}
/**
* 消费者线程
*/
public class ConsumerThread extends Thread {
private final MessageQueue messageQueue;
private final static Random random = new Random(System.currentTimeMillis());
public ConsumerThread(MessageQueue messageQueue, int seq) {
super("Consumer-" + seq);
this.messageQueue = messageQueue;
}
@Override
public void run() {
while (true) {
try {
Message message = messageQueue.take();
System.out.println(Thread.currentThread().getName() + " take a message " + message.getData());
Thread.sleep(random.nextInt(1000));
} catch (InterruptedException e) {
break;
}
}
}
}
/*
* 生产者线程
**/
public class ProducerThread extends Thread {
private final MessageQueue messageQueue;
private final static Random random = new Random(System.currentTimeMillis());
private final static AtomicInteger counter = new AtomicInteger(0);
public ProducerThread(MessageQueue messageQueue, int seq) {
super("PRODUCER-" + seq);
this.messageQueue = messageQueue;
}
@Override
public void run() {
while (true) {
try {
Message message = new Message("Message-" + counter.getAndIncrement());
messageQueue.put(message);
System.out.println(Thread.currentThread().getName() + " put message " + message.getData());
Thread.sleep(random.nextInt(1000));
} catch (InterruptedException e) {
break;
}
}
}
}
//测试
public class ProducerAndConsumerClient {
public static void main(String[] args) {
final MessageQueue messageQueue = new MessageQueue();
IntStream.rangeClosed(1,2).forEach(i->new ProducerThread(messageQueue,i).start());
IntStream.rangeClosed(1,2).forEach(i->new ConsumerThread(messageQueue,i).start());
}
}
java多线程:生产者消费者设计模式
最新推荐文章于 2024-07-15 15:40:07 发布