手撸Mybatis01: 创建简单映射器代理工厂

抛开mybatis-plus不说了, 正常情况下都是写个mapper接口类,然后绑定一下映射文件mapper.xml, mybatis就自动帮我们映射好了. 不然一般接口类都是去实现类的.

而这里就是通过代理工厂, 帮我们代理了这个接口的实现类.

所谓代理,一般就是先搞个类实现一下InvocationHandler接口的invoke方法.

invoke方法就可以针对这个对象的所有方法进行拦截,获取方法和参数,在代理中进行处理.

MapperProxy

这个代理类就是专门用来处理Mapper接口的代理实现

package com.linnine.mybatis.binding;

public class MapperProxy<T> implements InvocationHandler, Serializable {

	 /**
     * 结果管理
     * key: 类名.方法名
     * value: 对应返回结果(这里String是先small版的简单处理)
     */
    private Map<String,String> sqlSession;

    /**
     * mapper类
     */
    private final  Class<T> mapperInterface;

    public MapperProxy(Map<String, String> sqlSession, Class<T> mapperInterface) {
        this.sqlSession = sqlSession;
        this.mapperInterface = mapperInterface;
    }

    /**
     * 不管什么方法都走进来
     * @param proxy 代理对象
     * @param method 执行的方法
     * @param args 执行需要的参数
     * @return 返回值
     * @throws Throwable
     */
    @Override
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        if (Object.class.equals(method.getDeclaringClass())){
            return method.invoke(this,args);
        }else {
            //key 就是类的全名+方法名 获取实现就是通过这个key获取结果
            return "你呗代理了!" + sqlSession.get(mapperInterface.getName()+"."+method.getName());
        }
    }
}

MapperProxyFactory

搞一个代理工厂封装一下.

package com.linnine.mybatis.binding;

public class MapperProxyFactory<T> {

    private final Class<T> mapperInterface;


    public MapperProxyFactory(Class<T> mapperInterface) {
        this.mapperInterface = mapperInterface;
    }

    public T newInstance(Map<String, String> sqlSession) {
        final MapperProxy<T> mapperProxy = new MapperProxy<>(sqlSession, mapperInterface);
        return (T) Proxy.newProxyInstance(mapperInterface.getClassLoader(), new Class[]{mapperInterface}, mapperProxy);
    }
}

然后就可以写一个测试类

@Test
public void testMapperProxyFactory(){
    MapperProxyFactory<IUserDao> factory = new MapperProxyFactory<>(IUserDao.class);
    Map<String, String> sqlSession = new HashMap<>();
    //把对应的处理方法名和结果塞进去
    sqlSession.put("com.linnine.mybatis.dao.IUserDao.queryUserName","模拟执行Mapper.xml中的sql语句 查询用户姓名");
    sqlSession.put("com.linnine.mybatis.dao.IUserDao.queryUserAge","模拟执行Mapper.xml中的sql语句 查询用户年龄");
    IUserDao userDao = factory.newInstance(sqlSession);
    String s = userDao.queryUserName("10001");
    System.out.println(s);

}

结果如下,通过代理类生成的userDao,在执行方法queryUserName()时,代理类通过类名拼接方法名,去map中捞去结果,处理后返回. 所以后面就是代理类在里面把返回的结果一一映射过去再返回