【add】新增data jdbc数据库表反向生成工具类
This commit is contained in:
nixj
parent
1e090f0825
commit
890a97417a
|
|
@ -0,0 +1,473 @@
|
|||
package com.chint.manage.util;
|
||||
|
||||
import java.io.*;
|
||||
import java.sql.*;
|
||||
import java.util.ArrayList;
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
/**
|
||||
* 自动生成实体类、实体映射XML文件、Mapper
|
||||
* 运行当前类即可在D：/workspace 生成文件
|
||||
* @Author：nxj
|
||||
* @Date：2024/5/23 13:23
|
||||
*/
|
||||
public class DataJdbcCreatorUtil {
|
||||
|
||||
//package路径
|
||||
private final String prefix="com.chint.manage";
|
||||
|
||||
public static void main(String[] args) {
|
||||
try {
|
||||
new DataJdbcCreatorUtil().generate();
|
||||
} catch (ClassNotFoundException e) {
|
||||
e.printStackTrace();
|
||||
} catch (SQLException e) {
|
||||
e.printStackTrace();
|
||||
} catch (IOException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 获取所有的表
|
||||
*
|
||||
* @return
|
||||
* @throws SQLException
|
||||
*/
|
||||
private List<String> getTables() throws SQLException {
|
||||
List<String> tables = new ArrayList<String>();
|
||||
tables.add("amap_route_relationship");
|
||||
return tables;
|
||||
}
|
||||
|
||||
/**
|
||||
* ********************************* 使用前必读*******************
|
||||
* *
|
||||
* *
|
||||
* **********************************************************
|
||||
*/
|
||||
|
||||
private final String type_char = "char";
|
||||
|
||||
private final String type_date = "date";
|
||||
|
||||
private final String type_timestamp = "timestamp";
|
||||
|
||||
private final String type_int = "int";
|
||||
|
||||
private final String type_bigint = "bigint";
|
||||
|
||||
private final String type_tinyint = "tinyint";
|
||||
private final String type_varchar = "varchar";
|
||||
private final String type_longtext = "longtext";
|
||||
private final String type_smallint = "smallint";
|
||||
private final String type_datetime = "datetime";
|
||||
private final String type_binary = "binary";
|
||||
private final String type_time = "time";
|
||||
|
||||
private final String type_text = "text";
|
||||
|
||||
private final String type_bit = "bit";
|
||||
|
||||
private final String type_double = "double";
|
||||
|
||||
private final String type_float = "float";
|
||||
|
||||
private final String type_decimal = "decimal";
|
||||
|
||||
private final String type_blob = "blob";
|
||||
|
||||
|
||||
/**
|
||||
* 路径
|
||||
*/
|
||||
|
||||
private final String bean_path = "d:/Workspace/model/";
|
||||
|
||||
private final String mapper_path = "d:/Workspace/mapper";
|
||||
|
||||
private final String xml_path = "d:/Workspace/xml/";
|
||||
|
||||
private final String service_path = "d:/Workspace/service/";
|
||||
|
||||
private final String service_impl_path = "d:/Workspace/service/impl";
|
||||
|
||||
/**
|
||||
* package路径
|
||||
*/
|
||||
private final String bean_package = prefix+".entity";
|
||||
|
||||
private final String mapper_package = prefix+".mapper";
|
||||
|
||||
private final String service_package = prefix+".service";
|
||||
|
||||
private final String service_impl_package = prefix+".service.impl";
|
||||
|
||||
/**
|
||||
* 数据库
|
||||
*/
|
||||
|
||||
private final String url = "jdbc:mysql://10.10.24.44:6603/itinerary_booking?useUnicode=true&characterEncoding=UTF-8&allowMultiQueries=true&useSSL=false&serverTimezone=GMT%2B8";
|
||||
|
||||
private final String driverName = "com.mysql.cj.jdbc.Driver";
|
||||
|
||||
private final String user = "tripbook";
|
||||
|
||||
private final String password = "W@Xgf25d&lRk*L0X#";
|
||||
|
||||
|
||||
private String tableName = null;
|
||||
|
||||
private String beanName = null;
|
||||
|
||||
private String mapperName = null;
|
||||
|
||||
private String _mapperName = null;
|
||||
|
||||
private String serviceName = null;
|
||||
|
||||
private String serviceImplName = null;
|
||||
|
||||
private String primaryKeyName = null;
|
||||
|
||||
private Connection conn = null;
|
||||
|
||||
private void init() throws ClassNotFoundException, SQLException {
|
||||
Class.forName(driverName);
|
||||
conn = DriverManager.getConnection(url, user, password);
|
||||
}
|
||||
|
||||
private void processTable(String table) {
|
||||
StringBuffer sb = new StringBuffer(table.length());
|
||||
String tableNew = table.toLowerCase();
|
||||
String[] tables = tableNew.split("_");
|
||||
String temp = null;
|
||||
for (int i = 0; i < tables.length; i++) {
|
||||
temp = tables[i].trim();
|
||||
sb.append(temp.substring(0, 1).toUpperCase()).append(temp.substring(1));
|
||||
}
|
||||
beanName = sb.toString();
|
||||
mapperName = "Jdbc"+beanName + "Repository";
|
||||
_mapperName = shotFirst(mapperName);
|
||||
serviceName = beanName + "Service";
|
||||
serviceImplName = serviceName + "Impl";
|
||||
}
|
||||
|
||||
/**
|
||||
* 首字母转小写
|
||||
*
|
||||
* @return
|
||||
*/
|
||||
private String shotFirst(String str) {
|
||||
char[] chars = new char[1];
|
||||
chars[0] = str.charAt(0);
|
||||
String temp = new String(chars);
|
||||
if (chars[0] >= 'A' && chars[0] <= 'Z') {
|
||||
str = str.replaceFirst(temp, temp.toLowerCase());
|
||||
}
|
||||
return str;
|
||||
}
|
||||
|
||||
private String processType(String type) {
|
||||
if (type.indexOf(type_char) > -1) {
|
||||
return "java.lang.String";
|
||||
} else if (type.indexOf(type_bigint) > -1) {
|
||||
return "java.lang.Long";
|
||||
} else if (type.indexOf(type_int) > -1) {
|
||||
return "java.lang.Integer";
|
||||
} else if (type.indexOf(type_date) > -1) {
|
||||
return "java.util.Date";
|
||||
} else if (type.indexOf(type_text) > -1) {
|
||||
return "java.lang.String";
|
||||
} else if (type.indexOf(type_timestamp) > -1) {
|
||||
return "java.util.Date";
|
||||
} else if (type.indexOf(type_bit) > -1) {
|
||||
return "java.lang.Boolean";
|
||||
} else if (type.indexOf(type_decimal) > -1) {
|
||||
return "java.math.BigDecimal";
|
||||
} else if (type.indexOf(type_blob) > -1) {
|
||||
return "byte[]";
|
||||
} else if (type.indexOf(type_double) > -1) {
|
||||
return "java.lang.Double";
|
||||
} else if (type.indexOf(type_float) > -1) {
|
||||
return "java.lang.Float";
|
||||
} else if (type.indexOf(type_tinyint) > -1) {
|
||||
return "java.lang.Integer";
|
||||
} else if (type.indexOf(type_smallint) > -1) {
|
||||
return "java.lang.Integer";
|
||||
} else if (type.indexOf(type_varchar) > -1) {
|
||||
return "java.lang.String";
|
||||
} else if (type.indexOf(type_longtext) > -1) {
|
||||
return "java.lang.String";
|
||||
} else if (type.indexOf(type_datetime) > -1) {
|
||||
return "java.util.Date";
|
||||
} else if (type.indexOf(type_binary) > -1) {
|
||||
return "java.lang.Long";
|
||||
} else if (type.indexOf(type_time) > -1) {
|
||||
return "java.util.Date";
|
||||
}
|
||||
|
||||
|
||||
return null;
|
||||
}
|
||||
|
||||
private String processField(String field) {
|
||||
StringBuffer sb = new StringBuffer(field.length());
|
||||
String[] fields = field.split("_");
|
||||
String temp = null;
|
||||
sb.append(fields[0]);
|
||||
for (int i = 1; i < fields.length; i++) {
|
||||
temp = fields[i].trim();
|
||||
sb.append(temp.substring(0, 1).toUpperCase()).append(
|
||||
temp.substring(1));
|
||||
}
|
||||
return sb.toString();
|
||||
}
|
||||
|
||||
/**
|
||||
* 构建类上面的注释
|
||||
*/
|
||||
private BufferedWriter buildClassComment(BufferedWriter bw, String text)
|
||||
throws IOException {
|
||||
bw.newLine();
|
||||
bw.newLine();
|
||||
bw.write("/**");
|
||||
bw.newLine();
|
||||
bw.write(" * ");
|
||||
bw.newLine();
|
||||
bw.write(" * @author xx " + text);
|
||||
bw.newLine();
|
||||
bw.write(" * ");
|
||||
bw.newLine();
|
||||
bw.write(" **/");
|
||||
return bw;
|
||||
}
|
||||
|
||||
/**
|
||||
* 生成实体类
|
||||
*
|
||||
* @param columns
|
||||
* @param types
|
||||
* @param comments
|
||||
* @throws IOException
|
||||
*/
|
||||
private void buildEntityBean(List<String> columns, List<String> types, List<String> comments, String tableComment) throws IOException {
|
||||
File folder = new File(bean_path);
|
||||
if (!folder.exists()) {
|
||||
folder.mkdirs();
|
||||
}
|
||||
|
||||
File beanFile = new File(bean_path, beanName + ".java");
|
||||
BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(beanFile)));
|
||||
bw.write("package " + bean_package + ";\n");
|
||||
bw.newLine();
|
||||
bw.write("import lombok.Data;");
|
||||
bw.newLine();
|
||||
bw.write("import org.springframework.data.relational.core.mapping.Table;");
|
||||
bw.newLine();
|
||||
bw.write("import java.io.Serial;");
|
||||
bw.newLine();
|
||||
bw.write("import java.io.Serializable;");
|
||||
bw = buildClassComment(bw, beanName + ".java");
|
||||
bw.newLine();
|
||||
bw.write("@Data");
|
||||
bw.newLine();
|
||||
bw.write("@Table(\""+tableName+"\")");
|
||||
bw.newLine();
|
||||
bw.write("public class " + beanName+" implements Serializable" + " {");
|
||||
bw.newLine();
|
||||
bw.write("@Serial");
|
||||
bw.newLine();
|
||||
bw.write("private static final long serialVersionUID = 1L;");
|
||||
bw.newLine();
|
||||
int size = columns.size();
|
||||
for (int i = 0; i < size; i++) {
|
||||
bw.write("\t/**" + comments.get(i) + "**/");
|
||||
bw.newLine();
|
||||
bw.write("\tprivate " + processType(types.get(i)) + " "
|
||||
+ processField(columns.get(i)) + ";");
|
||||
bw.newLine();
|
||||
bw.newLine();
|
||||
}
|
||||
bw.newLine();
|
||||
String tempField = null;
|
||||
String _tempField = null;
|
||||
String tempType = null;
|
||||
for (int i = 0; i < size; i++) {
|
||||
tempType = processType(types.get(i));
|
||||
_tempField = processField(columns.get(i));
|
||||
if (i == 0) {
|
||||
primaryKeyName = _tempField;
|
||||
}
|
||||
}
|
||||
bw.write("}");
|
||||
bw.newLine();
|
||||
bw.flush();
|
||||
bw.close();
|
||||
}
|
||||
|
||||
/**
|
||||
* 构建Service文件
|
||||
*
|
||||
* @throws IOException
|
||||
*/
|
||||
private void buildService() throws IOException {
|
||||
File folder = new File(service_path);
|
||||
if (!folder.exists()) {
|
||||
folder.mkdirs();
|
||||
}
|
||||
|
||||
File serviceFile = new File(service_path, serviceName + ".java");
|
||||
BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(serviceFile), "utf-8"));
|
||||
bw.write("package " + service_package + ";");
|
||||
bw.newLine();
|
||||
bw.newLine();
|
||||
bw.write("import " + this.bean_package + "." + beanName + ";");
|
||||
bw.newLine();
|
||||
bw = buildClassComment(bw, serviceName);
|
||||
bw.newLine();
|
||||
bw.write("public interface " + serviceName + " {");
|
||||
bw.newLine();
|
||||
bw.newLine();
|
||||
bw.newLine();
|
||||
bw.newLine();
|
||||
bw.write("}");
|
||||
bw.flush();
|
||||
bw.close();
|
||||
}
|
||||
|
||||
/**
|
||||
* 构建ServiceImpl文件
|
||||
*
|
||||
* @throws IOException
|
||||
*/
|
||||
private void buildServiceImpl() throws IOException {
|
||||
File folder = new File(service_impl_path);
|
||||
if (!folder.exists()) {
|
||||
folder.mkdirs();
|
||||
}
|
||||
File daoFile = new File(service_impl_path, serviceImplName + ".java");
|
||||
BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(daoFile), "utf-8"));
|
||||
bw.write("package " + service_impl_package + ";");
|
||||
bw.newLine();
|
||||
bw.newLine();
|
||||
bw.write("import org.springframework.beans.factory.annotation.Autowired;");
|
||||
bw.newLine();
|
||||
bw.write("import " + mapper_package + "." + mapperName + ";");
|
||||
bw.newLine();
|
||||
bw.write("import org.springframework.stereotype.Service;");
|
||||
bw.newLine();
|
||||
bw.write("import " + service_package + "." + serviceName + ";");
|
||||
bw.newLine();
|
||||
bw.write("import " + bean_package + "." + beanName + ";");
|
||||
bw.newLine();
|
||||
bw = buildClassComment(bw, serviceName);
|
||||
bw.newLine();
|
||||
bw.write("@Service(\"" + this.shotFirst(serviceName) + "\")");
|
||||
bw.newLine();
|
||||
bw.write("public class " + serviceImplName + " implements " + serviceName + " {");
|
||||
bw.newLine();
|
||||
bw.newLine();
|
||||
bw.write("\t@Autowired");
|
||||
bw.newLine();
|
||||
bw.write("\tprivate " + mapperName + " " + this.shotFirst(mapperName) + ";");
|
||||
bw.newLine();
|
||||
bw.newLine();
|
||||
bw.write("}");
|
||||
bw.flush();
|
||||
bw.close();
|
||||
}
|
||||
|
||||
/**
|
||||
* 构建Mapper文件
|
||||
*
|
||||
* @throws IOException
|
||||
*/
|
||||
private void buildMapper() throws IOException {
|
||||
File folder = new File(mapper_path);
|
||||
if (!folder.exists()) {
|
||||
folder.mkdirs();
|
||||
}
|
||||
|
||||
File mapperFile = new File(mapper_path, mapperName + ".java");
|
||||
BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(
|
||||
new FileOutputStream(mapperFile), "utf-8"));
|
||||
bw.write("package " + mapper_package + ";");
|
||||
bw.newLine();
|
||||
bw.write("import " + bean_package + "." + beanName + ";");
|
||||
bw.newLine();
|
||||
bw.write("import org.springframework.data.repository.CrudRepository;");
|
||||
bw.newLine();
|
||||
bw.write("import org.springframework.stereotype.Repository;");
|
||||
bw.newLine();
|
||||
bw = buildClassComment(bw, mapperName + "数据库操作接口类");
|
||||
bw.newLine();
|
||||
bw.write("@Repository");
|
||||
bw.newLine();
|
||||
bw.write("public interface " + mapperName + " extends CrudRepository<" + beanName + ",Long> {");
|
||||
bw.newLine();
|
||||
bw.newLine();
|
||||
bw.write("}");
|
||||
bw.flush();
|
||||
bw.close();
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取所有的数据库表注释
|
||||
*/
|
||||
private Map<String, String> getTableComment() throws SQLException {
|
||||
Map<String, String> maps = new HashMap<String, String>();
|
||||
PreparedStatement pstate = conn.prepareStatement("show table status");
|
||||
ResultSet results = pstate.executeQuery();
|
||||
while (results.next()) {
|
||||
String tableName = results.getString("NAME");
|
||||
String comment = results.getString("COMMENT");
|
||||
maps.put(tableName, comment);
|
||||
}
|
||||
return maps;
|
||||
}
|
||||
|
||||
public void generate() throws ClassNotFoundException, SQLException,
|
||||
IOException {
|
||||
init();
|
||||
String prefix = "show full fields from ";
|
||||
List<String> columns = null;
|
||||
List<String> types = null;
|
||||
List<String> comments = null;
|
||||
PreparedStatement pstate = null;
|
||||
List<String> tables = getTables();
|
||||
Map<String, String> tableComments = getTableComment();
|
||||
|
||||
for (String table : tables) {
|
||||
System.out.println("======>>>" + table);
|
||||
if (!table.equalsIgnoreCase("test")) {
|
||||
columns = new ArrayList<String>();
|
||||
types = new ArrayList<String>();
|
||||
comments = new ArrayList<String>();
|
||||
pstate = conn.prepareStatement(prefix + table);
|
||||
ResultSet results = pstate.executeQuery();
|
||||
while (results.next()) {
|
||||
columns.add(results.getString("FIELD"));
|
||||
types.add(results.getString("TYPE"));
|
||||
comments.add(results.getString("COMMENT"));
|
||||
}
|
||||
tableName = table;
|
||||
if (table.startsWith("t_") || table.equalsIgnoreCase("user") || table.equalsIgnoreCase("support_gen")) {
|
||||
System.out.println("==skip table====>>>" + table);
|
||||
continue;
|
||||
}
|
||||
|
||||
processTable(table);
|
||||
String tableComment = tableComments.get(tableName);
|
||||
buildEntityBean(columns, types, comments, tableComment);
|
||||
buildMapper();
|
||||
buildService();
|
||||
buildServiceImpl();
|
||||
}
|
||||
}
|
||||
conn.close();
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue