xin.xihc.jba.scan.TableManager Maven / Gradle / Ivy
package xin.xihc.jba.scan;
import xin.xihc.jba.annotation.*;
import xin.xihc.jba.core.utils.SQLUtils;
import xin.xihc.jba.db.DB_MySql_Opera;
import xin.xihc.jba.scan.tables.properties.ColumnProperties;
import xin.xihc.jba.scan.tables.properties.IndexProperties;
import xin.xihc.jba.scan.tables.properties.TableProperties;
import xin.xihc.utils.common.CommonUtil;
import java.lang.reflect.Field;
import java.util.*;
import java.util.stream.Collectors;
/**
* 数据库表工具类
*
* @author Leo.Xi
* @version 1.2.0
* @date 2018年1月12日
* @since 1.0.0
*/
public class TableManager {
/** 有序的记录表对象类型 - 表属性 */
private final static LinkedHashMap, TableProperties> tbls = new LinkedHashMap<>(16);
/**
* 获取表的列表
*/
public static List getTables() {
return tbls.values().stream().sorted(Comparator.comparing(TableProperties::getOrder)).collect(Collectors.toList());
}
/**
* 添加表数据
*
* @param clazz 表对象的类型
* @param tblName 表名
* @return 表的属性
*/
private static TableProperties addTable(Class clazz, String tblName) {
TableProperties pp = new TableProperties();
pp.setTableName(tblName);
tbls.put(clazz, pp);
return pp;
}
/**
* 获取表的属性
*
* @param clazz 对象类Class
* @return 表属性信息
* @author Leo.Xi
* @date 2019/6/26
* @since 0.0.1
*/
public static TableProperties getTable(Class clazz) {
if (tbls.containsKey(clazz)) {
return tbls.get(clazz);
} else {
throw new RuntimeException(String.format("【%s】对应的表不存在", clazz.getName()));
}
}
/**
* 获取对应表的列名列表
*
* @param clazz 表对象的类型
* @return 表的列的列表
* @author Leo.Xi
* @date 2019/6/26
* @since 0.0.1
*/
public static List getColumnNames(Class clazz, final String prefix, final String suffix) {
if (tbls.containsKey(clazz)) {
return tbls.get(clazz).getColumns().values().stream()
.sorted(Comparator.comparing(ColumnProperties::order))
.map(x -> CommonUtil.null2Str(prefix, "") + x.colName() +
CommonUtil.null2Str(suffix, ""))
.collect(Collectors.toList());
}
return new LinkedList<>();
}
/**
* 将列名直接转为sql
* example：id,name,age,sex
*
* @param clazz 表对象的类型
* @return 表的列的sql
* @author Leo.Xi
* @date 2019/6/26
* @since 0.0.1
*/
public static String columns2Sql(Class clazz, final String prefix, final String suffix) {
if (tbls.containsKey(clazz)) {
List columns = tbls.get(clazz).getColumns().values().stream()
.sorted(Comparator.comparing(ColumnProperties::order))
.map(x -> CommonUtil.null2Str(prefix, "") + x.colName() +
CommonUtil.null2Str(suffix, ""))
.collect(Collectors.toList());
return " " + String.join(",", columns) + " ";
}
return " * ";
}
/**
* 获取索引前缀
*
* @param type 索引类型
* @return 获取索引前缀
* @author Leo.Xi
* @date 2019/6/26
* @since 0.0.1
*/
private static String getIndexPrefix(Index.IndexType type) {
String prefix = "idx_";
switch (type) {
case Unique:
prefix = "uk_";
break;
case FullText:
prefix = "full_";
break;
default:
break;
}
return prefix;
}
/**
* 转换成IndexProperties
*
* @param groupIndex 分组索引注解
* @return IndexProperties
* @author Leo.Xi
* @date 2019/6/26
* @since 0.0.1
*/
private static IndexProperties convert2IndexProp(GroupIndex groupIndex) {
IndexProperties idx = new IndexProperties();
idx.setColumnNames(groupIndex.value());
if (CommonUtil.isNullEmpty(groupIndex.name())) {
// 只取每个列名的首个_
String indexName = getIndexPrefix(groupIndex.type()) + Arrays.stream(groupIndex.value())
.map(x -> x.split("_")[0]).collect(Collectors.joining("_"));
idx.setIndexName(indexName);
} else {
idx.setIndexName(groupIndex.name());
}
idx.setType(groupIndex.type());
idx.setRemark(groupIndex.remark());
return idx;
}
/**
* 扫描表注解
*
* @param obj 表对象
* @author Leo.Xi
* @date 2019/6/26
* @since 0.0.1
*/
static TableProperties scanTableAnnotations(Object obj) {
Table table = obj.getClass().getAnnotation(Table.class);
TableProperties tblProp;
if (CommonUtil.isNotNullEmpty(table.value())) {
tblProp = addTable(obj.getClass(), table.value());
} else {
tblProp = addTable(obj.getClass(), obj.getClass().getSimpleName());
}
// 获取表注释
tblProp.setRemark(table.remark());
tblProp.setTableBean(obj);
tblProp.setIgnore(table.ignore());
tblProp.setOrder(table.order());
// 增加扫描@GroupIndex注解
GroupIndex groupIndex = obj.getClass().getAnnotation(GroupIndex.class);
if (null != groupIndex) {
tblProp.addIndex(convert2IndexProp(groupIndex));
}
GroupIndex.List groupIndexList = obj.getClass().getAnnotation(GroupIndex.List.class);
if (null != groupIndexList && groupIndexList.value().length > 0) {
for (GroupIndex index : groupIndexList.value()) {
if (null != index) {
tblProp.addIndex(convert2IndexProp(index));
}
}
}
return tblProp;
}
/**
* 扫描字段注解
*
* @param tblProp 表属性
* @param field 字段信息
* @author Leo.Xi
* @date 2019/6/26
* @since 0.0.1
*/
static void scanFieldAnnotations(TableProperties tblProp, Field field) {
Column column = field.getAnnotation(Column.class);
ColumnProperties colP = new ColumnProperties();
colP.type(field.getType());
if (null == column) {
colP.colName(SQLUtils.underscoreName(field.getName()));
// 没有的给默认值
colP.length(0)
.precision(0)
.policy(Column.Policy.NONE)
.primary(false)
.notNull(false)
.defaultValue("")
.remark("")
.order(0);
} else {
colP.colName(SQLUtils.underscoreName(field.getName()))
.defaultValue(column.defaultValue())
.notNull(column.notNull())
.remark(column.remark()).charset(column.charset()).order(column.order());
colP.policy(column.policy());
colP.length(0);
if (CommonUtil.isNotNullEmpty(column.value())) {
colP.colName(column.value());
}
/** 如果是guid为主键长度默认为32 */
if (colP.policy() == Column.Policy.GUID || colP.policy() == Column.Policy.GUID_UP) {
colP.length(32);
}
if (column.length() > 0) {
colP.length(column.length());
}
if (column.precision() > 0) {
colP.precision(column.precision());
}
if (column.primary()) {
colP.primary(true);
colP.notNull(true);
}
}
// 是否自动更新时间戳
OnUpdateCurrentTimestamp onUpdateCurrentTimestamp = field.getAnnotation(OnUpdateCurrentTimestamp.class);
if (null != onUpdateCurrentTimestamp) {
colP.onUpdateCurrentTimestamp(DB_MySql_Opera.ON_UPDATE_APPLIED.contains(field.getType()));
}
// 添加到缓存
tblProp.addColumn(field.getName(), colP);
// 记录索引
Index index = field.getAnnotation(Index.class);
if (null != index) {
IndexProperties idx = new IndexProperties();
idx.setColumnNames(new String[]{colP.colName()});
idx.setIndexName(CommonUtil.isNullEmpty(index.value()) ? getIndexPrefix(index.type()) +
colP.colName() : index.value());
idx.setType(index.type());
idx.setRemark(index.remark());
tblProp.addIndex(idx);
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy