CompletableFuture使用

Author: dolyw

SpringBoot/AsyncDemo/src/main/java/com/example/async/config/ThreadPoolConfig.java

@@ -57,9 +57,9 @@ public ThreadPoolTaskExecutor threadPoolTaskExecutor() {
         // 最大线程数
         threadPoolTaskExecutor.setMaxPoolSize(maxPoolSize);
         // 队列容量
-        threadPoolTaskExecutor.setQueueCapacity(keepAliveSeconds);
+        threadPoolTaskExecutor.setQueueCapacity(queueCapacity);
         // 活跃时间
-        threadPoolTaskExecutor.setKeepAliveSeconds(queueCapacity);
+        threadPoolTaskExecutor.setKeepAliveSeconds(keepAliveSeconds);
         // 主线程等待子线程执行时间
         threadPoolTaskExecutor.setAwaitTerminationSeconds(awaitTerminationSeconds);
         // 线程名字前缀

SpringBoot/AsyncDemo/src/main/resources/application.yml

@@ -10,7 +10,7 @@ asyncThreadPool:
   # 最大线程数
   maxPoolSize: 30
   # 队列容量
-  queueCapacity: 150
+  queueCapacity: 15
   # 活跃时间
   keepAliveSeconds: 60
   # 主线程等待子线程执行时间

SpringBoot/ThreadPool/src/main/java/com/example/config/JdkThreadPoolConfig.java

@@ -7,6 +7,7 @@
 import org.springframework.context.annotation.Configuration;
 import org.springframework.scheduling.concurrent.CustomizableThreadFactory;
 
+import java.util.concurrent.LinkedBlockingQueue;
 import java.util.concurrent.SynchronousQueue;
 import java.util.concurrent.ThreadPoolExecutor;
 import java.util.concurrent.TimeUnit;
@@ -64,6 +65,7 @@ public ThreadPoolExecutor jdkThreadPoolExecutor() {
                 corePoolSize,
                 maxPoolSize,
                 keepAliveSeconds, TimeUnit.SECONDS,
+                // new LinkedBlockingQueue<>(queueCapacity),
                 new ResizableCapacityLinkedBlockingQueue<>(queueCapacity),
                 new CustomizableThreadFactory(threadNamePrefix),
                 new ThreadPoolExecutor.CallerRunsPolicy());

SpringBoot/ThreadPool/src/main/java/com/example/controller/CompletableFutureController.java

@@ -0,0 +1,139 @@
+package com.example.controller;
+
+import com.example.helper.BusinessHelper;
+import com.example.util.SmsUtil;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+import java.util.concurrent.CompletableFuture;
+import java.util.concurrent.Future;
+import java.util.concurrent.ThreadPoolExecutor;
+
+/**
+ * AsyncController
+ *
+ * @author wliduo[i@dolyw.com]
+ * @date 2020/5/19 14:46
+ */
+@RestController
+@RequestMapping("/cf")
+public class CompletableFutureController {
+
+    /**
+     * logger
+     */
+    private final static Logger logger = LoggerFactory.getLogger(CompletableFutureController.class);
+
+    @Autowired
+    private BusinessHelper businessHelper;
+
+    @Autowired
+    private SmsUtil smsUtil;
+
+    @Autowired
+    private ThreadPoolExecutor jdkThreadPoolExecutor;
+
+    @Autowired
+    private ThreadPoolExecutor ioThreadPoolExecutor;
+
+    /**
+     * 无返回值
+     *
+     * @param
+     * @return void
+     * @throws
+     * @author wliduo[i@dolyw.com]
+     * @date 2020/5/20 9:53
+     */
+    @GetMapping("/run1")
+    public String run1() throws Exception {
+        logger.info("run1开始执行");
+        CompletableFuture completableFuture = CompletableFuture.runAsync(() -> {
+            try {
+                businessHelper.handle1();
+            } catch (Exception e) {
+
+            }
+        });
+        CompletableFuture completableFuture2 = CompletableFuture.runAsync(() -> {
+            try {
+                businessHelper.handle2();
+            } catch (Exception e) {
+
+            }
+        }, jdkThreadPoolExecutor);
+        // Thread.sleep(5000);
+        // 一个执行完就结束
+        CompletableFuture.anyOf(completableFuture, completableFuture2).get();
+        // 全部执行完才结束
+        CompletableFuture.allOf(completableFuture, completableFuture2).get();
+        logger.info("run1执行完成");
+        return "run1 success";
+    }
+
+    /**
+     * 无返回值
+     *
+     * @param
+     * @return void
+     * @throws
+     * @author wliduo[i@dolyw.com]
+     * @date 2020/5/20 9:53
+     */
+    @GetMapping("/run2")
+    public String run2() throws Exception {
+        logger.info("run2开始执行");
+        CompletableFuture<String> completableFuture = CompletableFuture.supplyAsync(() -> {
+            try {
+                return businessHelper.handle1();
+            } catch (Exception e) {
+                return "";
+            }
+        }, ioThreadPoolExecutor);
+        CompletableFuture<String> completableFuture2 = CompletableFuture.supplyAsync(() -> {
+            try {
+                // return businessHelper.handle2();
+                return this.handle3();
+            } catch (Exception e) {
+                return "";
+            }
+        }, jdkThreadPoolExecutor);
+        // Thread.sleep(5000);
+        // 一个执行完就结束
+        // CompletableFuture.anyOf(completableFuture, completableFuture2).get();
+        // 全部执行完才结束
+        CompletableFuture.allOf(completableFuture, completableFuture2).get();
+
+        logger.info(Boolean.toString(completableFuture.isDone()));
+        logger.info(Boolean.toString(completableFuture2.isDone()));
+        logger.info("run2执行完成");
+        return "run2 success";
+    }
+
+    public String handle3() throws Exception {
+        Thread.sleep(3000L);
+        logger.info("handle3");
+        return "handle3";
+    }
+
+    /**
+     * 工具类异步
+     *
+     * @param
+     * @return java.lang.String
+     * @throws
+     * @author wliduo[i@dolyw.com]
+     * @date 2020/5/20 10:59
+     */
+    @GetMapping("/sms")
+    public String sms() throws Exception {
+        logger.info("sms开始执行");
+        smsUtil.send("15912347896", "135333");
+        logger.info("sms执行完成");
+        return "send sms success";
+    }
+}

SpringBoot/ThreadPool/src/main/java/com/example/helper/BusinessHelper.java

@@ -0,0 +1,30 @@
+package com.example.helper;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.stereotype.Component;
+
+/**
+ * BusinessHelper
+ *
+ * @author wliduo[i@dolyw.com]
+ * @date 2022/1/6 14:15
+ */
+@Component
+public class BusinessHelper {
+
+    private final static Logger logger = LoggerFactory.getLogger(BusinessHelper.class);
+
+    public String handle1() throws Exception {
+        Thread.sleep(5000L);
+        logger.info("handle1");
+        return "handle1";
+    }
+
+    public String handle2() throws Exception {
+        Thread.sleep(3000L);
+        logger.info("handle2");
+        return "handle2";
+    }
+
+}

SpringBoot/ThreadPool/src/main/java/com/example/util/SmsUtil.java

@@ -34,4 +34,11 @@ public void sendCode(String phone, String code) throws Exception {
         logger.info("发送成功: {}", phone);
     }
 
+    public void send(String phone, String code) throws Exception {
+        logger.info("开始发送验证码...");
+        // 模拟调用接口发验证码的耗时
+        Thread.sleep(3000);
+        logger.info("发送成功: {}", phone);
+    }
+
 }