1. 介绍背景
在Java多线程应用中,队列的使用率很高,多数生产消费模型的首选数据结构就是队列。
Java提供的线程安全的Queue可以分为阻塞队列和非阻塞队列,其中阻塞队列的典型例子是BlockingQueue,非阻塞队列的典型例子是ConcurrentLinkedQueue,在实际应用中要根据实际需要选用阻塞队列或者非阻塞队列。
2. LinkedBlockingQueue
由于LinkedBlockingQueue实现是线程安全的,实现了先进先出等特性,是作为生产者消费者的首选,LinkedBlockingQueue 可以指定容量,也可以不指定,不指定的话,默认最大是Integer.MAX_VALUE,其中主要用到put和take方法,put方法在队列满的时候会阻塞直到有队列成员被消费,take方法在队列空的时候会阻塞,直到有队列成员被放进来。
package com.lky.test;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.LinkedBlockingQueue;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.junit.Test;
/**
* @Title: testBlockQueue.java
* @Package com.lky.test
* @Description:多线程模拟实现生产者消费者模型(阻塞队列)
* @author lky
* @date 2015年10月24日 下午5:08:01
* @version V1.0
*/
public class testBlockQueue {
private Log log = LogFactory.getLog(testBlockQueue.class);
/**
* @Title: testBlockQueue.java
* @Package com.lky.test
* @Description: 定义阻塞队列
* @author lky
* @date 2015年10月24日 下午5:07:28
* @version V1.0
*/
public class Basket {
// 队列的最大容量为3
BlockingQueue<String> basket = new LinkedBlockingQueue<String>(3);
// 如果队列不满,则放入,否则阻塞等待
public void produce(String apple) throws InterruptedException {
basket.put(apple);
}
// 如果队列不为空,则取出,否则阻塞等待
public String consumer() throws InterruptedException {
return basket.take();
}
}
/**
* @Title: testBlockQueue.java
* @Package com.lky.test
* @Description: 定义生产者
* @author lky
* @date 2015年10月24日 下午5:18:17
* @version V1.0
*/
public class Produce implements Runnable {
private Basket basket;
private String fruit;
public Produce(String fruit, Basket basket) {
this.basket = basket;
this.fruit = fruit;
}
@Override
public void run() {
try {
while (true) {
log.info("[" + Thread.currentThread().getName() + "]" + "开始生产apple----->" + this.fruit);
basket.produce(fruit);
log.info("apple生产完毕!!!!");
Thread.sleep(1000);
}
} catch (Exception e) {
log.error("生产苹果异常!!!!!");
}
}
}
/**
* @Title: testBlockQueue.java
* @Package com.lky.test
* @Description: 定义消费者
* @author lky
* @date 2015年10月24日 下午5:24:31
* @version V1.0
*/
public class Consumer implements Runnable {
private Basket basket;
public Consumer(Basket basket) {
this.basket = basket;
}
@Override
public void run() {
try {
while (true) {
String fruit = basket.consumer();
log.info("[" + Thread.currentThread().getName() + "]" + "取到一个水果: " + fruit);
Thread.sleep(1000);
}
} catch (Exception e) {
log.error("消费者取苹果异常!!!!");
}
}
}
@Test
public void test() {
System.out.println(Runtime.getRuntime().availableProcessors());//获取当前系统的cpu数目
Basket basket = new Basket();
Produce produce1 = new Produce("apple", basket);
Produce produce2 = new Produce("banna", basket);
Consumer consumer = new Consumer(basket);
// 新建一个线程池
ExecutorService service = Executors.newCachedThreadPool();
service.submit(produce1);
service.submit(produce2);
service.submit(consumer);
try {
Thread.sleep(20000);
} catch (Exception e) {
log.error("程序异常错误!!!!");
}
service.shutdown();
}
}
3. ConcurrentLinkedQueue
是Queue的一个安全实现.Queue中元素按FIFO原则进行排序.采用CAS操作,来保证元素的一致性。
package com.lky.test;
import java.util.concurrent.ConcurrentLinkedQueue;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
/**
* @Title: testNBlockQueue.java
* @Package com.lky.test
* @Description: 多线程模拟实现生产者消费者模型(非阻塞式队列)
* @author lky
* @date 2015年10月24日 下午8:02:14
* @version V1.0
*/
public class testNBlockQueue {
private static Log log = LogFactory.getLog(testNBlockQueue.class);
private static ConcurrentLinkedQueue<Integer> queue = new ConcurrentLinkedQueue<Integer>();
private static int count = 2;
private static CountDownLatch latch = new CountDownLatch(count);
private static class Poll implements Runnable {
@Override
public void run() {
while (!queue.isEmpty()) {
log.info(Thread.currentThread().getName() + "消费一个商品: " + queue.poll());
}
latch.countDown();
}
}
public static void main(String args[]) throws InterruptedException {
long timeStart = System.currentTimeMillis();
ExecutorService eService = Executors.newFixedThreadPool(4);
// 生产商品
for (int i = 0; i < 100000; ++i) {
queue.offer(i);
}
// 消费者
for (int i = 0; i < count; ++i) {
eService.submit(new Poll());
}
latch.await();// 使得主线程阻塞,直到latch.getCount()为0
System.out.println("Cost time: " + (System.currentTimeMillis() - timeStart));
eService.shutdown();
}
}
4. 使用场景
在Java多线程应用中,队列的使用率很高,多数生产消费模型的首选数据结构就是队列(先进先出)。Java提供的线程安全的Queue可以分为阻塞队列和非阻塞队列,其中阻塞队列的典型例子是BlockingQueue,非阻塞队列的典型例子是ConcurrentLinkedQueue,在实际应用中要根据实际需要选用阻塞队列或者非阻塞队列。