|
43 | 43 |
|
44 | 44 | public class OrmLiteDaoHandler extends BaseAnnotationHandler<EComponentHolder> { |
45 | 45 |
|
46 | | - private TargetAnnotationHelper helper; |
47 | | - private DeclaredType daoParametrizedType; |
48 | | - private DeclaredType runtimeExceptionDaoParametrizedType; |
| 46 | + private TargetAnnotationHelper helper; |
| 47 | + private DeclaredType daoParametrizedType; |
| 48 | + private DeclaredType runtimeExceptionDaoParametrizedType; |
49 | 49 | private boolean injectedOnDestroy = false; |
50 | 50 |
|
51 | | - public OrmLiteDaoHandler(ProcessingEnvironment processingEnvironment) { |
52 | | - super(OrmLiteDao.class, processingEnvironment); |
53 | | - helper = new TargetAnnotationHelper(processingEnv, getTarget()); |
54 | | - } |
| 51 | + public OrmLiteDaoHandler(ProcessingEnvironment processingEnvironment) { |
| 52 | + super(OrmLiteDao.class, processingEnvironment); |
| 53 | + helper = new TargetAnnotationHelper(processingEnv, getTarget()); |
| 54 | + } |
55 | 55 |
|
56 | | - @Override |
57 | | - public void validate(Element element, AnnotationElements validatedElements, IsValid valid) { |
58 | | - validatorHelper.hasOrmLiteJars(element, valid); |
| 56 | + @Override |
| 57 | + public void validate(Element element, AnnotationElements validatedElements, IsValid valid) { |
| 58 | + validatorHelper.hasOrmLiteJars(element, valid); |
59 | 59 |
|
60 | | - validatorHelper.enclosingElementHasEnhancedComponentAnnotation(element, validatedElements, valid); |
| 60 | + validatorHelper.enclosingElementHasEnhancedComponentAnnotation(element, validatedElements, valid); |
61 | 61 |
|
62 | | - validatorHelper.isNotPrivate(element, valid); |
| 62 | + validatorHelper.isNotPrivate(element, valid); |
63 | 63 |
|
64 | | - validatorHelper.extendsOrmLiteDao(element, valid); |
| 64 | + validatorHelper.extendsOrmLiteDao(element, valid); |
65 | 65 |
|
66 | | - validatorHelper.hasASqlLiteOpenHelperParameterizedType(element, valid); |
67 | | - } |
| 66 | + validatorHelper.hasASqlLiteOpenHelperParameterizedType(element, valid); |
| 67 | + } |
68 | 68 |
|
69 | | - @Override |
70 | | - public void process(Element element, EComponentHolder holder) { |
71 | | - createDaoParametrizedTypes(); |
| 69 | + @Override |
| 70 | + public void process(Element element, EComponentHolder holder) { |
| 71 | + createDaoParametrizedTypes(); |
72 | 72 |
|
73 | | - String fieldName = element.getSimpleName().toString(); |
| 73 | + String fieldName = element.getSimpleName().toString(); |
74 | 74 |
|
75 | | - JClass modelClass = getEntityAndIdClass(element, holder); |
76 | | - JExpression modelClassDotClass = modelClass.dotclass(); |
| 75 | + JClass modelClass = getEntityAndIdClass(element, holder); |
| 76 | + JExpression modelClassDotClass = modelClass.dotclass(); |
77 | 77 |
|
78 | | - TypeMirror databaseHelperTypeMirror = helper.extractAnnotationParameter(element, "helper"); |
79 | | - JFieldVar databaseHelperRef = holder.getDatabaseHelperRef(databaseHelperTypeMirror); |
| 78 | + TypeMirror databaseHelperTypeMirror = helper.extractAnnotationParameter(element, "helper"); |
| 79 | + JFieldVar databaseHelperRef = holder.getDatabaseHelperRef(databaseHelperTypeMirror); |
80 | 80 |
|
81 | 81 | // Create (onCreate or init) |
82 | 82 | createGetDaoMethod(element, holder, fieldName, modelClassDotClass, databaseHelperRef); |
@@ -116,47 +116,47 @@ private void createGetDaoMethod(Element element, EComponentHolder holder, String |
116 | 116 | } |
117 | 117 |
|
118 | 118 | private boolean elementExtendsRuntimeExceptionDao(Element element) { |
119 | | - TypeMirror elementType = element.asType(); |
120 | | - return helper.isSubtype(elementType, runtimeExceptionDaoParametrizedType); |
121 | | - } |
122 | | - |
123 | | - private JClass getEntityAndIdClass(Element element, EComponentHolder holder) { |
124 | | - if (isSubtypeOfDao(element.asType())) { |
125 | | - DeclaredType declaredType = (DeclaredType) element.asType(); |
126 | | - List<? extends TypeMirror> typeArguments = declaredType.getTypeArguments(); |
127 | | - if (typeArguments.size() == 2) { |
128 | | - return holder.refClass(typeArguments.get(0).toString()); |
129 | | - } |
130 | | - } |
131 | | - |
132 | | - List<? extends TypeMirror> superTypes = helper.directSupertypes(element.asType()); |
133 | | - for (TypeMirror superType : superTypes) { |
134 | | - |
135 | | - if (superType.getKind() == TypeKind.DECLARED && isSubtypeOfDao(superType)) { |
136 | | - |
137 | | - DeclaredType superDeclaredType = (DeclaredType) superType; |
138 | | - List<? extends TypeMirror> typeArguments = superDeclaredType.getTypeArguments(); |
139 | | - if (typeArguments.size() == 2) { |
140 | | - return holder.refClass(typeArguments.get(0).toString()); |
141 | | - } |
142 | | - } |
143 | | - } |
144 | | - return null; |
145 | | - } |
146 | | - |
147 | | - private boolean isSubtypeOfDao(TypeMirror type) { |
148 | | - return helper.isSubtype(type, daoParametrizedType) || helper.isSubtype(type, runtimeExceptionDaoParametrizedType); |
149 | | - } |
150 | | - |
151 | | - /* |
152 | | - * This method is not in the constructor because it prevents tests from |
153 | | - * succeeding |
154 | | - */ |
155 | | - private void createDaoParametrizedTypes() { |
156 | | - TypeMirror wildcardType = helper.getTypeUtils().getWildcardType(null, null); |
157 | | - TypeElement daoTypeElement = helper.typeElementFromQualifiedName(CanonicalNameConstants.DAO); |
158 | | - daoParametrizedType = helper.getTypeUtils().getDeclaredType(daoTypeElement, wildcardType, wildcardType); |
159 | | - TypeElement runtimeExceptionDaoTypeElement = helper.typeElementFromQualifiedName(CanonicalNameConstants.RUNTIME_EXCEPTION_DAO); |
160 | | - runtimeExceptionDaoParametrizedType = helper.getTypeUtils().getDeclaredType(runtimeExceptionDaoTypeElement, wildcardType, wildcardType); |
161 | | - } |
| 119 | + TypeMirror elementType = element.asType(); |
| 120 | + return helper.isSubtype(elementType, runtimeExceptionDaoParametrizedType); |
| 121 | + } |
| 122 | + |
| 123 | + private JClass getEntityAndIdClass(Element element, EComponentHolder holder) { |
| 124 | + if (isSubtypeOfDao(element.asType())) { |
| 125 | + DeclaredType declaredType = (DeclaredType) element.asType(); |
| 126 | + List<? extends TypeMirror> typeArguments = declaredType.getTypeArguments(); |
| 127 | + if (typeArguments.size() == 2) { |
| 128 | + return holder.refClass(typeArguments.get(0).toString()); |
| 129 | + } |
| 130 | + } |
| 131 | + |
| 132 | + List<? extends TypeMirror> superTypes = helper.directSupertypes(element.asType()); |
| 133 | + for (TypeMirror superType : superTypes) { |
| 134 | + |
| 135 | + if (superType.getKind() == TypeKind.DECLARED && isSubtypeOfDao(superType)) { |
| 136 | + |
| 137 | + DeclaredType superDeclaredType = (DeclaredType) superType; |
| 138 | + List<? extends TypeMirror> typeArguments = superDeclaredType.getTypeArguments(); |
| 139 | + if (typeArguments.size() == 2) { |
| 140 | + return holder.refClass(typeArguments.get(0).toString()); |
| 141 | + } |
| 142 | + } |
| 143 | + } |
| 144 | + return null; |
| 145 | + } |
| 146 | + |
| 147 | + private boolean isSubtypeOfDao(TypeMirror type) { |
| 148 | + return helper.isSubtype(type, daoParametrizedType) || helper.isSubtype(type, runtimeExceptionDaoParametrizedType); |
| 149 | + } |
| 150 | + |
| 151 | + /* |
| 152 | + * This method is not in the constructor because it prevents tests from |
| 153 | + * succeeding |
| 154 | + */ |
| 155 | + private void createDaoParametrizedTypes() { |
| 156 | + TypeMirror wildcardType = helper.getTypeUtils().getWildcardType(null, null); |
| 157 | + TypeElement daoTypeElement = helper.typeElementFromQualifiedName(CanonicalNameConstants.DAO); |
| 158 | + daoParametrizedType = helper.getTypeUtils().getDeclaredType(daoTypeElement, wildcardType, wildcardType); |
| 159 | + TypeElement runtimeExceptionDaoTypeElement = helper.typeElementFromQualifiedName(CanonicalNameConstants.RUNTIME_EXCEPTION_DAO); |
| 160 | + runtimeExceptionDaoParametrizedType = helper.getTypeUtils().getDeclaredType(runtimeExceptionDaoTypeElement, wildcardType, wildcardType); |
| 161 | + } |
162 | 162 | } |
0 commit comments