手写源码：MyBatis代理Mapper的核心实现

前言

基于前面写的文章：MyBatis精髓揭秘：Mapper代理实现的黑盒探索，里面详细的介绍了 MyBatis 代理的实现逻辑，整体来看就是基于 JDK 动态代理的实现，虽然我们在使用的时候没有创建任何的实现类，但是基于动态代理技术，我们可以无中生有。

本文我们就基于这个核心思想，手写一份超精简的 MyBatis 源码。

前提准备

准备数据库表并创建实体

/**
 * 账户实体
 *
 * @author 薛伟
 */
public class Account implements Serializable {
​
    private Integer id;
​
    private String name;
​
    private String password;
​
    public Account() {
    }
​
    public Account(Integer id, String name, String password) {
        this.id = id;
        this.name = name;
        this.password = password;
    }
​
    public Integer getId() {
        return id;
    }
​
    public void setId(Integer id) {
        this.id = id;
    }
​
    public String getName() {
        return name;
    }
​
    public void setName(String name) {
        this.name = name;
    }
​
    public String getPassword() {
        return password;
    }
​
    public void setPassword(String password) {
        this.password = password;
    }
​
    @Override
    public String toString() {
        return "Account{" + "id=" + id + ", name='" + name + '\'' + ", password='" + password + '\'' + '}';
    }
}

创建数据访问层

/**
 * 账户数据库访问
 *
 * @author 薛伟
 */
@Mapper
public interface AccountDao {
​
    /**
     * 查询全部
     */
    List<Account> getAll();
}

创建XML配置文件

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
​
<!-- 配置 namespace -->
<mapper namespace="world.xuewei.mybatis.dao.AccountDao">
​
    <select id="getAll" resultType="Account">
        select *
        from account;
    </select>
</mapper>

我们这里只准备了一个最简单的查询方法。

创建测试类

public class DaoTest {
​
    private SqlSession sqlSession;
​
    @Before
    public void before() throws IOException {
        InputStream inputStream = Resources.getResourceAsStream("mybatis-config.xml");
        SqlSessionFactory sessionFactory = new SqlSessionFactoryBuilder().build(inputStream);
        sqlSession = sessionFactory.openSession();
    }
​
    @After
    public void after() {
        sqlSession.commit();
    }
    
    // 这里编写测试类
}

代码实现

首先我们可以基于原生的 ibatis 和 MyBatis 来调用上面创建的 AccountDao.getAll 方法实现查询效果。

/**
  * 测试原生 iBatis 
  */
@Test
public void sqlSessionTest() {
    List<Account> list = sqlSession.selectList("world.xuewei.mybatis.dao.AccountDao.getAll");
    System.out.println(list);
}
​
/**
  * 测试原生 MyBatis 
  */
@Test
public void testGetAll() {
    AccountDao accountDao = sqlSession.getMapper(AccountDao.class);
    System.out.println(accountDao.getAll());
}

接下来就是代理的核心实现

/**
  * 测试自定义代理实现
  */
@Test
public void proxyTest() {
    ClassLoader classLoader = this.getClass().getClassLoader();
    Class<?>[] interfaces = new Class[]{AccountDao.class};
    MyMapperProxy handler = new MyMapperProxy(sqlSession, AccountDao.class);
    AccountDao dao = (AccountDao) Proxy.newProxyInstance(classLoader, interfaces, handler);
    System.out.println(dao.getAll());
}

上面的代码我们为 AccountDao 创建了代理对象，调用方法的增强逻辑交个 去处理。初始化 MyMapperProxy 时将当前的 SqlSession 对象和要 Mapper 的类型传递过去。接下来我们看一下 MyMapperProxy 是如何实现的。

/**
 * Mapper 代理增强
 *
 * @author XUEW
 */
public class MyMapperProxy implements InvocationHandler {
​
    private final SqlSession sqlSession;
​
    private final Class<?> daoClass;
​
    public MyMapperProxy(SqlSession sqlSession, Class<?> daoClass) {
        this.sqlSession = sqlSession;
        this.daoClass = daoClass;
    }
​
    @Override
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        return sqlSession.selectList(daoClass.getName() + "." + method.getName(), args);
    }
}

我们这个示例只有一个查询方法，并且没有参数，所以这里的 invoke 方法是非常简单的。直接就调用了 sqlSession 的查询方法。

尽管代码相当的简单，但是 MyBatis 关于代理的核心实现就是这样的，只不过他设计的更精美，且考虑问题更加全面。