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

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

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

@Mapper
public interface FileManagerMapper extends BaseMapper {
 
}

在实际场景中，我们需要给每个表中都加一些公共字段，往往值都跟环境变量有关系，如果每个对象都要把这些公共字段都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包起来就可以了，不要影响正常流程。