/* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.activiti.engine.impl.variable;
import java.lang.reflect.InvocationTargetException;
import java.math.BigDecimal;
import java.math.BigInteger;
import java.util.HashMap;
import java.util.Map;
import javax.persistence.EntityManager;
import org.activiti.engine.ActivitiException;
import org.activiti.engine.impl.context.Context;
import org.activiti.engine.impl.util.ReflectUtil;
/**
* @author Frederik Heremans
*/
public class JPAEntityMappings {
private Map<String, EntityMetaData> classMetaDatamap;
private JPAEntityScanner enitityScanner;
public JPAEntityMappings() {
classMetaDatamap = new HashMap<String, EntityMetaData>();
enitityScanner = new JPAEntityScanner();
}
public boolean isJPAEntity(Object value) {
if (value != null) {
// EntityMetaData will be added for all classes, even those who are not
// JPA-entities to prevent unneeded annotation scanning
return getEntityMetaData(value.getClass()).isJPAEntity();
}
return false;
}
private EntityMetaData getEntityMetaData(Class<?> clazz) {
EntityMetaData metaData = classMetaDatamap.get(clazz.getName());
if (metaData == null) {
// Class not present in meta-data map, create metaData for it and add
metaData = scanClass(clazz);
classMetaDatamap.put(clazz.getName(), metaData);
}
return metaData;
}
private EntityMetaData scanClass(Class<?> clazz) {
return enitityScanner.scanClass(clazz);
}
public String getJPAClassString(Object value) {
if(value == null) {
throw new ActivitiException("null value cannot be saved");
}
EntityMetaData metaData = getEntityMetaData(value.getClass());
if(!metaData.isJPAEntity()) {
throw new ActivitiException("Object is not a JPA Entity: class='" + value.getClass() + "', " + value);
}
// Extract the ID from the Entity instance using the metaData
return metaData.getEntityClass().getName();
}
public String getJPAIdString(Object value) {
EntityMetaData metaData = getEntityMetaData(value.getClass());
if(!metaData.isJPAEntity()) {
throw new ActivitiException("Object is not a JPA Entity: class='" + value.getClass() + "', " + value);
}
Object idValue = getIdValue(value, metaData);
return getIdString(idValue);
}
private Object getIdValue(Object value, EntityMetaData metaData) {
try {
if (metaData.getIdMethod() != null) {
return metaData.getIdMethod().invoke(value);
} else if (metaData.getIdField() != null) {
return metaData.getIdField().get(value);
}
} catch (IllegalArgumentException iae) {
throw new ActivitiException("Illegal argument exception when getting value from id method/field on JPAEntity", iae);
} catch (IllegalAccessException iae) {
throw new ActivitiException("Cannot access id method/field for JPA Entity", iae);
} catch (InvocationTargetException ite) {
throw new ActivitiException("Exception occured while getting value from id field/method on JPAEntity: " +
ite.getCause().getMessage(), ite.getCause());
}
// Fall trough when no method and field is set
throw new ActivitiException("Cannot get id from JPA Entity, no id method/field set");
}
public Object getJPAEntity(String className, String idString) {
Class<?> entityClass = null;
entityClass = ReflectUtil.loadClass(className);
EntityMetaData metaData = getEntityMetaData(entityClass);
if(metaData == null) {
throw new ActivitiException("Class is not a JPA-entity: " + className);
}
// Create primary key of right type
Object primaryKey = createId(metaData, idString);
return findEntity(entityClass, primaryKey);
}
private Object findEntity(Class< ? > entityClass, Object primaryKey) {
EntityManager em = Context
.getCommandContext()
.getSession(EntityManagerSession.class)
.getEntityManager();
Object entity = em.find(entityClass, primaryKey);
if(entity == null) {
throw new ActivitiException("Entity does not exist: " + entityClass.getName() + " - " + primaryKey);
}
return entity;
}
public Object createId(EntityMetaData metaData, String string) {
Class<?> type = metaData.getIdType();
// According to JPA-spec all primitive types (and wrappers) are supported, String, util.Date, sql.Date,
// BigDecimal and BigInteger
if(type == Long.class || type == long.class) {
return Long.parseLong(string);
} else if(type == String.class) {
return string;
} else if(type == Byte.class || type == byte.class) {
return Byte.parseByte(string);
} else if(type == Short.class || type == short.class) {
return Short.parseShort(string);
} else if(type == Integer.class || type == int.class) {
return Integer.parseInt(string);
} else if(type == Float.class || type == float.class) {
return Float.parseFloat(string);
} else if(type == Double.class || type == double.class) {
return Double.parseDouble(string);
} else if(type == Character.class || type == char.class) {
return new Character(string.charAt(0));
} else if(type == java.util.Date.class){
return new java.util.Date(Long.parseLong(string));
} else if(type == java.sql.Date.class) {
return new java.sql.Date(Long.parseLong(string));
} else if(type == BigDecimal.class) {
return new BigDecimal(string);
} else if(type == BigInteger.class) {
return new BigInteger(string);
} else {
throw new ActivitiException("Unsupported Primary key type for JPA-Entity: " + type.getName());
}
}
public String getIdString(Object value) {
if(value == null) {
throw new ActivitiException("Value of primary key for JPA-Entity cannot be null");
}
// Only java.sql.date and java.util.date require custom handling, the other types
// can just use toString()
if(value instanceof java.util.Date){
return "" + ((java.util.Date) value).getTime();
} else if(value instanceof java.sql.Date) {
return "" + ((java.sql.Date) value).getTime();
} else if(value instanceof Long || value instanceof String || value instanceof Byte
|| value instanceof Short || value instanceof Integer || value instanceof Float
|| value instanceof Double || value instanceof Character || value instanceof BigDecimal
|| value instanceof BigInteger) {
return value.toString();
} else {
throw new ActivitiException("Unsupported Primary key type for JPA-Entity: " + value.getClass().getName());
}
}
}