请帮我实现使用多线程进行数据库插入大量数据的java代码

时间: 2024-05-10 20:18:37 浏览: 10
好的,以下是使用多线程进行数据库插入大量数据的Java代码示例: ```java import java.sql.Connection; import java.sql.DriverManager; import java.sql.PreparedStatement; import java.sql.SQLException; import java.util.ArrayList; import java.util.List; import java.util.concurrent.ExecutorService; import java.util.concurrent.Executors; public class InsertDataMultiThread { private static final String DB_DRIVER = "com.mysql.jdbc.Driver"; private static final String DB_CONNECTION = "jdbc:mysql://localhost:3306/mydatabase"; private static final String DB_USER = "root"; private static final String DB_PASSWORD = "password"; private static final int THREAD_POOL_SIZE = 10; private static final int BATCH_SIZE = 1000; public static void main(String[] args) { Connection dbConnection = null; try { // 获取数据库连接 dbConnection = getDBConnection(); // 设置自动提交为 false dbConnection.setAutoCommit(false); // 创建一个线程池 ExecutorService executor = Executors.newFixedThreadPool(THREAD_POOL_SIZE); // 创建一个 PreparedStatement 对象 PreparedStatement preparedStatement = dbConnection.prepareStatement("INSERT INTO mytable (col1, col2, col3) VALUES (?, ?, ?)"); // 构造数据列表 List<Data> dataList = new ArrayList<>(); for (int i = 0; i < 1000000; i++) { Data data = new Data("value1", "value2", "value3"); dataList.add(data); } // 将数据列表分割成多个批次 List<List<Data>> batches = splitIntoBatches(dataList, BATCH_SIZE); // 依次将每个批次提交到线程池中执行 for (List<Data> batch : batches) { Runnable task = new InsertDataTask(preparedStatement, batch); executor.execute(task); } // 关闭线程池 executor.shutdown(); // 等待所有任务执行完毕 while (!executor.isTerminated()) {} // 提交事务 dbConnection.commit(); System.out.println("Data inserted successfully."); } catch (SQLException e) { System.err.println(e.getMessage()); } finally { try { if (dbConnection != null) { // 恢复自动提交 dbConnection.setAutoCommit(true); // 关闭数据库连接 dbConnection.close(); } } catch (SQLException e) { System.err.println(e.getMessage()); } } } private static Connection getDBConnection() throws SQLException { Connection dbConnection = null; try { Class.forName(DB_DRIVER); } catch (ClassNotFoundException e) { System.err.println(e.getMessage()); } dbConnection = DriverManager.getConnection(DB_CONNECTION, DB_USER, DB_PASSWORD); return dbConnection; } private static <T> List<List<T>> splitIntoBatches(List<T> list, int batchSize) { List<List<T>> batches = new ArrayList<>(); int size = list.size(); for (int i = 0; i < size; i += batchSize) { batches.add(list.subList(i, Math.min(size, i + batchSize))); } return batches; } private static class InsertDataTask implements Runnable { private PreparedStatement preparedStatement; private List<Data> dataList; public InsertDataTask(PreparedStatement preparedStatement, List<Data> dataList) { this.preparedStatement = preparedStatement; this.dataList = dataList; } @Override public void run() { try { for (Data data : dataList) { preparedStatement.setString(1, data.getCol1()); preparedStatement.setString(2, data.getCol2()); preparedStatement.setString(3, data.getCol3()); preparedStatement.addBatch(); } preparedStatement.executeBatch(); } catch (SQLException e) { System.err.println(e.getMessage()); } } } private static class Data { private String col1; private String col2; private String col3; public Data(String col1, String col2, String col3) { this.col1 = col1; this.col2 = col2; this.col3 = col3; } public String getCol1() { return col1; } public void setCol1(String col1) { this.col1 = col1; } public String getCol2() { return col2; } public void setCol2(String col2) { this.col2 = col2; } public String getCol3() { return col3; } public void setCol3(String col3) { this.col3 = col3; } } } ``` 在这个示例代码中,我们使用了线程池来并发执行大量的数据库插入操作。每个线程负责将一个批次的数据插入到数据库中。我们将数据列表分割成多个批次,每个批次的大小为 BATCH_SIZE,然后依次将每个批次提交到线程池中执行。在每个线程中,我们使用 PreparedStatement 对象来执行批量插入操作,可以有效地减少数据库连接和交互的次数。当所有任务执行完毕后,我们提交事务并关闭数据库连接。 如果你想要使用这个示例代码,需要将 DB_CONNECTION、DB_USER 和 DB_PASSWORD 替换成你自己的数据库连接信息,同时需要创建一个名为 mytable 的表,并且包含三个文本列 col1、col2 和 col3。

相关推荐

最新推荐

recommend-type

java使用influxDB数据库的详细代码

主要为大家介绍了java使用influxDB数据库的详细代码,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
recommend-type

mybatis 批量将list数据插入到数据库的实现

主要介绍了mybatis 批量将list数据插入到数据库的实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
recommend-type

Java实现excel大数据量导入

主要为大家详细介绍了Java实现excel大数据量导入,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
recommend-type

Java实验8 数据库.doc

本专栏主要为Java程序设计(基础)实验报告和Java程序设计(进阶)...进阶篇有反射、泛型、注解、网络编程、多线程、序列化、数据库、Servlet、JSP、XML解析、单例模式与枚举。本专栏主要为Java入门者提供实验参考。
recommend-type

zigbee-cluster-library-specification

最新的zigbee-cluster-library-specification说明文档。
recommend-type

管理建模和仿真的文件

管理Boualem Benatallah引用此版本:布阿利姆·贝纳塔拉。管理建模和仿真。约瑟夫-傅立叶大学-格勒诺布尔第一大学,1996年。法语。NNT:电话:00345357HAL ID:电话:00345357https://theses.hal.science/tel-003453572008年12月9日提交HAL是一个多学科的开放存取档案馆,用于存放和传播科学研究论文,无论它们是否被公开。论文可以来自法国或国外的教学和研究机构,也可以来自公共或私人研究中心。L’archive ouverte pluridisciplinaire
recommend-type

【实战演练】MATLAB用遗传算法改进粒子群GA-PSO算法

![MATLAB智能算法合集](https://static.fuxi.netease.com/fuxi-official/web/20221101/83f465753fd49c41536a5640367d4340.jpg) # 2.1 遗传算法的原理和实现 遗传算法(GA)是一种受生物进化过程启发的优化算法。它通过模拟自然选择和遗传机制来搜索最优解。 **2.1.1 遗传算法的编码和解码** 编码是将问题空间中的解表示为二进制字符串或其他数据结构的过程。解码是将编码的解转换为问题空间中的实际解的过程。常见的编码方法包括二进制编码、实数编码和树形编码。 **2.1.2 遗传算法的交叉和
recommend-type

openstack的20种接口有哪些

以下是OpenStack的20种API接口: 1. Identity (Keystone) API 2. Compute (Nova) API 3. Networking (Neutron) API 4. Block Storage (Cinder) API 5. Object Storage (Swift) API 6. Image (Glance) API 7. Telemetry (Ceilometer) API 8. Orchestration (Heat) API 9. Database (Trove) API 10. Bare Metal (Ironic) API 11. DNS
recommend-type

JSBSim Reference Manual

JSBSim参考手册,其中包含JSBSim简介,JSBSim配置文件xml的编写语法,编程手册以及一些应用实例等。其中有部分内容还没有写完,估计有生之年很难看到完整版了,但是内容还是很有参考价值的。
recommend-type

"互动学习:行动中的多样性与论文攻读经历"

多样性她- 事实上SCI NCES你的时间表ECOLEDO C Tora SC和NCESPOUR l’Ingén学习互动,互动学习以行动为中心的强化学习学会互动,互动学习,以行动为中心的强化学习计算机科学博士论文于2021年9月28日在Villeneuve d'Asq公开支持马修·瑟林评审团主席法布里斯·勒菲弗尔阿维尼翁大学教授论文指导奥利维尔·皮耶昆谷歌研究教授:智囊团论文联合主任菲利普·普雷教授,大学。里尔/CRISTAL/因里亚报告员奥利维耶·西格德索邦大学报告员卢多维奇·德诺耶教授,Facebook /索邦大学审查员越南圣迈IMT Atlantic高级讲师邀请弗洛里安·斯特鲁布博士,Deepmind对于那些及时看到自己错误的人...3谢谢你首先,我要感谢我的两位博士生导师Olivier和Philippe。奥利维尔,"站在巨人的肩膀上"这句话对你来说完全有意义了。从科学上讲,你知道在这篇论文的(许多)错误中,你是我可以依