工作上需要将部分csv数据导入到sqlserver数据库,所以自己写了个小程序来完成。本来可以用php或C++等自己更加熟练的语言来做,但是因为自己最近在练习java,所以最终还是决定使用java。
在运行过程中发现自己没注意的问题,就是insert效率的问题。一开始我按之前的习惯,把数据库操作的部分封装成一个函数在每次读取csv一行数据的循环体内进行调用,在导入数据时发现耗时较长,后来仔细分析一下,才发现这样调用会在循环体每执行一次时,都进行一次数据库连接的打开关闭操作,从而影响运行效率,所以最终取消了数据库操作的函数。
代码如下:
import java.io.BufferedReader;
import java.io.*;
import java.util.*;
import java.io.IOException;
import java.sql.*;
public class test{
public static void main(String[] args){
csv2sql("test.csv");
}
private static void csv2sql(String filename){
try{
BufferedReader br = new BufferedReader(new FileReader(filename));
String line = null;
String connString = "jdbc:sqlserver://127.0.0.1:1433;database=test;user=sa;password=test";
Connection con = null;
Class.forName("com.microsoft.sqlserver.jdbc.SQLServerDriver");
con = DriverManager.getConnection(connStr);
while ((line=br.readLine())!=null){
String item[] = line.split(",");
String sql = "insert into table (column1,column2) values ('"+item[0]+"','"+item[1]+"')";
Statement stmt = con.createStatement();
stmt.executeUpdate(sql);
stmt.close();
}
}
catch (IOException e){
e.printStackTrace();
}
}
}