wang-hao-jie
2022-06-06 4e837c1e8c6f8a7252fb95776a1530ab737bb684
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
package cn.exrick.xboot.core.config.mybatisplus;
 
import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.reflection.MetaObject;
import org.springframework.security.authentication.AnonymousAuthenticationToken;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.stereotype.Component;
 
import java.util.Date;
 
/**
 * 字段填充审计
 * @author Exrick
 */
@Slf4j
@Component
public class MyMetaObjectHandler implements MetaObjectHandler {
 
    @Override
    public void insertFill(MetaObject metaObject) {
 
        setFieldUser("createBy", metaObject);
        this.setFieldValByName("createTime", new Date(), metaObject);
    }
 
    @Override
    public void updateFill(MetaObject metaObject) {
 
        setFieldUser("updateBy", metaObject);
        this.setFieldValByName("updateTime", new Date(), metaObject);
    }
 
    public void setFieldUser(String field, MetaObject metaObject) {
 
        Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
        if (authentication != null && authentication.isAuthenticated() && authentication.getName() != null
                && !(authentication instanceof AnonymousAuthenticationToken)) {
            this.setFieldValByName(field, authentication.getName(), metaObject);
        }
    }
}