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

Added DeclaredType resolution to TableElement

Browse Source
This commit is contained in:
Collin Smith
2020-12-15 23:54:58 -08:00
parent 44282ad9ba
commit 1e3e2dd457
1 changed files with 23 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
table/annotation-processor/src/main/java/com/riiablo/table/annotation/TableElement.java
View File

@ -1,6 +1,8 @@
package com.riiablo.table.annotation;
import javax.lang.model.element.Element;
import javax.lang.model.element.ElementKind;
import javax.lang.model.element.ExecutableElement;
import javax.lang.model.element.TypeElement;
import javax.lang.model.type.DeclaredType;
import javax.lang.model.type.MirroredTypeException;
@ -10,17 +12,20 @@ final class TableElement {
static TableElement get(Context context, Element element) {
Table annotation = element.getAnnotation(Table.class);
final TypeElement tableElement, tableImplElement;
final DeclaredType declaredType;
if (annotation == null) {
// Only need tableElement if generating Table impl
tableElement = context.elementUtils.getTypeElement(com.riiablo.table.Table.class.getCanonicalName());
declaredType = context.typeUtils.getDeclaredType(tableElement, element.asType());
tableImplElement = null;
} else {
// Only need tableImplElement if @Table present
tableImplElement = getTableImpl(context, annotation);
tableElement = null;
declaredType = null;
}
return new TableElement(annotation, tableElement, tableImplElement);
return new TableElement(annotation, declaredType, tableElement, tableImplElement);
}
static TypeElement getTableImpl(Context context, Table annotation) {
@ -35,22 +40,39 @@ final class TableElement {
}
final Table annotation;
final DeclaredType declaredType;
final TypeElement tableElement; // Class<Table>
final TypeElement tableImplElement; // Class<? extends Table<?>>
TableElement(
Table annotation,
DeclaredType declaredType,
TypeElement tableElement,
TypeElement tableImplElement) {
this.annotation = annotation;
this.declaredType = declaredType;
this.tableElement = tableElement;
this.tableImplElement = tableImplElement;
}
ExecutableElement getMethod(CharSequence methodName) {
for (Element e : tableElement.getEnclosedElements()) {
if (e.getKind() == ElementKind.METHOD) {
ExecutableElement methodElement = (ExecutableElement) e;
if (methodElement.getSimpleName().contentEquals(methodName)) {
return methodElement;
}
}
}
throw new AssertionError(tableElement + " does not contain " + methodName);
}
@Override
public String toString() {
return new ToStringBuilder(this)
.append("annotation", annotation)
.append("declaredType", declaredType)
.append("tableElement", tableElement)
.append("tableImplElement", tableImplElement)
.toString();