使用@Async实现异步调用：资源优雅关闭

2018-03-22 17:16:09 +08:00
parent fd635d96b3
commit 4bf616ec0c
5 changed files with 181 additions and 0 deletions
@@ -0,0 +1,60 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <modelVersion>4.0.0</modelVersion>
+
+    <groupId>com.didispace</groupId>
+    <artifactId>Chapter4-1-4</artifactId>
+    <version>1.0.0</version>
+    <packaging>jar</packaging>
+
+    <name>Chapter4-1-4</name>
+    <description>Spring Boot project</description>
+
+    <parent>
+        <groupId>org.springframework.boot</groupId>
+        <artifactId>spring-boot-starter-parent</artifactId>
+        <version>1.5.10.RELEASE</version>
+        <relativePath/> <!-- lookup parent from repository -->
+    </parent>
+
+    <properties>
+        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
+        <java.version>1.8</java.version>
+    </properties>
+
+    <dependencies>
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter</artifactId>
+        </dependency>
+
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-test</artifactId>
+            <scope>test</scope>
+        </dependency>
+
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-data-redis</artifactId>
+        </dependency>
+
+        <dependency>
+            <groupId>org.projectlombok</groupId>
+            <artifactId>lombok</artifactId>
+            <version>1.16.20</version>
+            <scope>provided</scope>
+        </dependency>
+    </dependencies>
+
+    <build>
+        <plugins>
+            <plugin>
+                <groupId>org.springframework.boot</groupId>
+                <artifactId>spring-boot-maven-plugin</artifactId>
+            </plugin>
+        </plugins>
+    </build>
+
+</project>
@@ -0,0 +1,36 @@
+package com.didispace;
+
+import org.springframework.boot.SpringApplication;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.scheduling.annotation.EnableAsync;
+import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
+
+import java.util.concurrent.Executor;
+
+@SpringBootApplication
+public class Application {
+
+    public static void main(String[] args) {
+        SpringApplication.run(Application.class, args);
+    }
+
+    @EnableAsync
+    @Configuration
+    class TaskPoolConfig {
+
+        @Bean("taskExecutor")
+        public Executor taskExecutor() {
+            ThreadPoolTaskScheduler executor = new ThreadPoolTaskScheduler();
+            executor.setPoolSize(20);
+            executor.setThreadNamePrefix("taskExecutor-");
+
+            executor.setWaitForTasksToCompleteOnShutdown(true);
+            executor.setAwaitTerminationSeconds(60);
+            return executor;
+        }
+
+    }
+
+}
@@ -0,0 +1,49 @@
+package com.didispace.async;
+
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.data.redis.core.StringRedisTemplate;
+import org.springframework.scheduling.annotation.Async;
+import org.springframework.stereotype.Component;
+
+/**
+ * @author 程序猿DD
+ * @version 1.0.0
+ * @date 16/5/16 下午12:58.
+ * @blog http://blog.didispace.com
+ */
+@Slf4j
+@Component
+public class Task {
+
+    @Autowired
+    private StringRedisTemplate stringRedisTemplate;
+
+    @Async("taskExecutor")
+    public void doTaskOne() throws Exception {
+        log.info("开始做任务一");
+        long start = System.currentTimeMillis();
+        log.info(stringRedisTemplate.randomKey());
+        long end = System.currentTimeMillis();
+        log.info("完成任务一，耗时：" + (end - start) + "毫秒");
+    }
+
+    @Async("taskExecutor")
+    public void doTaskTwo() throws Exception {
+        log.info("开始做任务二");
+        long start = System.currentTimeMillis();
+        log.info(stringRedisTemplate.randomKey());
+        long end = System.currentTimeMillis();
+        log.info("完成任务二，耗时：" + (end - start) + "毫秒");
+    }
+
+    @Async("taskExecutor")
+    public void doTaskThree() throws Exception {
+        log.info("开始做任务三");
+        long start = System.currentTimeMillis();
+        log.info(stringRedisTemplate.randomKey());
+        long end = System.currentTimeMillis();
+        log.info("完成任务三，耗时：" + (end - start) + "毫秒");
+    }
+
+}
@@ -0,0 +1,2 @@
+spring.redis.pool.max-wait=5000
+spring.redis.pool.max-active=10
@@ -0,0 +1,34 @@
+package com.didispace;
+
+import com.didispace.async.Task;
+import lombok.SneakyThrows;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
+
+
+@RunWith(SpringJUnit4ClassRunner.class)
+@SpringBootTest
+public class ApplicationTests {
+
+    @Autowired
+    private Task task;
+
+    @Test
+    @SneakyThrows
+    public void test() {
+
+        for (int i = 0; i < 10000; i++) {
+            task.doTaskOne();
+            task.doTaskTwo();
+            task.doTaskThree();
+
+            if (i == 9999) {
+                System.exit(0);
+            }
+        }
+    }
+
+}