mirrors/riiablo
1
0
Fork 0
mirror of https://github.com/collinsmith/riiablo.git synced 2025-08-03 00:20:07 +07:00
Code Issues Actions Packages Projects Releases Wiki Activity

Changed Context#getAnnotationMirror to use the annotation class argument

Browse Source
This commit is contained in:
collinsmith
2020-12-15 13:31:48 -08:00
parent 06461b76c7
commit 9656bcc382
2 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
table/annotation-processor/src/main/java/com/riiablo/table/annotation/Context.java
View File

@ -27,9 +27,17 @@ class Context {
elementUtils = processingEnvironment.getElementUtils();
}
/**
* @deprecated use {@link #getAnnotationMirror(Element, ClassName)} with cached value
*/
@Deprecated
AnnotationMirror getAnnotationMirror(Element element, Class<? extends Annotation> annotationClass) {
return getAnnotationMirror(element, ClassName.get(annotationClass));
}
AnnotationMirror getAnnotationMirror(Element element, ClassName annotationClass) {
for (AnnotationMirror annotationMirror : element.getAnnotationMirrors()) {
if (Constants.PRIMARY_KEY.equals(ClassName.get(annotationMirror.getAnnotationType()))) {
if (annotationClass.equals(ClassName.get(annotationMirror.getAnnotationType()))) {
return annotationMirror;
}
}

2
table/annotation-processor/src/main/java/com/riiablo/table/annotation/PrimaryKeyElement.java
View File

@ -44,7 +44,7 @@ class PrimaryKeyElement {
primaryKeyElement = firstAcceptableElement;
}
AnnotationMirror primaryKeyMirror = context.getAnnotationMirror(primaryKeyElement, PrimaryKey.class);
AnnotationMirror primaryKeyMirror = context.getAnnotationMirror(primaryKeyElement, Constants.PRIMARY_KEY);
return new PrimaryKeyElement(primaryKeyElement, primaryKeyMirror);
}