跟着小傅哥手写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。