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

news/2024/5/19 0:49:23 标签: 数据库, jdbc

相关代码

声明常量

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 = "数据库密码";

主方法

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();
            }
        }
    }

设置属性值

 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));
        }
    }

是否为小驼峰

 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;
    }

小驼峰转为下划线

 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();
    }

实体

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

}

调用方法

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参数,例如: 

String url = "jdbc:mysql://localhost:3306/your_database_name?useUnicode=true&characterEncoding=UTF-8";


 


http://www.niftyadmin.cn/n/5111530.html

相关文章

laravel框架介绍(一)

laravel框架是一套简洁&#xff0c;优雅的PHPWeb开发框架。

空中计算(Over-the-Air Computation)学习笔记

文章目录 写在前面 写在前面 本文是论文A Survey on Over-the-Air Computation的阅读笔记&#xff1a; 通信和计算通常被视为独立的任务。 从工程的角度来看&#xff0c;这种方法是非常有效的&#xff0c;因为可以执行孤立的优化。 然而&#xff0c;对于许多面向计算的应用程序…

【C语言精髓之指针】结构体指针(->与.两个运算符的区别)

/*** file * author jUicE_g2R(qq:3406291309)————彬(bin-必应)* 通信与信息专业大二在读 * copyright 2023.10* COPYRIGHT 原创技术笔记&#xff1a;转载需获得博主本人同意&#xff0c;且需标明转载源* language C/C* IDE Base on Mic…

Rust 中的String与所有权机制

文章目录 一、string二、所有权2.1 所有权与作用域2.2 对所有权的操作2.2.1 转移2.2.3 拷贝2.2.3 传递 2.3 引用2.3.1 借用2.3.2 可变引用 一、string 之前学习过 Rust 只有几种基础的数据类型&#xff0c;但是没有常用的字符串也就是String&#xff0c;今天来学习一下 String…

使用VGG框架实现从二分类到多分类

一.数据集的准备 与之前的不同&#xff0c;这一次我们不使用开源数据集&#xff0c;而是自己来制作数据集。重点需要解决的问题是对数据进行预处理&#xff0c;如每一个图片的大小均不同&#xff0c;需要进行resize&#xff0c;还需要对每一张图片打标签等操作。 数据集文件 …

【单元测试】--基础知识

一、什么是单元测试 单元测试是软件开发中的一种测试方法&#xff0c;用于验证代码中的单个组件&#xff08;通常是函数、方法或类&#xff09;是否按预期工作。它旨在隔离和测试代码的最小单元&#xff0c;以确保其功能正确&#xff0c;提高代码质量和可维护性。单元测试通常…

Java SSL/TLS证书认证逻辑

前言 最近做项目使用httpclient转发https请求&#xff0c;但是遇到一些坑&#xff0c;尤其是证书的认证&#xff0c;证书认证一般都是单向的&#xff0c;除非相互访问&#xff0c;证书一般基于host&#xff0c;但是如果访问需要ip&#xff0c;那么JDK默认的认证就会不通过&…

【面试HOT100】链表树

系列综述&#xff1a; &#x1f49e;目的&#xff1a;本系列是个人整理为了秋招面试的&#xff0c;整理期间苛求每个知识点&#xff0c;平衡理解简易度与深入程度。 &#x1f970;来源&#xff1a;材料主要源于LeetCodeHot100进行的&#xff0c;每个知识点的修正和深入主要参考…