文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

如何使用mybatis的interceptor修改执行sql和传入参数

2023-06-21 21:47

关注

这篇文章主要介绍“如何使用mybatis的interceptor修改执行sql和传入参数”,在日常操作中,相信很多人在如何使用mybatis的interceptor修改执行sql和传入参数问题上存在疑惑,小编查阅了各式资料,整理出简单好用的操作方法,希望对大家解答”如何使用mybatis的interceptor修改执行sql和传入参数”的疑惑有所帮助!接下来,请跟着小编一起来学习吧!

mybatis interceptor修改执行sql以及传入参数

项目中途遇到业务需求更改,在查询某张表时需要增加条件,由于涉及的sql语句多而且依赖其他服务的jar,逐个修改sql语句和接口太繁杂。项目使用mybatis框架,因此借鉴PageHelper插件尝试使用mybatis的Interceptor来实现改需求。

总体思路

1、Interceptor 代码实现

package org.cnbi.project.other.sql.intercept;import cn.hutool.core.util.NumberUtil;import com.cnbi.cloud.common.core.exception.ServiceException;import com.github.pagehelper.Page;import com.github.pagehelper.util.ExecutorUtil;import com.github.pagehelper.util.MetaObjectUtil;import org.apache.ibatis.builder.annotation.ProviderSqlSource;import org.apache.ibatis.cache.CacheKey;import org.apache.ibatis.executor.Executor;import org.apache.ibatis.mapping.BoundSql;import org.apache.ibatis.mapping.MappedStatement;import org.apache.ibatis.mapping.ParameterMapping;import org.apache.ibatis.plugin.*;import org.apache.ibatis.reflection.MetaObject;import org.apache.ibatis.session.ResultHandler;import org.apache.ibatis.session.RowBounds;import org.cnbi.project.other.sql.aop.PeriodHolder;import java.util.*;import java.util.regex.Matcher;import java.util.regex.Pattern;@Intercepts({        @Signature(type = Executor.class, method = "query", args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class}),        @Signature(type = Executor.class, method = "query", args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class, CacheKey.class, BoundSql.class})})public class ParamInterceptor implements Interceptor {    private final static Pattern DW_DIMCOMPANY = Pattern.compile("dw_dimcompany", Pattern.CASE_INSENSITIVE);    private final static String REPLACE_TXT = "(select * from dw_dimcompany where cisdel = '0' and START_PERIOD <= ? and END_PERIOD > ?)";    @Override    public Object intercept(Invocation invocation) throws Throwable {        Object[] args = invocation.getArgs();        MappedStatement ms = (MappedStatement) args[0];        Object parameter = args[1];        RowBounds rowBounds = (RowBounds) args[2];        ResultHandler resultHandler = (ResultHandler) args[3];        Executor executor = (Executor) invocation.getTarget();        CacheKey cacheKey;        BoundSql boundSql;        if(args.length == 4){            boundSql = ms.getBoundSql(parameter);        } else {            boundSql = (BoundSql) args[5];        }        //获取sql语句,使用正则忽略大小写匹配        String sql = boundSql.getSql();        Matcher matcher = DW_DIMCOMPANY.matcher(sql);        //没有需要替换的表名则放行        if(!matcher.find()){            return invocation.proceed();        }       //收集占位符个数(即paramIndex 的size)以及占位符次序(slot:即参数在ParameterMappings中的顺序)        int index = 0;        ArrayList<Integer> paramIndex = new ArrayList<>();        while(matcher.find(index)){            index = matcher.end();            String sqlPart = sql.substring(0, index);            int slot = index - sqlPart.replace("?", "").length() + paramIndex.size() ;            paramIndex.add(slot);            paramIndex.add(slot + 1);        }        //替换子查询        String companyPeriodSql = matcher.replaceAll(REPLACE_TXT);        cacheKey = args.length == 4 ? executor.createCacheKey(ms, parameter, rowBounds, boundSql) : (CacheKey) args[4];        //处理参数        Object parameterObject = processParameterObject(ms, parameter, boundSql, cacheKey, paramIndex);        BoundSql companyPeriodBoundSql = new BoundSql(ms.getConfiguration(), companyPeriodSql, boundSql.getParameterMappings(), parameterObject);        Map<String, Object> additionalParameters = ExecutorUtil.getAdditionalParameter(boundSql);        //设置动态参数        for (String key : additionalParameters.keySet()) {            companyPeriodBoundSql.setAdditionalParameter(key, additionalParameters.get(key));        }        return executor.query(ms, parameterObject, RowBounds.DEFAULT, resultHandler, cacheKey, companyPeriodBoundSql);    }    public Object processParameterObject(MappedStatement ms, Object parameterObject, BoundSql boundSql, CacheKey pageKey, ArrayList<Integer> paramIndex) {        //处理参数        Map<String, Object> paramMap = null;        if (parameterObject == null) {            paramMap = new HashMap<>();        } else if (parameterObject instanceof Map) {            //解决不可变Map的情况            paramMap = new HashMap<>();            paramMap.putAll((Map) parameterObject);        } else {            paramMap = new HashMap<>();            // sqlSource为ProviderSqlSource时,处理只有1个参数的情况            if (ms.getSqlSource() instanceof ProviderSqlSource) {                String[] providerMethodArgumentNames = ExecutorUtil.getProviderMethodArgumentNames((ProviderSqlSource) ms.getSqlSource());                if (providerMethodArgumentNames != null && providerMethodArgumentNames.length == 1) {                    paramMap.put(providerMethodArgumentNames[0], parameterObject);                    paramMap.put("param1", parameterObject);                }            }            //动态sql时的判断条件不会出现在ParameterMapping中,但是必须有,所以这里需要收集所有的getter属性            //TypeHandlerRegistry可以直接处理的会作为一个直接使用的对象进行处理            boolean hasTypeHandler = ms.getConfiguration().getTypeHandlerRegistry().hasTypeHandler(parameterObject.getClass());            MetaObject metaObject = MetaObjectUtil.forObject(parameterObject);            //需要针对注解形式的MyProviderSqlSource保存原值            if (!hasTypeHandler) {                for (String name : metaObject.getGetterNames()) {                    paramMap.put(name, metaObject.getValue(name));                }            }            //下面这段方法,主要解决一个常见类型的参数时的问题            if (boundSql.getParameterMappings() != null && boundSql.getParameterMappings().size() > 0) {                for (ParameterMapping parameterMapping : boundSql.getParameterMappings()) {                    String name = parameterMapping.getProperty();                    if (!name.equals(GLOBALPERIOD)                            && paramMap.get(name) == null) {                        if (hasTypeHandler                                || parameterMapping.getJavaType().equals(parameterObject.getClass())) {                            paramMap.put(name, parameterObject);                            break;                        }                    }                }            }        }        return processPageParameter(ms, paramMap, boundSql, pageKey, paramIndex);    }    private final static String GLOBALPERIOD = "globalPeriod";    public Object processPageParameter(MappedStatement ms, Map<String, Object> paramMap, BoundSql boundSql, CacheKey pageKey, ArrayList<Integer> paramIndex) {        paramMap.put(GLOBALPERIOD, getPeriod());        //处理pageKey        pageKey.update(getPeriod());        //处理参数配置        handleParameter(boundSql, ms, paramIndex);        return paramMap;    }    protected void handleParameter(BoundSql boundSql, MappedStatement ms,  ArrayList<Integer> paramIndex) {        if (boundSql.getParameterMappings() != null) {            List<ParameterMapping> newParameterMappings = new ArrayList<>(boundSql.getParameterMappings());            for (Integer index : paramIndex) {                if(index < newParameterMappings.size()) {                    newParameterMappings.add(index, new ParameterMapping.Builder(ms.getConfiguration(), GLOBALPERIOD, String.class).build());                }else{                    newParameterMappings.add(new ParameterMapping.Builder(ms.getConfiguration(), GLOBALPERIOD, String.class).build());                }            }            MetaObject metaObject = MetaObjectUtil.forObject(boundSql);            metaObject.setValue("parameterMappings", newParameterMappings);        }    }    private final static String Q = "Q";    private final static String H = "H";    private String getPeriod(){        //使用threadlocal保存从request中获取的参数,此处不再描述        String period = PeriodHolder.getPeriod();        if(NumberUtil.isNumber(period)){            return period;        }else if(period.contains(Q)){            return period.substring(0, 4) + Integer.parseInt(period.substring(5)) * 3;        }else if(period.contains(H)){            return period.substring(0, 4) + Integer.parseInt(period.substring(5)) * 6;        }else{            throw new ServiceException("非法期间:" + period);        }    }    @Override    public Object plugin(Object target) {        return Plugin.wrap(target, this);    }    @Override    public void setProperties(Properties properties) {        //nothing to do...    }}

2、AutoConfiguration代码实现

package org.cnbi.project.autoconfig;import com.github.pagehelper.autoconfigure.PageHelperAutoConfiguration;import org.apache.ibatis.session.SqlSessionFactory;import org.cnbi.project.other.sql.intercept.ParamInterceptor;import org.mybatis.spring.boot.autoconfigure.MybatisAutoConfiguration;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.boot.autoconfigure.AutoConfigureAfter;import org.springframework.context.annotation.Configuration;import javax.annotation.PostConstruct;import java.util.Iterator;import java.util.List;@AutoConfigureAfter({MybatisAutoConfiguration.class, PageHelperAutoConfiguration.class})@Configurationpublic class ParamIntecepterAutoConfiguration {    @Autowired    private List<SqlSessionFactory> sqlSessionFactoryList;    public ParamIntecepterAutoConfiguration() {    }    @PostConstruct    public void addParamInterceptor() {        ParamInterceptor interceptor = new ParamInterceptor();        Iterator var3 = this.sqlSessionFactoryList.iterator();        while(var3.hasNext()) {            SqlSessionFactory sqlSessionFactory = (SqlSessionFactory)var3.next();            sqlSessionFactory.getConfiguration().addInterceptor(interceptor);        }    }}

mybatis interceptor 处理查询参数及查询结果

拦截器:拦截update,query方法

处理查询参数及返回结果。

@Intercepts({        @Signature(type=Executor.class,method="update",args={MappedStatement.class,Object.class}),        @Signature(type=Executor.class,method="query",args={MappedStatement.class,Object.class,RowBounds.class,ResultHandler.class})})public class EncryptInterceptor implements Interceptor {    public static final Logger logger = LoggerFactory.getLogger(EncryptInterceptor.class);      @Override    public Object intercept(Invocation invocation) throws Throwable {        dealParameter(invocation);        Object returnValue = invocation.proceed();        dealReturnValue(returnValue);        return returnValue;    }     //查询参数加密处理    private void dealParameter(Invocation invocation) {        MappedStatement statement = (MappedStatement) invocation.getArgs()[0];        String mapperl = ConfigUtils.get("mybaits.mapper.path");        String methodName = statement.getId().substring(statement.getId().indexOf(mapperl) + mapperl.length() + 1);        if (methodName.startsWith("UserBaseMapper")){            if(methodName.equals("UserBaseMapper.updateDriver")){                ((Driver) invocation.getArgs()[1]).encrypt();            }        }        logger.info("Mybatis Encrypt parameters Interceptor, method: {}, args: {}", methodName, invocation.getArgs()[1]);    }     //查询结果解密处理    private void dealReturnValue(Object returnValue){        if(returnValue instanceof ArrayList<?>){            List<?> list = (ArrayList<?>)returnValue;            for(Object val: list){                if(val instanceof Passenger){///                    //TODO                }                logger.info("Mybatis Decrypt result Interceptor, result object: {}", ToStringBuilder.reflectionToString(val));            }        }    }     @Override    public Object plugin(Object target) {        return Plugin.wrap(target, this);    }     @Override    public void setProperties(Properties properties) {     }}

添加xml配置

<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">              <property name="typeAliasesPackage" value="com.xx.model"/>              <property name="dataSource" ref="dataSource"/>              <!-- 自动扫描mapping.xml文件 -->              <property name="mapperLocations" value="classpath*:mybatis/*.xml"></property>              <property name="plugins">//拦截器插件                     <array>                            <bean class="com.github.pagehelper.PageHelper">                                   <property name="properties">                                          <value>dialect=hsqldb</value>                                   </property>                            </bean>                            <bean class="com.xx.interceptor.EncryptInterceptor">                                   <property name="properties">                                          <value>property-key=property-value</value>                                   </property>                            </bean>                     </array>              </property>       </bean>

到此,关于“如何使用mybatis的interceptor修改执行sql和传入参数”的学习就结束了,希望能够解决大家的疑惑。理论与实践的搭配能更好的帮助大家学习,快去试试吧!若想继续学习更多相关知识,请继续关注编程网网站,小编会继续努力为大家带来更多实用的文章!

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯