jpa Auditor 自动赋值与自定义 @CreatedBy @LastModifiedBy @CreatedDate @LastModifiedDate

2021/09/08 11:17
阅读数 76

 

在spring jpa audit 中,在字段或者方法上使用注解@CreatedDate@CreatedBy@LastModifiedDate@LastModifiedBy,当进行实体插入或者更新可以自动赋值

  • @CreatedDate 创建时间

  • @CreatedBy 创建人

  • @LastModifiedDate 更新时间

  • @LastModifiedBy 更新人

使用:

1.定义实体类,并使用注解标注字段

import lombok.Data;
import org.springframework.data.annotation.*;
import org.springframework.data.mongodb.core.mapping.Field;

import java.time.LocalDateTime;

@Data
public class BaseEntity {
  @Id private String id;
  @Field @CreatedBy private String createUserId;

  @Field @LastModifiedBy private String updateUserId;

  @Field @CreatedDate private LocalDateTime createTime; // 创建时间

  @Field @LastModifiedDate private LocalDateTime updateTime; // 修改时间
}

2.添加 AuditorAware配置,设置默认用户

@Configuration
@EnableMongoAuditing(auditorAwareRef = "jpaAuditorAware")//使用mongo,也可以使用其他,如jpa(mysql)
public class JpaAuditorAware implements AuditorAware<String> {
  @Override
  public String getCurrentAuditor() {
    return "system";
  }
}

这里是直接设置了一个默认值,正常来说,应该使用springsecurity或者shiro,从请求token中获取当前登录用户,如:

public final class SecurityUtils {

  private SecurityUtils() {}

  /**
   * 根据 Authorization 获取当前登录的用户
   *
   * @return 返回用户id
   */
  public static String getCurrentUserId() {
    SecurityContext securityContext = SecurityContextHolder.getContext();
    Authentication authentication = securityContext.getAuthentication();
    String userId = null;
    if (authentication != null) {
      if (authentication.getPrincipal() instanceof UserDetails) {
        UserDetails springSecurityUser = (UserDetails) authentication.getPrincipal();
        userId = springSecurityUser.getUsername();
      } else if (authentication.getPrincipal() instanceof String) {
        userId = (String) authentication.getPrincipal();
      }
    }
    return userId;
  }
}

//设置Auditor
@Component
public class SpringSecurityAuditorAware implements AuditorAware<String> {

  @Override
  public String getCurrentAuditor() {
    String userId= SecurityUtils.getCurrentUserId();
    return userId;
  }
}

3.新建 User类,继承BaseEntity

@Data
@Document(collection = "stu")
public class Stu extends BaseEntity
{ String name; String clazz; }

4.UserRepository 继承MongoRepository,连接mongo数据库

测试:

@RequestMapping("/user")
  public User saveUser(String name) {
    User user = new User();
    user.setName(name);
    return userRepo.save(user);
  }

 

发现4个字段都自动赋值了。

但是有个问题,有些场景是这样的:

User user = new User();
    user.setName(name);
    user.setCreateUserId("hahaha");//手动设置userId

等执行完数据库插入后,发现createUserId的值不是hahaha,还是上面默认的system 

 

解决方法:实现Auditable接口,通过重载来自定义这些方法

@Data
public class Base extends BaseEntity implements Auditable<String, String> { 

  @Override
  public String getCreatedBy() {
    return this.getCreateUserId();
  }

  @Override
  public void setCreatedBy(String s) {
    //如果已经设置了createUserId,则取当前设置的;否则,使用当前登录的用户id(即参数s)   下同。
    String createUserId = !StringUtils.isEmpty(getCreateUserId()) ? getCreateUserId() : s; setCreateUserId(createUserId);
  }

  @Override
  public DateTime getCreatedDate() {
    return new DateTime(
        this.getCreateTime().atZone(ZoneId.systemDefault()).toInstant().toEpochMilli());
  }

  @Override
  public void setCreatedDate(DateTime dateTime) {
    setCreateTime(
        Instant.ofEpochMilli(dateTime.getMillis())
            .atZone(ZoneId.systemDefault())
            .toLocalDateTime());
  }

  @Override
  public String getLastModifiedBy() {
    return this.getUpdateUserId();
  }

  @Override
  public void setLastModifiedBy(String s) {
    String createUserId = !StringUtils.isEmpty(getUpdateUserId()) ? getUpdateUserId() : s;
    setUpdateUserId(createUserId);
  }

  @Override
  public DateTime getLastModifiedDate() {
    return new DateTime(
        this.getUpdateTime().atZone(ZoneId.systemDefault()).toInstant().toEpochMilli());
  }

  @Override
  public void setLastModifiedDate(DateTime dateTime) {
    setUpdateTime(
        Instant.ofEpochMilli(dateTime.getMillis())
            .atZone(ZoneId.systemDefault())
            .toLocalDateTime());
  }

  @Override
  public boolean isNew() {
    return this.getId() == null;
  }
}

 

测试:新建实体类stu,继承Base

@Data
@Document(collection = "stu")
public class Stu extends Base {
  String name;
  String clazz;
}

web rest类:

@RequestMapping("/stu")
  public String saveStu(String name) throws JsonProcessingException {
    Stu stu = new Stu();
    stu.setName(name);
    stu.setClazz(random.nextInt() + "");
    stu.setCreateUserId(name);//自定义createUserId
    stu = stuRepo.save(stu);
    return om.writeValueAsString(stu);
  }

 

展开阅读全文
打赏
0
0 收藏
分享
加载中
更多评论
打赏
0 评论
0 收藏
0
分享
返回顶部
顶部