Add SpringRepository and SpringScope annotations for generating JdbcMapper implementations with spring annotations

This commit is contained in:
Travis Burtrum 2018-10-23 01:22:29 -04:00
parent 8698c2740d
commit 20c5cb6f8d
3 changed files with 100 additions and 0 deletions

View File

@ -1,6 +1,8 @@
package com.moparisthebest.jdbc.codegen; package com.moparisthebest.jdbc.codegen;
import com.moparisthebest.jdbc.*; import com.moparisthebest.jdbc.*;
import com.moparisthebest.jdbc.codegen.spring.SpringRepository;
import com.moparisthebest.jdbc.codegen.spring.SpringScope;
import javax.annotation.processing.*; import javax.annotation.processing.*;
import javax.lang.model.SourceVersion; import javax.lang.model.SourceVersion;
@ -10,6 +12,7 @@ import javax.lang.model.util.Elements;
import javax.lang.model.util.Types; import javax.lang.model.util.Types;
import javax.tools.Diagnostic; import javax.tools.Diagnostic;
import java.io.*; import java.io.*;
import java.lang.annotation.Annotation;
import java.sql.Blob; import java.sql.Blob;
import java.sql.Clob; import java.sql.Clob;
import java.sql.Connection; import java.sql.Connection;
@ -49,6 +52,7 @@ public class JdbcMapperProcessor extends AbstractProcessor {
} }
static Types types; static Types types;
static Elements elements;
static Messager messager; static Messager messager;
public static Types getTypes() { public static Types getTypes() {
@ -88,6 +92,7 @@ public class JdbcMapperProcessor extends AbstractProcessor {
java8 = RELEASE_8 != null && processingEnv.getSourceVersion().ordinal() >= RELEASE_8.ordinal(); java8 = RELEASE_8 != null && processingEnv.getSourceVersion().ordinal() >= RELEASE_8.ordinal();
types = processingEnv.getTypeUtils(); types = processingEnv.getTypeUtils();
elements = processingEnv.getElementUtils();
messager = processingEnv.getMessager(); messager = processingEnv.getMessager();
final Elements elements = processingEnv.getElementUtils(); final Elements elements = processingEnv.getElementUtils();
sqlExceptionType = elements.getTypeElement(SQLException.class.getCanonicalName()).asType(); sqlExceptionType = elements.getTypeElement(SQLException.class.getCanonicalName()).asType();
@ -227,6 +232,26 @@ public class JdbcMapperProcessor extends AbstractProcessor {
w.write("import java.sql.*;\n\n"); w.write("import java.sql.*;\n\n");
w.write("import static com.moparisthebest.jdbc.util.ResultSetUtil.*;\n"); w.write("import static com.moparisthebest.jdbc.util.ResultSetUtil.*;\n");
w.write("import static com.moparisthebest.jdbc.TryClose.tryClose;\n\n"); w.write("import static com.moparisthebest.jdbc.TryClose.tryClose;\n\n");
final SpringRepository springRepository = getLowestClassOrPackageAnnotation(genClass, SpringRepository.class);
if(springRepository != null) {
w.write("@org.springframework.stereotype.Repository");
if(!springRepository.value().isEmpty())
w.append("(\"").append(escapeJavaString(springRepository.value())).append("\")");
w.write("\n");
}
final SpringScope springScope = getLowestClassOrPackageAnnotation(genClass, SpringScope.class);
if(springScope != null) {
w.write("@org.springframework.context.annotation.Scope(");
if(!springScope.scopeName().isEmpty())
w.append("\n\tscopeName = \"").append(escapeJavaString(springScope.scopeName())).append("\",");
String proxyMode = springScope.proxyMode();
if(proxyMode.isEmpty())
proxyMode = isInterface ? "INTERFACES" : "TARGET_CLASS";
w.append("\n\tproxyMode = org.springframework.context.annotation.ScopedProxyMode.").append(proxyMode).append(")\n");
}
w.write("public class "); w.write("public class ");
w.write(className); w.write(className);
if (isInterface) { if (isInterface) {
@ -1084,6 +1109,28 @@ public class JdbcMapperProcessor extends AbstractProcessor {
} }
} }
private static final Pattern lastPackage = Pattern.compile("\\.?[^.]+$");
private static <A extends Annotation> A getLowestClassOrPackageAnnotation(final Element typeElement, final Class<A> annotationType) {
A ret = typeElement.getAnnotation(annotationType);
if(ret != null)
return ret;
PackageElement packageElement = elements.getPackageOf(typeElement);
while(packageElement != null) {
ret = packageElement.getAnnotation(annotationType);
if(ret != null)
return ret;
if(packageElement.isUnnamed())
break;
packageElement = elements.getPackageElement(lastPackage.matcher(packageElement.getQualifiedName()).replaceFirst(""));
}
return null;
}
private static String escapeJavaString(final String s) {
return s.replace("\"", "\\\"")
.replace("\n", "\\n");
}
public static String toString(final Throwable e) { public static String toString(final Throwable e) {
StringWriter sw = null; StringWriter sw = null;
PrintWriter pw = null; PrintWriter pw = null;

View File

@ -0,0 +1,21 @@
package com.moparisthebest.jdbc.codegen.spring;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
/**
* This puts a @org.springframework.stereotype.Repository annotation on the generated class
* <p>
* If put on a package, applies to all classes generated under that package, if put on a
* class or interface only applies to that class/interface, most specific class or package wins.
*/
@Retention(RetentionPolicy.SOURCE)
@Target({ElementType.TYPE, ElementType.PACKAGE})
public @interface SpringRepository {
/**
* Used for @org.springframework.stereotype.Repository.value
*/
String value() default "";
}

View File

@ -0,0 +1,32 @@
package com.moparisthebest.jdbc.codegen.spring;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
/**
* This puts a @org.springframework.context.annotation.Scope annotation on the generated class
* <p>
* If put on a package, applies to all classes generated under that package, if put on a
* class or interface only applies to that class/interface, most specific class or package wins.
*/
@Retention(RetentionPolicy.SOURCE)
@Target({ElementType.TYPE, ElementType.PACKAGE})
public @interface SpringScope {
/**
* Used for @org.springframework.context.annotation.Scope.scopeName
* <p>
* Defaults to "request", which is, as of Spring 5, the value of:
* org.springframework.web.context.WebApplicationContext.SCOPE_REQUEST
*/
String scopeName() default "request";
/**
* Used for @org.springframework.context.annotation.Scope.proxyMode
* <p>
* If empty string default is set, defaults to setting ScopedProxyMode.INTERFACES
* when implementing an interface, or ScopedProxyMode.TARGET_CLASS otherwise
*/
String proxyMode() default "";
}