Skip to content
This repository was archived by the owner on Feb 26, 2023. It is now read-only.

allow injection of custom RuntimeExceptionDao #1252

Merged
WonderCsabo merged 2 commits into from
Dec 9, 2014
Merged

allow injection of custom RuntimeExceptionDao #1252

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
c28d6ee
allow injection of custom RuntimeExceptionDao classes
dodgex Nov 19, 2014
82e1362
add test case for injecting OrmLiteDao that extends from RuntimeExcep…
dodgex Nov 19, 2014
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 10 additions & 4 deletions ...ndroidannotations/src/main/java/org/androidannotations/handler/BaseAnnotationHandler.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,8 +15,10 @@
*/
package org.androidannotations.handler;

import com.sun.codemodel.JClass;
import com.sun.codemodel.JCodeModel;
import javax.annotation.processing.ProcessingEnvironment;
import javax.lang.model.element.Element;
import javax.lang.model.type.TypeMirror;

import org.androidannotations.helper.AndroidManifest;
import org.androidannotations.helper.IdAnnotationHelper;
import org.androidannotations.helper.IdValidatorHelper;
Expand All @@ -27,8 +29,8 @@
import org.androidannotations.process.ProcessHolder;
import org.androidannotations.rclass.IRClass;

import javax.annotation.processing.ProcessingEnvironment;
import javax.lang.model.element.Element;
import com.sun.codemodel.JClass;
import com.sun.codemodel.JCodeModel;

public abstract class BaseAnnotationHandler<T extends GeneratedClassHolder> implements AnnotationHandler<T> {

Expand Down Expand Up @@ -89,6 +91,10 @@ public JClass refClass(String fullyQualifiedClassName) {
return processHolder.refClass(fullyQualifiedClassName);
}

public JClass refClass(TypeMirror typeMirror) {
return processHolder.refClass(typeMirror.toString());
}

public JClass refClass(Class<?> clazz) {
return processHolder.refClass(clazz);
}
Expand Down
100 changes: 31 additions & 69 deletions ...ns/androidannotations/src/main/java/org/androidannotations/handler/OrmLiteDaoHandler.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,33 +15,42 @@
*/
package org.androidannotations.handler;

import com.sun.codemodel.*;
import static com.sun.codemodel.JExpr._new;
import static com.sun.codemodel.JExpr.cast;
import static com.sun.codemodel.JExpr.ref;

import javax.annotation.processing.ProcessingEnvironment;
import javax.lang.model.element.Element;
import javax.lang.model.type.TypeMirror;

import org.androidannotations.annotations.OrmLiteDao;
import org.androidannotations.helper.APTCodeModelHelper;
import org.androidannotations.helper.CanonicalNameConstants;
import org.androidannotations.helper.OrmLiteHelper;
import org.androidannotations.helper.TargetAnnotationHelper;
import org.androidannotations.holder.EComponentHolder;
import org.androidannotations.model.AnnotationElements;
import org.androidannotations.process.IsValid;

import javax.annotation.processing.ProcessingEnvironment;
import javax.lang.model.element.Element;
import javax.lang.model.element.TypeElement;
import javax.lang.model.type.DeclaredType;
import javax.lang.model.type.TypeKind;
import javax.lang.model.type.TypeMirror;
import java.util.List;

import static com.sun.codemodel.JExpr.ref;
import com.sun.codemodel.JBlock;
import com.sun.codemodel.JCatchBlock;
import com.sun.codemodel.JClass;
import com.sun.codemodel.JExpression;
import com.sun.codemodel.JFieldVar;
import com.sun.codemodel.JTryBlock;
import com.sun.codemodel.JVar;

public class OrmLiteDaoHandler extends BaseAnnotationHandler<EComponentHolder> {

private TargetAnnotationHelper helper;
private DeclaredType daoParametrizedType;
private DeclaredType runtimeExceptionDaoParametrizedType;
private final OrmLiteHelper ormLiteHelper;
private final TargetAnnotationHelper helper;
private final APTCodeModelHelper codeModelHelper;

public OrmLiteDaoHandler(ProcessingEnvironment processingEnvironment) {
super(OrmLiteDao.class, processingEnvironment);
helper = new TargetAnnotationHelper(processingEnv, getTarget());
ormLiteHelper = new OrmLiteHelper(helper);
codeModelHelper = new APTCodeModelHelper();
}

@Override
Expand All @@ -52,37 +61,30 @@ public void validate(Element element, AnnotationElements validatedElements, IsVa

validatorHelper.isNotPrivate(element, valid);

validatorHelper.extendsOrmLiteDao(element, valid);
validatorHelper.extendsOrmLiteDao(element, valid, ormLiteHelper);

validatorHelper.hasASqlLiteOpenHelperParameterizedType(element, valid);
}

@Override
public void process(Element element, EComponentHolder holder) {
createDaoParametrizedTypes();

String fieldName = element.getSimpleName().toString();

JClass modelClass = getEntityAndIdClass(element, holder);
JClass modelClass = refClass(ormLiteHelper.getEntityType(element));
JClass idClass = refClass(ormLiteHelper.getEntityIdType(element));
JExpression modelClassDotClass = modelClass.dotclass();

JClass daoClass = refClass(CanonicalNameConstants.DAO).narrow(modelClass, idClass);

TypeMirror databaseHelperTypeMirror = helper.extractAnnotationParameter(element, "helper");
JFieldVar databaseHelperRef = holder.getDatabaseHelperRef(databaseHelperTypeMirror);

JBlock initBody = holder.getInitBody();

JExpression injectExpr;
JExpression injectExpr = databaseHelperRef.invoke("getDao").arg(modelClassDotClass);
if (elementExtendsRuntimeExceptionDao(element)) {

injectExpr = classes().RUNTIME_EXCEPTION_DAO//
.staticInvoke("createDao") //
.arg(databaseHelperRef.invoke("getConnectionSource")) //
.arg(modelClassDotClass);

} else {

injectExpr = databaseHelperRef.invoke("getDao").arg(modelClassDotClass);

JClass daoImplClass = codeModelHelper.typeMirrorToJClass(element.asType(), holder);
injectExpr = _new(daoImplClass).arg(cast(daoClass, injectExpr));
}

JTryBlock tryBlock = initBody._try();
Expand All @@ -100,46 +102,6 @@ public void process(Element element, EComponentHolder holder) {

private boolean elementExtendsRuntimeExceptionDao(Element element) {
TypeMirror elementType = element.asType();
return helper.isSubtype(elementType, runtimeExceptionDaoParametrizedType);
}

private JClass getEntityAndIdClass(Element element, EComponentHolder holder) {
if (isSubtypeOfDao(element.asType())) {
DeclaredType declaredType = (DeclaredType) element.asType();
List<? extends TypeMirror> typeArguments = declaredType.getTypeArguments();
if (typeArguments.size() == 2) {
return holder.refClass(typeArguments.get(0).toString());
}
}

List<? extends TypeMirror> superTypes = helper.directSupertypes(element.asType());
for (TypeMirror superType : superTypes) {

if (superType.getKind() == TypeKind.DECLARED && isSubtypeOfDao(superType)) {

DeclaredType superDeclaredType = (DeclaredType) superType;
List<? extends TypeMirror> typeArguments = superDeclaredType.getTypeArguments();
if (typeArguments.size() == 2) {
return holder.refClass(typeArguments.get(0).toString());
}
}
}
return null;
}

private boolean isSubtypeOfDao(TypeMirror type) {
return helper.isSubtype(type, daoParametrizedType) || helper.isSubtype(type, runtimeExceptionDaoParametrizedType);
}

/*
* This method is not in the constructor because it prevents tests from
* succeeding
*/
private void createDaoParametrizedTypes() {
TypeMirror wildcardType = helper.getTypeUtils().getWildcardType(null, null);
TypeElement daoTypeElement = helper.typeElementFromQualifiedName(CanonicalNameConstants.DAO);
daoParametrizedType = helper.getTypeUtils().getDeclaredType(daoTypeElement, wildcardType, wildcardType);
TypeElement runtimeExceptionDaoTypeElement = helper.typeElementFromQualifiedName(CanonicalNameConstants.RUNTIME_EXCEPTION_DAO);
runtimeExceptionDaoParametrizedType = helper.getTypeUtils().getDeclaredType(runtimeExceptionDaoTypeElement, wildcardType, wildcardType);
return helper.isSubtype(elementType, ormLiteHelper.getRuntimeExceptionDaoParametrizedType());
}
}
95 changes: 95 additions & 0 deletions ...tations/androidannotations/src/main/java/org/androidannotations/helper/OrmLiteHelper.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,15 @@

import static com.sun.codemodel.JExpr._null;

import java.util.List;

import javax.lang.model.element.Element;
import javax.lang.model.element.TypeElement;
import javax.lang.model.type.DeclaredType;
import javax.lang.model.type.TypeKind;
import javax.lang.model.type.TypeMirror;
import javax.lang.model.util.Types;

import org.androidannotations.holder.HasLifecycleMethods;
import org.androidannotations.process.ProcessHolder;

Expand All @@ -25,6 +34,92 @@

public class OrmLiteHelper {

private final TargetAnnotationHelper helper;

private DeclaredType daoParametrizedType;
private DeclaredType runtimeExceptionDaoParametrizedType;

private TypeElement daoTypeElement;
private TypeElement runtimeExceptionDaoTypeElement;

public OrmLiteHelper(TargetAnnotationHelper helper) {
this.helper = helper;
}

public TypeMirror getEntityType(Element element) {
return getEntityOrIdType(element, 0);
}

public TypeMirror getEntityIdType(Element element) {
return getEntityOrIdType(element, 1);
}

private TypeMirror getEntityOrIdType(Element element, int index) {
if (isSubtypeOfDao(element.asType())) {
DeclaredType declaredType = (DeclaredType) element.asType();
List<? extends TypeMirror> typeArguments = declaredType.getTypeArguments();
if (typeArguments.size() == 2) {
return typeArguments.get(index);
}
}

List<? extends TypeMirror> superTypes = helper.directSupertypes(element.asType());
for (TypeMirror superType : superTypes) {

if (superType.getKind() == TypeKind.DECLARED && isSubtypeOfDao(superType)) {

DeclaredType superDeclaredType = (DeclaredType) superType;
List<? extends TypeMirror> typeArguments = superDeclaredType.getTypeArguments();
if (typeArguments.size() == 2) {
return typeArguments.get(index);
}
}
}
return null;
}

private boolean isSubtypeOfDao(TypeMirror type) {
return helper.isSubtype(type, daoParametrizedType) || helper.isSubtype(type, runtimeExceptionDaoParametrizedType);
}

public DeclaredType getDaoParametrizedType() {
if (daoParametrizedType == null) {
createDaoParametrizedTypes();
}
return daoParametrizedType;
}

public DeclaredType getRuntimeExceptionDaoParametrizedType() {
if (runtimeExceptionDaoParametrizedType == null) {
createDaoParametrizedTypes();
}
return runtimeExceptionDaoParametrizedType;
}

public DeclaredType getTypedDao(Element element) {
if (daoTypeElement == null) {
createDaoParametrizedTypes();
}
return helper.getTypeUtils().getDeclaredType(daoTypeElement, getEntityType(element), getEntityIdType(element));
}

public DeclaredType getTypedRuntimeExceptionDao(Element element) {
if (runtimeExceptionDaoTypeElement == null) {
createDaoParametrizedTypes();
}
return helper.getTypeUtils().getDeclaredType(runtimeExceptionDaoTypeElement, getEntityType(element), getEntityIdType(element));
}

private void createDaoParametrizedTypes() {
daoTypeElement = helper.typeElementFromQualifiedName(CanonicalNameConstants.DAO);
runtimeExceptionDaoTypeElement = helper.typeElementFromQualifiedName(CanonicalNameConstants.RUNTIME_EXCEPTION_DAO);

Types typeUtils = helper.getTypeUtils();
TypeMirror wildcardType = typeUtils.getWildcardType(null, null);
daoParametrizedType = helper.getTypeUtils().getDeclaredType(daoTypeElement, wildcardType, wildcardType);
runtimeExceptionDaoParametrizedType = helper.getTypeUtils().getDeclaredType(runtimeExceptionDaoTypeElement, wildcardType, wildcardType);
}

public static void injectReleaseInDestroy(JFieldVar databaseHelperRef, HasLifecycleMethods holder, ProcessHolder.Classes classes) {
JBlock destroyBody = holder.getOnDestroyBeforeSuperBlock();

Expand Down
Loading