栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Java

mybtis自定义拦截器实现忽略大小写的模糊查询

Java 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

mybtis自定义拦截器实现忽略大小写的模糊查询

项目场景:

springboot项目中需要实现忽略大小写的模糊查询,底层数据库为pgsql

场景分析:

项目集成了mybatis,模糊查询一般使用like关键字,且pgsql本身提供ilike关键字实现忽略大小写的模糊匹配,因此可以考虑采用mybatis的拦截器,对特定sql进行修改。

解决方案:
package com.xxx;

import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.executor.statement.StatementHandler;
import org.apache.ibatis.mapping.BoundSql;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.SqlCommandType;
import org.apache.ibatis.plugin.*;
import org.apache.ibatis.reflection.DefaultReflectorFactory;
import org.apache.ibatis.reflection.MetaObject;
import org.apache.ibatis.reflection.SystemMetaObject;
import org.springframework.stereotype.Component;
import java.lang.reflect.Field;
import java.sql.Connection;
import java.util.Properties;


@Slf4j
@Component
@Intercepts({@Signature(type = StatementHandler.class, method = "prepare", args = {Connection.class, Integer.class})})
public class FuzzyQueryInterceptor implements Interceptor{
    private static final String FUZZY_QUERY_KEY = "like";
    private static final String IGNORE_FUZZY_QUERY_KEY = "ilike";

    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        //拦截StatementHandler,获取Mybatis对象属性
        StatementHandler statementHandler = (StatementHandler) invocation.getTarget();
        MetaObject metaObject = MetaObject
                .forObject(statementHandler, SystemMetaObject.DEFAULT_OBJECT_FACTORY, SystemMetaObject.DEFAULT_OBJECT_WRAPPER_FACTORY,
                        new DefaultReflectorFactory());
        MappedStatement mappedStatement = (MappedStatement) metaObject.getValue("delegate.mappedStatement");
        //执行mapper方法的全路径名
        String id = mappedStatement.getId();
        //只拦截查询语句,将like关键字替换成ilike,以忽略大小写
        if (SqlCommandType.SELECT.equals(mappedStatement.getSqlCommandType())){
            BoundSql boundSql = statementHandler.getBoundSql();
            String sql = boundSql.getSql();
            if (sql.contains(FUZZY_QUERY_KEY) || sql.contains(FUZZY_QUERY_KEY.toUpperCase())){
                String mSql = sql.contains(FUZZY_QUERY_KEY) ? sql.replace(FUZZY_QUERY_KEY,IGNORE_FUZZY_QUERY_KEY)
                        : sql.replace(FUZZY_QUERY_KEY.toUpperCase(),IGNORE_FUZZY_QUERY_KEY.toUpperCase());
                log.debug(id + ":原始查询语句 ==> " + sql);
                log.debug(id + ":替换后的查询语句 ==> " + mSql);
                //通过反射修改sql语句
                Field field = boundSql.getClass().getDeclaredField("sql");
                field.setAccessible(true);
                field.set(boundSql, mSql);
            }
        }
        //执行结果
        return invocation.proceed();
    }

    @Override
    public Object plugin(Object target) {
        return Plugin.wrap(target, this);
    }

    @Override
    public void setProperties(Properties properties) {
    }
}

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/874054.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号