java：对mysql切面实现创建人创建时间更新人更新时间的赋值

背景：日常开发中，对mysql基础数据表进行设计时，几乎每个表都要有创建人创建时间更新人更新时间这四个标准字段，那么每次进行curd的时候，如果每次都要对此4个字段进行赋值，代码重复率会非常高，而且开发效率会降低，一旦漏掉操作容易导致bug。

github:GitHub - ChaShiyu/rutui-mysql-aop: 对mysql切面实现创建人创建时间更新人更新时间的赋值

方案：在controller方法上通过自定义注解实现aop切面。

pom依赖

<!-- 核心包，千万别引错了，不要引入aspectjrt，aspectjrt只是基础包，无法实现pointCut -->
<dependency>
    <groupId>org.aspectj</groupId>
    <artifactId>aspectjweaver</artifactId>
</dependency>
<!-- fastjson -->
<dependency>
    <groupId>com.alibaba</groupId>
    <artifactId>fastjson</artifactId>
    <version>1.2.76</version>
</dependency>
<!-- lombok -->
<dependency>
    <groupId>org.projectlombok</groupId>
    <artifactId>lombok</artifactId>
    <optional>true</optional>
</dependency>

步骤：

1、定义基类：

@Data
public class BaseInfo implements Serializable {
    private Long creatorId;
    private String creatorName;
    private Date createTime;
    private Long updaterId;
    private String updaterName;
    private Date updateTime;
}

2、定义DTO：

@Data
public class Member extends BaseEntity {
    private String name;
}

3、自定义切面注解：

@Documented
@Target({ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface BaseInfoAnnotation {
    OperateTypeEnum value();
}

4、操作类型枚举：

public enum OperateTypeEnum {
    ADD,
    DELETE,
    UPDATE;
}

5、自定义 aop 操作：

@Aspect
@Component
public class BaseInfoAop {

    @Pointcut("@annotation(cn.rutui.mysqlaop.aop.BaseInfoAnnotation))")
    public void setBaseInfo() {
    }

    @Around("setBaseInfo()")
    public Object doAround(ProceedingJoinPoint joinPoint) throws Throwable {
        MethodSignature signature = (MethodSignature) joinPoint.getSignature();
        BaseInfoAnnotation annotation = signature.getMethod().getAnnotation(BaseInfoAnnotation.class);
        OperateTypeEnum value = annotation.value();
        Object[] args = joinPoint.getArgs();
        if (args.length > 0) {
            BaseInfo arg = (BaseInfo) args[0];
            Map<String, String> userInfo = getUserInfo();
            if (value == OperateTypeEnum.ADD) {
                arg.setCreatorId(Long.valueOf(userInfo.get("id")));
                arg.setCreatorName(userInfo.get("name"));
                arg.setOptTime(new Date());
            } else if (value == OperateTypeEnum.UPDATE || value == OperateTypeEnum.DELETE) {
                arg.setCreatorId(Long.valueOf(userInfo.get("id")));
                arg.setCreatorName(userInfo.get("name"));
                arg.setOptTime(new Date());
            }
        }
        return joinPoint.proceed(args);
    }

    private Map<String, String> getUserInfo() {
        // get user info by session
        HashMap<String, String> map = new HashMap<>();
        map.put("id", "10");
        map.put("name", "namename");
        return map;
    }
}

6、定义Controller

@RestController
@RequestMapping("/member")
public class MemberController {
    @PostMapping("")
    @BaseInfoAnnotation(OperateTypeEnum.ADD)
    public String save(@RequestBody Member member) {
        System.out.println(JSON.toJSONString(member));
        return "SUCCESS";
    }
}

7、结果

{"creatorId":10,"creatorName":"namename","optTime":1670494115667}