向mysql数据库快速插入两万条数据库

时间:2024-04-06 12:53:37

向mysql数据库快速插入两万条数据库

package test.java;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.util.Date;

import com.topband.cloud.common.utils.IdGen;

public class test {

    public static void main(String[] args) throws Exception {
        final String url = "jdbc:mysql://127.0.0.1/myhome";
        final String name = "com.mysql.jdbc.Driver";
        final String user = "root";
        final String password = "root";
        Connection conn = null;
        Class.forName(name);//指定连接类型
        conn = DriverManager.getConnection(url, user, password);//获取连接
        if (conn!=null) {
            System.out.println("获取连接成功");
            insert(conn);
        }else {
            System.out.println("获取连接失败");
        }
    }
    
    public static void insert(Connection conn) {
        // 开始时间
        Long begin = new Date().getTime();
        // sql前缀
        String prefix = "INSERT INTO sheep_inside_code (id,org_id) VALUES ";
        try {
            // 保存sql后缀
            StringBuffer suffix = new StringBuffer();
            // 设置事务为非自动提交
            conn.setAutoCommit(false);
            // 比起st,pst会更好些
            PreparedStatement pst = (PreparedStatement) conn.prepareStatement("");//准备执行语句
            // 外层循环,总提交事务次数
            for (int i = 1; i <= 5; i++) {
                suffix = new StringBuffer();
                // 第j次提交步长
                if(i == 1) {
                    for (int j = 1; j < 10; j++) {
                        // 构建SQL后缀
                        suffix.append("('" + IdGen.uuid()+"'," +"'e2e0374d570442d9a3db0c99d6d3f72e"+"'),");
                    }
                }else if(i == 2) {
                    for (int j = 10; j < 100; j++) {
                        // 构建SQL后缀
                        suffix.append("('" + IdGen.uuid()+"'," +"'e2e0374d570442d9a3db0c99d6d3f72e"+"'),");
                    }
                }else if(i == 3) {
                    for (int j = 100; j < 1000; j++) {
                        // 构建SQL后缀
                        suffix.append("('" + IdGen.uuid()+"'," +"'e2e0374d570442d9a3db0c99d6d3f72e"+"'),");
                    }
                }else if(i == 4) {
                    for (int j = 1000; j < 10000; j++) {
                        // 构建SQL后缀
                        suffix.append("('" + IdGen.uuid()+"'," +"'e2e0374d570442d9a3db0c99d6d3f72e"+"'),");
                    }
                }else {
                    for (int j = 10000; j <= 20000; j++) {
                        // 构建SQL后缀
                        suffix.append("('" + IdGen.uuid()+"'," +"'e2e0374d570442d9a3db0c99d6d3f72e"+"'),");
                    }
                }


                // 构建完整SQL
                String sql = prefix + suffix.substring(0, suffix.length() - 1);
                // 添加执行SQL
                pst.addBatch(sql);
                // 执行操作
                pst.executeBatch();
                // 提交事务
                conn.commit();
                // 清空上一次添加的数据
                suffix = new StringBuffer();
            }
            // 头等连接
            pst.close();
            conn.close();
        } catch (SQLException e) {
            e.printStackTrace();
        }
        // 结束时间
        Long end = new Date().getTime();
        // 耗时
        System.out.println("2万条数据插入花费时间 : " + (end - begin) / 1000 + " s");
        System.out.println("插入完成");
    }
}