xiaogaotongxue__ 2023-04-03 17:22 采纳率: 73.3%
浏览 25

mybatisplus的拦截器获取connection对象

BoundSql没有getConfiguration(),是版本问题吗?
我想要在mybatisPlus的拦截器中对insert,update方法进行拦截,实现对要插入的数据进行长度校验。
还有其他办法可以获取connection吗

img

版本:

img

//基于MyBatis,你可以使用拦截器(Interceptor)来实现通用的数据长度校验。以下是一个示例拦截器的实现:
//        1. 自定义拦截器类,并实现Interceptor接口。在intercept方法中,读取数据库中每个字段的最大长度限制,然后对要插入的数据进行校验并截取。
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.ResultSet;
import java.util.Properties;

import org.apache.commons.lang3.StringUtils;
import org.apache.ibatis.executor.statement.StatementHandler;
import org.apache.ibatis.mapping.BoundSql;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.plugin.*;

import javax.security.auth.login.Configuration;

@Intercepts({
        @Signature(type = StatementHandler.class, method = "update", args = {java.sql.Statement.class})
})
public class DataLengthInterceptor implements Interceptor {
    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        System.out.println("==================================hyg==========================================");
        StatementHandler statementHandler = (StatementHandler) invocation.getTarget();
        Connection connection = statementHandler.getBoundSql().getConfiguration().getEnvironment().getDataSource().getConnection();
        try {
            // 获取数据库中每个字段的最大长度限制
            DatabaseMetaData meta = connection.getMetaData();
            ResultSet resultSet = meta.getColumns(null, null, null, null);
            while (resultSet.next()) {
                String tableName = resultSet.getString("TABLE_NAME");
                String columnName = resultSet.getString("COLUMN_NAME");
                int maxLength = resultSet.getInt("COLUMN_SIZE");
                // 对要插入的数据进行校验并截取
                String sql = statementHandler.getBoundSql().getSql();
                if (sql.toLowerCase().contains(tableName.toLowerCase())
                        && sql.toLowerCase().contains(columnName.toLowerCase())) {
                    Object parameterObject = statementHandler.getBoundSql().getParameterObject();
                        if (parameterObject instanceof String) {
                        String value = (String) parameterObject;
                        String truncatedValue = StringUtils.truncate(value, maxLength);
                        statementHandler.getBoundSql().setAdditionalParameter(columnName, truncatedValue);
                    }
                }
            }
        } finally {
            connection.close();
        }
        return invocation.proceed();
    }
    @Override
    public Object plugin(Object target) {
        return Plugin.wrap(target, this);
    }
    @Override
    public void setProperties(Properties properties) {
        // You can provide some properties here if needed
    }
}

各位有什么解决办法吗?

  • 写回答

2条回答 默认 最新

  • a1767028198 2023-04-03 18:27
    关注

    你的代码有问题,不能直接通过datasource去获取connection,你这压根就没有考虑事务复用connection的情况
    推荐拦截prepare方法,直接拿到了connecttion

    img

    评论

报告相同问题?

问题事件

  • 创建了问题 4月3日

悬赏问题

  • ¥15 关于#微信#的问题:微信实名不绑卡 可以实现吗 有没有专家 可以解决
  • ¥15 (标签-考研|关键词-set)
  • ¥15 求修改代码,图书管理系统
  • ¥15 请问有没求偏多标签数据集yeast,reference,recreation,scene,health数据集。
  • ¥15 传感网应用开发单片机实训
  • ¥15 Delphi 关于sAlphaImageList使用问题
  • ¥15 寻找将CAJ格式文档转txt文本的方案
  • ¥15 shein测试开发会问些啥我是写java的
  • ¥15 关于#单片机#的问题:我有个课程项目设计,我想在STM32F103veTX单片机,M3主控模块上设计一个程序,在Keil uVision5(C语言)上代码该怎么编译?(嫌钱少我可以加钱,急急急)
  • ¥15 opnet仿真网络协议遇到问题