[+] 结构调整

taoyao-test/pom.xml

@@ -0,0 +1,27 @@
+<?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>
+
+	<parent>
+		<groupId>com.acgist</groupId>
+		<artifactId>taoyao</artifactId>
+		<version>1.0.0</version>
+	</parent>
+
+	<artifactId>taoyao-test</artifactId>
+	<packaging>jar</packaging>
+
+	<name>taoyao-test</name>
+	<description>测试：测试工具</description>
+
+	<dependencies>
+		<dependency>
+			<groupId>org.springframework.boot</groupId>
+			<artifactId>spring-boot-starter-test</artifactId>
+			<optional>true</optional>
+		</dependency>
+	</dependencies>
+
+</project>

taoyao-test/src/main/java/com/acgist/taoyao/test/annotation/CostedTest.java

@@ -0,0 +1,42 @@
+package com.acgist.taoyao.test.annotation;
+
+import java.lang.annotation.Documented;
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Inherited;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+import java.util.concurrent.TimeUnit;
+
+/**
+ * 多线程测试
+ * 
+ * @author acgist
+ */
+@Target(ElementType.METHOD)
+@Inherited
+@Retention(RetentionPolicy.RUNTIME)
+@Documented
+public @interface CostedTest {
+
+	/**
+	 * @return 执行次数
+	 */
+	int count() default 1;
+
+	/**
+	 * @return 线程数量
+	 */
+	int thread() default 1;
+	
+	/**
+	 * @return 超时时间
+	 */
+	long timeout() default 1000;
+	
+	/**
+	 * @return 超时时间单位
+	 */
+	TimeUnit timeUnit() default TimeUnit.MILLISECONDS;
+	
+}

taoyao-test/src/main/java/com/acgist/taoyao/test/annotation/CostedTestTestExecutionListener.java

@@ -0,0 +1,55 @@
+package com.acgist.taoyao.test.annotation;
+
+import java.lang.reflect.InvocationTargetException;
+import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Executors;
+import java.util.concurrent.TimeUnit;
+
+import org.springframework.test.context.TestContext;
+import org.springframework.test.context.TestExecutionListener;
+
+import lombok.extern.slf4j.Slf4j;
+
+/**
+ * 多线程测试监听
+ * 
+ * @author acgist
+ */
+@Slf4j
+public class CostedTestTestExecutionListener implements TestExecutionListener {
+	
+	@Override
+	public void afterTestMethod(TestContext testContext) throws Exception {
+		final CostedTest costedTest = testContext.getTestMethod().getDeclaredAnnotation(CostedTest.class);
+		final int count  = costedTest.count();
+		final int thread  = costedTest.thread();
+		final long timeout = costedTest.timeout();
+		final TimeUnit timeUnit = costedTest.timeUnit();
+		final long aTime = System.currentTimeMillis();
+		if(thread == 1) {
+			for (int index = 0; index < count; index++) {
+				testContext.getTestMethod().invoke(testContext.getTestInstance());
+			}
+		} else {
+			final CountDownLatch countDownLatch = new CountDownLatch(count);
+			final ExecutorService executor = Executors.newFixedThreadPool(thread);
+			for (int index = 0; index < count; index++) {
+				executor.execute(() -> {
+					try {
+						testContext.getTestMethod().invoke(testContext.getTestInstance());
+					} catch (IllegalAccessException | IllegalArgumentException | InvocationTargetException e) {
+						log.error("多线程测试异常", e);
+					} finally {
+						countDownLatch.countDown();
+					}
+				});
+			}
+			countDownLatch.await(timeout, timeUnit);
+		}
+		final long zTime = System.currentTimeMillis();
+		final long costed = zTime - aTime;
+		log.info("多线程测试消耗时间：{}", costed);
+	}
+	
+}

taoyao-test/src/main/java/com/acgist/taoyao/test/annotation/TaoyaoTest.java

@@ -0,0 +1,31 @@
+package com.acgist.taoyao.test.annotation;
+
+import java.lang.annotation.Documented;
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Inherited;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.core.annotation.AliasFor;
+import org.springframework.test.context.TestExecutionListeners;
+import org.springframework.test.context.TestExecutionListeners.MergeMode;
+
+/**
+ * 测试启动
+ * 
+ * @author acgist
+ */
+@Target(ElementType.TYPE)
+@Inherited
+@Retention(RetentionPolicy.RUNTIME)
+@Documented
+@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
+@TestExecutionListeners(listeners = CostedTestTestExecutionListener.class, mergeMode = MergeMode.MERGE_WITH_DEFAULTS)
+public @interface TaoyaoTest {
+
+	@AliasFor(annotation = SpringBootTest.class)
+	Class<?>[] classes() default {};
+
+}