本文主要是通过代理和反射来模拟Mybatis此ORM框架是如何获得SQL语句及相关参数等。当我们拿到这些东西,那么一切就变得很简单啦,想做啥就做啥啦。
与君共勉
😁
一、基础知识
我们都知道Mybatis框架主要就是靠代理和反射来进行操作,那么对于动态代理,怎么都需要了解一点点哈。
Proxy.newProxyInstance(ClassLoader loader, Class<?>[] interfaces,InvocationHandler h)
newProxyInstance,方法有三个参数:
-
loader: 用哪个类加载器去加载代理对象
-
interfaces:动态代理类需要实现的接口
-
h:动态代理方法在执行时,会调用h里面的invoke方法去执行
二、代码
一个User类 为了测试加上的。
public class User {
private Integer id;
private String username;
private String password;
}
具体代码哦:加一下 Mybatis 的依赖就 可以直接copy测试啦哦
import org.apache.ibatis.annotations.Select;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.lang.reflect.Parameter;
import java.lang.reflect.Proxy;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
* @Author: crush
* @Date: 2021-05-27 11:32
* version 1.0
*/
interface UserMapper {
/**
* @return
*/
@Select("select * from user where id=#{id} and name=#{name}")
List<User> selectUser(Integer id,String name);
}
/**
* @author crush
*/
public class Application {
public static void main(String[] args) {
// newProxyInstance 这里用的是jdk的代理
UserMapper userMapper = (UserMapper) Proxy.newProxyInstance(Application.class.getClassLoader(), new Class[]{UserMapper.class}, new InvocationHandler() {
@Override
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
System.out.println("参数==> " + Arrays.toString(args));
System.out.println("UserMapper执行的方法名==> " + method.getName());
Map<String, Object> nameArgMap = buildMethodArgNameMap(method, args);
System.out.println("方法上的参数名==> "+nameArgMap.toString());
// 这咯是通过反射获取方法上的注解
Select annotation = method.getAnnotation(Select.class);
if (annotation != null) {
// 这里是获取 注解的值
String[] value = annotation.value();
String sql = value[0];
//解析sql 语句
sql = parseSQL(sql, nameArgMap );
System.out.println(sql);
System.out.println("返回的类型==>"+method.getReturnType());
System.out.println("返回的泛型==>"+method.getGenericReturnType());
}
return null;
}
});
userMapper.selectUser(1,"crush");
}
public static String parseSQL(String sql, Map<String, Object> nameArgMap ) {
StringBuilder stringBuilder = new StringBuilder();
int length = sql.length();
for (int i = 0; i < length; i++) {
char c = sql.charAt(i);
// 此处是判断'#{'
if (c == '#') {
int nextIndex = i + 1;
char nextChar = sql.charAt(nextIndex);
if (nextChar != '{') {
throw new RuntimeException(String.format("这里应该为#{\nsql:%s\nindex:%d",
stringBuilder.toString(), nextIndex));
}
StringBuilder argSB = new StringBuilder();
//此处是 让argSB 获取到 #{} 中的值 手动打印一下查看
i = parseSQLArg(argSB, sql, nextIndex);
String argName = argSB.toString();
Object argValue = nameArgMap.get(argName);
stringBuilder.append(argValue.toString());
continue;
}
stringBuilder.append(c);
}
return stringBuilder.toString();
}
private static int parseSQLArg(StringBuilder argSB, String sql, int nextIndex) {
nextIndex++;
for (; nextIndex < sql.length(); nextIndex++) {
char c = sql.charAt(nextIndex);
if (c != '}') {
argSB.append(c);
continue;
}
if (c == '}') {
return nextIndex;
}
}
throw new RuntimeException(String.format("缺少右括号\nindex:%d", nextIndex));
}
public static Map<String, Object> buildMethodArgNameMap(Method method, Object[] args) {
Map<String, Object> nameArgMap = new HashMap<String, Object>();
Parameter[] parameters = method.getParameters();
int[] index = {0};
Arrays.asList(parameters).forEach(parameter -> {
String name = parameter.getName();
System.out.println("接口方法上参数名字==》"+name);
nameArgMap.put(name, args[index[0]]);
index[0]++;
});
return nameArgMap;
}
}
自言自语
之前只是学习和使用mybatis,知道是代理和反射实现的框架原理,但是从来没有真正去了解。看过之后发现,学习带有好奇心,做什么都会有激情,事半功倍哦。
加油哦。