使用PreparedStatement实现批量数据的操作

使用PreparedStatement实现批量数据的操作

package com.sammery.Blob;

import java.sql.Connection;
import java.sql.PreparedStatement;
import org.junit.Test;

import com.sammery.util.JDBCUtils;

/*
 * 使用PreparedStatement实现批量数据的操作
 * 
 * update、delete本身具有批量操作的效果。
 * 此时的批量操作，主要指的是批量插入。使用PreparedStatement如何实现更高效的批量插入？
 * 
 * 题目：想goods表中插入20000条数据
 *  CREATE TABLE goods (
	id INT PRIMARY KEY AUTO_INCREMENT,
	`name` VARCHAR(25)
	);
 * 方式一：使用Statement
 * 	Connection connection = JDBCUtils.getConnection();
	Statement statement = connection.createStatement();
	for (int i = 0; i <= 20000; i++) {
		String sql = "INSERT INTO goods(`name`) VALUES(`name_"+i+"`)";
		statement.execute(sql);
	}
 * */

public class InsertTest {
	//方式二：使用PreparedStatement批量插入数据
	//20000条运行时间：60374毫秒
	@Test
	public void testInsert2() throws Exception{
		Connection connection = null;
		PreparedStatement ps = null;
		try {
			long start = System.currentTimeMillis();
			
			connection = JDBCUtils.getConnection();
			String sql = "INSERT INTO goods(`name`) VALUES(?)";
			ps = connection.prepareStatement(sql);
			for (int i = 1; i <= 20000; i++) {
				ps.setObject(1, "name_"+i);
				ps.execute();
			}
			long end = System.currentTimeMillis();
			
			System.out.println("运行时间："+(end-start)+"毫秒");
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			JDBCUtils.closeResource(connection, ps);
		}
	}
	
	
	/*
	 * 批量操作方式三：
	 * 1.addBatch()、executeBatch()、clearBatch()
	 * 2.mysql服务器默认是关闭批处理的，我们需要通过一个参数，让MySQL开启批处理的支持。
	 * 		?rewriteBatchedStatements=true 写在配置文件的url后面
	 * 3.使用更新的MySQL驱动：mysql-connector-java-5.1.37-bin.jar
	 * 
	 * 20000运行时间：731毫秒
	 * 1000000运行时间：13088毫秒
	 * */
	@Test
	public void testInsert3() throws Exception{
		Connection connection = null;
		PreparedStatement ps = null;
		try {
			long start = System.currentTimeMillis();
			
			connection = JDBCUtils.getConnection();
			String sql = "INSERT INTO goods(`name`) VALUES(?)";
			ps = connection.prepareStatement(sql);
			for (int i = 1; i <= 1000000; i++) {
				ps.setObject(1, "name_"+i);
				
				//1."攒"sql
				ps.addBatch();
				if(i % 500 == 0){
					//2.攒够500条执行
					ps.executeBatch();
					
					//3.清空Batch
					ps.clearBatch();
				}
			}
			long end = System.currentTimeMillis();
			
			System.out.println("运行时间："+(end-start)+"毫秒");
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			JDBCUtils.closeResource(connection, ps);
		}
	}
	
	/*
	 * 批量操作方式四：设置连接不允许自动提交数据
	 * 1000000运行时间：6241毫秒
	 * */
	@Test
	public void testInsert4() throws Exception{
		Connection connection = null;
		PreparedStatement ps = null;
		try {
			long start = System.currentTimeMillis();
			
			connection = JDBCUtils.getConnection();
			
			//设置不允许自动提交数据
			connection.setAutoCommit(false);
			
			String sql = "INSERT INTO goods(`name`) VALUES(?)";
			ps = connection.prepareStatement(sql);
			for (int i = 1; i <= 1000000; i++) {
				ps.setObject(1, "name_"+i);
				
				//1."攒"sql
				ps.addBatch();
				if(i % 500 == 0){
					//2.攒够500条执行
					ps.executeBatch();
					
					//3.清空Batch
					ps.clearBatch();
				}
			}
			
			//提交数据
			connection.commit();
			
			long end = System.currentTimeMillis();
			
			System.out.println("运行时间："+(end-start)+"毫秒");
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			JDBCUtils.closeResource(connection, ps);
		}
	}

}