com.starmcc.qmdata.base.QmDataDto Maven / Gradle / Ivy
package com.starmcc.qmdata.base;
import com.starmcc.qmdata.exception.QmDataDtoException;
import com.starmcc.qmdata.note.Id;
import com.starmcc.qmdata.note.Param;
import com.starmcc.qmdata.note.Style;
import com.starmcc.qmdata.note.Table;
import com.starmcc.qmdata.util.QmDataStyleTools;
import java.lang.reflect.Field;
import java.util.*;
/**
* @author qm
* @date 2018年11月24日 上午2:31:35
* @Description 数据持久层封装DTO
*/
public final class QmDataDto {
private final Object bean;
private String tableName;
private Table table;
private List> params = null;
private Map primaryKey = null;
/**
* 构造一个Dto
*
* @param bean
*/
protected QmDataDto(T bean, boolean isPrimaryKey) {
this.bean = bean;
final Class clamm = bean.getClass();
// 获取该实体的表名
table = clamm.getAnnotation(Table.class);
if (table == null || table.name() == null || "".equals(table.name())) {
tableName = clamm.getSimpleName();
if (table != null && table.style() == Style.UNDERLINE) {
tableName = QmDataStyleTools.transformNameByUnderline(tableName);
}
} else {
tableName = table.name();
}
// 获取该实体的字段进行操作
final Field[] fields = clamm.getDeclaredFields();
// 如果该字段为空则返回
if (fields == null) {
throw new QmDataDtoException("检测不到该实体的字段集!");
}
// 遍历字段进行参数封装
for (Field field : fields) {
// 开放字段权限public
if (!field.isAccessible()) {
field.setAccessible(true);
}
// 判断是否需要主键策略
if (isPrimaryKey && primaryKey == null) {
// 序列化该主键
if (setPrimaryKey(field)) {
continue;
}
}
// 序列化该字段
setFiledToList(field);
}
return;
}
/**
* 设置主键
*
* @param id
* @return
*/
private boolean setPrimaryKey(Field filed) {
Id idKey = filed.getAnnotation(Id.class);
if (idKey == null) {
return false;
}
Object obj = null;
try {
obj = filed.get(bean);
} catch (IllegalAccessException e) {
throw new QmDataDtoException("读取实体类主键字段发生了异常！");
}
if (obj != null) {
// 不等于null
// 判断是否设置别名
primaryKey = new HashMap<>(16);
if (idKey.name() == null || "".equals(idKey.name())) {
String key = filed.getName();
if (table != null && table.style() == Style.UNDERLINE) {
primaryKey.put("key", QmDataStyleTools.transformNameByUnderline(key));
} else {
primaryKey.put("key", key);
}
} else {
primaryKey.put("key", idKey.name());
}
primaryKey.put("value", obj);
}
return true;
}
/**
* 设置字段
*
* @param field
* @return
*/
private void setFiledToList(Field field) {
Param param = field.getAnnotation(Param.class);
// 判断是否有该注解，如果存在并且except等于true则不加入该字段。
if (param != null && param.except()) {
return;
}
Object obj = null;
try {
obj = field.get(bean);
// 如果值是null则不需要这个值了。
if (obj == null) {
return;
}
} catch (IllegalAccessException e) {
throw new QmDataDtoException("获取字段失败！");
}
// 开始获取字段并加入字段列表
Map fieldMap = new HashMap(16);
if (param == null || param.name() == null || "".equals(param.name())) {
if (table != null && table.style() == Style.UNDERLINE) {
fieldMap.put("key", QmDataStyleTools.transformNameByUnderline(field.getName()));
} else {
fieldMap.put("key", field.getName());
}
} else {
fieldMap.put("key", param.name());
}
fieldMap.put("value", obj);
if (params == null) {
params = new ArrayList<>();
}
params.add(fieldMap);
return;
}
/**
* 获取实体类参数封装Map
*
* @return
*/
protected LinkedHashMap getParamsMap() {
LinkedHashMap resMap = new LinkedHashMap<>();
resMap.put("primaryKey", primaryKey);
resMap.put("params", params);
resMap.put("tableName", tableName);
System.out.println("注入MyBatis数据：" + resMap.toString());
return resMap;
}
}