17.1.0

FirokOtaku · Nov 5, 2021 · 3eb013b · 3eb013b
1 parent 0ec5d83
commit 3eb013b
Show file tree

Hide file tree

Showing 26 changed files with 800 additions and 429 deletions.
diff --git a/pom.xml b/pom.xml
@@ -6,8 +6,15 @@
 
     <groupId>firok.spring</groupId>
     <artifactId>mvci</artifactId>
-    <version>17.0.0</version>
+    <version>17.1.0</version>
     <description>An annotation processing tool to generate MVC code from JavaBean(s).</description>
+    <dependencies>
+        <dependency>
+            <groupId>org.projectlombok</groupId>
+            <artifactId>lombok</artifactId>
+            <version>1.18.22</version>
+        </dependency>
+    </dependencies>
 
     <properties>
         <java.version>17</java.version>
@@ -35,6 +42,21 @@
             </plugin>
 
         </plugins>
+
+        <resources>
+<!--            <resource>-->
+<!--                <directory>${basedir}/src/java</directory>-->
+<!--                <includes>-->
+<!--                    <include>**/*</include>-->
+<!--                </includes>-->
+<!--            </resource>-->
+            <resource>
+                <directory>${basedir}/src/main/resources</directory>
+                <includes>
+                    <include>*.txt</include>
+                </includes>
+            </resource>
+        </resources>
     </build>
 
 

diff --git a/src/main/java/firok/spring/mvci/AvailableValues.java b/src/main/java/firok/spring/mvci/AvailableValues.java
@@ -0,0 +1,21 @@
+package firok.spring.mvci;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+/**
+ * <p>
+ *     标注于注解字段, 标明此字段可接受的值.
+ * </p>
+ * <p>
+ *     marked on annotation fields to indicate what values could be taken.
+ * </p>
+ */
+@Retention(RetentionPolicy.CLASS)
+@Target(ElementType.METHOD)
+public @interface AvailableValues
+{
+	String[] value();
+}
diff --git a/src/main/java/firok/spring/mvci/Constants.java b/src/main/java/firok/spring/mvci/Constants.java
@@ -1,24 +1,29 @@
 package firok.spring.mvci;
 
-import firok.spring.mvci.internal.DefaultControllerTemplate;
-import firok.spring.mvci.internal.DefaultMapperTemplate;
-import firok.spring.mvci.internal.DefaultServiceImplTemplate;
-import firok.spring.mvci.internal.DefaultServiceTemplate;
-
 public final class Constants
 {
-	public static final String DEFAULT_MAPPER_TEMPLATE = DefaultMapperTemplate.VALUE;
+	public static final String BEAN_NAME_FULL = "##BEAN_NAME_FULL##";
+	public static final String BEAN_NAME_SHORT = "##BEAN_NAME_SHORT##";
+	public static final String BEAN_PACKAGE = "##BEAN_PACKAGE##";
+	public static final String BASE_PACKAGE = "##BASE_PACKAGE##";
 
-	public static final String DEFAULT_SERVICE_TEMPLATE = DefaultServiceTemplate.VALUE;
+	public static final String MAPPER_NAME = "##MAPPER_NAME##";
+	public static final String SERVICE_NAME = "##SERVICE_NAME##";
+	public static final String SERVICE_IMPL_NAME = "##SERVICE_IMPL_NAME##";
+	public static final String CONTROLLER_NAME = "##CONTROLLER_NAME##";
 
-	public static final String DEFAULT_SERVICE_IMPL_TEMPLATE = DefaultServiceImplTemplate.VALUE;
+	public static final String MAPPER_PACKAGE = "##MAPPER_PACKAGE##";
+	public static final String SERVICE_PACKAGE = "##SERVICE_PACKAGE##";
+	public static final String SERVICE_IMPL_PACKAGE = "##SERVICE_IMPL_PACKAGE##";
+	public static final String CONTROLLER_PACKAGE = "##CONTROLLER_PACKAGE##";
 
-	public static final String DEFAULT_CONTROLLER_TEMPLATE = DefaultControllerTemplate.VALUE;
+	public static final String TRUE = "##TRUE##";
+	public static final String FALSE = "##FALSE##";
+	public static final String DEFAULT = "##DEFAULT##";
 
-	/**
-	 * 如果标注为 DISABLE, 将不会生成指定结构
-	 */
-	public static final String DISABLE = "##DISABLE##";
+	public static final String PREFER_TRUE = "##PERFER_TRUE##";
+	public static final String PREFER_FALSE = "##PERFER_FALSE##";
+	public static final String PREFER_DEFAULT = "##PREFER_DEFAULT##";
 
-	public static final String DEFAULT = "##DEFAULT##";
+	public static final String CUSTOM = "##CUSTOM##";
 }
diff --git a/src/main/java/firok/spring/mvci/MVCIntrospectProcessor.java b/src/main/java/firok/spring/mvci/MVCIntrospectProcessor.java
@@ -1,21 +1,27 @@
 package firok.spring.mvci;
 
-import firok.spring.mvci.internal.IntrospectContext;
+import firok.spring.mvci.internal.BeanContext;
+import lombok.SneakyThrows;
 
 import javax.annotation.processing.*;
 import javax.lang.model.SourceVersion;
-import javax.lang.model.element.Element;
+import javax.lang.model.element.PackageElement;
 import javax.lang.model.element.TypeElement;
 import javax.lang.model.util.Elements;
 import javax.lang.model.util.Types;
 import javax.tools.Diagnostic;
 import javax.tools.JavaFileObject;
 import java.io.Writer;
+import java.lang.annotation.Annotation;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Objects;
 import java.util.Set;
 
-import static firok.spring.mvci.Constants.DISABLE;
-
-@SupportedAnnotationTypes("firok.spring.mvci.MVCIntrospective")
+@SupportedAnnotationTypes({
+		"firok.spring.mvci.MVCIntrospective",
+		"firok.spring.mvci.MVCConfig",
+})
 @SupportedSourceVersion(SourceVersion.RELEASE_17)
 public class MVCIntrospectProcessor extends AbstractProcessor
 {
@@ -50,67 +56,128 @@ public synchronized void init(ProcessingEnvironment processingEnv) {
 		messager.printMessage(Diagnostic.Kind.NOTE,"MVC Introspector ADT 初始化完成");
 	}
 
-	private void genJavaSource(String location,String content) throws Exception
+	/**
+	 * 省得出问题 还是加上个锁吧
+	 */
+	private static final Object LOCK_JFO_API = new Object();
+	public Writer createSourceFileWrite(String location) throws Exception
+	{
+		JavaFileObject jfo;
+		synchronized (LOCK_JFO_API) { jfo = filer.createSourceFile(location); return jfo.openWriter(); }
+
+	}
+
+	@SneakyThrows
+	private void checkAnnotation(Annotation anno)
 	{
-		JavaFileObject jfo = filer.createSourceFile(location);
-		try(Writer jw = jfo.openWriter())
+		var classAnno = anno.getClass();
+		for(var field : classAnno.getDeclaredMethods()) // 注解字段 = 方法
 		{
-			jw.write(content);
+//			printNote("field:"+field.getName());
+			AvailableValues annoAVs;
+			synchronized (LOCK_REFLECT_API) { annoAVs = field.getAnnotation(AvailableValues.class); }
+			if(annoAVs == null) {
+//				printNote("continue;");
+				continue;
+			}
+
+			Object fieldValue = field.invoke(anno);
+			assert fieldValue != null;
+
+			boolean passCheck = false;
+			for(var av : annoAVs.value())
+			{
+				if(Constants.CUSTOM.equals(av) || Objects.equals(av, fieldValue))
+				{
+					passCheck = true;
+					break;
+				}
+			}
+
+			if(!passCheck)
+			{
+				printError("错误的值[%s]于[%s]注解的[%s]字段".formatted(
+						fieldValue,
+						classAnno.getSimpleName(),
+						field.getName()
+				));
+			}
 		}
 	}
 
+	/**
+	 * JDK的反射api看起来不是线程安全的, 加一个全局锁省得炸裂
+	 */
+	public static final Object LOCK_REFLECT_API = new Object();
+
 	@Override
 	public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv)
 	{
-		Set<? extends Element> setElements = roundEnv.getElementsAnnotatedWith(MVCIntrospective.class);
-		printNote("读取并处理 MVC 结构");
-		for(Element ele : setElements)
+		try
 		{
-			if(ele instanceof TypeElement te)
-			{
-				var anno = ele.getAnnotation(MVCIntrospective.class);
-				var context = new IntrospectContext(te,anno);
+			var setAnno = roundEnv.getElementsAnnotatedWith(MVCIntrospective.class);
+			if(setAnno.isEmpty()) return true;
 
-				try
+			setAnno.stream().parallel().forEach(ele -> {
+				if(ele instanceof TypeElement eleBean)
 				{
-					String content; // content to generate
-					String qn; // qualified name
-					GEN_MAPPER:
-					{
-						if(DISABLE.equals(anno.mapperTemplate())) break GEN_MAPPER;
-						content = context.pipeline(anno.mapperTemplate());
-						qn = context.mapperPackage + "." + context.mapperName;
-						genJavaSource(qn,content);
-					}
-					GEN_SERVICE:
-					{
-						if(DISABLE.equals(anno.serviceTemplate())) break GEN_SERVICE;
-						content = context.pipeline(anno.serviceTemplate());
-						qn = context.servicePackage + "." + context.serviceName;
-						genJavaSource(qn,content);
-					}
-					GEN_SERVICE_IMPL:
-					{
-						if(DISABLE.equals(anno.serviceImplTemplate())) break GEN_SERVICE_IMPL;
-						content = context.pipeline(anno.serviceImplTemplate());
-						qn = context.serviceImplPackage + "." + context.serviceImplName;
-						genJavaSource(qn,content);
-					}
-					GEN_CONTROLLER:
+					List<MVCIntrospective> configs = new ArrayList<>();
+
+					MVCIntrospective annoBean;
+					synchronized (LOCK_REFLECT_API) { annoBean = eleBean.getAnnotation(MVCIntrospective.class); }
+					checkAnnotation(annoBean);
+					configs.add(annoBean);
+
+					// 读取相关的配置
+
+					final String beanFullName = eleBean.getQualifiedName().toString(); // a.b.c.d.E
+					String beanPackageName = beanFullName.substring(0, beanFullName.lastIndexOf('.')); // a.b.c.d
+					if(beanPackageName.lastIndexOf('.') > 0) do
 					{
-						if(DISABLE.equals(anno.controllerTemplate())) break GEN_CONTROLLER;
-						content = context.pipeline(anno.controllerTemplate());
-						qn = context.controllerPackage + "." + context.controllerName;
-						genJavaSource(qn,content);
+						// 寻找配置注解
+						PackageElement elePackage;
+						synchronized (LOCK_REFLECT_API) { elePackage = elementUtils.getPackageElement(beanPackageName); }
+
+						if(elePackage != null)
+						{
+							MVCIntrospective annoConfig;
+							synchronized (LOCK_REFLECT_API) { annoConfig = elePackage.getAnnotation(MVCIntrospective.class); }
+
+							if(annoConfig != null)
+							{
+								checkAnnotation(annoConfig);
+								configs.add(annoConfig);
+							}
+						}
+
+						// 准备寻找上级package
+						final int indexLastDot = beanPackageName.lastIndexOf('.');
+						if(indexLastDot > 0)
+							beanPackageName = beanPackageName.substring(0, indexLastDot);
+						else // 当前package限定名已经是根package了
+							beanPackageName = null;
 					}
-				}
-				catch (Exception e)
-				{
-					printError("为 [%s] 生成结构失败: %s".formatted(context.beanNameFull,e.getLocalizedMessage()));
-				}
+					while(beanPackageName != null);
 
-				printNote("为 [%s] 生成结构成功".formatted(context.beanNameFull));
-			}
+					// 此时, 列表内的配置是由近而远
+					// 在根据配置调整生成时, 前面的配置会覆盖后面的配置
+
+					// 初始化上下文
+					var context = new BeanContext(configs, eleBean, this);
+
+					// 开始生成
+					context.generate();
+				}
+			});
+		}
+		catch (Exception e)
+		{
+//			if(e.getStackTrace() != null)
+//				for(var st : e.getStackTrace())
+//				{
+//					printWarning(st.toString());
+//				}
+			printError(e);
 		}
 
 		return true;