mybatis源码第四课

最新推荐文章于 2023-02-10 00:28:37 发布

weixin_42815539

最新推荐文章于 2023-02-10 00:28:37 发布

阅读量67

点赞数

分类专栏： mybatis源码解析文章标签： mybatis

本文链接：https://blog.csdn.net/weixin_42815539/article/details/114994283

版权

mybatis源码解析专栏收录该内容

3 篇文章 0 订阅

订阅专栏

mybatis插件扩展

mybatis只能在这四个点扩展
在这里插入图片描述

1.Executor

在这里插入图片描述

2.StatementHandler
在这里插入图片描述

3.ParameterHandler
4.ResultSetHandler

示例代理StatementHandler对象的prepare方法

public class PageInterceptor implements Interceptor {
    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        return null;
    }

    @Override
    public Object plugin(Object target) {
        if (target instanceof StatementHandler){
            return Proxy.newProxyInstance(
                    PageInterceptor.class.getClassLoader(),
                    new Class[]{StatementHandler.class},
                    new PageHandler((StatementHandler) target));
        }
        return target;
    }

    @Override
    public void setProperties(Properties properties) {

    }


    private class PageHandler implements InvocationHandler{

        StatementHandler statementHandler;

        public PageHandler(StatementHandler statementHandler) {
            this.statementHandler = statementHandler;
        }

        @Override
        public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
            if (method.getName().equalsIgnoreCase("prepare")){
                setParametersProxy();
            }
            return method.invoke(statementHandler,args);
        }

        private void setParametersProxy() throws NoSuchFieldException, IllegalAccessException {
            appendPageSql(((Page)statementHandler.getBoundSql()
                    .getParameterObject()));

            }

        private void appendPageSql(Page page) throws NoSuchFieldException, IllegalAccessException {
            BoundSql boundSql = statementHandler.getBoundSql();
            String format = String.format(" limit %s,%s", page.getNumber(), page.getSize());
            setFieldValue("sql",boundSql,boundSql.getSql()+format);
        }

        private void setFieldValue(String sql, BoundSql boundSql, String s) throws NoSuchFieldException, IllegalAccessException {
            Field sqlField = boundSql.getClass().getDeclaredField(sql);
            sqlField.setAccessible(true);
            sqlField.set(boundSql,s);
        }
    }
}