All Downloads are FREE. Search and download functionalities are using the official Maven repository.
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.starmcc.qmdata.model.QmDataModel Maven / Gradle / Ivy
package com.starmcc.qmdata.model;
import com.starmcc.qmdata.exception.QmDataModelException;
import com.starmcc.qmdata.note.*;
import com.starmcc.qmdata.util.QmDataStyleTools;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.util.StringUtils;
import java.io.Serializable;
import java.lang.reflect.Field;
import java.util.*;
/**
* @author qm
* @date 2018年11月24日 上午2:31:35
* @Description 数据持久层封装Model
*/
public final class QmDataModel implements Serializable {
private static final long serialVersionUID = 6857822692226391769L;
private static final Logger LOG = LoggerFactory.getLogger(QmDataModel.class);
public static final int PRIVATE = 0x00000002;
private T bean;
private boolean isPrimaryKey;
private Style style;
private String tableName;
private String whereSql;
private String orderByValue;
private Map primaryKey = null;
private List> params = null;
private Map paramsMap = new HashMap<>();
public static QmDataModel getInstance(T bean, boolean isPrimaryKey) {
return QmDataModel.getInstance(bean, null, null, isPrimaryKey);
}
public static QmDataModel getInstance(T bean, String whereSql, boolean isPrimaryKey) {
return QmDataModel.getInstance(bean, whereSql, null, isPrimaryKey);
}
public static QmDataModel getInstance(T bean, String whereSql, String orderBy, boolean isPrimaryKey) {
return new QmDataModel(bean, whereSql, orderBy, isPrimaryKey);
}
private QmDataModel() {
}
/**
* 构造一个Model
*
* @param bean
*/
private QmDataModel(T bean, String whereSql, String orderBy, boolean isPrimaryKey) {
// 构建模型
this.buildModel(bean, whereSql, orderBy, isPrimaryKey);
// 构建 paramsMap
this.buildParamsMap();
}
/**
* 获取实体类参数封装Map
*
* @return
*/
public Map getParamsMap() {
return paramsMap;
}
/**
* 构建模型
*
* @param bean
* @param whereSql
* @param orderBy
* @param isPrimaryKey
*/
private void buildModel(T bean, String whereSql, String orderByValue, boolean isPrimaryKey) {
if (Objects.isNull(bean)) {
return;
}
this.bean = bean;
this.isPrimaryKey = isPrimaryKey;
Table table = this.bean.getClass().getAnnotation(Table.class);
OrderBy orderBy = this.bean.getClass().getAnnotation(OrderBy.class);
// 获取实体类风格
if (Objects.isNull(table.style())) {
this.style = Style.UNDERLINE;
} else {
this.style = table.style();
}
// 获取该实体的表名
if (Objects.nonNull(table) && !StringUtils.isEmpty(table.name())) {
this.tableName = table.name();
} else {
this.tableName = this.bean.getClass().getSimpleName();
if (Objects.nonNull(table.style()) && style == Style.UNDERLINE) {
this.tableName = QmDataStyleTools.transformNameByUnderline(this.tableName);
}
}
this.whereSql = whereSql;
// 获取orderby SQL语句
if (!StringUtils.isEmpty(orderByValue)) {
// 如果不等于空，则直接使用参数列表提供的orderBy语句
this.orderByValue = orderByValue;
} else if (Objects.nonNull(orderBy) && !StringUtils.isEmpty(orderBy.value())) {
// 如果参数列表的orderBy 等于空，且实体类中的OrderBy注解不为空，则直接使用orderBy注解提供的语句。
this.orderByValue = orderBy.value();
}
// 获取该实体的字段进行操作
final Field[] fields = this.bean.getClass().getDeclaredFields();
// 如果该字段为空则返回
if (Objects.isNull(fields)) {
throw new QmDataModelException("检测不到该实体的字段集!");
}
// 遍历字段进行参数封装
for (Field field : fields) {
// 必须是私有的属性
if (PRIVATE != field.getModifiers()) {
continue;
}
// 开放字段权限public
field.setAccessible(true);
// 判断是否需要主键策略
if (isPrimaryKey && Objects.isNull(this.primaryKey)) {
// 序列化该主键
if (this.setPrimaryKey(field)) {
continue;
}
}
// 序列化该字段
this.setFiledToList(field);
}
}
/**
* 构建 params Map
*/
private void buildParamsMap() {
paramsMap.put("primaryKey", this.primaryKey);
paramsMap.put("params", this.params);
paramsMap.put("tableName", this.tableName);
paramsMap.put("whereSql", this.whereSql);
paramsMap.put("orderBy", this.orderByValue);
LOG.debug("注入MyBatis数据：" + paramsMap.toString());
}
/**
* 设置主键
*
* @param id
* @return
*/
private boolean setPrimaryKey(Field field) {
Id idKey = field.getAnnotation(Id.class);
if (Objects.isNull(idKey)) {
return false;
}
Object obj = null;
try {
obj = field.get(bean);
} catch (IllegalAccessException e) {
throw new QmDataModelException("读取实体类主键字段发生了异常！", e);
}
if (Objects.nonNull(obj)) {
// 不等于null
// 判断是否设置别名
this.primaryKey = new HashMap<>(16);
if (StringUtils.isEmpty(idKey.name())) {
String key = field.getName();
if (this.style == Style.UNDERLINE) {
this.primaryKey.put("key", QmDataStyleTools.transformNameByUnderline(key));
} else {
this.primaryKey.put("key", key);
}
} else {
this.primaryKey.put("key", idKey.name());
}
this.primaryKey.put("value", obj);
}
return true;
}
/**
* 设置字段
*
* @param field
* @return
*/
private void setFiledToList(Field field) {
Param param = field.getAnnotation(Param.class);
// 判断是否有该注解，如果存在并且except等于true则不加入该字段。
if (Objects.nonNull(param) && param.except()) {
return;
}
Object obj = null;
try {
obj = field.get(bean);
// 如果值是null则不需要这个值了。
if (Objects.isNull(obj)) {
return;
}
} catch (IllegalAccessException e) {
throw new QmDataModelException("获取字段失败！");
}
// 开始获取字段并加入字段列表
Map fieldMap = new HashMap(16);
if (Objects.isNull(param) || StringUtils.isEmpty(param.name())) {
if (this.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 (Objects.isNull(this.params)) {
this.params = new ArrayList<>();
}
this.params.add(fieldMap);
return;
}
}
