JDBC:DAO(二)DAO实现增删改查(CRUD)

目前这个connection是独立获取的，以后要加入事务及连接池就使用参数传入

public class DAO {

    // INSERT, UPDATE, DELETE 操作都可以包含在其中
    public void update(String sql, Object... args) {
        Connection connection = null;
        PreparedStatement preparedStatement = null;

        try {
            connection = JDBCTools.getConnection();
            preparedStatement = connection.prepareStatement(sql);

            for (int i = 0; i < args.length; i++) {
                preparedStatement.setObject(i + 1, args[i]);
            }

            preparedStatement.executeUpdate();
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            JDBCTools.releaseDB(null, preparedStatement, connection);
        }
    }

    // 查询一条记录, 返回对应的对象
    public <T> T get(Class<T> clazz, String sql, Object... args) {
        List<T> result = getForList(clazz, sql, args);
        if(result.size() > 0){
            return result.get(0);
        }
        
        return null;
    }
    ///熊诗言修改的，没必要弄的那么麻烦，还整个List<Map<String, Object>>这么复杂///
     /**
     * 传入 SQL 语句和 Class 对象, 返回 SQL 语句查询到的记录对应的 Class 类的对象的集合
     * @param clazz: 对象的类型
     * @param sql: SQL 语句
     * @param args: 填充 SQL 语句的占位符的可变参数.
     * @return
     */
    public <T> List<T> getObjects(Class<T> clazz,
            String sql, Object... args) {

        List<T> list = new ArrayList<>();

        Connection connection = null;
        PreparedStatement preparedStatement = null;
        ResultSet resultSet = null;

        try {
            //1. 得到结果集
            connection = JDBCTools.getConnection();
            preparedStatement = connection.prepareStatement(sql);

            for (int i = 0; i < args.length; i++) {
                preparedStatement.setObject(i + 1, args[i]);
            }

            resultSet = preparedStatement.executeQuery();
            
            List<String> columnLabels = getColumnLabels(resultSet);
            
            T bean = null;
            while(resultSet.next()){
                bean = clazz.newInstance();
                for (String columnLabel : columnLabels) {
                    Object value = resultSet.getObject(columnLabel);
                    BeanUtils.setProperty(bean, columnLabel, value);
                }
                list.add(bean);
            }

        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            JDBCTools.releaseDB(resultSet, preparedStatement, connection);
        }

        return list;
    }
    ///熊诗言修改的，没必要弄的那么麻烦，还整个List<Map<String, Object>>这么复杂///
    /**
     * 传入 SQL 语句和 Class 对象, 返回 SQL 语句查询到的记录对应的 Class 类的对象的集合
     * @param clazz: 对象的类型
     * @param sql: SQL 语句
     * @param args: 填充 SQL 语句的占位符的可变参数.
     * @return
     */
    public <T> List<T> getForList(Class<T> clazz,
            String sql, Object... args) {

        List<T> list = new ArrayList<>();

        Connection connection = null;
        PreparedStatement preparedStatement = null;
        ResultSet resultSet = null;

        try {
            //1. 得到结果集
            connection = JDBCTools.getConnection();
            preparedStatement = connection.prepareStatement(sql);

            for (int i = 0; i < args.length; i++) {
                preparedStatement.setObject(i + 1, args[i]);
            }

            resultSet = preparedStatement.executeQuery();
            
            //2. 处理结果集, 得到 Map 的 List, 其中一个 Map 对象
            //就是一条记录. Map 的 key 为 reusltSet 中列的别名, Map 的 value
            //为列的值.
            List<Map<String, Object>> values =
                    handleResultSetToMapList(resultSet);
            
            //3. 把 Map 的 List 转为 clazz 对应的 List
            //其中 Map 的 key 即为 clazz 对应的对象的 propertyName,
            //而 Map 的 value 即为 clazz 对应的对象的 propertyValue
            list = transfterMapListToBeanList(clazz, values);

        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            JDBCTools.releaseDB(resultSet, preparedStatement, connection);
        }

        return list;
    }

    public <T> List<T> transfterMapListToBeanList(Class<T> clazz,
            List<Map<String, Object>> values) throws InstantiationException,
            IllegalAccessException, InvocationTargetException {

        List<T> result = new ArrayList<>();

        T bean = null;

        if (values.size() > 0) {
            for (Map<String, Object> m : values) {
                bean = clazz.newInstance();
                for (Map.Entry<String, Object> entry : m.entrySet()) {
                    String propertyName = entry.getKey();
                    Object value = entry.getValue();

                    BeanUtils.setProperty(bean, propertyName, value);
                }
                // 13. 把 Object 对象放入到 list 中.
                result.add(bean);
            }
        }

        return result;
    }

    /**
     * 处理结果集, 得到 Map 的一个 List, 其中一个 Map 对象对应一条记录
     *
     * @param resultSet
     * @return
     * @throws SQLException
     */
    public List<Map<String, Object>> handleResultSetToMapList(
            ResultSet resultSet) throws SQLException {
        // 5. 准备一个 List<Map<String, Object>>:
        // 键: 存放列的别名, 值: 存放列的值. 其中一个 Map 对象对应着一条记录
        List<Map<String, Object>> values = new ArrayList<>();

        List<String> columnLabels = getColumnLabels(resultSet);
        Map<String, Object> map = null;

        // 7. 处理 ResultSet, 使用 while 循环
        while (resultSet.next()) {
            map = new HashMap<>();

            for (String columnLabel : columnLabels) {
                Object value = resultSet.getObject(columnLabel);
                map.put(columnLabel, value);
            }

            // 11. 把一条记录的一个 Map 对象放入 5 准备的 List 中
            values.add(map);
        }
        return values;
    }

    /**
     * 获取结果集的 ColumnLabel 对应的 List
     *
     * @param rs
     * @return
     * @throws SQLException
     */
    private List<String> getColumnLabels(ResultSet rs) throws SQLException {
        List<String> labels = new ArrayList<>();

        ResultSetMetaData rsmd = rs.getMetaData();
        for (int i = 0; i < rsmd.getColumnCount(); i++) {
            labels.add(rsmd.getColumnLabel(i + 1));
        }

        return labels;
    }

    // 返回某条记录的某一个字段的值 或 一个统计的值(一共有多少条记录等.)
    public <E> E getForValue(String sql, Object... args) {
        
        //1. 得到结果集: 该结果集应该只有一行, 且只有一列
        Connection connection = null;
        PreparedStatement preparedStatement = null;
        ResultSet resultSet = null;

        try {
            //1. 得到结果集
            connection = JDBCTools.getConnection();
            preparedStatement = connection.prepareStatement(sql);

            for (int i = 0; i < args.length; i++) {
                preparedStatement.setObject(i + 1, args[i]);
            }

            resultSet = preparedStatement.executeQuery();
            
            if(resultSet.next()){
                return (E) resultSet.getObject(1);
            }
        } catch(Exception ex){
            ex.printStackTrace();
        } finally{
            JDBCTools.releaseDB(resultSet, preparedStatement, connection);
        }
        //2. 取得结果
        
        return null;
    }

}

单元测试类

public class DAOTest {

    DAO dao = new DAO();
    
    @Test
    public void testUpdate() {
        String sql = "INSERT INTO customers(name, " +
                "email, birth) VALUES(?,?,?)";
        dao.update(sql, "XiaoMing", "xiaoming@atguigu.com",
                new Date(new java.util.Date().getTime()));
    }

    @Test
    public void testGet() {
        String sql = "SELECT flow_id flowId, type, exam_card examCard, " +
                "id_card idCard, student_name studentName, location, " +
                "grade FROM examstudent WHERE flow_id = ?";
        Student student = dao.get(Student.class, sql, 5);
        
        System.out.println(student);
    }

    @Test
    public void testGetForList() {
        String sql = "SELECT flow_id flowId, type, exam_card examCard, " +
                "id_card idCard, student_name studentName, location, " +
                "grade FROM examstudent";
        
        List<Student> students = dao.getForList(Student.class, sql);
        System.out.println(students);
    }

    @Test
    public void testGetForValue() {
        String sql = "SELECT exam_card FROM examstudent " +
                "WHERE flow_id = ?";
        String examCard = dao.getForValue(sql, 5);
        System.out.println(examCard);
        
        sql = "SELECT max(grade) FROM examstudent";
        int grade = dao.getForValue(sql);
        System.out.println(grade);
    }

}