my thread pool

zhangml · zhangml · commit 819a3b53251d · 2019-06-18T15:19:24.000+08:00
diff --git a/src/main/java/com/example/concurrency/threadPool/MyThreadPool.java b/src/main/java/com/example/concurrency/threadPool/MyThreadPool.java
@@ -0,0 +1,84 @@
+package com.example.concurrency.threadPool;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.concurrent.BlockingQueue;
+import java.util.concurrent.LinkedBlockingQueue;
+
+/**
+ * 描述:
+ * 简化的线程池，仅用来说明工作原理
+ *
+ * @author zed
+ * @since 2019-06-18 3:10 PM
+ */
+
+public class MyThreadPool{
+    /**
+     * 利用阻塞队列实现生产者 - 消费者模式
+     */
+    private BlockingQueue<Runnable> workQueue;
+    /**
+     * 保存内部工作线程
+     */
+    private List<WorkerThread> threads = new ArrayList<>();
+    // 构造方法
+    public MyThreadPool(int poolSize, BlockingQueue<Runnable> workQueue){
+        this.workQueue = workQueue;
+        // 创建工作线程
+        for(int idx=0; idx<poolSize; idx++){
+            WorkerThread work = new WorkerThread();
+            work.start();
+            threads.add(work);
+        }
+    }
+
+    /**
+     * 提交任务
+     * @param command command
+     */
+    public void execute(Runnable command){
+        try{
+            workQueue.put(command);
+
+        }catch (InterruptedException e){
+            Thread.currentThread().interrupt();
+
+        }
+    }
+
+    /**
+     * 工作线程负责消费任务，并执行任务
+     */
+    class WorkerThread extends Thread{
+        @Override
+        public void run() {
+            // 循环取任务并执行
+            while(true){
+                try{
+                    Runnable task = workQueue.take();
+                    task.run();
+                }catch (InterruptedException e){
+                    Thread.currentThread().interrupt();
+                }
+
+            }
+        }
+    }
+
+    public static void main(String[] args) {
+        // 创建有界阻塞队列
+        BlockingQueue<Runnable> workQueue = new LinkedBlockingQueue<>(2);
+        // 创建线程池
+        MyThreadPool pool = new MyThreadPool(10, workQueue);
+        // 提交任务
+        pool.execute(()->{
+            System.out.println("hello");
+        });
+
+
+    }
+
+}
+
+
