com.centit.support.database.orm.OrmUtils 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.algorithm.UuidOpt;
import com.centit.support.common.KeyValuePair;
import com.centit.support.compiler.VariableFormula;
import com.centit.support.database.jsonmaptable.GeneralJsonObjectDao;
import com.centit.support.database.jsonmaptable.JsonObjectDao;
import com.centit.support.database.metadata.SimpleTableField;
import com.centit.support.database.utils.DatabaseAccess;
import java.io.IOException;
import java.lang.reflect.Field;
import java.sql.*;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
* Created by codefan on 17-8-27.
* @author [email protected]
*/
@SuppressWarnings("unused")
public abstract class OrmUtils {
private OrmUtils() {
throw new IllegalAccessError("Utility class");
}
public static void setObjectFieldValue(Object object, SimpleTableField field,
Object newValue)
throws NoSuchFieldException, IOException {
if (newValue instanceof Clob) {
if(field.getJavaType() == "Clob"){
field.setObjectFieldValue(object,
/*(Clob)*/ newValue );
}else {
field.setObjectFieldValue(object,
DatabaseAccess.fetchClobString((Clob) newValue));
}
}else if (newValue instanceof Blob) {
if(field.getJavaType() == "Blob"){
field.setObjectFieldValue(object,
/*(Blob)*/ newValue );
}else {
field.setObjectFieldValue(object,
DatabaseAccess.fetchBlobBytes((Blob) newValue));
}
} else {
field.setObjectFieldValue(object, newValue);
}
}
private static T makeObjectValueByGenerator(T object, TableMapInfo mapInfo,
JsonObjectDao sqlDialect, GeneratorTime generatorTime)
throws SQLException, NoSuchFieldException, IOException {
List> valueGenerators = mapInfo.getValueGenerators();
if(valueGenerators == null || valueGenerators.size()<1 )
return object;
for(KeyValuePair ent : valueGenerators) {
ValueGenerator valueGenerator = ent.getValue();
if ( valueGenerator.occasion().matchTime(generatorTime)){
SimpleTableField filed = mapInfo.findFieldByName(ent.getKey());
Object fieldValue = ReflectionOpt.forceGetProperty(object, filed.getPropertyName());
if( fieldValue == null || valueGenerator.condition() == GeneratorCondition.ALWAYS ){
switch (valueGenerator.strategy()){
case UUID:
filed.setObjectFieldValue(object, UuidOpt.getUuidAsString32());
break;
case SEQUENCE:
//GeneratorTime.READ 读取数据时不能用 SEQUENCE 生成值
if(sqlDialect!=null) {
setObjectFieldValue(object, filed,
sqlDialect.getSequenceNextValue(valueGenerator.value()));
}
break;
case CONSTANT:
setObjectFieldValue(object, filed, valueGenerator.value());
break;
case FUNCTION:
setObjectFieldValue(object, filed,
VariableFormula.calculate(valueGenerator.value(),object));
break;
}
}
}
}
return object;
}
public static T prepareObjectForInsert(T object, TableMapInfo mapInfo,JsonObjectDao sqlDialect)
throws SQLException, NoSuchFieldException, IOException {
return makeObjectValueByGenerator(object, mapInfo, sqlDialect,GeneratorTime.NEW);
}
public static T prepareObjectForUpdate(T object, TableMapInfo mapInfo,JsonObjectDao sqlDialect)
throws SQLException, NoSuchFieldException, IOException {
return makeObjectValueByGenerator(object, mapInfo, sqlDialect, GeneratorTime.UPDATE);
}
public static T prepareObjectForMerge(T object, TableMapInfo mapInfo,JsonObjectDao sqlDialect)
throws SQLException, NoSuchFieldException, IOException {
Map objectMap = OrmUtils.fetchObjectDatabaseField(object,mapInfo);
if(! GeneralJsonObjectDao.checkHasAllPkColumns(mapInfo,objectMap)){
return makeObjectValueByGenerator(object, mapInfo, sqlDialect, GeneratorTime.NEW);
}else {
return makeObjectValueByGenerator(object, mapInfo, sqlDialect, GeneratorTime.UPDATE);
}
}
public static Map fetchObjectField(Object object)
{
if(object instanceof Map) {
return (Map) object;
}
Field[] objFields = object.getClass().getDeclaredFields();
Map fields = new HashMap<>(objFields.length*2);
for(Field field :objFields){
Object value = ReflectionOpt.forceGetFieldValue(object,field);
fields.put(field.getName() ,value);
}
return fields;
}
public static Map fetchObjectDatabaseField(Object object, TableMapInfo tableInfo) {
List tableFields = tableInfo.getColumns();
if(tableFields == null)
return null;
Map fields = new HashMap<>(tableFields.size()*2+6);
for(SimpleTableField column : tableFields){
Object value = column.getObjectFieldValue(object);
//ReflectionOpt.getFieldValue(object, column.getPropertyName());
if(value!=null){
fields.put(column.getPropertyName(),value);
}
}
tableFields = tableInfo.getLazyColumns();
if(tableFields != null) {
for (SimpleTableField column : tableFields) {
Object value = column.getObjectFieldValue(object);
//ReflectionOpt.getFieldValue(object, column.getPropertyName());
if (value != null) {
fields.put(column.getPropertyName(), value);
}
}
}
return fields;
}
private static T insideFetchFieldsFormResultSet(ResultSet rs, T object, TableMapInfo mapInfo )
throws SQLException, NoSuchFieldException, IOException {
ResultSetMetaData resMeta = rs.getMetaData();
int fieldCount = resMeta.getColumnCount();
for (int i = 1; i <= fieldCount; i++) {
String columnName = resMeta.getColumnName(i);
SimpleTableField filed = mapInfo.findFieldByColumn(columnName);
if (filed != null) {
setObjectFieldValue(object, filed, rs.getObject(i));
}
}
return makeObjectValueByGenerator(object, mapInfo, null, GeneratorTime.READ);
//return object;
}
public static T fetchObjectFormResultSet(ResultSet rs, Class clazz)
throws SQLException, IllegalAccessException, InstantiationException, NoSuchFieldException, IOException {
TableMapInfo mapInfo = JpaMetadata.fetchTableMapInfo(clazz);
if(mapInfo == null)
return null;
if(rs.next()) {
return insideFetchFieldsFormResultSet(rs, clazz.newInstance(), mapInfo);
}else {
return null;
}
}
public static T fetchFieldsFormResultSet(ResultSet rs, T object, TableMapInfo mapInfo )
throws SQLException, NoSuchFieldException, IOException {
if(rs.next()) {
object = insideFetchFieldsFormResultSet(rs, object, mapInfo);
}
return object;
}
public static List fetchObjectListFormResultSet(ResultSet rs, Class clazz)
throws SQLException, IllegalAccessException, InstantiationException, NoSuchFieldException, IOException {
TableMapInfo mapInfo = JpaMetadata.fetchTableMapInfo(clazz);
if(mapInfo == null)
return null;
ResultSetMetaData resMeta = rs.getMetaData();
int fieldCount = resMeta.getColumnCount();
SimpleTableField[] fields = new SimpleTableField[fieldCount+1];
for(int i=1;i<=fieldCount;i++) {
String columnName = resMeta.getColumnName(i);
fields[i] = mapInfo.findFieldByColumn(columnName);
}
List listObj = new ArrayList<>();
while(rs.next()){
T object = clazz.newInstance();
for(int i=1;i<=fieldCount;i++){
if(fields[i] != null){
setObjectFieldValue(object, fields[i], rs.getObject(i));
}
}
listObj.add(makeObjectValueByGenerator(object, mapInfo, null, GeneratorTime.READ));
}
return listObj;
}
}