添加阻塞队列

byhieg · byhieg · commit d200fb2c02e3 · 2017-05-03T17:23:12.000+08:00
diff --git a/src/main/java/cn/byhieg/threadtutorial/concurrent/blocking/ArrayBlock.java b/src/main/java/cn/byhieg/threadtutorial/concurrent/blocking/ArrayBlock.java
@@ -0,0 +1,21 @@
+package cn.byhieg.threadtutorial.concurrent.blocking;
+
+import java.util.concurrent.ArrayBlockingQueue;
+import java.util.concurrent.BlockingQueue;
+
+/**
+ * Created by byhieg on 17/5/3.
+ * Mail to byhieg@gmail.com
+ */
+public class ArrayBlock {
+
+    private BlockingQueue<String> blockingQueue;
+
+    public ArrayBlock(){
+        blockingQueue = new ArrayBlockingQueue<String>(1024);
+    }
+
+    public BlockingQueue<String> getBlockingQueue() {
+        return blockingQueue;
+    }
+}
diff --git a/src/main/java/cn/byhieg/threadtutorial/concurrent/blocking/Costumer.java b/src/main/java/cn/byhieg/threadtutorial/concurrent/blocking/Costumer.java
@@ -0,0 +1,31 @@
+package cn.byhieg.threadtutorial.concurrent.blocking;
+
+import java.util.concurrent.BlockingQueue;
+
+/**
+ * Created by byhieg on 17/5/3.
+ * Mail to byhieg@gmail.com
+ */
+public class Costumer extends Thread{
+
+    private BlockingQueue<String> blockingQueue;
+
+    public Costumer(ArrayBlock arrayBlock) {
+        blockingQueue = arrayBlock.getBlockingQueue();
+        this.setName("Costumer");
+    }
+
+    @Override
+    public void run() {
+        super.run();
+        while (true) {
+            try {
+                Thread.sleep(2000);
+                String str = blockingQueue.take();
+                System.out.println(getName() + " 取出数据 " + str);
+            } catch (InterruptedException e) {
+                e.printStackTrace();
+            }
+        }
+    }
+}
diff --git a/src/main/java/cn/byhieg/threadtutorial/concurrent/blocking/Producer.java b/src/main/java/cn/byhieg/threadtutorial/concurrent/blocking/Producer.java
@@ -0,0 +1,30 @@
+package cn.byhieg.threadtutorial.concurrent.blocking;
+
+import java.util.concurrent.BlockingQueue;
+
+/**
+ * Created by byhieg on 17/5/3.
+ * Mail to byhieg@gmail.com
+ */
+public class Producer extends Thread {
+
+    private BlockingQueue<String> blockingQueue;
+    @Override
+    public void run() {
+        super.run();
+        for (int i = 0 ; i < 5;i++) {
+            try {
+                System.out.println(getName() + " 生产数据");
+                blockingQueue.put(i + "");
+                Thread.sleep(1000);
+            } catch (InterruptedException e) {
+                e.printStackTrace();
+            }
+        }
+    }
+
+    public Producer(ArrayBlock arrayBlock){
+        this.setName("Producer");
+        blockingQueue = arrayBlock.getBlockingQueue();
+    }
+}
diff --git a/src/test/java/cn/byhieg/threadtutorialtest/concurrenttest/blockingtest/ArrayBlockTest.java b/src/test/java/cn/byhieg/threadtutorialtest/concurrenttest/blockingtest/ArrayBlockTest.java
@@ -0,0 +1,34 @@
+package cn.byhieg.threadtutorialtest.concurrenttest.blockingtest;
+
+import cn.byhieg.threadtutorial.concurrent.blocking.ArrayBlock;
+import cn.byhieg.threadtutorial.concurrent.blocking.Costumer;
+import cn.byhieg.threadtutorial.concurrent.blocking.Producer;
+import junit.framework.TestCase;
+
+import javax.swing.*;
+
+/**
+ * Created by byhieg on 17/5/3.
+ * Mail to byhieg@gmail.com
+ */
+public class ArrayBlockTest extends TestCase {
+    ArrayBlock block;
+    public void setUp() throws Exception {
+        super.setUp();
+        block = new ArrayBlock();
+    }
+
+    public void tearDown() throws Exception {
+    }
+
+
+    public void testBlocking() throws Exception {
+        Producer producer = new Producer(block);
+        Costumer costumer = new Costumer(block);
+        producer.start();
+        costumer.start();
+        producer.join();
+        costumer.join();
+
+    }
+}
