Java高效解决方案：批量导入CSV文件到MySQL数据库

这篇文是在参加CCF时导入.csv文件时自己总结的，虽然NavicatForMysql可以导入.csv文件，可是当我导入的时候不知道是文件太大还是什么原因，总是会出现失败。然后就用java写了一个批量导入数据的类去导入该.csv文件，这里也没有考虑代码的结构，只是为了快速的完成这个工作。做一个总结。

package com.cqu.price_prediction.farm;

import java.io.File;

import java.io.FileNotFoundException;

import java.sql.Connection;

import java.sql.DriverManager;

import java.sql.PreparedStatement;

import java.sql.SQLException;

import java.util.Scanner;

public class Read

{

private static Connection con;

public static void main(String[] args) throws FileNotFoundException, SQLException

{

long startTime = System.currentTimeMillis();

File file = new File("H:/AgriculturalProduct/data/farming.csv");

Scanner in = new Scanner(file);

getConnect();

System.out.println("数据库连接成功");

insert_data(in);

long EndTime = System.currentTimeMillis();

long time = (EndTime - startTime) / 1000;

System.out.println("导入数据共用时：" + time);

}

private static void insert_data(Scanner in) throws SQLException

{

int count = 0;

String sql = "insert into farming (province,market,type,name,standard,area,color,unit,minprice,avgprice,maxprice,entertime,time)"

+ "values(?,?,?,?,?,?,?,?,?,?,?,?,?)";

con.setAutoCommit(false);

PreparedStatement pstmt = con.prepareStatement(sql);

in.next();

while (in.hasNext())

{

String temp1 = in.nextLine();

String[] temp = temp1.split(",");

if (temp.length < 13)

continue;

if (temp.length == 13)

{

pstmt.setString(1, temp[0]);

pstmt.setString(2, temp[1]);

pstmt.setString(3, temp[2]);

pstmt.setString(4, temp[3]);

pstmt.setString(5, temp[4]);

pstmt.setString(6, temp[5]);

pstmt.setString(7, temp[6]);

pstmt.setString(8, temp[7]);

pstmt.setString(9, temp[8]);

pstmt.setString(10, temp[9]);

pstmt.setString(11, temp[10]);

pstmt.setString(12, temp[11]);

pstmt.setString(13, temp[12]);

}

pstmt.addBatch();

count++;

if (count == 20000)

{

count = execute(pstmt, count);

}

}

pstmt.executeBatch();

con.commit();

}

public static int execute(PreparedStatement pstmt, int count) throws SQLException

{

pstmt.executeBatch();

con.commit();

return 0;

}

private static void getConnect()

{

try

{

Class.forName("com.mysql.jdbc.Driver");

con = DriverManager.getConnection(

"jdbc:mysql://localhost:3306/agricultural_price_prediction?useUnicode=true&characterEncoding=utf8&useServerPrepStmts=false&rewriteBatchedStatements=true",

"root", "123456");

}

catch (ClassNotFoundException | SQLException e)

{

// TODO Auto-generated catch block

e.printStackTrace();

}

}

}