测试异步

Author: saleson

spring-cloud-gray-samples/spring-cloud-gray-service-b-sample/src/main/java/cn/springcloud/gray/service/b/ServiceBApplication.java

@@ -6,6 +6,7 @@
 import org.springframework.cloud.client.discovery.EnableDiscoveryClient;
 import org.springframework.cloud.netflix.feign.EnableFeignClients;
 import org.springframework.core.env.Environment;
+import org.springframework.scheduling.annotation.EnableAsync;
 
 import java.net.InetAddress;
 import java.net.UnknownHostException;
@@ -16,6 +17,7 @@
 @SpringBootApplication
 @EnableDiscoveryClient
 @EnableFeignClients
+@EnableAsync
 public class ServiceBApplication {
     private static final org.slf4j.Logger log = LoggerFactory.getLogger(ServiceBApplication.class);
 

spring-cloud-gray-samples/spring-cloud-gray-service-b-sample/src/main/java/cn/springcloud/gray/service/b/configuration/TaskConfiguration.java

@@ -0,0 +1,34 @@
+package cn.springcloud.gray.service.b.configuration;
+
+import cn.springcloud.gray.concurrent.GrayTaskDecorator;
+import cn.springcloud.gray.request.RequestLocalStorage;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
+
+import java.util.concurrent.Executor;
+import java.util.concurrent.ThreadPoolExecutor;
+
+/**
+ * Description:异步任务线程池
+ *
+ * @author JianqiangZhao
+ * @date 2019/1/14
+ */
+@Configuration
+public class TaskConfiguration {
+    @Bean("taskExecutor")
+    public Executor taskExecutor(RequestLocalStorage requestLocalStorage) {
+        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
+        executor.setCorePoolSize(100);
+        executor.setMaxPoolSize(400);
+        executor.setQueueCapacity(400);
+        executor.setKeepAliveSeconds(60);
+        executor.setWaitForTasksToCompleteOnShutdown(true);
+        executor.setAwaitTerminationSeconds(60);
+        executor.setThreadNamePrefix("taskExecutor-");
+        executor.setRejectedExecutionHandler(new ThreadPoolExecutor.DiscardPolicy());
+        executor.setTaskDecorator(new GrayTaskDecorator(requestLocalStorage));
+        return executor;
+    }
+}

spring-cloud-gray-samples/spring-cloud-gray-service-b-sample/src/main/java/cn/springcloud/gray/service/b/rest/TestResource.java

@@ -2,6 +2,7 @@
 
 import cn.springcloud.gray.service.b.feign.Test2Client;
 import cn.springcloud.gray.service.b.feign.TestClient;
+import cn.springcloud.gray.service.b.service.TestService;
 import com.google.common.collect.ImmutableMap;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -11,6 +12,8 @@
 
 import javax.servlet.http.HttpServletRequest;
 import java.util.Map;
+import java.util.concurrent.ExecutionException;
+import java.util.concurrent.Future;
 
 /**
  * Created by saleson on 2017/10/18.
@@ -25,6 +28,8 @@ public class TestResource {
     private TestClient testClient;
     @Autowired
     private Test2Client test2Client;
+    @Autowired
+    private TestService testService;
 
 
     @RequestMapping(value = "/testGet", method = RequestMethod.GET)
@@ -80,6 +85,15 @@ public Map<String, Object> feignGet(
         return ImmutableMap.of("feignGet", "success.", "service-a-result", map);
     }
 
+    @RequestMapping(value = "/feignGetAsync", method = RequestMethod.GET)
+    @ResponseBody
+    public Map<String, Object> feignGetAsync(
+            @RequestParam(value = "version", required = false) String version,
+            HttpServletRequest request) throws ExecutionException, InterruptedException {
+        Future<Map> map = testService.get(version);
+        return ImmutableMap.of("feignGet", "success.", "service-a-result", map.get());
+    }
+
     @RequestMapping(value = "/feign2Get", method = RequestMethod.GET)
     @ResponseBody
     public Map<String, Object> feign2Get(

spring-cloud-gray-samples/spring-cloud-gray-service-b-sample/src/main/java/cn/springcloud/gray/service/b/service/TestService.java

@@ -0,0 +1,21 @@
+package cn.springcloud.gray.service.b.service;
+
+import cn.springcloud.gray.service.b.feign.TestClient;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.scheduling.annotation.Async;
+import org.springframework.scheduling.annotation.AsyncResult;
+import org.springframework.stereotype.Service;
+
+import java.util.Map;
+import java.util.concurrent.Future;
+
+@Service
+public class TestService {
+    @Autowired
+    private TestClient testClient;
+
+    @Async("taskExecutor")
+    public Future<Map> get(String version) {
+        return AsyncResult.forValue(testClient.testGet(version));
+    }
+}