Java手寫持久層框架的詳細(xì)代碼

更新時(shí)間：2022年08月20日 09:34:33 作者：So,Cool

這篇文章主要介紹了Java手寫持久層框架,本文適合有一定java基礎(chǔ)的同學(xué)，通過(guò)自定義持久層框架，可以更加清楚常用的mybatis等開源框架的原理，需要的朋友可以參考下

前言

本文適合有一定java基礎(chǔ)的同學(xué)，通過(guò)自定義持久層框架，可以更加清楚常用的mybatis等開源框架的原理。

JDBC操作回顧及問(wèn)題分析

學(xué)習(xí)java的同學(xué)一定避免不了接觸過(guò)jdbc，讓我們來(lái)回顧下初學(xué)時(shí)期接觸的jdbc操作吧

以下代碼連接數(shù)據(jù)庫(kù)查詢用戶表信息，用戶表字段分別為用戶id，用戶名username。

Connection connection = null;
        PreparedStatement preparedStatement = null;
        ResultSet resultSet = null;
        User user = new User();
        try {
            // 加載數(shù)據(jù)庫(kù)驅(qū)動(dòng)
            //Class.forName("com.mysql.jdbc.Driver");
            Class.forName("com.mysql.cj.jdbc.Driver");
            // 通過(guò)驅(qū)動(dòng)管理類獲取數(shù)據(jù)庫(kù)鏈接
            connection = DriverManager.getConnection("jdbc:mysql://localhost:3306/mybatis?characterEncoding=utf-8", "root", "mimashi3124");
            // 定義sql語(yǔ)句？表示占位符
            String sql = "select * from user where username = ?";
            // 獲取預(yù)處理statement
            preparedStatement = connection.prepareStatement(sql);
            // 設(shè)置參數(shù)，第?個(gè)參數(shù)為sql語(yǔ)句中參數(shù)的序號(hào)(從1開始)，第?個(gè)參數(shù)為設(shè)置的參數(shù)值
            preparedStatement.setString(1, "蓋倫");
            // 向數(shù)據(jù)庫(kù)發(fā)出sql執(zhí)?查詢，查詢出結(jié)果集
            resultSet = preparedStatement.executeQuery();
            // 遍歷查詢結(jié)果集
            while (resultSet.next()) {
                int id = resultSet.getInt("id");
                String username = resultSet.getString("username");
                // 封裝User
                user.setId(id);
                user.setUsername(username);
            }
            System.out.println(user);
        } catch (
                Exception e) {
            e.printStackTrace();
        } finally {
            // 釋放資源
            if (resultSet != null) {
                try {
                    resultSet.close();
                } catch (SQLException e) {
                    e.printStackTrace();
                }
            }
            if (preparedStatement != null) {
                try {
                    preparedStatement.close();
                } catch (SQLException e) {
                    e.printStackTrace();
                }
            }
            if (connection != null) {
                try {
                    connection.close();
                } catch (SQLException e) {
                    e.printStackTrace();
                }
            }
        }

查看代碼我們可以發(fā)現(xiàn)使用JDBC操作數(shù)據(jù)庫(kù)存在以下問(wèn)題：

數(shù)據(jù)庫(kù)連接創(chuàng)建、釋放頻繁造成系統(tǒng)資源浪費(fèi)，從?影響系統(tǒng)性能。
Sql語(yǔ)句我們是寫在代碼里的，代碼不容易維護(hù)，實(shí)際應(yīng)?中sql變化的可能較?，sql變動(dòng)需要改變 java代碼。
使?preparedStatement向占有位符號(hào)傳參數(shù)存在硬編碼，因?yàn)閟ql語(yǔ)句的where條件不?定，可能多也可能少，修改sql還要修改代碼，系統(tǒng)不易維護(hù)。
對(duì)結(jié)果集解析存在硬編碼(查詢列名)，sql變化導(dǎo)致解析代碼變化，系統(tǒng)不易維護(hù)，如果能將數(shù)據(jù) 庫(kù)記錄封裝成pojo對(duì)象解析?較?便

問(wèn)題解決思路

使?數(shù)據(jù)庫(kù)連接池初始化連接資源，避免資源浪費(fèi)
將sql語(yǔ)句抽取到xml配置中，這種sql的變動(dòng)只用關(guān)注xml文件，不比去一堆java代碼里改寫sql
參數(shù)硬編碼問(wèn)題可以使用反射、內(nèi)省等技術(shù)、自動(dòng)將實(shí)體與表字段進(jìn)行映射。

自己動(dòng)手寫個(gè)持久層框架

接下來(lái)，我們來(lái)一個(gè)個(gè)解決上面的問(wèn)題

數(shù)據(jù)庫(kù)連接池我們可以直接使用c3p0提供的ComboPooledDataSource即可

為了解決sql硬編碼問(wèn)題，我們要把sql寫到xml文件中，那自然是要定義一個(gè)xml文件了。

光有sql肯定不行，畢竟我們要先連接數(shù)據(jù)庫(kù)，sql語(yǔ)句才有存在的意義。所以xml中得先定義數(shù)據(jù)配置信息，然后才是sql語(yǔ)句。

1.定義配置xml文件

我們新建一個(gè)sqlMapConfig.xml，定義數(shù)據(jù)源信息、并且增加兩個(gè)sql語(yǔ)句，parameterType為sql執(zhí)行參數(shù)，resultType為方法返回實(shí)體。

代碼如下(數(shù)據(jù)庫(kù)不同版本使用驅(qū)動(dòng)類可能不同)：

<configuration>
    <!--數(shù)據(jù)庫(kù)連接信息-->
    <property name="driverClass" value="com.mysql.cj.jdbc.Driver"/>
<!--    <property name="driverClass" value="com.mysql.jdbc.Driver"/>-->
    <property name="jdbcUrl" value="jdbc:mysql://localhost:3306/mybatis?serverTimezone=Asia/Shanghai"/>
    <property name="username" value="root"/>
    <property name="password" value="mimashi3124"/>

	<select id="selectOne" parameterType="org.example.pojo.User"
            resultType="org.example.pojo.User">
        select * from user where id = #{id} and username =#{username}
    </select>

    <select id="selectList" resultType="org.example.pojo.User">
        select * from user
    </select>
</configuration>

現(xiàn)在xml文件數(shù)據(jù)庫(kù)信息也有了，sql語(yǔ)句定義也有了，還有什么問(wèn)題呢？

我們實(shí)際中對(duì)sql的操作會(huì)涉及到不同的表，所以我們改進(jìn)一下，把每個(gè)表的sql語(yǔ)句單獨(dú)放在一個(gè)xml里，這樣結(jié)構(gòu)更清晰就容易維護(hù)。

優(yōu)化以后的xml配置現(xiàn)在是這樣了

sqlMapConfig.xml

<configuration>
    <!--數(shù)據(jù)庫(kù)連接信息-->
    <property name="driverClass" value="com.mysql.cj.jdbc.Driver"/>
	<!-- <property name="driverClass" value="com.mysql.jdbc.Driver"/>-->
    <property name="jdbcUrl" value="jdbc:mysql://localhost:3306/mybatis?serverTimezone=Asia/Shanghai"/>
    <property name="username" value="root"/>
    <property name="password" value="mimashi3124"/>
    <!--引?sql配置信息-->
	<mapper resource="mapper.xml"></mapper>
</configuration>

mapper.xml

<mapper namespace="user">
    <select id="selectOne" parameterType="org.example.pojo.User"
            resultType="org.example.pojo.User">
        select * from user where id = #{id} and username =#{username}
    </select>

    <select id="selectList" resultType="org.example.pojo.User">
        select * from user
    </select>
</mapper>

順便定義一下業(yè)務(wù)實(shí)體User

public class User {
    private int id;
    private String username;
    
    public int getId() {
        return id;
    }

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

    public String getUsername() {
        return username;
    }

    public void setUsername(String username) {
        this.username = username;
    }

    @Override
    public String toString() {
        return "User{" +
                "id=" + id +
                ", username='" + username + '\'' +
                '}';
    }
}

2.讀取配置文件

讀取完成以后以流的形式存在，不好操作，所以我們要做解析拿到信息，創(chuàng)建實(shí)體對(duì)象來(lái)存儲(chǔ)。

Configuration : 存放數(shù)據(jù)庫(kù)基本信息、Map<唯?標(biāo)識(shí)，Mapper> 唯?標(biāo)識(shí)：namespace + "." + idMappedStatement：存放sql語(yǔ)句、輸?參數(shù)類型、輸出參數(shù)類型

xml解析我們使用dom4j

首先引入maven依賴

代碼如下(mysql驅(qū)動(dòng)版本根據(jù)實(shí)際使用mysql版本調(diào)整)：

<properties>
        <maven.compiler.source>8</maven.compiler.source>
        <maven.compiler.target>8</maven.compiler.target>
    </properties>

    <dependencies>
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <version>8.0.22</version>
        </dependency>
        <dependency>
            <groupId>c3p0</groupId>
            <artifactId>c3p0</artifactId>
            <version>0.9.1.2</version>
        </dependency>
        <dependency>
            <groupId>log4j</groupId>
            <artifactId>log4j</artifactId>
            <version>1.2.12</version>
        </dependency>
        <dependency>
            <groupId>junit</groupId>
            <artifactId>junit</artifactId>
            <version>4.10</version>
        </dependency>
        <dependency>
            <groupId>dom4j</groupId>
            <artifactId>dom4j</artifactId>
            <version>1.6.1</version>
        </dependency>
        <dependency>
            <groupId>jaxen</groupId>
            <artifactId>jaxen</artifactId>
            <version>1.1.6</version>
        </dependency>
    </dependencies>

數(shù)據(jù)庫(kù)配置實(shí)體 Configuration

public class Configuration {
    //數(shù)據(jù)源
    private DataSource dataSource;
    //map集合： key:statementId value:MappedStatement
    private Map<String,MappedStatement> mappedStatementMap = new HashMap<>();

    public DataSource getDataSource() {
        return dataSource;
    }

    public Configuration setDataSource(DataSource dataSource) {
        this.dataSource = dataSource;
        return this;
    }

    public Map<String, MappedStatement> getMappedStatementMap() {
        return mappedStatementMap;
    }

    public Configuration setMappedStatementMap(Map<String, MappedStatement> mappedStatementMap) {
        this.mappedStatementMap = mappedStatementMap;
        return this;
    }
}

Sql語(yǔ)句信息實(shí)體

public class MappedStatement {

    //id
    private String id;
    //sql語(yǔ)句
    private String sql;
    //輸?參數(shù)
    private String parameterType;
    //輸出參數(shù)
    private String resultType;

    public String getId() {
        return id;
    }

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

    public String getSql() {
        return sql;
    }

    public MappedStatement setSql(String sql) {
        this.sql = sql;
        return this;
    }

    public String getParameterType() {
        return parameterType;
    }

    public MappedStatement setParameterType(String parameterType) {
        this.parameterType = parameterType;
        return this;
    }

    public String getResultType() {
        return resultType;
    }

    public MappedStatement setResultType(String resultType) {
        this.resultType = resultType;
        return this;
    }
}

順便定義一個(gè)Resources類來(lái)讀取xml文件流

public class Resources {
    public static InputStream getResourceAsSteam(String path) {
        return Resources.class.getClassLoader().getResourceAsStream(path);
    }
}

接下來(lái)就是實(shí)際的解析了，因?yàn)榻馕龃a比較多，我們考慮封裝類單獨(dú)處理解析

定義XMLConfigBuilder類解析數(shù)據(jù)庫(kù)配置信息

public class XMLConfigBuilder {

    private Configuration configuration;


    public XMLConfigBuilder() {
        this.configuration = new Configuration();
    }

    public Configuration parserConfiguration(InputStream inputStream) throws DocumentException, PropertyVetoException, ClassNotFoundException {
        Document document = new SAXReader().read(inputStream);
        Element rootElement = document.getRootElement();
        List<Element> propertyElements = rootElement.selectNodes("http://property");
        Properties properties = new Properties();
        for (Element propertyElement : propertyElements) {
            String name = propertyElement.attributeValue("name");
            String value = propertyElement.attributeValue("value");
            properties.setProperty(name,value);
        }
        //解析到數(shù)據(jù)庫(kù)配置信息，設(shè)置數(shù)據(jù)源信息
        ComboPooledDataSource comboPooledDataSource = new ComboPooledDataSource();
        comboPooledDataSource.setDriverClass(properties.getProperty("driverClass"));
        comboPooledDataSource.setJdbcUrl(properties.getProperty("jdbcUrl"));
        comboPooledDataSource.setUser(properties.getProperty("username"));
        comboPooledDataSource.setPassword(properties.getProperty("password"));
		
        configuration.setDataSource(comboPooledDataSource);

		//將configuration傳入XMLMapperBuilder中做sql語(yǔ)句解析。
        XMLMapperBuilder xmlMapperBuilder = new XMLMapperBuilder(this.configuration);
        List<Element> mapperElements = rootElement.selectNodes("http://mapper");
        for (Element mapperElement : mapperElements) {
            String mapperPath = mapperElement.attributeValue("resource");
            InputStream resourceAsStream = this.getClass().getClassLoader().getResourceAsStream(mapperPath);
            xmlMapperBuilder.parse(resourceAsStream);
        }
        return configuration;
    }
}

定義XMLMapperBuilder類解析數(shù)據(jù)庫(kù)配置信息

public class XMLMapperBuilder {

    private Configuration configuration;

    public XMLMapperBuilder(Configuration configuration) {
        this.configuration = configuration;
    }
    public void parse(InputStream inputStream) throws DocumentException,
            ClassNotFoundException {
        Document document = new SAXReader().read(inputStream);
        Element rootElement = document.getRootElement();
        String namespace = rootElement.attributeValue("namespace");
        List<Element> select = rootElement.selectNodes("select");
        for (Element element : select) { //id的值
            String id = element.attributeValue("id");
            String parameterType = element.attributeValue("parameterType"); //輸?參數(shù)
            String resultType = element.attributeValue("resultType"); //返回參數(shù)
            //statementId，后續(xù)調(diào)用通過(guò)statementId，找到對(duì)應(yīng)的sql執(zhí)行
            String key = namespace + "." + id;
            //sql語(yǔ)句
            String textTrim = element.getTextTrim();
            //封裝 mappedStatement
            MappedStatement mappedStatement = new MappedStatement();
            mappedStatement.setId(id);
            mappedStatement.setParameterType(parameterType);
            mappedStatement.setResultType(resultType);
            mappedStatement.setSql(textTrim);
            //填充 configuration
            configuration.getMappedStatementMap().put(key, mappedStatement);
        }
    }
}

現(xiàn)在我們可以通過(guò)調(diào)用配置解析的方法拿到Configuration對(duì)象了。但是我們實(shí)際使用，肯定是希望我給你配置信息、sql語(yǔ)句，再調(diào)用你的方法就返回結(jié)果了。
所以我們還需要定義一個(gè)數(shù)據(jù)庫(kù)操作接口（類）

3.定義sql操作接口SqlSession

public interface SqlSession {
	//查詢多個(gè)
    public <E> List<E> selectList(String statementId, Object... param) throws Exception;

	//查詢一個(gè)
    public <T> T selectOne(String statementId,Object... params) throws Exception;
}

對(duì)操作接口SqlSession做具體實(shí)現(xiàn)，這里主要是通過(guò)statementId找到對(duì)應(yīng)的sql信息，進(jìn)行執(zhí)行

代碼中simpleExcutor做真正的數(shù)據(jù)庫(kù)語(yǔ)句執(zhí)行、返回參數(shù)封裝等操作

public class DefaultSqlSession implements SqlSession {

    private Configuration configuration;
    private Executor simpleExcutor = new SimpleExecutor();

    public DefaultSqlSession(Configuration configuration) {
        this.configuration = configuration;
    }

    @Override
    public <E> List<E> selectList(String statementId, Object... param) throws Exception {
        MappedStatement mappedStatement =
                configuration.getMappedStatementMap().get(statementId);
        List<E> query = simpleExcutor.query(configuration, mappedStatement, param);
        return query;
    }

    @Override
    public <T> T selectOne(String statementId, Object... params) throws Exception {
        List<Object> objects = selectList(statementId, params);
        if (objects.size() == 1) {
            return (T) objects.get(0);
        } else {
            throw new RuntimeException("返回結(jié)果過(guò)多");
        }
    }
}

4.編寫數(shù)據(jù)庫(kù)執(zhí)行邏輯

數(shù)據(jù)庫(kù)操作類DefaultSqlSession中的selectList方法調(diào)用到了simpleExcutor.query()方法

public class SimpleExecutor implements Executor {

    private Connection connection = null;

    @Override
    public <E> List<E> query(Configuration configuration, MappedStatement mappedStatement, Object[] params) throws Exception {
        //獲取連接
        connection = configuration.getDataSource().getConnection();
        // select * from user where id = #{id} and username = #{username} String sql =
        String sql = mappedStatement.getSql();
        //對(duì)sql進(jìn)?處理
        BoundSql boundSql = getBoundSql(sql);
        // 3.獲取預(yù)處理對(duì)象：preparedStatement
        PreparedStatement preparedStatement = connection.prepareStatement(boundSql.getSqlText());

        // 4. 設(shè)置參數(shù)
        //獲取到了參數(shù)的全路徑
        String parameterType = mappedStatement.getParameterType();
        Class<?> parameterTypeClass = getClassType(parameterType);

        List<ParameterMapping> parameterMappingList = boundSql.getParameterMappingList();
        for (int i = 0; i < parameterMappingList.size(); i++) {
            ParameterMapping parameterMapping = parameterMappingList.get(i);
            String content = parameterMapping.getContent();

            //反射
            Field declaredField = parameterTypeClass.getDeclaredField(content);
            //暴力訪問(wèn)
            declaredField.setAccessible(true);
            Object o = declaredField.get(params[0]);

            preparedStatement.setObject(i+1,o);

        }
        // 5. 執(zhí)行sql
        ResultSet resultSet = preparedStatement.executeQuery();
        String resultType = mappedStatement.getResultType();
        Class<?> resultTypeClass = getClassType(resultType);

        ArrayList<Object> objects = new ArrayList<>();

        // 6. 封裝返回結(jié)果集
        while (resultSet.next()){
            Object o =resultTypeClass.newInstance();
            //元數(shù)據(jù)
            ResultSetMetaData metaData = resultSet.getMetaData();
            for (int i = 1; i <= metaData.getColumnCount(); i++) {
                // 字段名
                String columnName = metaData.getColumnName(i);
                // 字段的值
                Object value = resultSet.getObject(columnName);
                //使用反射或者內(nèi)省，根據(jù)數(shù)據(jù)庫(kù)表和實(shí)體的對(duì)應(yīng)關(guān)系，完成封裝
                PropertyDescriptor propertyDescriptor = new PropertyDescriptor(columnName, resultTypeClass);
                Method writeMethod = propertyDescriptor.getWriteMethod();
                writeMethod.invoke(o,value);
            }
            objects.add(o);
        }
        return (List<E>) objects;
    }

    @Override
    public void close() throws SQLException {

    }

    private Class<?> getClassType(String parameterType) throws ClassNotFoundException {
        if(parameterType!=null){
            Class<?> aClass = Class.forName(parameterType);
            return aClass;
        }
        return null;

    }

    private BoundSql getBoundSql(String sql) {
        //標(biāo)記處理類：主要是配合通?標(biāo)記解析器GenericTokenParser類完成對(duì)配置?件等的解 析?作，其中
        //TokenHandler主要完成處理
        ParameterMappingTokenHandler parameterMappingTokenHandler = new
                ParameterMappingTokenHandler();
        //GenericTokenParser :通?的標(biāo)記解析器，完成了代碼?段中的占位符的解析，然后再根 據(jù)給定的
       // 標(biāo)記處理器(TokenHandler)來(lái)進(jìn)?表達(dá)式的處理
        //三個(gè)參數(shù)：分別為openToken (開始標(biāo)記)、closeToken (結(jié)束標(biāo)記)、handler (標(biāo)記處 理器)
        GenericTokenParser genericTokenParser = new GenericTokenParser("#{", "}",
                parameterMappingTokenHandler);
        String parse = genericTokenParser.parse(sql);
        List<ParameterMapping> parameterMappings =
                parameterMappingTokenHandler.getParameterMappings();
        return new BoundSql(parse, parameterMappings);
    }
}

上面的注釋比較詳細(xì)，流程為

根據(jù)對(duì)應(yīng)的statementId獲取到要執(zhí)行的sql語(yǔ)句、調(diào)用參數(shù)、返回參數(shù)。
對(duì)sql的占位符進(jìn)行解析、調(diào)用參數(shù)進(jìn)行設(shè)置
根據(jù)解析到的入?yún)⒆侄?，通過(guò)反射獲取到對(duì)應(yīng)的值，進(jìn)行sql語(yǔ)句參數(shù)設(shè)定
執(zhí)行sql語(yǔ)句，使用反射、內(nèi)省，根據(jù)數(shù)據(jù)庫(kù)表和實(shí)體的對(duì)應(yīng)關(guān)系，完成對(duì)象屬性的設(shè)置，最終返回結(jié)果。

通過(guò)以上步驟，我們獲取到了數(shù)據(jù)庫(kù)配置、sql語(yǔ)句信息。定義了數(shù)據(jù)庫(kù)操作類SqlSession，但是我們并沒有在什么地方調(diào)用解析配置文件。

我們還需要一個(gè)東西把兩者給串起來(lái)，這里我們可以使用工廠模式來(lái)生成SqlSession

使用工廠模式創(chuàng)建SqlSession

public interface SqlSessionFactory {
    public SqlSession openSession();
}

public class DefaultSqlSessionFactory implements SqlSessionFactory{

    private Configuration configuration;

    public DefaultSqlSessionFactory(Configuration configuration) {
        this.configuration = configuration;
    }

    @Override
    public SqlSession openSession() {
        return new DefaultSqlSession(configuration);
    }
}

同時(shí)為了屏蔽構(gòu)建SqlSessionFactory工廠類時(shí)獲取Configuration的解析過(guò)程，我們可以使用構(gòu)建者模式來(lái)獲得一個(gè)SqlSessionFactory類。

public class SqlSessionFactoryBuilder {
    public SqlSessionFactory build(InputStream inputStream) throws PropertyVetoException, DocumentException, ClassNotFoundException {
        XMLConfigBuilder xmlConfigerBuilder = new XMLConfigBuilder();
        Configuration configuration = xmlConfigerBuilder.parserConfiguration(inputStream);
        SqlSessionFactory sqlSessionFactory = new DefaultSqlSessionFactory(configuration);
        return sqlSessionFactory;
    }
}

5.調(diào)用測(cè)試

終于好了，通過(guò)以上幾個(gè)步驟我們現(xiàn)在可以具體調(diào)用執(zhí)行代碼了。

 public static void main(String[] args) throws Exception {
        InputStream resourceAsSteam = Resources.getResourceAsSteam("sqlMapConfig.xml");
        SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().build(resourceAsSteam);
        SqlSession sqlSession = sqlSessionFactory.openSession();
        User user = new User();
        user.setId(1);
        user.setUsername("蓋倫");
        User user2 = sqlSession.selectOne("user.selectOne", user);

        System.out.println(user2);
        List<User> users = sqlSession.selectList("user.selectList");
        for (User user1 : users) {
            System.out.println(user1);
        }
    }

代碼正確執(zhí)行，輸出

?定義框架優(yōu)化

上述?定義框架，解決了JDBC操作數(shù)據(jù)庫(kù)帶來(lái)的?些問(wèn)題：例如頻繁創(chuàng)建釋放數(shù)據(jù)庫(kù)連接，硬編
碼，?動(dòng)封裝返回結(jié)果集等問(wèn)題，現(xiàn)在我們繼續(xù)來(lái)分析剛剛完成的?定義框架代碼，有沒有什么問(wèn)題呢？

問(wèn)題如下：

dao的實(shí)現(xiàn)類中存在重復(fù)的代碼，整個(gè)操作的過(guò)程模板重復(fù)(創(chuàng)建sqlsession,調(diào)?sqlsession? 法，關(guān)閉sqlsession)
dao的實(shí)現(xiàn)類中存在硬編碼，調(diào)?sqlsession的?法時(shí)，參數(shù)statement的id硬編碼

我們可以使用代理模式，生成代理對(duì)象，在調(diào)用之前獲取到執(zhí)行方法的方法名、具體類。這樣我們就能獲取到statementId。

為SqlSession類新增getMappper方法，獲取代理對(duì)象

public interface SqlSession {
    public <E> List<E> selectList(String statementId, Object... param) throws Exception;

    public <T> T selectOne(String statementId,Object... params) throws Exception;

    //為Dao接口生成代理實(shí)現(xiàn)類
    public <T> T getMapper(Class<?> mapperClass);
}

public class DefaultSqlSession implements SqlSession {

    private Configuration configuration;
    private Executor simpleExcutor = new SimpleExecutor();

    public DefaultSqlSession(Configuration configuration) {
        this.configuration = configuration;
    }

    @Override
    public <E> List<E> selectList(String statementId, Object... param) throws Exception {
        MappedStatement mappedStatement =
                configuration.getMappedStatementMap().get(statementId);
        List<E> query = simpleExcutor.query(configuration, mappedStatement, param);
        return query;
    }

    @Override
    public <T> T selectOne(String statementId, Object... params) throws Exception {
        List<Object> objects = selectList(statementId, params);
        if (objects.size() == 1) {
            return (T) objects.get(0);
        } else {
            throw new RuntimeException("返回結(jié)果過(guò)多");
        }
    }

    @Override
    public <T> T getMapper(Class<?> mapperClass) {
        Object proxyInstance = Proxy.newProxyInstance(DefaultSqlSession.class.getClassLoader(), new Class[]{mapperClass}, new InvocationHandler() {
            @Override
            public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
            	// selectOne
                String methodName = method.getName();
                // className:namespace
                String className = method.getDeclaringClass().getName();
                //statementId
                String statementId = className+'.'+methodName;
                Type genericReturnType = method.getGenericReturnType();
                //判斷是否實(shí)現(xiàn)泛型類型參數(shù)化
                if (genericReturnType instanceof ParameterizedType){
                    List<Object> objects = selectList(statementId,args);
                    return objects;
                }
                return selectOne(statementId,args);
            }
        });
        return (T) proxyInstance;
    }
}

定義業(yè)務(wù)數(shù)據(jù)dao接口

public interface IUserDao {
    //查詢所有用戶
    public List<User> findAll() throws Exception;
    //根據(jù)條件進(jìn)行用戶查詢
    public User findByCondition(User user) throws Exception;
}

接下來(lái)我們只需獲取到代理對(duì)象，調(diào)用方法即可。

public class Main2 {
    public static void main(String[] args) throws Exception {
        InputStream resourceAsSteam = Resources.getResourceAsSteam("sqlMapConfig.xml");
        SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().build(resourceAsSteam);
        SqlSession sqlSession = sqlSessionFactory.openSession();
		//獲取到代理對(duì)象
        IUserDao userDao = sqlSession.getMapper(IUserDao.class);
        List<User> all = userDao.findAll();
        for (User user1 : all) {
            System.out.println(user1);
        }
    }
}