package com.lunhan.xxx.common.orm2.gencolumn;
|
|
import org.apache.commons.lang3.StringUtils;
|
|
import java.io.File;
|
import java.io.FileWriter;
|
import java.io.IOException;
|
import java.io.PrintWriter;
|
import java.sql.*;
|
import java.util.LinkedHashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
/**
|
* 生成数据库列 MySql
|
*/
|
public class GenColumnMySql {
|
private final Connection conn;
|
private String outFilePath;
|
|
public GenColumnMySql(String url, String name, String password, String driver) throws ClassNotFoundException, SQLException {
|
//加载驱动
|
Class.forName(driver);
|
//获得连接
|
conn = DriverManager.getConnection(url, name, password);
|
}
|
|
public boolean run(String packageName, List<String> importList, String outPath) throws Exception {
|
if (conn == null || conn.isClosed()) {
|
return false;
|
}
|
|
outFilePath = new File("").getAbsolutePath() + "/" + outPath + "/" + packageName.replace(".", "/");
|
System.out.println("outFilePath：" + outFilePath);
|
File directory = new File(outFilePath);
|
directory.mkdirs();
|
|
try {
|
Map<String, String> tableInfo = this.getTableInfo(conn.getCatalog());
|
for (Map.Entry<String, String> item : tableInfo.entrySet()) {
|
String tableName = item.getKey();
|
String tableComment = item.getValue();
|
Map<String, String> columnInfoMap = this.getColumnInfo(tableName);
|
this.createColumnFile(tableName, tableComment, columnInfoMap, packageName, importList);
|
}
|
} finally {
|
conn.close();
|
}
|
return true;
|
}
|
|
private Map<String, String> getTableInfo(String tableSchema) throws SQLException {
|
Map<String, String> tableInfoMap = new LinkedHashMap<>();
|
String sql = " SELECT TABLE_NAME ,TABLE_COMMENT " +
|
" FROM information_schema.`TABLES` " +
|
" WHERE TABLE_SCHEMA = ? ";
|
try (PreparedStatement pStmt = conn.prepareStatement(sql)) {
|
pStmt.setString(1, tableSchema);
|
try (ResultSet rs = pStmt.executeQuery()) {
|
while (rs.next()) {
|
String tableName = rs.getString("TABLE_NAME");
|
String tableComment = rs.getString("TABLE_COMMENT");
|
tableInfoMap.put(tableName, tableComment);
|
}
|
}
|
}
|
return tableInfoMap;
|
}
|
|
private Map<String, String> getColumnInfo(String tableName) throws SQLException {
|
Map<String, String> columnInfoMap = new LinkedHashMap<>();
|
String sql = " SELECT COLUMN_NAME ,COLUMN_COMMENT " +
|
" FROM information_schema.`COLUMNS` " +
|
" WHERE TABLE_NAME = ? ";
|
try (PreparedStatement pStmt = conn.prepareStatement(sql)) {
|
pStmt.setString(1, tableName);
|
try (ResultSet rs = pStmt.executeQuery()) {
|
while (rs.next()) {
|
String columnName = rs.getString("COLUMN_NAME");
|
String columnComment = rs.getString("COLUMN_COMMENT");
|
columnInfoMap.put(columnName, columnComment);
|
}
|
}
|
}
|
return columnInfoMap;
|
}
|
|
private void createColumnFile(String tableName, String tableComment, Map<String, String> columnInfoMap, String packageName, List<String> importList) throws IOException {
|
String fileName = this.underline2Camel(tableName, true);
|
String filePath = this.outFilePath + "/" + fileName + ".java";
|
|
File columnFile = new File(filePath);
|
columnFile.createNewFile();
|
|
StringBuilder sb = new StringBuilder();
|
sb.append("/*\n" +
|
"# __----~~~~~~~~~~~------___\n" +
|
"# . . ~~//====...... __--~ ~~\n" +
|
"# -. \\_|// |||\\\\ ~~~~~~::::... /~\n" +
|
"# ___-==_ _-~o~ \\/ ||| \\\\ _/~~-\n" +
|
"# __---~~~.==~||\\=_ -_--~/_-~|- |\\\\ \\\\ _/~\n" +
|
"# _-~~ .=~ | \\\\-_ '-~7 /- / || \\ /\n" +
|
"# .~ .~ | \\\\ -_ / /- / || \\ /\n" +
|
"# / ____ / | \\\\ ~-_/ /|- _/ .|| \\ /\n" +
|
"# |~~ ~~|--~~~~--_ \\ ~==-/ | \\~--===~~ .\\\n" +
|
"# ' ~-| /| |-~\\~~ __--~~\n" +
|
"# |-~~-_/ | | ~\\_ _-~ /\\\n" +
|
"# / \\ \\__ \\/~ \\__\n" +
|
"# _--~ _/ | .-~~____--~-/ ~~==.\n" +
|
"# ((->/~ '.|||' -_| ~~-/ , . _||\n" +
|
"# -_ ~\\ ~~---l__i__i__i--~~_/\n" +
|
"# _-~-__ ~) \\--______________--~~\n" +
|
"# //.-~~~-~_--~- |-------~~~~~~~~\n" +
|
"# //.-~~~--\\\n" +
|
"# 神兽保佑\n" +
|
"# 代码无BUG! \n" +
|
"*/");
|
sb.append("\r\n");
|
sb.append(String.format("package %s;%n", packageName));
|
sb.append("\r\n");
|
for (String item : importList) {
|
sb.append(String.format("import %s;%n", item));
|
}
|
sb.append("\r\n");
|
sb.append("/**\r\n");
|
sb.append(String.format(" * %s%n", tableComment));
|
sb.append(" */\r\n");
|
sb.append(String.format("public enum %s implements ColumnBasic {%n", fileName));
|
int index = 1;
|
for (Map.Entry<String, String> item : columnInfoMap.entrySet()) {
|
String columnName = item.getKey();
|
String columnComment = item.getValue();
|
sb.append(" /**\r\n");
|
sb.append(String.format(" * %s%n", columnComment.replace("\r\n", "\r\n * ")));
|
sb.append(" */\r\n");
|
sb.append(String.format(" %s(\"%s\", %s),%n", columnName.toUpperCase(), columnName, index++));
|
}
|
sb.replace(sb.lastIndexOf(","), sb.length(), ";\r\n");
|
sb.append("\r\n");
|
sb.append(" private String columnName;\r\n");
|
sb.append(" private Integer columnIndex;\r\n");
|
sb.append(String.format(" public static final String TABLE_NAME = \"%s\";%n", tableName));
|
sb.append("\r\n");
|
sb.append(String.format(" %s(String columnName, Integer columnIndex) {%n", fileName));
|
sb.append(" this.columnName = columnName;\r\n");
|
sb.append(" this.columnIndex = columnIndex;\r\n");
|
sb.append(" }\r\n");
|
sb.append("\r\n");
|
sb.append(" @Override\r\n");
|
sb.append(" public String getColumnName() {\r\n");
|
sb.append(" return ColumnBuilder.getColumnName(TABLE_NAME, this.columnName);\r\n");
|
sb.append(" }\r\n");
|
sb.append("}");
|
|
try (FileWriter fw = new FileWriter(columnFile.getAbsolutePath());
|
PrintWriter pw = new PrintWriter(fw)) {
|
pw.println(sb.toString());
|
pw.flush();
|
}
|
}
|
|
private String underline2Camel(String line, boolean... firstIsUpperCase) {
|
String str = "";
|
if (StringUtils.isBlank(line)) {
|
return str;
|
} else {
|
StringBuilder sb = new StringBuilder();
|
String[] strArr;
|
// 不包含下划线，且第二个参数是空的
|
if (!line.contains("_") && firstIsUpperCase.length == 0) {
|
sb.append(line.substring(0, 1).toLowerCase()).append(line.substring(1));
|
str = sb.toString();
|
} else if (!line.contains("_") && firstIsUpperCase.length != 0) {
|
if (!firstIsUpperCase[0]) {
|
sb.append(line.substring(0, 1).toLowerCase()).append(line.substring(1));
|
str = sb.toString();
|
} else {
|
sb.append(line.substring(0, 1).toUpperCase()).append(line.substring(1));
|
str = sb.toString();
|
}
|
} else if (line.contains("_") && firstIsUpperCase.length == 0) {
|
strArr = line.split("_");
|
for (String s : strArr) {
|
sb.append(s.substring(0, 1).toUpperCase()).append(s.substring(1));
|
}
|
str = sb.toString();
|
str = str.substring(0, 1).toLowerCase() + str.substring(1);
|
} else if (line.contains("_") && firstIsUpperCase.length != 0) {
|
strArr = line.split("_");
|
for (String s : strArr) {
|
sb.append(s.substring(0, 1).toUpperCase()).append(s.substring(1));
|
}
|
if (!firstIsUpperCase[0]) {
|
str = sb.toString();
|
str = str.substring(0, 1).toLowerCase() + str.substring(1);
|
} else {
|
str = sb.toString();
|
}
|
}
|
}
|
return str;
|
}
|
|
// public static void main(String[] args) throws Exception {
|
// String url = "jdbc:mysql://81.68.189.38:3306/mall-test1?serverTimezone=UTC&useUnicode=true&characterEncoding=utf8&useSSL=false";
|
// String name = "root";
|
// String password = "root";
|
// String driver = "com.mysql.jdbc.Driver";
|
// String outPath = "framework-common/src/test/java";
|
// String packageName = "com.ctrip.tour.cq.framework.common.test.dbcoulmns.orms";
|
// List<String> importList = new ArrayList() {{
|
// add("com.ctrip.tour.cq.framework.common.orm2.ColumnBuilder");
|
// add("com.ctrip.tour.cq.framework.common.orm2.enums.ColumnBasic");
|
// }};
|
// new GenColumnMySql(url, name, password, driver).run(packageName, importList, outPath);
|
// }
|
}
|
