com.centit.support.database.orm.JpaMetadata Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of centit-database Show documentation
Show all versions of centit-database Show documentation
数据库操作通用方法和函数,从以前的util包中分离出来,并且整合了部分sys-module中的函数
package com.centit.support.database.orm;
import com.centit.support.algorithm.ReflectionOpt;
import com.centit.support.database.metadata.SimpleTableField;
import com.centit.support.database.metadata.SimpleTableReference;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import javax.persistence.*;
import java.lang.reflect.Field;
import java.util.HashSet;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
/**
* Created by codefan on 17-8-27.
*/
@SuppressWarnings("unused")
public abstract class JpaMetadata {
private JpaMetadata() {
throw new IllegalAccessError("Utility class");
}
private static final Logger logger = LoggerFactory.getLogger(JpaMetadata.class);
private static final ConcurrentHashMap ORM_JPA_METADATA_CLASSPATH =
new ConcurrentHashMap<>(100);
private static final ConcurrentHashMap ORM_JPA_METADATA_CLASSNAME =
new ConcurrentHashMap<>(100);
private static final ConcurrentHashMap ORM_JPA_METADATA_TABLENAME =
new ConcurrentHashMap<>(100);
public static TableMapInfo fetchTableMapInfo(Class> type){
String className = type.getName();
TableMapInfo mapInfo = ORM_JPA_METADATA_CLASSPATH.get(className);
if(mapInfo == null){
mapInfo = obtainMapInfoFromClass(type);
if(mapInfo!=null){
ORM_JPA_METADATA_CLASSPATH.put(className,mapInfo);
ORM_JPA_METADATA_TABLENAME.put(mapInfo.getTableName(),mapInfo);
ORM_JPA_METADATA_CLASSNAME.put(/*type.getSimpleName()*/
className.substring(className.lastIndexOf(".")+1),mapInfo);
}
}
return mapInfo;
}
/**
* 将属性名称转换为字段名称
* @param type 类型
* @param propertyName 属性名称
* @return 字段名称
*/
public static String translatePropertyNameToColumnName(Class> type, String propertyName ){
TableMapInfo mapInfo = JpaMetadata.fetchTableMapInfo(type);
SimpleTableField field = mapInfo.findFieldByName(propertyName);
return field==null?propertyName:field.getColumnName();
}
/**
* 将属性名称转换为字段名称;
* 这里有一个问题就是这个类一定是已经扫描过得,不然会找不到这个对象
* @param propertyName 表名或者类名.属性名称
* @param tableAlias 表的笔名
* @return 字段名称
*/
public static String translatePropertyNameToColumnName(String propertyName , String tableAlias){
int n = propertyName.indexOf('.');
if(n<0) {
return propertyName;
}
String tableName = propertyName.substring(0,n);
String fieldName = propertyName.substring(n+1);
TableMapInfo mapInfo = ORM_JPA_METADATA_CLASSNAME.get(tableName);
if(mapInfo ==null){
mapInfo = ORM_JPA_METADATA_TABLENAME.get(tableName);
}
if(mapInfo ==null){
return propertyName;
}
tableName = mapInfo.getTableName();
SimpleTableField field = mapInfo.findFieldByName(propertyName);
if(field!=null){
fieldName = field.getColumnName();
}
if(StringUtils.isNotBlank(tableAlias)){
return tableAlias +"."+fieldName;
}
return tableName +"."+fieldName;
}
/**
* 将属性名称转换为字段名称;
* 这里有一个问题就是这个类一定是已经扫描过得,不然会找不到这个对象
* @param propertyName 表名或者类名.属性名称
* @return 字段名称
*/
public static String translatePropertyNameToColumnName(String propertyName ){
return translatePropertyNameToColumnName(propertyName, null);
}
private static SimpleTableField obtainColumnFromField(Class> objType, Field field) {
SimpleTableField column = new SimpleTableField();
Column colInfo = field.getAnnotation(Column.class);
column.setColumnName(colInfo.name());
//column.setColumnType( colInfo.);
column.setJavaType(field.getType());
column.setPropertyName(field.getName());
column.setMaxLength(colInfo.length());
column.setScale(colInfo.scale());
column.setPrecision(colInfo.precision());
column.setObjectField(field);
column.setObjectGetFieldValueFunc(ReflectionOpt.getGetterMethod(objType, field.getType(), field.getName()));
column.setObjectSetFieldValueFunc(ReflectionOpt.getSetterMethod(objType, field.getType(), field.getName()));
return column;
}
public static TableMapInfo obtainMapInfoFromClass(Class> objType){
if(!objType.isAnnotationPresent( Table.class ))
return null;
Table tableInfo = objType.getAnnotation(Table.class);
TableMapInfo mapInfo = new TableMapInfo();
mapInfo.setTableName( tableInfo.name() );
mapInfo.setSchema( tableInfo.schema());
Field[] objFields = objType.getDeclaredFields();
for(Field field :objFields){
if(field.isAnnotationPresent(Column.class)){
SimpleTableField column = obtainColumnFromField(objType, field);
if(field.isAnnotationPresent(Id.class) ){
mapInfo.addColumn(column);
mapInfo.addPkColumns(column.getPropertyName());
}else if( field.isAnnotationPresent(Lazy.class) ){
mapInfo.addLazyColumn(column);
} else {
mapInfo.addColumn(column);
}
if(field.isAnnotationPresent(ValueGenerator.class) ){
ValueGenerator valueGenerator = field.getAnnotation(ValueGenerator.class);
mapInfo.addValueGenerator(
column.getPropertyName(),
valueGenerator);
}
if(field.isAnnotationPresent(OrderBy.class) ){
OrderBy orderBy = field.getAnnotation(OrderBy.class);
mapInfo.appendOrderBy(column, orderBy.value());
}
}else if(field.isAnnotationPresent(EmbeddedId.class)){
EmbeddedId embeddedId = field.getAnnotation(EmbeddedId.class);
mapInfo.setPkName(field.getName());
for(Field idField : field.getType().getDeclaredFields()){
if(idField.isAnnotationPresent(Column.class)) {
SimpleTableField column = obtainColumnFromField(objType, idField);
mapInfo.addColumn(column);
mapInfo.addPkColumns(column.getPropertyName());
if(idField.isAnnotationPresent(ValueGenerator.class) ){
ValueGenerator valueGenerator = idField.getAnnotation(ValueGenerator.class);
mapInfo.addValueGenerator(
column.getPropertyName(),
valueGenerator);
}
if(idField.isAnnotationPresent(OrderBy.class) ){
OrderBy orderBy = idField.getAnnotation(OrderBy.class);
mapInfo.appendOrderBy(column, orderBy.value());
}
}
}
}else if ( field.isAnnotationPresent(OneToOne.class)
|| field.isAnnotationPresent(OneToMany.class) ) {
SimpleTableReference reference = new SimpleTableReference();
if (field.isAnnotationPresent(OneToOne.class)) {
OneToOne oneToOne = field.getAnnotation(OneToOne.class);
Class targetClass = oneToOne.targetEntity();
if(targetClass ==null || targetClass.equals(void.class) )
targetClass = field.getType();
reference.setTargetEntityType(targetClass);
}else if (field.isAnnotationPresent(ManyToOne.class)) {
ManyToOne manyToOne = field.getAnnotation(ManyToOne.class);
Class targetClass = manyToOne.targetEntity();
if(targetClass ==null || targetClass.equals(void.class) )
targetClass = field.getType();
reference.setTargetEntityType(targetClass);
}else if (field.isAnnotationPresent(OneToMany.class)) {
OneToMany oneToMany = field.getAnnotation(OneToMany.class);
reference.setTargetEntityType(oneToMany.targetEntity());
}else if (field.isAnnotationPresent(ManyToMany.class)) {
ManyToMany manyToMany = field.getAnnotation(ManyToMany.class);
reference.setTargetEntityType(manyToMany.targetEntity());
}
if(reference.getTargetEntityType() !=null
&& ! reference.getTargetEntityType().equals(void.class) ) {
boolean haveJoinColumns = false;
if (field.isAnnotationPresent(JoinColumn.class)) {
JoinColumn joinColumn = field.getAnnotation(JoinColumn.class);
reference.setReferenceName(field.getName());
reference.setReferenceType(field.getType());
reference.addReferenceColumn(joinColumn.name(), joinColumn.referencedColumnName());
haveJoinColumns = true;
} else if (field.isAnnotationPresent(JoinColumns.class)) {
JoinColumns joinColumns = field.getAnnotation(JoinColumns.class);
reference.setReferenceName(field.getName());
reference.setReferenceType(field.getType());
for (JoinColumn joinColumn : joinColumns.value()) {
reference.addReferenceColumn(joinColumn.name(), joinColumn.referencedColumnName());
}
haveJoinColumns = true;
}
if(haveJoinColumns) {
reference.setObjectField(field);
reference.setObjectGetFieldValueFunc(ReflectionOpt.getGetterMethod(objType, field.getType(), field.getName()));
reference.setObjectSetFieldValueFunc(ReflectionOpt.getSetterMethod(objType, field.getType(), field.getName()));
mapInfo.addReference(reference);
}
}
}
}
// 整理 引用字段
if( mapInfo.getReferences() != null && mapInfo.getReferences().size()>0) {
Set errorRef = new HashSet<>(mapInfo.getReferences().size());
for (SimpleTableReference reference : mapInfo.getReferences()) {
TableMapInfo refTalbeInfo = fetchTableMapInfo(reference.getTargetEntityType());
if (refTalbeInfo == null) {
errorRef.add(reference);
continue;
}
Set keySet = new HashSet<>(reference.getReferenceColumns().keySet());
for (String keyName : keySet) {
SimpleTableField field = mapInfo.findFieldByName(keyName);
SimpleTableField refField = refTalbeInfo.findFieldByName(reference.getReferenceColumns().get(keyName));
if (field == null || refField == null) {
errorRef.add(reference);
break;
}
if (!keyName.equals(field.getPropertyName())) {
reference.getReferenceColumns().remove(keyName);
}
reference.getReferenceColumns().put(field.getPropertyName(), refField.getPropertyName());
}
}
if(errorRef.size()>0) {
mapInfo.getReferences().removeAll(errorRef);
}
}
return mapInfo;
}
}