mybatis

描述

在《MyBatis手寫框架（二）》中，我們已經(jīng)將數(shù)據(jù)庫連接信息轉(zhuǎn)移到配置文件里，另外，我們還把獲取連接的方式修改為使用數(shù)據(jù)源去管理連接。
現(xiàn)在，我們要使 SQL 語句的執(zhí)行及返回更為通用，而不是固定一個(gè)方法面對(duì)一個(gè)業(yè)務(wù)問題。

    @Test
    public void testSelectUser() {
        
        Connection connection = null;
        PreparedStatement preparedStatement = null;
        ResultSet rs = null;

        try {
                        ...
            String sql = "select * from user_info where name = ?";

            preparedStatement = connection.prepareStatement(sql);
            preparedStatement.setString(1, "tester");

            // 向數(shù)據(jù)庫發(fā)出 sql 執(zhí)行查詢，查詢出結(jié)果集
            rs = preparedStatement.executeQuery();

            // 遍歷查詢結(jié)果集
            while (rs.next()) {
                System.out.println(rs.getString("id") + " " + rs.getString("name"));
            }
            
        } catch (Exception e) {
            e.printStackTrace();
        }
                ...

• 把 SQL 語句調(diào)整到配置文件
• 把 resultset 封裝到一個(gè) pojo 里返回

實(shí)現(xiàn)

實(shí)現(xiàn)的步聚如下：

1. 在配置文件添加配置項(xiàng)
2. 調(diào)整代碼

在配置文件添加配置項(xiàng)

db.driver=com.mysql.jdbc.Driver
db.url=jdbc:mysql://localhost:3306/tester?characterEncoding=utf-8
db.username=root
db.password=123456

db.sql.queryUserByName=select * from user_info where name = ?
db.sql.queryUserByName.resultType=com.edugroup.pojo.UserInfo

• 將 SQL 語句移至配置文件中
• 在配置文件中指定 resultType

調(diào)整代碼

    public <T> List<T> selectList(String statementId, Object param) throws Exception {
        Connection conn = null;
        PreparedStatement pstat = null;
        ResultSet rs = null;
        List<T> results = new ArrayList<>();
        
        try {
            //獲取連接并執(zhí)行查詢
            conn = dataSource.getConnection();
            String sql = props.getProperty("db.sql." + statementId);
            pstat = conn.prepareStatement(sql);
            pstat.setObject(1, param);
            rs = pstat.executeQuery();
            
            //處理返回的結(jié)果集
            String resultType = props.getProperty("db.sql." + statementId +".resultType");
            Class<?> clazz = Class.forName(resultType);
            ResultSetMetaData rsMetaData = rs.getMetaData();
            
            while(rs.next()) {
                Object result = clazz.newInstance();
                for(int i = 1; i <= rsMetaData.getColumnCount(); i++) {
                    Field field = clazz.getDeclaredField(rsMetaData.getColumnName(i));
                    field.setAccessible(true);
                    field.set(result, rs.getObject(i));
                }
                results.add((T) result);
            }
            
        } catch (Exception e) {
            e.printStackTrace();
            throw e;
        } finally {
            if(pstat != null) { pstat.close(); }
            if(rs != null) { rs.close(); }
        }
        return results;
    }

• 從配置文件中讀取 SQL 語句
• 利用反射將 resultset 放到 pojo 里

碼字不易，感謝點(diǎn)贊打賞