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

Fixed build issue, see #142 #issuecomment-939136245

Browse Source 
https://github.com/collinsmith/riiablo/issues/142#issuecomment-939136245
This commit is contained in:
Collin Smith
2021-10-08 15:08:10 -07:00
parent 0a6547b1b6
commit 7e7b340fa2
1 changed files with 50 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
excel/annotation-processor/src/main/java/com/riiablo/excel/PrimaryKeyAnnotatedElement.java Normal file
View File

@ -0,0 +1,50 @@
package com.riiablo.excel;
import com.squareup.javapoet.ClassName;
import com.squareup.javapoet.TypeName;
import java.util.Arrays;
import javax.lang.model.element.AnnotationMirror;
import javax.lang.model.element.Element;
import javax.lang.model.element.ElementKind;
import javax.lang.model.element.VariableElement;
import org.apache.commons.lang3.ArrayUtils;
import com.riiablo.excel.annotation.PrimaryKey;
public class PrimaryKeyAnnotatedElement {
static final TypeName STRING = ClassName.get("java.lang", "String");
static final TypeName[] VALID_TYPES = new TypeName[] { TypeName.INT, STRING };
static PrimaryKeyAnnotatedElement get(Element element) {
AnnotationMirror a = ElementUtils.getAnnotationMirror(element, PrimaryKey.class);
if (a == null) return null;
if (element.getKind() != ElementKind.FIELD) {
throw new GenerationException(
String.format("only fields can be @%s", PrimaryKey.class.getCanonicalName()),
element, a);
}
TypeName elementType = ClassName.get(element.asType());
if (!ArrayUtils.contains(VALID_TYPES, elementType)) {
throw new GenerationException(
String.format("@%s must be one of %s",
PrimaryKey.class.getCanonicalName(),
Arrays.toString(VALID_TYPES)),
element, a);
}
PrimaryKey annotation = element.getAnnotation(PrimaryKey.class);
return new PrimaryKeyAnnotatedElement((VariableElement) element, a, annotation);
}
PrimaryKey annotation;
AnnotationMirror mirror;
VariableElement element;
PrimaryKeyAnnotatedElement(VariableElement element, AnnotationMirror mirror, PrimaryKey annotation) {
this.element = element;
this.mirror = mirror;
this.annotation = annotation;
}
}