• mybatis-3.5.0使用插件拦截sql以及通用字段赋值

    1、添加插件配置类

    1. import org.apache.commons.lang3.StringUtils;
    2. import org.apache.commons.lang3.reflect.FieldUtils;
    3. import org.apache.ibatis.executor.Executor;
    4. import org.apache.ibatis.mapping.BoundSql;
    5. import org.apache.ibatis.mapping.MappedStatement;
    6. import org.apache.ibatis.mapping.SqlCommandType;
    7. import org.apache.ibatis.mapping.SqlSource;
    8. import org.apache.ibatis.plugin.*;
    9. import org.apache.ibatis.session.defaults.DefaultSqlSession;
    10. import org.slf4j.Logger;
    11. import org.slf4j.LoggerFactory;
    12. import org.springframework.stereotype.Component;
    13. import tk.mybatis.mapper.annotation.LogicDelete;
    14. import tk.mybatis.mapper.annotation.Version;
    15.  
    16. import java.io.File;
    17. import java.lang.reflect.Field;
    18. import java.util.*;
    19. import java.util.regex.Matcher;
    20. import java.util.regex.Pattern;
    21.  
    22. @Component
    23. @Intercepts({@Signature(
    24.         type = Executor.class,
    25.         method = "update",
    26.         args = {MappedStatement.class, Object.class}
    27. )})
    28. public class MybatisInterception implements Interceptor {
    29.  
    30.     public MybatisInterception() {
    31.     }
    32.  
    33.     public Object intercept(Invocation invocation) throws Throwable {
    34.         MappedStatement mappedStatement = (MappedStatement)invocation.getArgs()[0];
    35.         SqlCommandType sqlCommandType = mappedStatement.getSqlCommandType();
    36.  
    37.         Object parameter = invocation.getArgs()[1];
    38.         if (parameter == null) {
    39.             return invocation.proceed();
    40.         } else {
    41.             Field[] declaredFields = this.getAllFields(parameter);
    42.             Date now = new Date();
    43.             String userId = ContextUtil.getUserId();
    44.             Integer dteptId = ContextUtil.getDteptId();
    45.             Field[] var9 = declaredFields;
    46.  
    47.             if (SqlCommandType.DELETE.equals(sqlCommandType)) {
    48.  
    49.                 BoundSql boundSql = mappedStatement.getBoundSql(parameter);
    50.  
    51.                 //转换DELETE语句为逻辑删除语句
    52.                 String logicDeleteSql = convertToLogicDelete(boundSql.getSql());
    53.  
    54.                 FieldUtils.writeField(boundSql, "sql", logicDeleteSql, true);
    55.  
    56.                 MappedStatement newMappedStatement = new MappedStatement.Builder(mappedStatement.getConfiguration(),
    57.                         mappedStatement.getId(), new BoundSqlSqlSource(boundSql), mappedStatement.getSqlCommandType())
    58.                         .build();
    59.  
    60.                 invocation.getArgs()[0] = newMappedStatement;
    61.  
    62.                 return invocation.proceed();
    63.             }
    64.  
    65.             if(parameter instanceof DefaultSqlSession.StrictMap) {
    66.                 DefaultSqlSession.StrictMap strictMap = (DefaultSqlSession.StrictMap) parameter;
    67.                 Object collection = strictMap.get("collection");
    68.                 if(collection instanceof Collection) {
    69.                     for (Object obj : ((Collection) collection)) {
    70.                         Field[] fields = this.getAllFields(obj);
    71.                         for (Field field : fields) {
    72.                             extracted(sqlCommandType, obj, now, userId, dteptId, field);
    73.                         }
    74.                     }
    75.                 }
    76.             } else {
    77.                 for (Field field : var9) {
    78.                     extracted(sqlCommandType, parameter, now, userId, dteptId, field);
    79.                 }
    80.             }
    81.             return invocation.proceed();
    82.         }
    83.     }
    84.  
    85.     private void extracted(SqlCommandType sqlCommandType, Object parameter, Date now, String userId, Integer dteptId, Field field) throws IllegalAccessException {
    86.         if (field.getAnnotation(NotAutoFill.class) == null) {
    87.             Class clazz;
    88.             if (SqlCommandType.INSERT.equals(sqlCommandType)) {
    89.                 if (field.getAnnotation(CreateTime.class) != null || Objects.equals(field.getName(), "createTime")) {
    90.                     field.setAccessible(true);
    91.                     field.set(parameter, now);
    92.                 }
    93.  
    94.                 if (field.getAnnotation(Version.class) != null || Objects.equals(field.getName(), "version")) {
    95.                     field.setAccessible(true);
    96.                     field.set(parameter, 1);
    97.                 }
    98.  
    99.                 if (field.getAnnotation(LogicDelete.class) != null || Objects.equals(field.getName(), "isDeleted")) {
    100.                     field.setAccessible(true);
    101.                     field.set(parameter, 0);
    102.                 }
    103.  
    104.                 if (Objects.equals(field.getName(), "status")) {
    105.                     field.setAccessible(true);
    106.                     if (field.getType() == Integer.class) {
    107.                         field.set(parameter, 0);
    108.                     }
    109.                 }
    110.  
    111.                 if (Objects.equals(field.getName(), "createBy")) {
    112.                     field.setAccessible(true);
    113.                     if (StringUtils.isNotEmpty(userId)) {
    114.                         clazz = field.getType();
    115.                         if (clazz == Integer.class) {
    116.                             field.set(parameter, Integer.parseInt(userId));
    117.                         } else {
    118.                             field.set(parameter, userId);
    119.                         }
    120.                     }
    121.                 }
    122.  
    123.                 if (Objects.equals(field.getName(), "createDeptId")) {
    124.                     field.setAccessible(true);
    125.                     field.set(parameter, dteptId);
    126.                 }
    127.             }
    128.  
    129.             if (SqlCommandType.INSERT.equals(sqlCommandType) || SqlCommandType.UPDATE.equals(sqlCommandType)) {
    130.                 if (field.getAnnotation(UpdateTime.class) != null || Objects.equals(field.getName(), "updateTime")) {
    131.                     field.setAccessible(true);
    132.                     field.set(parameter, now);
    133.                 }
    134.  
    135.                 if (Objects.equals(field.getName(), "updateBy")) {
    136.                     field.setAccessible(true);
    137.                     clazz = field.getType();
    138.                     if (StringUtils.isNotEmpty(userId)) {
    139.                         if (clazz == Integer.class) {
    140.                             field.set(parameter, Integer.parseInt(userId));
    141.                         } else {
    142.                             field.set(parameter, userId);
    143.                         }
    144.                     }
    145.                 }
    146.             }
    147.         }
    148.     }
    149.  
    150.     public static Class findEntityClassByTableName(String tableName, String basePackage) {
    151.         ClassPathScanningCandidateComponentProvider scanner =
    152.                 new ClassPathScanningCandidateComponentProvider(false);
    153.         scanner.addIncludeFilter(new AnnotationTypeFilter(Table.class));
    154.         Set beanDefinitions = scanner.findCandidateComponents(basePackage);
    155.         for (BeanDefinition beanDefinition : beanDefinitions) {
    156.             try {
    157.                 Class clazz = Class.forName(beanDefinition.getBeanClassName());
    158.                 Table table = clazz.getAnnotation(Table.class);
    159.                 if (table != null && table.name().equals(tableName)) {
    160.                     return clazz;
    161.                 }
    162.             } catch (ClassNotFoundException e) {
    163.                continue;
    164.             }
    165.         }
    166.         return null;
    167.     }
    168.  
    169.     private String convertToLogicDelete(String sql) throws NoSuchFieldException {
    170.         Pattern pattern = Pattern.compile("DELETE\\s+FROM\\s+(\w+)\\s+WHERE\\s+(.*)", Pattern.CASE_INSENSITIVE | Pattern.DOTALL);
    171.         Matcher matcher = pattern.matcher(sql);
    172.  
    173.         if (matcher.matches()) {
    174.             String tableName = matcher.group(1); //获取表名
    175.  
    176.             Class classByTableName = findEntityClassByTableName(tableName, "com.demo.entity");
    177.             if(classByTableName == null) {
    178.                 return sql;
    179.             }
    180.  
    181.             Field isDeletedField = getDeletedField(classByTableName);
    182.  
    183.             if(isDeletedField == null || isDeletedField.getAnnotation(LogicDelete.class) == null) {
    184.                 return sql;
    185.             }
    186.  
    187.             String whereClause = matcher.group(2).trim(); //获取WHERE子句并去除两端空白字符
    188.  
    189.             // 构建逻辑删除的UPDATE语句
    190.             StringBuilder logicDeleteSql = new StringBuilder();
    191.             logicDeleteSql.append("UPDATE ").append(tableName).append(" SET ").append("is_deleted").append(" = ").append(1);
    192.  
    193.             //如果WHERE子句不为空或不全为空白字符，则加上WHERE子句
    194.             if (!whereClause.isEmpty()) {
    195.                 logicDeleteSql.append(" WHERE ").append(whereClause);
    196.             }
    197.             return logicDeleteSql.toString();
    198.         } else {
    199.             // 如果SQL语句格式不匹配，则返回原始SQL或抛出异常，这里为了简单起见，直接返回原始 SQL
    200.             return sql;
    201.         }
    202.     }
    203.  
    204.     public Object plugin(Object o) {
    205.         return Plugin.wrap(o, this);
    206.     }
    207.  
    208.     public void setProperties(Properties properties) {
    209.     }
    210.  
    211.     private Field getDeletedField(Class clazz) {
    212.         String isDeleted =  "isDeleted";
    213.         Field isDeletedField = null;
    214.         for (Class c = clazz; c != null; c = c.getSuperclass()) {
    215.             Field[] declaredFields = c.getDeclaredFields();
    216.             for (Field field : declaredFields) {
    217.                 if(field.getName().equals(isDeleted)) {
    218.                     isDeletedField = field;
    219.                     break;
    220.                 }
    221.             }
    222.         }
    223.         return isDeletedField;
    224.     }
    225.  
    226.     private Field[] getAllFields(Object object) {
    227.         Class clazz = object.getClass();
    228.         ArrayList fieldList;
    229.         for(fieldList = new ArrayList(); clazz != null; clazz = clazz.getSuperclass()) {
    230.             fieldList.addAll(new ArrayList(Arrays.asList(clazz.getDeclaredFields())));
    231.         }
    232.         Field[] fields = new Field[fieldList.size()];
    233.         fieldList.toArray(fields);
    234.         return fields;
    235.     }
    236.  
    237.     public class BoundSqlSqlSource implements SqlSource {
    238.  
    239.         private BoundSql boundSql;
    240.  
    241.         public BoundSqlSqlSource(BoundSql boundSql) {
    242.             this.boundSql = boundSql;
    243.         }
    244.  
    245.         @Override
    246.         public BoundSql getBoundSql(Object parameterObject) {
    247.             return boundSql;
    248.         }
    249.     }
    250. }
      251.  

      2、修改mybatis-config.xml
       

      切身经历但不一定是真相：项目已经有拦截器和mybatis-config.xml比如在公共依赖中，再新建mybatis-config.xml会存在覆盖xml配置的情况，虽然拦截器都生效了。如果命名为mybatis-config-xxx.xml等拦截器就没有生效。通过采用配置类的形式，暂时未发现不良影响。
       

      1. "1.0" encoding="utf-8"?>
      2. configuration PUBLIC "-//mybatis.org//DTD Config 3.0//EN" "http://mybatis.org/dtd/mybatis-3-config.dtd">
      3. <configuration>
      4.     
      5.     <plugins>
      6.         <plugin interceptor="xxx.xxx.MybatisInterceptor">
      7.             
      8.             
      9.         plugin>
      10.     plugins>
      11.  
      12.     
       

      3、配置类，一般叫MybatisConfig，显然不可重名。于是另起炉灶。
       

      1. @Configuration
      2. @AutoConfigureAfter({PageHelperAutoConfiguration.class})
      3. public class MybatisFeatureConfig {
      4.     @Autowired
      5.     private List sqlSessionFactoryList;
      6.  
      7.     public MybatisFeatureConfig() {
      8.     }
      9.  
      10.     @PostConstruct
      11.     public void addMyInterceptor() {
      12.         MybatisInterception e = new MybatisInterception();
      13.         Iterator var2 = this.sqlSessionFactoryList.iterator();
      14.  
      15.         while(var2.hasNext()) {
      16.             SqlSessionFactory sqlSessionFactory = (SqlSessionFactory)var2.next();
      17.             sqlSessionFactory.getConfiguration().addInterceptor(e);
      18.         }
      19.  
      20.     }
      21. }
       

      4、基本思路，如果使用mybatis-plus我尚未注意过批量操作的问题，所以不知会有什么坑
      
                
                    
                

                
    251. 

                    相关阅读:
      
                    
如何用神经网络预测数据,人工神经网络分析方法                            
      
【U8+】用友U8-UFO录入关键字，计算后乱码                            
      
[python-大语言模型]从浅到深一系列学习笔记记录                            
      
ELK介绍以及搭建                            
      
包分配并不是个好制度                            
      
c++学习--第二部分                            
      
ifconfig看不见自己外网地址？                            
      
AI加速（八）| 循环展开Unrooling——你肯定能学会的程序加速方法                            
      
ceph 线程池分析                            
      
第十四届蓝桥杯模拟赛（第二期）——C语言版                            
      
                          
                
      252. 
                
    252. 
                    原文地址：https://blog.csdn.net/z275598733/article/details/137949421
                
      253.