利用的是SpringBoot的Aop思想和自定义注解和反射机制的方法来实现
项目中我涉及公共字段的有createTime、updateTime、createUser、updateUser
/**
* 数据库操作类型 使用的是枚举方法
*/
public enum OperationType {
/**
* 更新操作
*/
UPDATE,
/**
* 插入操作
*/
INSERT
}
/**
* 自定义注解,用于标识某个方法需要进行功能字段自动填充处理
*/
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface AutoFill {
//数据库操作类型:UPDATE INSERT
OperationType value();
}
/**
* 自定义切面,实现公共字段字段填充处理逻辑
*/
@Aspect
@Component
@Slf4j
public class AutoFillAspect {
/**
* 切入点
*/
@Pointcut("execution(* com.sky.mapper.*.*(..)) && @annotation(com.sky.annotation.AutoFill)")
public void autoFillPointCut(){}
/**
* 前置通知
*/
@Before("autoFillPointCut()")
public void autoFill(JoinPoint joinPoint){
log.info("开始进行公共字段的填充...");
//获取到当前被拦截方法上的数据库操作类型
MethodSignature signature = (MethodSignature) joinPoint.getSignature();
AutoFill autoFill = signature.getMethod().getAnnotation(AutoFill.class);
OperationType type = autoFill.value();
//获取到当前被拦截方法的参数--实体对象
Object[] args = joinPoint.getArgs();
if(args == null || args.length!=0){
return;
}
Object enity = args[0];
//准备赋值的数据
Long id = BaseContext.getCurrentId();
LocalDateTime localDateTime = LocalDateTime.now();
//根据当前不同的操作类型,为对应额属性通过反射来赋值
if(type == OperationType.INSERT){
//为四个公共字段赋值
try {
Method setCreateTime = enity.getClass().getMethod(AutoFillConstant.SET_CREATE_TIME, LocalDateTime.class);
Method setUpdateTime = enity.getClass().getMethod(AutoFillConstant.SET_UPDATE_TIME, LocalDateTime.class);
Method setCreateUser = enity.getClass().getMethod(AutoFillConstant.SET_CREATE_USER, Long.class);
Method setUpdateUser = enity.getClass().getMethod(AutoFillConstant.SET_UPDATE_USER, Long.class);
setCreateTime.invoke(enity,localDateTime);
setUpdateTime.invoke(enity,localDateTime);
setCreateUser.invoke(enity,id);
setUpdateUser.invoke(enity,id);
} catch (Exception e) {
e.printStackTrace();
}
}else if(type == OperationType.UPDATE){
//为两个公共字段赋值
try {
Method setUpdateTime = enity.getClass().getMethod(AutoFillConstant.SET_UPDATE_TIME,LocalDateTime.class);
Method setUpdateUser = enity.getClass().getMethod(AutoFillConstant.SET_UPDATE_USER,Long.class);
setUpdateTime.invoke(enity,localDateTime);
setUpdateUser.invoke(enity,id);
} catch (Exception e) {
e.printStackTrace();
}
}
}
@Insert("insert into category (type, name, sort, status, create_time, update_time, create_user, update_user) " +
"values (#{type},#{name},#{sort},#{status},#{createTime},#{updateTime},#{createUser},#{updateUser});")
@AutoFill(value = OperationType.INSERT)
void save(Category category);
@Delete("delete from category where id = #{id}")
void deleteById(Long id);
@AutoFill(value = OperationType.UPDATE)
void update(Category category);