利用aop+反射拦截mybatisplus的insert方法补充公共属性

当我们持久层框架使用mybatis的时候，可以继承mybatisplus中的BaseMapper接口，这样的话我们就可以避免编写mapper.xml，就可以获得CRUD功能，下面放一张源码截图

可以看到，该接口中定义了一些简单的增删改查方法，我们可以直接使用，只需要在我们自定义的mapper接口类中继承该接口，如下：

  

   

    

     

    
    

     

      @Mapper
     
    
   

    

     

    
    

     

      public 
      interface 
      FileManagerMapper 
      extends 
      BaseMapper<FileManager> {
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      }
     
    
  

在实际场景中，我们需要给每个表中都加一些公共字段，往往值都跟环境变量有关系，如果每个对象都要把这些公共字段都set一下，不仅繁琐，而且还可能会遗漏。

因为这里只有接口定义，实现是mybatis框架执行的，所以我们没有办法通过java继承的方式set字段，这里考虑使用spring框架中的aop来拦截insert方法，然后利用反射动态的赋值。

话不多说，直接上代码：

1、首先我们定义一个aop配置类，注解加上就可以了

  

   

    

     

    
    

     

      @Aspect
     
    
   

    

     

    
    

     

      @Component
     
    
   

    

     

    
    

     

      public 
      class 
      MybatisAop {
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      }
     
    
  

2、然后我们定义一个切点

  

   

    

     

    
    

     

      @Pointcut("execution(* com.cxc.mapper.*.insert(..))")
     
    
   

    

     

    
    

     
    
      private 
      void 
      insert
      () {
     
    
   

    

     

    
    

     

      }
     
    
  

注意，这里的拦截的方法是你自己定义的mapper接口类中的方法，而不是com.baomidou.mybatisplus.core.mapper.BaseMapper接口类中的方法

3、定义增强方法，因为我们只需要对属性赋值，所以我们使用前置通知即可，假设我们要补充的字段名是loginName

  

   

    

     

    
    

     

      @Before(value = "insert()")
     
    
   

    

     

    
    

     
    
      public 
      void 
      insertBefore
      (JoinPoint point) {
     
    
   

    

     

    
    

     

              Object[] args = point.getArgs();
     
    
   

    

     

    
    

     
        
      Object 
      arg 
      = args[
      0];
     
    
   

    

     

    
    

     
        
      try {
     
    
   

    

     

    
    

     

                  Field[] fields = arg.getClass().getDeclaredFields();
     
    
   

    

     

    
    

     
            
      for (
      int 
      i 
      = 
      0; i < fields.length; i++) {
     
    
   

    

     

    
    

     
                
      if(fields[i].getName().equals(
      "loginName")) {
     
    
   

    

     

    
    

     

                          fields[i].set(arg,
      "张三");
     
    
   

    

     

    
    

     

                      }
     
    
   

    

     

    
    

     

                  }
     
    
   

    

     

    
    

     

              }
      catch (Exception e) {
     
    
   

    

     

    
    

     

                  e.printStackTrace();
     
    
   

    

     

    
    

     

              }
     
    
   

    

     

    
    

     

      }
     
    
  
 

  
 

注意，使用Field的方式只能给public定义的属性赋值，如果是非public属性，我们需要使用set方法赋值

  

   

    

     

    
    

     

      @Before(value = "insert()")
     
    
   

    

     

    
    

     
    
      public 
      void 
      insertBefore
      (JoinPoint point) {
     
    
   

    

     

    
    

     

              Object[] args = point.getArgs();
     
    
   

    

     

    
    

     
        
      Object 
      arg 
      = args[
      0];
     
    
   

    

     

    
    

     
        
      try {
     
    
   

    

     

    
    

     
            
      Method 
      setTyun_id 
      = arg.getClass().getMethod(
      "setLoginName", String.class);
     
    
   

    

     

    
    

     

                  setTyun_id.invoke(arg, 
      "张三");
     
    
   

    

     

    
    

     

              }
      catch (Exception e) {
     
    
   

    

     

    
    

     

                  e.printStackTrace();
     
    
   

    

     

    
    

     

              }
     
    
   

    

     

    
    

     

          }
     
    
  

这里也需要注意一下，如果我们的类中没有set方法，会抛异常，所以我们使用try-catch包起来就可以了，不要影响正常流程。