Mybatis拦截提取SQL文

最新推荐文章于 2024-08-25 17:06:06 发布

科大刘德华

最新推荐文章于 2024-08-25 17:06:06 发布

阅读量999

点赞数

分类专栏： mybatis 文章标签： mybatis sql java

本文链接：https://blog.csdn.net/apple_52952947/article/details/129199758

版权

mybatis 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

该文章介绍了一个Mybatis的拦截器实现，主要功能是对Executor类的update方法进行拦截，当执行SQL更新时，检查参数并截断超过1000个字符的字段值，同时记录操作日志，包括用户信息、SQL内容和操作时间。如果SQL不涉及tbl_sys_operation_log表，则会创建一个新的操作日志条目。

摘要由CSDN通过智能技术生成

@Intercepts({ @Signature(type = Executor.class, method = "update", args = { MappedStatement.class, Object.class }), })
public class MybatisInterceptor implements Interceptor {
	@Override
	public Object intercept(Invocation invocation) throws Throwable {
		try {
			MappedStatement mappedStatement = (MappedStatement) invocation.getArgs()[0];
			Object parameter = null;
			if (invocation.getArgs().length > 1) {
				parameter = invocation.getArgs()[1];
			}
			LoginUser user = LoginUser.getLoginUser();
			BoundSql boundSql = mappedStatement.getBoundSql(parameter);
			String sql = boundSql.getSql().replaceAll("[\\s]+", " ");
			List<ParameterMapping> parameterMappings = boundSql.getParameterMappings();
			if (parameterMappings != null && parameterMappings.size() > 0) {
				Configuration configuration = mappedStatement.getConfiguration();
				Object parameterObject = boundSql.getParameterObject();
				MetaObject metaObject = configuration.newMetaObject(parameterObject);
				for (ParameterMapping parameterMapping : parameterMappings) {
					String propertyName = parameterMapping.getProperty();
					if (metaObject.hasGetter(propertyName)) {
						Object obj = metaObject.getValue(propertyName);
						String columValue = "";
						if (obj != null && obj.toString().length() > 1000) {
							columValue = obj.toString().substring(0, 999);
						} else if (obj != null && obj.toString().length() < 1000) {
							columValue = obj.toString();
						}
						sql = sql.replaceFirst("\\?", columValue);
					} else if (boundSql.hasAdditionalParameter(propertyName)) {
						Object obj = boundSql.getAdditionalParameter(propertyName);
						String columValue = "";
						if (obj != null && obj.toString().length() > 1000) {
							columValue = obj.toString().substring(0, 999);
						} else if (obj != null && obj.toString().length() < 1000) {
							columValue = obj.toString();
						}
						sql = sql.replaceFirst("\\?", columValue);
					}
				}
			}
			log.info("SQL: {} , user: {}, date : {}", sql, null != user ? user.getName() : "", new Date());
			if (!sql.contains("tbl_sys_operation_log")) {
				DboSysOperationLog log =  new DboSysOperationLog();
				if (user != null) {
					log.setUserId(user.getId());
				} else {
					log.setUserId(8505);
				}
				log.setSqlText(sql);
				log.setOperateTime(new Date());
				log.setModuleName("maintenance");
				DboSysOperationLogService.insert(log);
			}
		} catch (Exception e) {
			e.printStackTrace();
		}
		return invocation.proceed();
	}

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

	@Override
	public void setProperties(Properties properties) {
	}

}

使用拦截器进行SQL的拦截保存