JDBC 实现批量插入-任意表名

相关代码

声明常量

cpp 复制代码
static final String JDBC_DRIVER = "com.mysql.jdbc.Driver";
static final String DB_URL = "jdbc:mysql://localhost:3306/your_database_name";

static final String USER = "数据库账号";
static final String PASS = "数据库密码";

主方法

cpp 复制代码
public void batchInsert(List<?> records, String tableName,Class<?> className) throws SQLException{
        Connection connection = null;
        PreparedStatement preparedStatement = null;
        try {
            // 注册 JDBC 驱动
            Class.forName(JDBC_DRIVER);
            connection = DriverManager.getConnection(DB_URL, USER, PASS);
            // 创建预处理语句对象
            StringBuilder columnNames = new StringBuilder();
            StringBuilder placeholders = new StringBuilder();

            Field[] fields = className.getDeclaredFields();
            for (int i = 0; i < fields.length; i++) {
                String fieldName = fields[i].getName();
                if (fieldName.equals("id")) {
                    continue;
                }
                if (isCamelCase(fieldName)) {
                    fieldName = toUnderscoreCase(fieldName);
                }
                columnNames.append(fieldName);
                placeholders.append("?");
                if (i < fields.length - 1) {
                    columnNames.append(", ");
                    placeholders.append(", ");
                }
            }
            String sql = "INSERT INTO " + tableName + " (" + columnNames.toString() + ") VALUES (" + placeholders.toString() + ")";
            preparedStatement = connection.prepareStatement(sql);

            // 添加批量数据
            for (Object entity : records) {
                setFieldValues(entity, preparedStatement);
                preparedStatement.addBatch();
            }

            // 执行批量插入
            int[] result = preparedStatement.executeBatch();

            // 输出结果
            for (int i = 0; i < result.length; i++) {
                log.error("插入结果:" + result[i]);
            }
            log.error("插入结果数量{}", result.length);

        } catch (Exception e) {
            log.error("插入结果:", e);
        } finally {
            // 关闭资源
            if (preparedStatement != null) {
                preparedStatement.close();
            }
            if (connection != null) {
                connection.close();
            }
        }
    }

设置属性值

cpp 复制代码
 private static void setFieldValues(Object obj, PreparedStatement preparedStatement) throws IllegalAccessException, SQLException {
        Field[] fields = obj.getClass().getDeclaredFields();

        fields = Arrays.stream(fields)
                .filter(field -> !field.getName().equals("id")).toArray(Field[]::new);

        for (int i = 0; i < fields.length; i++) {

            fields[i].setAccessible(true);

            preparedStatement.setObject(i + 1, fields[i].get(obj));
        }
    }

是否为小驼峰

cpp 复制代码
 private static boolean isCamelCase(String str) {
         if(str != null && !str.isEmpty()){
             String regex = "^[a-z]+([A-Z][a-z0-9]*)*$";
             return str.matches(regex);
         }
        return false;
    }

小驼峰转为下划线

cpp 复制代码
 private static String toUnderscoreCase(String str) {
        StringBuilder result = new StringBuilder();
        for (int i = 0; i < str.length(); i++) {
            char c = str.charAt(i);
            if (Character.isUpperCase(c)) {
                if (i > 0) {
                    result.append('_');
                }
                result.append(Character.toLowerCase(c));
            } else {
                result.append(c);
            }
        }
        return result.toString();
    }

实体

cpp 复制代码
@Data
public class ProductEntity implements Serializable {
    /**
     * 主键id
     */
    private BigInteger id;
    /**
     * 用户id
     */
    private String userId;
   /**
     * 商品名称
     */
    private String productName;

}

调用方法

cpp 复制代码
public void copyData(UserDataModel model) throws SQLException {
	
        List<ProductEntity> products = prodctMapper.selectAllInfos(model.getFromUserId());
        products.forEach(record -> record.setUserId(Long.valueOf(model.getToUserId())));
        batchInsert(products, "product_record",ProductEntity.class);
}

说明

如果实体属性与数据库字段一致,不需要做驼峰转下划线这一操作,否则就需要进行转化。

补充

如果数据库内容有表情的时候,确保你的Java程序使用正确的字符集连接MySQL数据库。在JDBC连接字符串中添加&useUnicode=true&characterEncoding=UTF-8参数,例如:

cpp 复制代码
String url = "jdbc:mysql://localhost:3306/your_database_name?useUnicode=true&characterEncoding=UTF-8";
相关推荐
JH30733 分钟前
Oracle与MySQL中CONCAT()函数的使用差异
数据库·mysql·oracle
蓝染-惣右介5 分钟前
【MyBatisPlus·最新教程】包含多个改造案例,常用注解、条件构造器、代码生成、静态工具、类型处理器、分页插件、自动填充字段
java·数据库·tomcat·mybatis
冷心笑看丽美人6 分钟前
Spring框架特性及包下载(Java EE 学习笔记04)
数据库
武子康1 小时前
Java-07 深入浅出 MyBatis - 一对多模型 SqlMapConfig 与 Mapper 详细讲解测试
java·开发语言·数据库·sql·mybatis·springboot
代码吐槽菌2 小时前
基于SSM的毕业论文管理系统【附源码】
java·开发语言·数据库·后端·ssm
路有瑶台2 小时前
MySQL数据库学习(持续更新ing)
数据库·学习·mysql
数字扫地僧2 小时前
WebLogic 版本升级的注意事项与流程
数据库
Viktor_Ye2 小时前
高效集成易快报与金蝶应付单的方案
java·前端·数据库
努力算法的小明3 小时前
SQL 复杂查询
数据库·sql
斗-匕3 小时前
MySQL 三大日志详解
数据库·mysql·oracle