基于Mybatis拦截器开发自定义拦截器,实现记录数据表最后更新时间

场景:
记录有过增删改操作的数据表及最后更新时间,实现数据增量同步。

解决方案:基于Mybatis拦截器开发自定义拦截器

1.实现Mybatis拦截器Interceptor,重写intercept方法

2.使用druid解析sql,获取sql中涉及的表

3.实现自定义逻辑

 拦截器代码如下:

package com.cms.filter;

import com.alibaba.druid.sql.SQLUtils;
import com.alibaba.druid.sql.ast.SQLStatement;
import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlSchemaStatVisitor;
import com.alibaba.druid.stat.TableStat;
import com.alibaba.druid.util.JdbcConstants;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.cms.common.utils.StringUtils;
import com.cms.common.utils.spring.SpringUtils;
import com.cms.domain.entity.systemData.SystemData;
import com.cms.mapper.systemData.SystemDataMapper;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.executor.Executor;
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.springframework.stereotype.Component;

import java.util.*;
import java.util.stream.Collectors;

/**
 * @author daixin
 * @version 1.0
 * @description: TODO
 * @date 2022/9/28 13:48
 */
@Slf4j
@Component
@Intercepts({@Signature(type = Executor.class, method = "update", args = {MappedStatement.class, Object.class})})
public class ExecutorDataVaryInterceptor implements Interceptor {

    /**
     * 拦截方法逻辑
     */
    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        //方法参数
        Object[] args = invocation.getArgs();
        MappedStatement mappedStatement = (MappedStatement) args[0];
        Object parameterObject = args[1];

        //获取Sql命令类型UNKNOWN, INSERT, UPDATE, DELETE, SELECT, FLUSH
        SqlCommandType sqlCommandType = mappedStatement.getSqlCommandType();
        String commandType = sqlCommandType.name();
        //获取拦截的Sql
        BoundSql boundSql = mappedStatement.getSqlSource().getBoundSql(parameterObject);
        String sql = boundSql.getSql();
        //判断新增和修改操作,执行自定义逻辑
        if("INSERT".equalsIgnoreCase(commandType)||"UPDATE".equalsIgnoreCase(commandType)){
            //解析Sql, 获取涉及的数据表
            List<String> tableNameList =  getTableNameBySql(sql);
            //判断sql执行的数据表是否在需要管理的白名单内
            QueryWrapper<SystemData> queryWrapper = new QueryWrapper<>();
            queryWrapper.lambda().in(SystemData::getTableName,tableNameList);
            SystemDataMapper systemDataMapper = SpringUtils.getBean(SystemDataMapper.class);
            List<SystemData> systemDataList = systemDataMapper.selectList(queryWrapper);
            //在白名单内的表, 管理更新时间
            if(!StringUtils.isEmpty(systemDataList)){
                List<String> updateTimeTables = systemDataList.stream().map(SystemData::getTableName).collect(Collectors.toList());
                systemDataMapper.saveUpdateTimeBatch(updateTimeTables);
            }
        }
        Object result = invocation.proceed();
        return result;
    }

    /**
     * 解析Sql,获取涉及的数据表
     * @param sql
     * @return
     */
    private List<String> getTableNameBySql(String sql) {
        try {
            String dbType = String.valueOf(JdbcConstants.MYSQL);
            List<String> tableNameList = new ArrayList<>();
            log.info("格式化后的sql:{}",SQLUtils.format(sql, dbType));

            List<SQLStatement> sqlStatementList = SQLUtils.parseStatements(sql, dbType);
            if (CollectionUtils.isEmpty(sqlStatementList)) {
                return Collections.emptyList();
            }
            for (SQLStatement sqlStatement : sqlStatementList) {
                MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor();
                sqlStatement.accept(visitor);
                Map<TableStat.Name, TableStat> tables = visitor.getTables();
                for (TableStat.Name name : tables.keySet()) {
                    String tableName = name.getName();
                    if (StringUtils.isNotBlank(tableName)) {
                        tableNameList.add(tableName);
                    }
                }
            }
            log.info("解析sql获取的表名:{}",tableNameList);
            return tableNameList;
        } catch (Exception e) {
            log.error("===================异常SQL=================={}",sql);
            log.error(e.getMessage(),e);
        }
        return Collections.emptyList();
    }

}

mybatis-config.xml注入拦截器

<plugins>
    <plugin interceptor="com.cms.filter.ExecutorDataVaryInterceptor">
    </plugin>
</plugins>

spring boot集成druid所需依赖

<dependency>
    <groupId>com.alibaba</groupId>
    <artifactId>druid-spring-boot-starter</artifactId>
    <version>${druid.version}</version>
</dependency>
<dependency>
    <groupId>com.alibaba</groupId>
    <artifactId>druid</artifactId>
    <version>1.2.8</version>
    <scope>compile</scope>
</dependency>

同理,也可实现日志收集等功能。 

  • 1
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
MyBatis拦截和自定义注解是MyBatis框架中的两个重要特性。下面我会分别解释它们的作用和用法。 MyBatis拦截是一种机制,可以在执行SQL语句的过程中对其进行拦截和修改。它提供了一种方便的方式来扩展和自定义MyBatis的功能。拦截可以在SQL语句执行前后、参数设置前后、结果集处理前后等关键点进行拦截,并对其进行修改或增强。 要实现一个MyBatis拦截,你需要实现`Interceptor`接口,并重写其中的方法。其中最重要的方法是`intercept`,它接收一个`Invocation`对象作为参数,通过该对象你可以获取到当前执行的SQL语句、参数等信息,并可以对其进行修改。另外还有`plugin`方法和`setProperties`方法用于对拦截进行初始化。 自定义注解是一种用于标记和配置特定功能的注解。在MyBatis中,你可以使用自定义注解来配置一些特殊的功能,比如动态SQL的条件判断、结果集映射等。通过自定义注解,你可以将一些常用的功能封装成注解,并在需要时直接使用。 要使用自定义注解,你需要先定义一个注解,并在相应的地方使用该注解。然后通过MyBatis的配置文件或者Java代码进行配置,告诉MyBatis如何处理这些注解。在MyBatis的执行过程中,它会根据注解的配置来动态生成相应的SQL语句或者进行特定的处理。 总结一下,MyBatis拦截和自定义注解是MyBatis框架中的两个重要特性。拦截可以对SQL语句进行拦截和修改,自定义注解可以用于配置一些特殊功能。它们都提供了一种扩展和自定义MyBatis功能的方式。如果你有具体的问题或者需要更详细的示例代码,欢迎继续提问!
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值