Mybatis框架中是如何获取到SQL语句的，让我们一起来模拟一下吧

宁在春

发布于 2022-10-31 14:26:18

2960

发布于 2022-10-31 14:26:18

文章被收录于专栏：关于Java学习@宁在春

本文主要是通过代理和反射来模拟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;
    }
}