JDBC反射批量插入数据(只需传入表名)

备注:1、实体类需要实现 Serializable接口,并且将生成 serialVersionUID 放在第一行;

           2、实体类中属性的顺序需和数据库保持一致;

           3、实体类中若需新增属性必须在原先实体类属性的末尾添加;


package com.dz;

import java.lang.reflect.Field;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.UUID;

/**
 * @Description:JDBC批量插入反射操作数据库
 * @Date 2019年11月2日 下午1:49:46 @Author:DZ Copyright (c) 2019
 */
public class JdbcUtil2 {
	public static final String DRIVER = "com.mysql.cj.jdbc.Driver"; // jdbc驱动
	public static final String URL = "jdbc:mysql://localhost:3306/test?&useSSL=false&serverTimezone=UTC";
	public static final String USER = "root";
	public static final String PWD = "root";

	public static final String BATCH_COUNT = "11";// 批量插入的数据量

	/**
	 * @Description:进行数据库的链接
	 * @return
	 * @Date 2019年11月2日 下午1:50:11 @Author:DZ
	 */
	public static Connection getConnection() {
		Connection conn = null;
		try {

			Class.forName(DRIVER);// 加载驱动

			conn = DriverManager.getConnection(URL, USER, PWD);// 创建链接

			if (!conn.isClosed()) {
				System.out.println("数据库连接成功");
			}

		} catch (ClassNotFoundException e) {
			System.out.println("数据库驱动没有安装");
			e.printStackTrace();
		} catch (SQLException e) {
			System.out.println("数据库连接失败");
			e.printStackTrace();
		}

		return conn;// 返回连接
	}

	/**
	 * @Description:关闭所有
	 * @param conn
	 * @param ps
	 * @param rs
	 * @Date 2019年11月2日 下午1:51:03 @Author:DZ
	 */
	public static void closeAll(Connection conn, PreparedStatement ps, ResultSet rs) {
		try {
			// 判断是否被操作
			if (rs != null) {
				rs.close();
			}
			if (ps != null) {
				ps.close();
			}
			if (conn != null) {
				conn.close();
			}
		} catch (SQLException e) {
			e.printStackTrace();
		}
	}

	/**
	 * @Description:关闭
	 * @param conn
	 * @param ps
	 * @Date 2019年11月2日 下午1:51:17 @Author:DZ
	 */
	public static void close(Connection conn, PreparedStatement ps) {
		try {
			// 判断是否被操作
			if (ps != null) {
				ps.close();
			}
			if (conn != null) {
				conn.close();
			}
		} catch (SQLException e) {
			e.printStackTrace();
		}
	}

	/**
	 * @Description:获取表的所有列
	 * @param tableName
	 * @return
	 * @Date 2019年11月2日 下午1:51:40 @Author:DZ
	 */
	public static List<String> getColumns(String tableName) {
		List<String> columnList = new ArrayList<String>();// 存放获取到的列集合
		Connection conn = JdbcUtil2.getConnection();// 获取数据库连接
		PreparedStatement ps = null;
		ResultSet rs = null;
		ResultSetMetaData rsmd = null;
		try {
			ps = conn.prepareStatement("SELECT * FROM " + tableName);// 查询数据库边表
			rs = ps.executeQuery();
			rsmd = rs.getMetaData(); // 获取字段名
			if (rsmd != null) {

				int count = rsmd.getColumnCount();

				for (int i = 1; i <= count; i++) {
					columnList.add(rsmd.getColumnName(i));
				}

			}

		} catch (SQLException e) {
			e.printStackTrace();
		} catch (Exception e) {
			e.printStackTrace();

		} finally {
			closeAll(conn, ps, rs);
		}
		return columnList;
	}

	/**
	 * @Description:获取数据库字段数量
	 * @param tableName
	 * @return
	 * @Date 2019年11月2日 下午2:02:45 @Author:DZ
	 */
	public static int getColumnCount(String tableName) {
		Connection conn = JdbcUtil2.getConnection();// 获取数据库连接
		PreparedStatement ps = null;
		ResultSet rs = null;
		ResultSetMetaData rsmd = null;
		int count = 0;
		try {
			ps = conn.prepareStatement("SELECT * FROM " + tableName);// 查询数据库边表
			rs = ps.executeQuery();
			rsmd = rs.getMetaData(); // 获取字段名
			if (rsmd != null) {
				count = rsmd.getColumnCount();
			}

		} catch (SQLException e) {
			e.printStackTrace();
		} catch (Exception e) {
			e.printStackTrace();

		} finally {
			closeAll(conn, ps, rs);
		}
		return count;
	}

	/**
	 * @Description:创建所需插入SQL
	 * @param TableName
	 * @return
	 * @Date 2019年11月2日 下午1:58:54 @Author:DZ
	 */
	public static String createSql(String TableName) {
		int columnCount = getColumnCount(TableName);
		StringBuffer buffer = new StringBuffer();

		for (int i = 0; i < columnCount; i++) {

			buffer.append("?");
			if (i < (columnCount - 1)) {
				buffer.append(",");
			}

		}
		String sql = "INSERT INTO " + TableName + " values (" + buffer + ")";
		return sql;
	}

	/**
	 * @Description:反射执行批量插入
	 * @param <T>
	 * @param objects
	 * @param sql
	 * @Date 2019年11月2日 下午1:52:11 @Author:DZ
	 */
	public static <T> void saveBatch(List<T> objects, String tableName) {
		long startTime = System.currentTimeMillis();
		String sql = createSql(tableName);
		Connection conn = JdbcUtil2.getConnection();
		PreparedStatement ps = null;
		int num = 0;// 数据插入的计数器
		int count = 0;// 批量插入的批次数
		try {
			List<String> columns = getColumns(tableName);// 表中的字段
			conn.setAutoCommit(false); // 设置手动提交
			ps = conn.prepareStatement(sql);
			for (int i = 0; i < objects.size(); i++) {
				num++;
				Object obj = objects.get(i);// 获取List中的每一个对象
				Class<?> c = obj.getClass();// 反射形成实际对象
				Field[] fields = c.getDeclaredFields();// 通过反射获取对象的属性数组
				for (int k = 0; k < columns.size(); k++) {
					String dataSourceColumn = columns.get(k);
					for (int j = 0; j < fields.length; j++) {
						Field field = fields[j];// 拿到每一个属性
						field.setAccessible(true);// 设置些属性是可以访问的
						String fieldName = field.getName();

						if (fieldName.contentEquals(dataSourceColumn.replace("_", ""))) {// 数据库表中的字段和实体类中的字段比较
							String type = field.getType().toString();// 得到此属性的类型
							if (type.endsWith("String")) {
								ps.setString(j, String.valueOf(field.get(obj)));
								break;
							} else if (type.endsWith("Date")) {
								ps.setDate(j, new java.sql.Date(((Date) field.get(obj)).getTime()));
								break;
							} else {
								try {
									throw new Exception("反射异常!");
								} catch (Exception e) {
									e.printStackTrace();
								}
							}
						}
					}

				}
				ps.addBatch();// 添加到批次

				if (num > Integer.valueOf(BATCH_COUNT) - 1) {
					count++;// 计算执行的批次数
					ps.executeBatch();// 提交批处理
					conn.commit();// 执行
					ps.clearBatch();// 清除
					System.out.println(num + " 条插入完成! ");
					System.gc();// 提示释放GC
					num = 0;
				}
			}
			ps.executeBatch();// 提交批处理
			System.out.println("插入最后 " + (objects.size() - count * Integer.valueOf(BATCH_COUNT) + "条成功!"));
			conn.commit();// 执行
			ps.clearBatch();

			long endTime = System.currentTimeMillis();
			System.out.println("耗时:" + (endTime - startTime) / 1000 + "s");
		} catch (SQLException e) {
			e.printStackTrace();
		} catch (IllegalAccessException e) {
			e.printStackTrace();
		} finally {
			close(conn, ps);// 关闭连接
		}

	}

	/**
	 * @Description:获取UUID
	 * @return
	 * @Date 2019年11月2日 下午2:52:12 @Author:DZ
	 */
	public static String getUUID() {
		String uuid = UUID.randomUUID().toString().replaceAll("-", "").toUpperCase();
		return uuid;
	}

	/**
	 * @Description:创建测试所需实体对象
	 * @return
	 * @Date 2019年11月2日 下午3:16:38 @Author:DZ
	 */
	public static List<User> CreateVo() {
		List<User> userList = new ArrayList<User>();

		for (int i = 0; i < 100; i++) {
			User user = new User(getUUID(), "xiaohua", "15", new Date());
			userList.add(user);
		}
		return userList;
	}

	public static void main(String[] args) {
		List<User> userList = CreateVo();
		JdbcUtil2.saveBatch(userList, "user");
		System.out.println("测试完成!");
	}
}

实体类:


package com.dz;

import java.io.Serializable;
import java.util.Date;

/**
 * @Description:用户实体类 
 * @Date 2019年11月2日 下午3:23:18 @Author:DZ 
 * Copyright (c) 2019 
 */
public class User implements Serializable {

	private static final long serialVersionUID = -6297509897955271853L;

	private String id;
	private String name;
	private String age;
	private Date date;
	private String version;
	private String pageSize;
	private String pageIndex;

	public User(String id, String name, String age, Date date) {
		this.id = id;
		this.name = name;
		this.age = age;
		this.date = date;
	}

	public User(String name, String age, Date date) {
		this.name = name;
		this.age = age;
		this.date = date;
	}

	public String getPageSize() {
		return pageSize;
	}

	public void setPageSize(String pageSize) {
		this.pageSize = pageSize;
	}

	public String getPageIndex() {
		return pageIndex;
	}

	public void setPageIndex(String pageIndex) {
		this.pageIndex = pageIndex;
	}

	public String getVersion() {
		return version;
	}

	public void setVersion(String version) {
		this.version = version;
	}

	public String getId() {
		return id;
	}

	public void setId(String id) {
		this.id = id;
	}

	public String getName() {
		return name;
	}

	public void setName(String name) {
		this.name = name;
	}

	public String getAge() {
		return age;
	}

	public void setAge(String age) {
		this.age = age;
	}

	public Date getDate() {
		return date;
	}

	public void setDate(Date date) {
		this.date = date;
	}
}

数据库脚本:


Mysql建表脚本
SET FOREIGN_KEY_CHECKS=0;
 
-- ----------------------------
-- Table structure for user
-- ----------------------------
DROP TABLE IF EXISTS `user`;
CREATE TABLE `user` (
  `id` int(11) NOT NULL AUTO_INCREMENT,
  `name` varchar(255) DEFAULT NULL,
  `age` varchar(11) DEFAULT NULL,
  `date` date DEFAULT NULL,
  PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=301 DEFAULT CHARSET=utf8;

 

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值