添加线程池进行测试

Author: saleson

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

@@ -0,0 +1,24 @@
+package cn.springcloud.gray.service.b.configuration;
+
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+import java.util.concurrent.ArrayBlockingQueue;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.ThreadPoolExecutor;
+import java.util.concurrent.TimeUnit;
+
+@Configuration
+public class ConcurrentConfiguration {
+
+//    @Bean
+//    public ExecutorServiceGrayAspect executorServiceGrayAspect() {
+//        return new ExecutorServiceGrayAspect();
+//    }
+
+
+    @Bean
+    public ExecutorService executorService() {
+        return new ThreadPoolExecutor(10, 10, 1, TimeUnit.MINUTES, new ArrayBlockingQueue<>(10));
+    }
+}

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

@@ -1,7 +1,6 @@
 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;
@@ -11,14 +10,11 @@
 
 /**
  * Description:异步任务线程池
- *
- * @author JianqiangZhao
- * @date 2019/1/14
  */
 @Configuration
 public class TaskConfiguration {
     @Bean("taskExecutor")
-    public Executor taskExecutor(RequestLocalStorage requestLocalStorage) {
+    public Executor taskExecutor() {
         ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
         executor.setCorePoolSize(100);
         executor.setMaxPoolSize(400);
@@ -28,7 +24,7 @@ public Executor taskExecutor(RequestLocalStorage requestLocalStorage) {
         executor.setAwaitTerminationSeconds(60);
         executor.setThreadNamePrefix("taskExecutor-");
         executor.setRejectedExecutionHandler(new ThreadPoolExecutor.DiscardPolicy());
-        executor.setTaskDecorator(new GrayTaskDecorator(requestLocalStorage));
+        executor.setTaskDecorator(new GrayTaskDecorator());
         return executor;
     }
 }

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

@@ -63,6 +63,7 @@ public Map<String, Object> restTemplateGet(HttpServletRequest request) {
      * test feign invoke service-a
      *
      * @param version 请求的版本
+     * @param request 请求
      * @return 内容
      */
     @RequestMapping(value = "/feignGet", method = RequestMethod.GET)

spring-cloud-gray-samples/spring-cloud-gray-service-b-sample/src/test/java/cn/springcloud/gray/service/b/ExecutorServiceGrayAspectTest.java

@@ -0,0 +1,120 @@
+package cn.springcloud.gray.service.b;
+
+import cn.springcloud.gray.request.GrayHttpTrackInfo;
+import cn.springcloud.gray.request.LocalStorageLifeCycle;
+import cn.springcloud.gray.request.RequestLocalStorage;
+import org.junit.After;
+import org.junit.Before;
+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.SpringRunner;
+
+import java.util.Arrays;
+import java.util.concurrent.ExecutionException;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.TimeUnit;
+import java.util.concurrent.TimeoutException;
+
+@RunWith(SpringRunner.class)
+@SpringBootTest(classes = ServiceBApplication.class)
+public class ExecutorServiceGrayAspectTest {
+
+    @Autowired
+    private LocalStorageLifeCycle localStorageLifeCycle;
+
+    @Autowired
+    private ExecutorService executorService;
+    @Autowired
+    private RequestLocalStorage requestLocalStorage;
+
+
+    @Before
+    public void before() {
+        localStorageLifeCycle.initContext();
+    }
+
+    @After
+    public void after() {
+        localStorageLifeCycle.closeContext();
+    }
+
+
+    @Test
+    public void testExecute() {
+        requestLocalStorage.setGrayTrackInfo(new GrayHttpTrackInfo());
+        executorService.execute(() -> {
+            System.out.println("323fd");
+        });
+    }
+
+    @Test
+    public void testSubmit() {
+        executorService.submit(() -> {
+            System.out.println("323fd");
+        });
+    }
+
+    @Test
+    public void testSubmit2() {
+        executorService.submit(() -> {
+            System.out.println("323fd");
+        }, "");
+    }
+
+    @Test
+    public void testSubmit3() {
+        executorService.submit(() -> {
+            System.out.println("323fd");
+            return "";
+        });
+    }
+
+
+    @Test
+    public void testInvokeAll() throws InterruptedException {
+        executorService.invokeAll(Arrays.asList(() -> {
+            System.out.println("323fd");
+            return "";
+        }, () -> {
+            System.out.println("323fd");
+            return "";
+        }));
+    }
+
+    @Test
+    public void testInvokeAll2() throws InterruptedException {
+        executorService.invokeAll(Arrays.asList(() -> {
+            System.out.println("323fd");
+            return "";
+        }, () -> {
+            System.out.println("323fd");
+            return "";
+        }), 10, TimeUnit.SECONDS);
+    }
+
+
+    @Test
+    public void testInvokeAny() throws InterruptedException, ExecutionException {
+        executorService.invokeAny(Arrays.asList(() -> {
+            System.out.println("323fd");
+            return "";
+        }, () -> {
+            System.out.println("323fd");
+            return "";
+        }));
+    }
+
+    @Test
+    public void testInvokeAny2() throws InterruptedException, ExecutionException, TimeoutException {
+        executorService.invokeAny(Arrays.asList(() -> {
+            System.out.println("323fd");
+            return "";
+        }, () -> {
+            System.out.println("323fd");
+            return "";
+        }), 10, TimeUnit.SECONDS);
+    }
+
+}