com.wesleyhome.dao.processor.method.FindByPKQueryMethodGenerator Maven / Gradle / Ivy

Go to download
/*
 * @(#)FindAllQueryMethodGenerator.java
 * 
 * (C) Copyright 2014 by Travelers
 * All Rights Reserved.
 * 
 * This software is the confidential and proprietary information
 * of the Travelers Corporation. ("Confidential Information").
 * Redistribution of the source code or binary form is not permitted
 * without prior authorization from Travelers.
 */
package com.wesleyhome.dao.processor.method;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import javax.lang.model.element.Element;
import javax.lang.model.type.TypeMirror;
import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Path;
import javax.persistence.criteria.Root;
import com.sun.codemodel.JBlock;
import com.sun.codemodel.JClass;
import com.sun.codemodel.JCodeModel;
import com.sun.codemodel.JDefinedClass;
import com.sun.codemodel.JExpr;
import com.sun.codemodel.JFieldRef;
import com.sun.codemodel.JFieldVar;
import com.sun.codemodel.JInvocation;
import com.sun.codemodel.JMethod;
import com.sun.codemodel.JMod;
import com.sun.codemodel.JVar;
import com.wesleyhome.dao.processor.EntityInformationMap;
import com.wesleyhome.dao.processor.model.EntityInfo;
import com.wesleyhome.dao.processor.model.MappingType;

/**
 * The FindAllQueryMethodGenerator class is a
 * 
 * @author
 * @since
 */
public class FindByPKQueryMethodGenerator extends BaseQueryMethodGenerator {

	/* (non-Javadoc)
	 * @see com.travelers.smart.dao.generator.processor.method.BaseQueryMethodGenerator#createQueryMethod(com.sun.codemodel.JCodeModel, com.sun.codemodel.JClass, com.sun.codemodel.JDefinedClass, javax.lang.model.element.Element, com.travelers.smart.dao.generator.processor.model.EntityInfo)
	 */
	@Override
	protected JMethod createQueryMethod(final JCodeModel model, final JClass entityJClass, final JDefinedClass _class,
		final Element fieldElement, final EntityInfo entityInfo, final EntityInformationMap entityInfoMap, MappingType mappingType) {
		String methodName = "findByPK";
		int queryMethodModifiers = _class.isInterface() ? JMod.NONE : JMod.PUBLIC;
		JMethod method = _class.method(queryMethodModifiers, entityJClass, methodName);
		List idElements = entityInfo.idElements;
		for (Element element : idElements) {
			addMethodParameter(model, method, element);
		}
		return method;
	}

	/**
	 * @param model
	 * @param method
	 * @param element
	 */
	private void addMethodParameter(final JCodeModel model, final JMethod method, final Element element) {
		TypeMirror fieldTypeMirror = element.asType();
		String fieldTypeString = fieldTypeMirror.toString();
		JClass fieldType = model.ref(fieldTypeString);
		addMethodParameters(method, fieldType, element.getSimpleName().toString());
	}

	/* (non-Javadoc)
	 * @see com.travelers.smart.dao.generator.processor.method.BaseQueryMethodGenerator#createMethodBody(com.sun.codemodel.JMethod, com.sun.codemodel.JClass, com.sun.codemodel.JCodeModel, javax.lang.model.element.Element, com.sun.codemodel.JFieldVar, com.travelers.smart.dao.generator.processor.model.EntityInfo)
	 */
	@Override
	protected void createMethodBody(final JMethod method, final JClass entityJClass, final JCodeModel model, final Element fieldElement,
		final JFieldVar entityManagerField, final EntityInfo entityInfo, final EntityInformationMap entityInfoMap, MappingType mappingType) {
		method.annotate(Override.class);
		JBlock body = method.body();
		JClass cbClass = model.ref(CriteriaBuilder.class);
		JInvocation invoke = JExpr._this().invoke("getCriteriaBuilder");
		JVar criteriaBuilderVar = body.decl(cbClass, "cb", invoke);
		JInvocation criteriaQueryInvoke = criteriaBuilderVar.invoke("createQuery");
		criteriaQueryInvoke.arg(entityJClass.staticRef("class"));
		JClass criteriaQueryRef = model.ref(CriteriaQuery.class);
		JClass cqNarrow = criteriaQueryRef.narrow(entityJClass);
		JVar cqVar = body.decl(cqNarrow, "cq", criteriaQueryInvoke);
		JClass rootRef = model.ref(Root.class);
		JClass rootNarrow = rootRef.narrow(entityJClass);
		JInvocation fromInvoke = cqVar.invoke("from");
		fromInvoke.arg(entityJClass.staticRef("class"));
		JVar rootVar = body.decl(rootNarrow, "root", fromInvoke);
		String entityClassName = entityJClass.fullName();
		String entityMetamodelClassName = entityClassName + "_";
		JClass metamodelClass = model.ref(entityMetamodelClassName);
		List predicateInvokes = new ArrayList<>();
		List predicateNames = new ArrayList<>();
		List params = method.params();
		List idElements = entityInfo.idElements;
		Iterator elmItr = idElements.iterator();
		Iterator paramItr = params.iterator();
		while (elmItr.hasNext()) {
			Element idElement = elmItr.next();
			JVar param = paramItr.next();
			String fieldName = idElement.getSimpleName().toString();
			JFieldRef fieldMetamodelRef = metamodelClass.staticRef(fieldName);
			JClass pathClass = model.ref(Path.class);
			JClass fieldTypeRef = model.ref(idElement.asType().toString());
			JClass pathNarrow = pathClass.narrow(fieldTypeRef);
			JInvocation getInvoke = rootVar.invoke("get");
			getInvoke.arg(fieldMetamodelRef);
			JInvocation predicateInvoke = JExpr.invoke("equals");
			JVar pathVar = body.decl(pathNarrow, fieldName + "Path", getInvoke);
			predicateInvoke.arg(pathVar).arg(param);
			predicateInvokes.add(predicateInvoke);
			predicateNames.add(fieldName + "Predicate");
		}
		JInvocation whereInvoke = cqVar.invoke("where");
		invokePredicates(model, body, whereInvoke, predicateInvokes, predicateNames);
		body.add(whereInvoke);
		JInvocation resultListInvoke = JExpr._this().invoke("getSingleResult");
		resultListInvoke.arg(cqVar);
		body._return(resultListInvoke);
	}
}