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 Java与Hbase相关问题
  • ¥20 bash代码推送不上去 git fetch origin master #失败了
  • ¥15 LOL外服加入了反作弊系统,现在游戏录像rofl文件离线都无法打开
  • ¥45 工程软件单片机设计课题要求
  • ¥15 在centos7安装conda
  • ¥15 c#调用yolo3 dll文件获取的数据对不上
  • ¥20 WPF 如何实现多语言,label 和cs(live Charts)中是否都能翻译
  • ¥15 STM32F103上电短路问题
  • ¥15 打开软件提示错误:failed to get wglChoosePixelFormatARB
  • ¥15 (标签-python|关键词-char)