();
/**
* 默认表主键
*/
private static final String DEFAULT_ID_NAME = "id";
/**
*
* 获取实体映射表信息
*
*
* @param clazz
* 反射实体类
* @return
*/
public static TableInfo getTableInfo(Class clazz) {
return tableInfoCache.get(clazz.getName());
}
/**
*
* 随机获取一个TableInfo
*
*
* @return
*/
public static TableInfo getRandomTableInfo() {
Collection tableInfos = tableInfoCache.values();
if (CollectionUtils.isNotEmpty(tableInfos)) {
Iterator iterator = tableInfos.iterator();
TableInfo tableInfo = iterator.next();
return tableInfo;
}
return null;
}
/**
*
* 实体类反射获取表信息【初始化】
*
*
* @param clazz
* 反射实体类
* @return
*/
public synchronized static TableInfo initTableInfo(MapperBuilderAssistant builderAssistant, Class clazz) {
TableInfo ti = tableInfoCache.get(clazz.getName());
if (ti != null) {
return ti;
}
TableInfo tableInfo = new TableInfo();
GlobalConfiguration globalCache = null;
if (null != builderAssistant) {
tableInfo.setCurrentNamespace(builderAssistant.getCurrentNamespace());
tableInfo.setConfigMark(builderAssistant.getConfiguration());
globalCache = GlobalConfiguration.GlobalConfig(builderAssistant.getConfiguration());
} else {
// 兼容测试场景
globalCache = GlobalConfiguration.DEFAULT;
}
/* 表名 */
TableName table = clazz.getAnnotation(TableName.class);
String tableName = clazz.getSimpleName();
if (table != null && StringUtils.isNotEmpty(table.value())) {
tableName = table.value();
} else {
// 开启字段下划线申明
if (globalCache.isDbColumnUnderline()) {
tableName = StringUtils.camelToUnderline(tableName);
}
// 大写命名判断
if (globalCache.isCapitalMode()) {
tableName = tableName.toUpperCase();
}
}
tableInfo.setTableName(tableName);
/* 表结果集映射 */
if (table != null && StringUtils.isNotEmpty(table.resultMap())) {
tableInfo.setResultMap(table.resultMap());
}
List fieldList = new ArrayList();
List list = getAllFields(clazz);
boolean existTableId = existTableId(list);
for (Field field : list) {
/**
* 主键ID 初始化
*/
if (existTableId) {
if (initTableId(globalCache, tableInfo, field, clazz)) {
continue;
}
} else if (initFieldId(globalCache, tableInfo, field, clazz)) {
continue;
}
/**
* 字段初始化
*/
if (initTableField(globalCache, fieldList, field, clazz)) {
continue;
}
/**
* 字段, 使用 camelToUnderline 转换驼峰写法为下划线分割法, 如果已指定 TableField , 便不会执行这里
*/
fieldList.add(new TableFieldInfo(globalCache, field.getName()));
}
/* 字段列表 */
tableInfo.setFieldList(fieldList);
/*
* 未发现主键注解，跳过注入
*/
if (null == tableInfo.getKeyColumn()) {
logger.warn(String.format("Warn: Could not find @TableId in Class: %s, initTableInfo Method Fail.",
clazz.getName()));
return null;
}
/*
* 注入
*/
tableInfoCache.put(clazz.getName(), tableInfo);
return tableInfo;
}
/**
*
* 判断主键注解是否存在
*
*
* @param list
* 字段列表
* @return
*/
public static boolean existTableId(List list) {
boolean exist = false;
for (Field field : list) {
TableId tableId = field.getAnnotation(TableId.class);
if (tableId != null) {
exist = true;
break;
}
}
return exist;
}
/**
*
* 主键属性初始化
*
*
* @param tableInfo
* @param field
* @param clazz
* @return true 继续下一个属性判断，返回 continue;
*/
private static boolean initTableId(GlobalConfiguration globalConfig, TableInfo tableInfo, Field field,
Class clazz) {
TableId tableId = field.getAnnotation(TableId.class);
if (tableId != null) {
if (tableInfo.getKeyColumn() == null) {
/*
* 主键策略（ 注解 > 全局 > 默认 ）
*/
if (IdType.INPUT != tableId.type()) {
tableInfo.setIdType(tableId.type());
} else {
tableInfo.setIdType(globalConfig.getIdType());
}
/* 字段 */
String column = field.getName();
if (StringUtils.isNotEmpty(tableId.value())) {
column = tableId.value();
tableInfo.setKeyRelated(true);
} else {
// 开启字段下划线申明
if (globalConfig.isDbColumnUnderline()) {
column = StringUtils.camelToUnderline(column);
}
// 全局大写命名
if (globalConfig.isCapitalMode()) {
column = column.toUpperCase();
}
}
tableInfo.setKeyColumn(column);
tableInfo.setKeyProperty(field.getName());
return true;
} else {
throwExceptionId(clazz);
}
}
return false;
}
/**
*
* 主键属性初始化
*
*
* @param tableInfo
* @param field
* @param clazz
* @return true 继续下一个属性判断，返回 continue;
*/
private static boolean initFieldId(GlobalConfiguration globalConfig, TableInfo tableInfo, Field field,
Class clazz) {
String column = field.getName();
if (globalConfig.isCapitalMode()) {
column = column.toUpperCase();
}
if (DEFAULT_ID_NAME.equalsIgnoreCase(column)) {
if (tableInfo.getKeyColumn() == null) {
tableInfo.setIdType(globalConfig.getIdType());
tableInfo.setKeyColumn(column);
tableInfo.setKeyProperty(field.getName());
return true;
} else {
throwExceptionId(clazz);
}
}
return false;
}
/**
*
* 发现设置多个主键注解抛出异常
*
*/
private static void throwExceptionId(Class clazz) {
StringBuffer errorMsg = new StringBuffer();
errorMsg.append("There must be only one, Discover multiple @TableId annotation in ");
errorMsg.append(clazz.getName());
throw new MybatisPlusException(errorMsg.toString());
}
/**
*
* 字段属性初始化
*
*
* @param fieldList
* @param clazz
* @return true 继续下一个属性判断，返回 continue;
*/
private static boolean initTableField(GlobalConfiguration globalCache, List fieldList, Field field,
Class clazz) {
/* 获取注解属性，自定义字段 */
TableField tableField = field.getAnnotation(TableField.class);
if (tableField != null) {
String columnName = field.getName();
if (StringUtils.isNotEmpty(tableField.value())) {
columnName = tableField.value();
}
Class fieldType = field.getType();
FieldStrategy validate = tableField.validate();
/* 字符串类型默认 FieldStrategy.NOT_EMPTY */
if (String.class.isAssignableFrom(fieldType) && FieldStrategy.NOT_NULL.equals(validate)) {
validate = FieldStrategy.NOT_EMPTY;
}
/*
* el 语法支持，可以传入多个参数以逗号分开
*/
String el = field.getName();
if (StringUtils.isNotEmpty(tableField.el())) {
el = tableField.el();
}
String[] columns = columnName.split(";");
String[] els = el.split(";");
if (null != columns && null != els && columns.length == els.length) {
for (int i = 0; i < columns.length; i++) {
fieldList.add(new TableFieldInfo(globalCache, columns[i], field.getName(), els[i], validate));
}
} else {
String errorMsg = "Class: %s, Field: %s, 'value' 'el' Length must be consistent.";
throw new MybatisPlusException(String.format(errorMsg, clazz.getName(), field.getName()));
}
return true;
}
return false;
}
/**
* 获取该类的所有属性列表
*
* @param clazz
* 反射类
* @return
*/
private static List getAllFields(Class clazz) {
List result = new LinkedList();
Field[] fields = clazz.getDeclaredFields();
for (Field field : fields) {
/* 过滤静态属性 */
if (Modifier.isStatic(field.getModifiers())) {
continue;
}
/* 过滤 transient关键字修饰的属性 */
if (Modifier.isTransient(field.getModifiers())) {
continue;
}
/* 过滤注解非表字段属性 */
TableField tableField = field.getAnnotation(TableField.class);
if (tableField == null || tableField.exist()) {
result.add(field);
}
}
/* 处理父类字段 */
Class superClass = clazz.getSuperclass();
if (superClass.equals(Object.class)) {
return result;
}
result.addAll(getAllFields(superClass));
return result;
}
/**
* 初始化SqlSessionFactory (供Mybatis原生调用)
*
* @param sqlSessionFactory
* @return
*/
public static void initSqlSessionFactory(SqlSessionFactory sqlSessionFactory) {
Configuration configuration = sqlSessionFactory.getConfiguration();
GlobalConfiguration globalCache = GlobalConfiguration.GlobalConfig(configuration);
if (globalCache == null) {
GlobalConfiguration defaultCache = GlobalConfiguration.defaults();
defaultCache.setSqlSessionFactory(sqlSessionFactory);
GlobalConfiguration.setGlobalConfig(configuration, defaultCache);
} else {
globalCache.setSqlSessionFactory(sqlSessionFactory);
}
}
}