com.star.template.db.table.Processor Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of utils Show documentation
Show all versions of utils Show documentation
some utility class for java develop
The newest version!
package com.star.template.db.table;
import com.star.collection.CollectionUtil;
import com.star.exception.TableException;
import com.star.jdbc.SqlRunner;
import com.star.properties.Config;
import com.star.string.StringUtil;
import com.star.template.db.constant.ColumnEnum;
import com.star.template.db.constant.Constant;
import com.star.template.db.constant.TableEnum;
import com.star.template.db.model.Column;
import com.star.template.db.model.ForeignKey;
import com.star.template.db.model.Table;
import com.star.template.db.util.ColumnUtil;
import com.star.template.db.util.DatabaseUtil;
import com.star.template.db.util.DatabaseVersionUtil;
import com.star.template.db.util.PropertyUtil;
import com.star.template.db.util.TypeUtil;
import java.sql.DatabaseMetaData;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.StringJoiner;
import java.util.StringTokenizer;
/**
* Created by win7 on 2017/2/25.
*/
/**
* Created by win7 on 2017/2/25.
*/
public class Processor {
/**
* 根据结果记创建表信息
*
* @param resultSet 结果姐
* @return 表信息
*/
public Table create(final ResultSet resultSet) {
try {
final String tableName = resultSet.getString(TableEnum.NAME.code);
final String remarks = getRemarks(tableName, resultSet);
final Table table = new Table();
table.setSqlName(tableName);
table.setRemarks(remarks);
final String name = StringUtil.toCamelCase(removePrefix(tableName));
table.setClassName(StringUtil.upperOrLowerFirst(name, true));
table.setVariableName(StringUtil.upperOrLowerFirst(name, false));
final List columns = retriveTableColumns(tableName);
table.setColumns(columns);
final List importedKeys = initForeginKeys(tableName, true);
table.setManyToOne(importedKeys);
final List exportedKeys = initForeginKeys(tableName, false);
table.setOneToMany(exportedKeys);
return table;
} catch (SQLException e) {
throw new TableException(StringUtil.format("create table failure,the reason is: {}", e.getMessage()), e);
}
}
private List initForeginKeys(final String sqlName, final boolean isImported) {
final DatabaseMetaData metaData = DatabaseUtil.getInstance().getMetaData();
final List foreignKeys = new ArrayList<>();
try (ResultSet resultSet = isImported ? metaData.getImportedKeys(PropertyUtil.getSchema(),
PropertyUtil.getCatalog(), sqlName) : metaData.getExportedKeys(PropertyUtil.getSchema(),
PropertyUtil.getCatalog(), sqlName)) {
while (resultSet.next()) {
final String pkTable = resultSet.getString(TableEnum.PKTABLE.code);
final String pkColumn = resultSet.getString(TableEnum.PKCOLUMN.code);
final String fkTable = resultSet.getString(TableEnum.FKTABLE.code);
final String fkColumn = resultSet.getString(TableEnum.FKCOLUMN.code);
final ForeignKey foreignKey = new ForeignKey();
String name = StringUtil.toCamelCase(removePrefix(pkTable));
foreignKey.setPkTable(pkTable);
foreignKey.setPkTableClass(StringUtil.upperOrLowerFirst(name, true));
foreignKey.setPkTableClassLower(StringUtil.upperOrLowerFirst(name, false));
foreignKey.setPkColumn(pkColumn);
foreignKey.setPkCloumnClass(StringUtil.upperOrLowerFirst(StringUtil.toCamelCase(pkColumn), true));
foreignKey.setPkColumnLower(StringUtil.upperOrLowerFirst(StringUtil.toCamelCase(pkColumn), false));
name = StringUtil.toCamelCase(removePrefix(fkTable));
foreignKey.setFkTable(name);
foreignKey.setFkTableClass(StringUtil.upperOrLowerFirst(name, true));
foreignKey.setFkTableLower(StringUtil.upperOrLowerFirst(name, false));
foreignKey.setFkColumn(fkColumn);
foreignKey.setFkColumnClass(StringUtil.upperOrLowerFirst(StringUtil.toCamelCase(fkColumn), true));
foreignKey.setFkColumnLower(StringUtil.upperOrLowerFirst(StringUtil.toCamelCase(fkColumn), true));
foreignKeys.add(foreignKey);
}
return foreignKeys;
} catch (SQLException e) {
throw new TableException(StringUtil.format("init table {}'s foreign key failure,the reason is" +
" {}", sqlName, e.getMessage()), e);
}
}
/**
* 获得所有表信息
*
* @return 表集合
*/
public List getAllTables() {
final DatabaseMetaData metaData = DatabaseUtil.getInstance().getMetaData();
try (ResultSet resultSet = metaData.getTables(PropertyUtil.getCatalog(), PropertyUtil.getSchema(), null,
null)) {
final List tables = new ArrayList<>();
while (resultSet.next()) {
tables.add(create(resultSet));
}
return tables;
} catch (SQLException e) {
throw new TableException(StringUtil.format("get all tables failure,the reason is: {}", e.getMessage()), e);
}
}
private String getOracleTableComments(final String tableName) {
String sql = "SELECT comments FROM user_tab_comments WHERE table_name= ?";
final SqlRunner sqlRunner = new SqlRunner();
return sqlRunner.query(DatabaseUtil.getInstance().getConnection(), sql, resultSet -> {
try {
return resultSet.getString(1);
} catch (SQLException e) {
throw new TableException(e);
}
}, tableName);
}
private List retriveTableColumns(final String tableName) {
final List primaryKeys = getTablePrimaryKeys(tableName);
//索引集合
final List indices = new ArrayList<>();
//列名+索引名（id+primary）
final Map uniqueIndices = new HashMap<>();
//索引名+列的集合（primary+id）
final Map> uniqueColumns = new HashMap<>();
final DatabaseMetaData metaData = DatabaseUtil.getInstance().getMetaData();
try (ResultSet indexRs = metaData.getIndexInfo(PropertyUtil.getSchema(),
PropertyUtil.getCatalog(), tableName, false, true)) {
while (indexRs.next()) {
final String columnName = indexRs.getString(ColumnEnum.COLUMNNAME.code);
if (!StringUtil.isBlank(columnName)) {
indices.add(columnName);
}
final String indexName = indexRs.getString(ColumnEnum.INDEX.code);
final boolean nonUnique = indexRs.getBoolean(ColumnEnum.NONUNIQUE.code);
if (!nonUnique && !StringUtil.isBlank(columnName) && !StringUtil.isBlank(indexName)) {
List column = uniqueColumns.get(indexName);
if (CollectionUtil.isEmpty(column)) {
column = new ArrayList<>();
uniqueColumns.put(indexName, column);
}
column.add(columnName);
uniqueIndices.put(columnName, indexName);
}
}
return getTableColumn(tableName, primaryKeys, indices, uniqueIndices, uniqueColumns);
} catch (SQLException e) {
throw new TableException(StringUtil.format("get {}'s index failure: {}", tableName, e.getMessage
()), e);
}
}
private List getTableColumn(final String tableName, final List primaryKeys, final List
indices, final Map uniqueIndices, final Map> uniqueColumns) {
final DatabaseMetaData metaData = DatabaseUtil.getInstance().getMetaData();
final List columns = new ArrayList<>();
try (ResultSet columnRs = metaData.getColumns(PropertyUtil.getSchema(), PropertyUtil.getCatalog(), tableName,
null)) {
while (columnRs.next()) {
final int sqlType = columnRs.getInt(ColumnEnum.DATATYPE.code);
final String sqlTypeName = columnRs.getString(ColumnEnum.TYPENAME.code);
final String sqlName = columnRs.getString(ColumnEnum.COLUMNNAME.code);
final String defaultValue = columnRs.getString(ColumnEnum.COLUMNDEF.code);
final String remarks = getRemarks(tableName, columnRs);
final boolean isNullable = DatabaseMetaData.columnNullable == columnRs.getInt(ColumnEnum.NULLABLE.code);
final int size = columnRs.getInt(ColumnEnum.COLUMNSIZE.code);
final int decimalDigits = columnRs.getInt(ColumnEnum.DECIMALDIGITS.code);
final boolean isPk = primaryKeys.contains(sqlName);
final boolean isIndexed = indices.contains(sqlName);
final String uniqueIndex = uniqueIndices.get(sqlName);
List uniqueList = null;
if (!StringUtil.isBlank(uniqueIndex)) {
uniqueList = uniqueColumns.get(uniqueIndex);
}
final boolean isUnique = !CollectionUtil.isEmpty(uniqueList) && uniqueList.contains(sqlName);
final Column column = new Column();
column.setSqlName(sqlName);
column.setSqlTypeName(sqlTypeName);
column.setDefaultValue(defaultValue);
column.setRemarks(remarks);
column.setIsNullable(isNullable);
column.setSize(size);
column.setDecimalDigits(decimalDigits);
column.setIsPK(isPk);
column.setIsIndexed(isIndexed);
column.setIsUnique(isUnique);
final String sqlTypeToJavaTYpe = TypeUtil.getJavaTypeBySqlType(sqlType, size, decimalDigits);
Config config = new Config("config.properties");
final String javaType = config.getString(Constant.MAPPING + sqlTypeToJavaTYpe, sqlTypeToJavaTYpe);
column.setJavaType(javaType);
final String columnName = StringUtil.toCamelCase(sqlName);
column.setColumnName(StringUtil.upperOrLowerFirst(columnName, true));
column.setColumnNameLower(StringUtil.upperOrLowerFirst(columnName, false));
final String columnAlias = StringUtil.defaultIfEmpty(column.getRemarks(), column.getColumnNameLower());
final StringTokenizer stringTokenizer = new StringTokenizer(columnAlias, "\t" + StringUtil
.CRLF + StringUtil.PAGEBREAKS);
final StringJoiner joiner = new StringJoiner(StringUtil.SPACE);
while (stringTokenizer.hasMoreElements()) {
joiner.add((String) stringTokenizer.nextElement());
}
column.setColumnAlias(joiner.toString());
column.setHibernateValidatorExprssion(ColumnUtil.getHibernateValidatorExpression(column));
columns.add(column);
}
return columns;
} catch (SQLException e) {
throw new TableException(StringUtil.format("get {}'s column failure: {}", tableName, e.getMessage
()), e);
}
}
private String getRemarks(final String tableName, final ResultSet columnRs) throws SQLException {
String remarks = columnRs.getString(ColumnEnum.REMARKS.code);
if (StringUtil.isBlank(remarks) && DatabaseVersionUtil.isOracle(DatabaseUtil.getInstance()
.getConnection().getMetaData())) {
remarks = getOracleTableComments(tableName);
}
return remarks;
}
private List getTablePrimaryKeys(final String tableName) {
final List primaryKeys = new ArrayList<>();
try (ResultSet primaryKeyRs = DatabaseUtil.getInstance().getMetaData().getPrimaryKeys(PropertyUtil.getSchema(),
PropertyUtil.getCatalog(), tableName)) {
while (primaryKeyRs.next()) {
final String columnName = primaryKeyRs.getString(ColumnEnum.COLUMNNAME.code);
primaryKeys.add(columnName);
}
return primaryKeys;
} catch (SQLException e) {
throw new TableException(StringUtil.format("get {}'s primary keys failure: {}", tableName, e.getMessage()
), e);
}
}
/**
* 如果有表前缀，删除表铅锤
*
* @return
*/
private String removePrefix(final String tableName) {
final String prefixes = PropertyUtil.getPrefixes();
String name = tableName;
final String[] prefixArray = StringUtil.split(prefixes, StringUtil.COMMA);
for (final String prefix : prefixArray) {
name = StringUtil.removePrefix(tableName, prefix, true);
}
return name;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy