Mybatis源码学习01

跟着小傅哥手写Mybatis

首先我们要知道什么是代理模式？
代理模式分为两种，jdk代理以及cglib代理，我们要用到的是jdk代理。

jdk代理代码如下：

 @Test
    public void jdkProxy(){
    /**
1)ClassLoader loader 类加载器，负责向内存中加载对象的。 使用反射获取对象的ClassLoader类a , a.getCalss().getClassLoader(), 目标对象的类加载器
2)Class<?>[] interfaces： 接口， 目标对象实现的接口，也是反射获取的。
3)InvocationHandler h : 我们自己写的，代理类要完成的功能

*/
        UserMapper userMapper= (UserMapper) Proxy.newProxyInstance(Thread.currentThread().getContextClassLoader(), new Class[]{UserMapper.class},
                (new InvocationHandler() {
                    @Override
                    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
                        System.out.println("代理执行了");
                        return "代理后返回值";
                    }
                }));

        String result= userMapper.getUserById("11");
        logger.info("执行完毕,结果为:{}",result);
    }
//接口
public interface UserMapper {
    String getUserById(String id);
}

创建代理工厂

package com.mybatis.binding;

import java.lang.reflect.Proxy;
import java.util.Calendar;
import java.util.Map;

/**
 * @author 促酒
 * @desc   映射器代理工厂
 * @date 2022/6/12 22:28
 */
public class MapperProxyFactory<T> {
    private  final Class<T> mapperInterface;

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

    public T getInstance(Map<String, String> sqlSession){
//        获取代理类
        MapperProxy<T> mapperProxy = new MapperProxy<>(sqlSession,mapperInterface);
        return (T) Proxy.newProxyInstance(mapperInterface.getClassLoader(), new Class[]{mapperInterface}, mapperProxy);
    }
}

创建代理类

package com.mybatis.binding;

import java.io.Serializable;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.util.Map;

/**
 * @author 促酒
 * @desc
 * @date 2022/6/12 22:27
 */
public class MapperProxy<T> implements InvocationHandler , Serializable {

    private static final Long serialVersionUID = -1234565342312134141L;
 
    private Map<String, String> sqlSession;
    private final Class<T> mapperInterface;

    public MapperProxy(Map<String, String> sqlSession, Class<T> mapperInterface) {
        this.sqlSession = sqlSession;
        this.mapperInterface = mapperInterface;
    }
    @Override
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        return "你被代理了！"+sqlSession.get(mapperInterface.getName()+"."+method.getName());
    }

}

测试

  @Test
    public void test_mapperProxyFactory(){
        MapperProxyFactory<UserMapper> factory =  new MapperProxyFactory(UserMapper.class);
        Map<String, String> sqlsession = new HashMap<>();
        sqlsession.put("com.lqf.dao.UserMapper.getUserById","模拟执行器中的sql");
        UserMapper instance = factory.getInstance(sqlsession);
        String userById = instance.getUserById("123");
        logger.info("执行结果：{}",userById);
    }

执行结果：

基本过程就是：

创建映射器工厂，然后执行获取实例化对象方法，在方法中先获取映射器，然后执行jdk动态代理。

在使用时，先传接口类（工厂参数）获取工厂对象，然后传一个sqlSession获取实列化对象，实现对方法的代理。当我们调用方法时，变执行了代理。

针对整个mybatis框架来说，sqlSession就是加载xml或注解声明的bean接口以后，生成的一个map集合，以key-value的形式存入sqlSession中，key为包名+类名+方法名，value为该方法对应的sql。