手写源码（四）：自己实现Mybatis

手写MyBatis

如题，这次我又来作死试试编写类似Mybatis的持久层框架了

MyBatis的难点

• 如何在没有实例的情况下创建Mapping接口的实现类并且调用接口中的方法
  • 使用字节技术创建子类
  • 使用匿名内部类
  • 使用动态代理创建对象(我们使用这个)

---

创建一个接口UserMapper,再创建一个实体类User

使用JDK的动态代理，创建一个代理处理器

public class InvocationHandlerMybatis implements InvocationHandler {
    /**
     *
     * @param proxy     代理对象
     * @param method    拦截的方法
     * @param args      方法上的参数
     * @return
     * @throws Throwable
     */
    @Override
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        System.out.println("开始代理");
        return 1;
    }
}

包装上面的代理

public class SqlSession {
    /**加载Mapper接口*/
    public static  T getMapper(Class clazz) {
        return (T) Proxy.newProxyInstance(clazz.getClassLoader(), new Class[]{clazz}, new InvocationHandlerMybatis());
    }
}

使用测试类测试一下

    public static void main(String[] args) {
        UserMapper userMapper = SqlSession.getMapper(UserMapper.class);
        int i = userMapper.insertUser("", "");
        System.out.println(i);
    }

测试结果如下

动态代理的测试结果

这样就可以实现拿到接口的方法参数并且自行控制对象的返回值

一、@Insert的实现步骤

• 判断方法上是否存在@Insert注解，存在的话，获取上面的SQL语句
• 获取方法的参数，和SQL参数经行匹配，并且替换SQL的参数（变成问号）
• 调用JDBC代码执行语句并获取返回值

具体实现如下

/**
 * @author libi
 * 用于动态代理，获取方法的参数并且给返回值
 */
public class InvocationHandlerMybatis implements InvocationHandler {
    /**
     * @param proxy     代理对象
     * @param method    拦截的方法
     * @param args      方法上的参数
     * @return          方法的返回值
     * @throws Throwable
     */
    @Override
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        System.out.println("开始代理");
        //判断方法上是否存在Insert注解
        ExtInsert extInsert = method.getDeclaredAnnotation(ExtInsert.class);
        if (extInsert != null) {
            //执行插入的操作，返回影响行数
            return doInsert(method, args, extInsert);
        }
        return null;
    }

    /**
     * 执行插入的操作
     * @param method
     * @param args
     * @param extInsert
     * @return
     */
    private int doInsert(Method method, Object[] args, ExtInsert extInsert) {
        //获取Sql语句
        String sql = extInsert.value();
        System.out.println("insert sql:" + sql);
        //获取方法参数和Sql语句进行匹配
        //定义一个Map，Key是参数名，Value是参数值
        ConcurrentHashMap map = new ConcurrentHashMap<>();
        //获取方法上的参数
        Parameter[] parameters = method.getParameters();
        for (int i = 0; i < parameters.length; i++) {
            //获取参数名称和参数的值
            ExtParam param = parameters[i].getDeclaredAnnotation(ExtParam.class);
            if (param != null) {
                String name = param.value();
                Object value = args[i];
                System.out.println("paramName:"+name+",paramValue:"+value);
                map.put(name, value);
            }
        }
        //怕打乱顺序而把sql语句的参数放在一个有序的数组里
        List