BeanTosql 工具类
功能
指定Bean包自动生成sql语句文件,并导入mysql数据库
pom依赖
</dependencies>
<!-- SLF4J API -->
<dependency>
<groupId>org.slf4j</groupId>
<artifactId>slf4j-api</artifactId>
<version>2.0.6</version> <!-- 请根据需要选择最新版本 -->
</dependency>
<!-- Logback 经典模块 -->
<dependency>
<groupId>ch.qos.logback</groupId>
<artifactId>logback-classic</artifactId>
<version>1.4.11</version> <!-- 请根据需要选择最新版本 -->
</dependency>
<!-- MySQL JDBC 驱动 -->
<dependency>
<groupId>mysql</groupId>
<artifactId>mysql-connector-java</artifactId>
<version>8.0.30</version> <!-- 请根据需要选择最新版本 -->
</dependency>
</dependencies>
JavaBeanSqlGenerator.java
package com.main;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.lang.reflect.Field;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
public class JavaBeanSqlGenerator {
private static final Logger logger = LoggerFactory.getLogger(JavaBeanSqlGenerator.class);
// MySQL数据库的JDBC URL、用户名和密码
private static final String JDBC_URL = "jdbc:mysql://localhost:3306/?allowMultiQueries=true&useSSL=false&serverTimezone=UTC";
private static final String JDBC_USER = "root";
private static final String JDBC_PASSWORD = "123456";
// 示例数据库名和字符集
private static final String databaseName = "test";
private static final String characterSet = "utf8mb4";
//实体类所在的package在磁盘上的绝对路径
//这里要修改
//表命名前缀
private static final String tablePrefix = "";
private static final String packageName = "C:/Users/zhaid/Desktop/网络数据采集技术-Java网络爬虫实战代码/Chapter 7/DongFangCrawler/src/main/java/com/model";
private static final String sqlName = "entityToSql";
//生成sql的文件夹
private static final String filePath = "src/main/java/com/model/" + sqlName;
//是否插入数据库,写true 直接插入msyql 数据库,谨慎操作,可能会清空数据库
private static boolean CREATETABLE_TO_DATABASE = false;
/**
* 用来存储Java等属性类型与sql中属性类型的对照
* </br>
* 例如:java.lang.Integer 对应 integer
*/
private static Map<String, String> map = new HashMap<>();
static {
map.put("class java.lang.String", "varchar(255)");
map.put("class java.lang.Integer", "int");
map.put("int", "int");
map.put("class java.lang.Long", "integer unsigned");
map.put("class java.lang.byte[]", "blob");
map.put("class java.lang.Boolean", "bit");
map.put("class java.math.BigInteger", "bigint unsigned");
map.put("class java.lang.Float", "float");
map.put("float", "float");
map.put("class java.lang.Double", "double");
map.put("class java.sql.Date", "datetime");
map.put("class java.sql.Time", "time");
map.put("class java.sql.Timestamp", "datetime");
map.put("class java.util.Date", "datetime");
map.put("class java.time.LocalDateTime", "datetime");
map.put("class java.time.LocalDate", "date");
map.put("class java.time.LocalTime", "time");
map.put("class java.lang.Byte", "tinyint");
map.put("class java.math.BigDecimal", "decimal");
}
public static void main(String[] args) {
//项目中实体类的路径,(可以不用修改)
//String prefix = "com.test.model.entity.";
String prefix = packageName.substring(packageName.indexOf("java") + 5).replace("/", ".") + ".";
//开始构造sql
sqlConstruction(packageName, filePath, prefix, tablePrefix);
}
/**
* 生成用于创建MySQL数据库的SQL语句,如果数据库已存在则先删除
*
* @param databaseName 要创建的数据库名
* @param characterSet 要使用的字符集
* @return 包含删除和创建数据库操作的SQL语句
*/
public static String generateCreateOrReplaceDatabaseSql(String databaseName, String characterSet) {
// 检查数据库名和字符集是否为空或null
if (databaseName == null || databaseName.isEmpty()) {
throw new IllegalArgumentException("数据库名不能为空");
}
if (characterSet == null || characterSet.isEmpty()) {
throw new IllegalArgumentException("字符集不能为空");
}
// 生成SQL语句
String sql = "DROP DATABASE IF EXISTS `" + databaseName + "`; \n" +
"CREATE DATABASE `" + databaseName + "` CHARACTER SET " + characterSet + ";\n"+
"USE `" + databaseName + "`;\n\n";
return sql;
}
/**
* 生成sql建库语句
* @param tablePrefix 表命名前缀
* @param packageName 实体类所在的package在磁盘上的绝对路径
* @param filePath 生成sql的文件夹
* @param prefix 项目中实体类的路径
*/
private static void sqlConstruction(String packageName, String filePath, String prefix, String tablePrefix) {
String className = "";
StringBuffer sqls = new StringBuffer();
//添加创建删除数据库语句
String databaseSql = generateCreateOrReplaceDatabaseSql(databaseName, characterSet);
sqls.append(databaseSql);
//获取包下的所有类名称
List<String> list = getAllClasses(packageName);
for (String str : list) {
className = prefix + str.substring(0, str.lastIndexOf("."));
String sql = generateSql(className, tablePrefix);
sqls.append(sql);
}
StringToSql(sqls.toString(), filePath + ".sql");
//执行插入数据库
if(CREATETABLE_TO_DATABASE){
extracted(sqls.toString());
}
}
/**
* 根据实体类生成建表语句
*
* @param className 全类名
* @param tablePrefix 表命名前缀
* @return
*/
public static String generateSql(String className, String tablePrefix) {
try {
Class<?> clz = Class.forName(className);
className = clz.getSimpleName();
// 表表名adminUser → tb_admin_user
className = tablePrefix + getStandardFields(className);
Field[] fields = clz.getDeclaredFields();
StringBuffer column = new StringBuffer();
String varchar = " CHARACTER SET "+characterSet+" COLLATE utf8mb4_general_ci DEFAULT NULL,";
for (Field f : fields) {
if ("class java.lang.String".equals(f.getType().toString())){
column.append(" \n `" + getStandardFields(f.getName()) + "`" + " " + map.get(f.getType().toString())).append(varchar);
}else {
column.append(" \n `" + getStandardFields(f.getName()) + "`" + " " + map.get(f.getType().toString())).append(",");
}
}
//已单独指定id列的生成语句,去掉多余id的拼接
String column1 = column.substring(column.indexOf(",") + 1);
StringBuffer sql = new StringBuffer();
sql.append("\n -- 表 " + className + "\n")
.append("\n DROP TABLE IF EXISTS `" + className + "`; ")
.append(" \n CREATE TABLE `" + className + "` (")
.append(" \n `id` int(11) NOT NULL AUTO_INCREMENT,")
.append(" " + column1)
.append(" \n PRIMARY KEY (`id`) USING BTREE,")
.append("\n INDEX `id`(`id`) USING BTREE")
.append(" \n ) ENGINE = InnoDB AUTO_INCREMENT = 1 CHARACTER SET = "+ characterSet + " COLLATE = utf8mb4_general_ci; \n");
return sql.toString();
} catch (ClassNotFoundException e) {
logger.debug("该类未找到!");
return "";
}
}
/**
* 转换为标准等sql字段 例如 adminUser → admin_user
*
* @param str 转换为字符串的字段名
* @return
*/
public static String getStandardFields(String str) {
StringBuffer sb = new StringBuffer();
for (int i = 0; i < str.length(); i++) {
char c = str.charAt(i);
//非首字母,在大写字母A到Z之间
if (i != 0 && (c >= 'A' && c <= 'Z')) {
sb.append("_");
}
sb.append(c);
}
return sb.toString();
}
/**
* 获取包下面等所有实体类名称,类似于获取 XXX.java
*
* @param packageName 全类名
* @return
*/
public static List<String> getAllClasses(String packageName) {
List<String> classList = new ArrayList();
String className = "";
File f = new File(packageName);
if (f.exists() && f.isDirectory()) {
File[] files = f.listFiles();
// 遍历实体类下面等所有.java文件 获取其类名
for (File file : files) {
className = file.getName();
classList.add(className);
}
return classList;
} else {
logger.debug("包路径未找到!");
return null;
}
}
/**
* 将生成等String字符串 写进sql文件
*
* @param str String字符串
* @param path sql文件路径路径
*/
public static void StringToSql(String str, String path) {
byte[] sourceByte = str.getBytes();
FileOutputStream os = null;
if (null != sourceByte) {
try {
//文件路径(路径+文件名)
File file = new File(path);
//文件不存在则创建文件,先创建目录
if (!file.exists()) {
File dir = new File(file.getParent());
dir.mkdirs();
file.createNewFile();
}
//文件输出流用于将数据写入文件
os = new FileOutputStream(file);
os.write(sourceByte);
os.flush();
System.out.println("生成成功!!");
} catch (Exception e) {
e.printStackTrace();
} finally {
// 关闭文件输出流
try {
if (os != null) {
os.close();
}
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
private static void extracted(String sql) {
Connection connection = null;
Statement statement = null;
try {
// 1. 读取SQL文件
// 2. 连接到MySQL数据库
connection = DriverManager.getConnection(JDBC_URL, JDBC_USER, JDBC_PASSWORD);
// 3. 创建Statement对象来执行SQL语句
statement = connection.createStatement();
// 4. 禁用auto-commit模式(可选,如果你想在出现错误时回滚事务)
connection.setAutoCommit(false);
// 5. 执行SQL语句(这里假设SQL文件中只包含一条SQL语句,或者多条以分号分隔的SQL语句)
// 注意:如果SQL文件包含多条语句,这里可能需要更复杂的逻辑来逐条执行
statement.executeUpdate(sql);
// 6. 提交事务(如果禁用了auto-commit)
connection.commit();
System.out.println("SQL文件执行成功!");
} catch (SQLException e) {
// 如果出现异常,则回滚事务(如果禁用了auto-commit)
if (connection != null) {
try {
connection.rollback();
} catch (SQLException ex) {
ex.printStackTrace();
}
}
e.printStackTrace();
} finally {
// 7. 清理环境,关闭资源
try {
if (statement != null) statement.close();
if (connection != null) connection.close();
} catch (SQLException se) {
se.printStackTrace();
}
}
}
}