手写自己的MyBatis框架-MapperProxy

我们要在Configuration 中通过getMapper()方法拿到这个代理对象,必须要有一个实现了InvocationHandler 的代理类。我们来创建它:MapperProxy。

提供一个invoke()方法。

// MapperProxy.java
public class MapperProxy implements InvocationHandler {
	@Override
	public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
		return null;
	}
}

invoke()的实现我们先留着,先返回null。MapperProxy 已经有了,我们回到Configuration.getMapper()完成获取代理对象的逻辑。

返回代理对象,直接使用JDK 的动态代理:第一个参数是类加载器,第二个参数是被代理类,第三个参数是代理类。

把返回结果强转为(T):

Configuration.java
public <T> T getMapper(Class<T> clazz, SqlSession sqlSession) {
	return (T)Proxy.newProxyInstance(this.getClass().getClassLoader(),
	new Class[]{clazz},
	new MapperProxy());
}

获取代理类的逻辑已经实现完了,我们可以在SqlSession 中通过getMapper()拿到代理对象了,也就是可以调用invoke()方法了。接下来去完成MapperProxy 的invoke()方法。

在MapperProxy 的invoke()方法里面又调用了SqlSession 的selectOne()方法。一个问题出现了:在MapperProxy 里面根本没有SqlSession 对象?

这两个对象的关系怎么建立起来?MapperProxy 怎么拿到一个SqlSession 对象?

很简单,我们可通过构造函数传入它。

先定义一个属性,然后在MapperProxy 的构造函数里面赋值:

// MapperProxy.java
private SqlSession sqlSession;
public MapperProxy(SqlSession sqlSession) {
	this.sqlSession = sqlSession;
}

因为修改了代理类的构造函数,这个时候Configuration 创建代理类的方法getMapper()也要修改。

问题:Configuration 的getMapper()方法参数中也没有SqlSession,没办法传给MapperProxy 的构造函数。怎么拿到SqlSession 呢?是直接new 一个吗?

不需要,可以在SqlSession 调用它的时候直接把自己传进来(红色是修改的地方):

// Configuration.java
public <T> T getMapper(Class clazz, SqlSession sqlSession) {
	return (T)Proxy.newProxyInstance(this.getClass().getClassLoader(),
	new Class[]{clazz},
	new MapperProxy(sqlSession));
}

那么SqlSession 的getMapper()方法也要修改(红色是修改的地方):

// SqlSession.java
public <T> T getMapper(Class clazz){
	return configuration.getMapper(clazz, this);
}

现在在MapperProxy 里面已经就可以拿到SqlSession 对象了,在invoke()方法里面我们会调用SqlSession 的selectOne()方法。我们继续来完成invoke()方法。

selectOne()方法有两个参数, statementId 和paramater,这两个我们怎么拿到呢?

statementId 其实就是接口的全路径+方法名,中间加一个英文的点。

paramater 可以从方法参数中拿到,这里我们只传了一个参数,用args[0]。

它要把statementId 和参数传给SqlSession:

// MapperProxy.java
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
	String mapperInterface = method.getDeclaringClass().getName();
	String methodName = method.getName();
	String statementId = mapperInterface + "." + methodName;
	return sqlSession.selectOne(statementId, args[0]);
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值