add 批量插入工具

潘钊
1 parent a2591f68
Showing 1 changed file with 37 additions and 5 deletions
src/main/java/com/bsth/util/BatchSaveUtils.java
@@ -4,6 +4,7 @@ import java.lang.reflect.Field;
 import java.sql.Connection;
 import java.sql.DriverManager;
 import java.sql.PreparedStatement;
+import java.sql.ResultSet;
 import java.util.ArrayList;
 import java.util.List;
@@ -44,7 +45,15 @@ public class BatchSaveUtils&lt;T&gt; {
 		pwd = t.getValue("spring.datasource.password");
 	}
-	public int saveListMysql(List<T> list, Class<T> clazz){
+	/**
+	 * 
+	 * @Title: saveListMysql 
+	 * @Description: TODO(批量对象入库，请自行准备好ID) 
+	 * @param @param list
+	 * @param @param clazz
+	 * @throws
+	 */
+	public int saveList(List<T> list, Class<T> clazz){
 		//获取泛型 T 的字节码
 		Table table = clazz.getAnnotation(Table.class);
 		if(null == table){
@@ -56,10 +65,12 @@ public class BatchSaveUtils&lt;T&gt; {
 		logger.info(sql);
 		//每5000条批量入库一次
+		Connection conn = null;
+		PreparedStatement ps = null;
 		try{
-			Connection conn = getConn();
+			conn = getConn();
 			conn.setAutoCommit(false);
-			PreparedStatement ps = conn.prepareStatement(sql);
+			ps = conn.prepareStatement(sql);
 			int fsize = fs.size(), count = 0;
 			for(T t : list){
@@ -68,15 +79,20 @@ public class BatchSaveUtils&lt;T&gt; {
 					ps.setObject(i + 1, fs.get(i).get(t));
 				}
+				ps.addBatch();
 				if(count % batchSize == 0){
 					ps.executeBatch();
+					conn.commit();
                     ps.clearBatch();
 				}
 			}
 			ps.executeBatch();
+			conn.commit();
 		}catch(Exception e){
 			logger.error("",e);
 			return -1;
+		}finally {
+			closeAll(conn, ps, null);
 		}
 		return 0;
@@ -84,7 +100,7 @@ public class BatchSaveUtils&lt;T&gt; {
 	public String createSql(Table table, List<Field> fs){
 		String sqlBefore = "insert into " + table.name() + "("
-				,sqlValues = "(";
+				,sqlValues = " values(";
 		for(Field field : fs){
 			sqlBefore += (propertyToField(field.getName()) + ",");
 			sqlValues += "?,";
@@ -98,7 +114,7 @@ public class BatchSaveUtils&lt;T&gt; {
 	public static void main(String[] args) {
 		List<ScheduleRealInfo> list = new ArrayList<>();
-		new BatchSaveUtils<ScheduleRealInfo>().saveListMysql(list, ScheduleRealInfo.class);
+		new BatchSaveUtils<ScheduleRealInfo>().saveList(list, ScheduleRealInfo.class);
 	}
 	/**
@@ -141,4 +157,20 @@ public class BatchSaveUtils&lt;T&gt; {
 		Connection conn = DriverManager.getConnection(url,uname,pwd);
 		return conn;
 	}
+	
+	public static void closeAll(Connection conn, PreparedStatement ps, ResultSet rs){
+		try {
+			if(conn != null){
+				conn.close();
+			}
+			if(ps != null){
+				ps.close();
+			}
+			if(rs != null){
+				rs.close();
+			}
+		} catch (Exception e) {
+			e.printStackTrace();
+		}
+	}
 }