mysql jdbc dao_MYSQL 之 JDBC（九）：增删改查（七）DAO的补充和重构

DAO重构后的代码

package com.litian.jdbc;

import org.apache.commons.beanutils.BeanUtils;

import java.sql.*;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

/**

* @author: Li Tian

* @contact: litian_cup@163.com

* @software: IntelliJ IDEA

* @file: DAO.java

* @time: 2020/3/26 18:37

* @desc: |

*/

public class DAO {

// insert, update, delete 操作都可以包含在其中

void update(String sql, Object... args) {

Connection conn = null;

PreparedStatement ps = null;

try {

conn = JDBCTools.getConnection();

ps = conn.prepareStatement(sql);

for (int i = 0; i < args.length; i++) {

ps.setObject(i + 1, args[i]);

}

ps.executeUpdate();

} catch (Exception e) {

e.printStackTrace();

} finally {

JDBCTools.release(null, ps, conn);

}

}

// 查询一条记录，返回对应的对象

T get(Class clazz, String sql, Object... args) {

List result = getForList(clazz, sql, args);

if (result.size() > 0) {

return result.get(0);

}

return null;

/* 根据下面的方法进行了改写

T entity = null;

Connection conn = null;

PreparedStatement ps = null;

ResultSet rs = null;

try {

// 1. 获取Connection

conn = JDBCTools.getConnection();

// 2. 获取PreparedStatement

ps = conn.prepareStatement(sql);

// 3. 填充占位符

for (int i = 0; i < args.length; i++) {

ps.setObject(i + 1, args[i]);

}

// 4. 进行查询，得到ResultSet

rs = ps.executeQuery();

// 5. 若ResultSet中有记录，准备一个Map: 键：存放列的别名；值：存放列的值

if (rs.next()) {

Map values = new HashMap<>();

// 6. 得到ResultSetMetaData对象

ResultSetMetaData rsmd = rs.getMetaData();

// 7. 处理ResultSet，把指针向下移动一个单位

// 8. 由ResultSetMetaData对象得到结果集中有多少列

int columnCount = rsmd.getColumnCount();

// 9. 由ResultSetMetaData得到每一列的别名，由ResultSet得到具体每一列的值

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

String columnLabel = rsmd.getColumnLabel(i + 1);

Object columnValue = rs.getObject(columnLabel);

// 10. 填充Map对象

values.put(columnLabel, columnValue);

}

// 11. 用反射创建Class对象的对象

entity = clazz.newInstance();

// 12. 遍历Map对象，用反射填充对象的属性值：属性名为Map中的Key，属性值为Map中的Value

for (Map.Entry entry : values.entrySet()) {

String propertyName = entry.getKey();

Object value = entry.getValue();

BeanUtils.setProperty(entity, propertyName, value);

}

}

} catch (Exception e) {

e.printStackTrace();

} finally {

JDBCTools.release(rs, ps, conn);

}

return entity;

*/

}

// 查询多条记录，返回对应的对象的集合

List getForList(Class clazz, String sql, Object... args) {

List list = new ArrayList<>();

Connection conn = null;

PreparedStatement ps = null;

ResultSet rs = null;

try {

// 1. 得到结果集

conn = JDBCTools.getConnection();

ps = conn.prepareStatement(sql);

for (int i = 0; i < args.length; i++) {

ps.setObject(i + 1, args[i]);

}

rs = ps.executeQuery();

// 2. 处理结果集，得到一个Map对应的List，其中一个Map对象就是一条记录，Map的Key为rs中列的别名，Value为列的值

List> values = handleResultSetToMapList(rs);

// 3. 把Map的List转为clazz对应的List，其中Map的key即clazz对应的对象的propertyName，value即为clazz对应对象的propertyValue

list = transferMapListToBeanList(clazz, values);

} catch (Exception e) {

e.printStackTrace();

} finally {

JDBCTools.release(rs, ps, conn);

}

return list;

}

public List transferMapListToBeanList(Class clazz, List> values) throws Exception {

// 12. 判断List是否为空集合，若不为空，则遍历List，得到一个个Map对象，再把一个Map对象转为一个Class参数对应的Object对象。

List result = new ArrayList();

T bean = null;

if (values.size() > 0) {

for (Map m : values) {

bean = clazz.newInstance();

for (Map.Entry entry : m.entrySet()) {

String propertyName = entry.getKey();

Object value = entry.getValue();

BeanUtils.setProperty(bean, propertyName, value);

}

result.add(bean);

}

}

return result;

}

/**

* 处理结果集，得到Map的一个List，其中一个Map对象对应一条记录

*

* @param rs

* @return

* @throws SQLException

*/

private List> handleResultSetToMapList(ResultSet rs) throws SQLException {

// 5. 若ResultSet中有记录，准备一个List>

// 键：存放列的别名；值：存放列的值，其中一个Map对象对应着一条记录

List> values = new ArrayList<>();

List columnLabels = getColumnLabels(rs);

// 7. 处理ResultSet，使用while循环

Map maps = null;

while (rs.next()) {

maps = new HashMap<>();

for (String columnLabel : columnLabels) {

Object value = rs.getObject(columnLabel);

maps.put(columnLabel, value);

}

// 11. 把一条记录的填充好的Map对象放入5准备的List中

values.add(maps);

}

return values;

}

/**

* 获取结果集ColumnLabel对应的结果集

*/

private List getColumnLabels(ResultSet rs) throws SQLException {

List labels = new ArrayList<>();

ResultSetMetaData rsmd = rs.getMetaData();

for (int i = 0; i < rsmd.getColumnCount(); i++) {

String columnLabel = rsmd.getColumnLabel(i + 1);

labels.add(columnLabel);

}

return labels;

}

// 返回某条记录的某一个字段的值或一个统计的值(一共有多少记录等。)

E getForValue(String sql, Object... args) {

Connection conn = null;

PreparedStatement ps = null;

ResultSet rs = null;

try {

// 1. 得到结果集

conn = JDBCTools.getConnection();

ps = conn.prepareStatement(sql);

for (int i = 0; i < args.length; i++) {

ps.setObject(i + 1, args[i]);

}

rs = ps.executeQuery();

// 2. 取得结果

if (rs.next()){

return (E) rs.getObject(1);

}

} catch (Exception e) {

e.printStackTrace();

} finally {

JDBCTools.release(rs, ps, conn);

}

return null;

}

}

测试

package com.litian.jdbc;

import java.sql.Date;

import java.sql.Timestamp;

import java.util.List;

/**

* @author: Li Tian

* @contact: litian_cup@163.com

* @software: IntelliJ IDEA

* @file: DAOTest.java

* @time: 2020/3/26 18:59

* @desc: |

*/

public class DAOTest {

public static void main(String[] args) {

DAO dao = new DAO();

// 测试update

// String sql = "insert into t_user(id, username, pwd, regTime, lastLoginTime) values(?,?,?,?,?)";

// dao.update(sql, 4, "李英俊", "123456", new Date(System.currentTimeMillis()), new Timestamp(System.currentTimeMillis()));

// 测试get

// String sql = "select id, username, pwd, regTime, lastLoginTime from t_user where id=?";

// User u = dao.get(User.class, sql, 4);

// System.out.println(u);

// 测试getForList

// String sql2 = "select id, username, pwd, regTime, lastLoginTime from t_user where id";

// List us = dao.getForList(User.class, sql2, 10);

// System.out.println(us);

// 测试getForValue

String sql3 = "select username from t_user where id = ?";

String cc = dao.getForValue(sql3, 1);

System.out.println(cc);

}

}