io.github.jinghui70.rainbow.dbaccess.object.PropInfoCache Maven / Gradle / Ivy
package io.github.jinghui70.rainbow.dbaccess.object;
import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.bean.PropDesc;
import cn.hutool.core.map.WeakConcurrentMap;
import cn.hutool.core.util.ReflectUtil;
import cn.hutool.core.util.StrUtil;
import io.github.jinghui70.rainbow.dbaccess.annotation.*;
import io.github.jinghui70.rainbow.dbaccess.enumSupport.EnumMapper;
import io.github.jinghui70.rainbow.dbaccess.fieldmapper.BlobByteArrayField;
import io.github.jinghui70.rainbow.dbaccess.fieldmapper.BlobObjectField;
import io.github.jinghui70.rainbow.dbaccess.fieldmapper.BlobStringField;
import io.github.jinghui70.rainbow.dbaccess.fieldmapper.FieldMapper;
import java.util.LinkedHashMap;
public class PropInfoCache {
private static final WeakConcurrentMap, LinkedHashMap> cache = new WeakConcurrentMap<>();
public static LinkedHashMap get(Class beanClass) {
return cache.computeIfAbsent(beanClass, (key) -> createPropInfo(beanClass));
}
public static LinkedHashMap createPropInfo(Class clazz) {
LinkedHashMap result = new LinkedHashMap<>();
BeanUtil.descForEach(clazz, propDesc -> {
if (propDesc.getField().getAnnotation(Transient.class) != null)
return;
Column column = propDesc.getField().getAnnotation(Column.class);
FieldMapper mapper = getMapper(column, propDesc);
String fieldName = column == null || StrUtil.isEmpty(column.name()) ?
StrUtil.toUnderlineCase(propDesc.getRawFieldName()) : column.name();
fieldName = fieldName.toLowerCase();
ArrayField arrayAnnotation = propDesc.getField().getAnnotation(ArrayField.class);
if (arrayAnnotation == null) {
Id id = propDesc.getField().getAnnotation(Id.class);
result.put(fieldName, new PropInfo(fieldName, propDesc, mapper, id));
} else {
if (mapper == null) mapper = checkEnumMapper(propDesc.getFieldClass().getComponentType());
String join = arrayAnnotation.underline() ? "_" : "";
for (int i = 0; i < arrayAnnotation.length(); i++) {
String field = String.format("%s%s%d", fieldName, join, i + arrayAnnotation.start());
result.put(field, new PropInfo(field, propDesc, mapper, i));
}
}
});
return result;
}
/**
* 根据字段配置，获取 FieldMapper 对象
*
* @param column
* @param propDesc
* @return
*/
@SuppressWarnings({"unchecked", "rawtypes"})
private static FieldMapper getMapper(Column column, PropDesc propDesc) {
Class fieldClass = propDesc.getFieldClass();
if (column == null) return checkEnumMapper(fieldClass);
Class mapperClass = column.mapper();
if (mapperClass != FieldMapper.class) {
return ReflectUtil.newInstance(mapperClass);
}
LobType lobType = column.lobType();
switch (lobType) {
case BLOB:
if (fieldClass == String.class)
return new BlobStringField();
if (fieldClass == byte[].class)
return new BlobByteArrayField();
return new BlobObjectField(fieldClass);
case CLOB: // 暂时没有必要做特殊处理，因为对象中的字符串要读到内存中，当做普通的字符串处理了
return null;
default:
return checkEnumMapper(fieldClass);
}
}
@SuppressWarnings({"rawtypes", "unchecked"})
private static FieldMapper checkEnumMapper(Class fieldClass) {
return fieldClass.isEnum() ? new EnumMapper(fieldClass) : null;
}
/**
* 清空全局的Bean属性缓存
*
* @since 5.7.21
*/
public static void clear() {
cache.clear();
}
}