如何手动覆盖JPA Audit中@CreatedBy注解的自动赋值？

问题根源在于JPA Auditing的默认逻辑：AuditingEntityListener会在实体持久化（persist）或更新（merge）的生命周期回调中，强制用AuditAware提供的值覆盖对应字段，无论你是否提前手动设置了值。以下是几种可行的解决方案：

方案1：自定义审计监听器，优先保留手动赋值 #

继承默认的AuditingEntityListener，重写回调方法，仅当字段未赋值时才执行自动审计逻辑，手动设置的值会被优先保留。

自定义监听器代码 #

public class CustomAuditingEntityListener extends AuditingEntityListener {

    @Override
    public void onPrePersist(Object entity) {
        if (entity instanceof BaseEntity baseEntity) {
            // 仅当createdBy/createdDate未赋值时，触发自动审计
            boolean needAuditCreate = baseEntity.getCreatedBy() == null || baseEntity.getCreatedDate() == null;
            // 始终确保modifiedBy/modifiedDate被更新（可根据需求调整）
            boolean needAuditModify = baseEntity.getModifiedBy() == null || baseEntity.getModifiedDate() == null;

            if (needAuditCreate) {
                super.onPrePersist(entity);
            } else if (needAuditModify) {
                // 单独更新修改相关字段
                super.setLastModifiedDate(entity, AuditingHandler.getLastModifiedDate(entity.getClass()));
                super.setLastModifiedBy(entity, AuditingHandler.getLastModifiedBy(entity.getClass()));
            }
        } else {
            super.onPrePersist(entity);
        }
    }

    @Override
    public void onPreUpdate(Object entity) {
        if (entity instanceof BaseEntity baseEntity) {
            // 如果手动设置了modifiedBy，仅自动更新modifiedDate；否则执行完整自动审计
            if (baseEntity.getModifiedBy() != null) {
                super.setLastModifiedDate(entity, AuditingHandler.getLastModifiedDate(entity.getClass()));
            } else {
                super.onPreUpdate(entity);
            }
        } else {
            super.onPreUpdate(entity);
        }
    }
}

修改实体类的监听器配置 #

将BaseEntity的@EntityListeners替换为自定义监听器：

@Data
@SuperBuilder
@EntityListeners({CustomAuditingEntityListener.class})
public abstract class BaseEntity {
    // 原有字段代码不变
}

方案2：临时切换审计上下文（适合特定业务场景） #

如果仅在少数场景下需要手动赋值，可以临时修改AuditAware的返回值，让它优先使用你指定的用户ID，操作完成后恢复原有上下文。

修改AuditAware实现 #

@Component
public class ApplicationAuditAware implements AuditorAware<Integer> {
    // 用ThreadLocal存储临时审计者，保证线程安全
    private final ThreadLocal<Integer> tempAuditor = new ThreadLocal<>();

    @Override
    public Optional<Integer> getCurrentAuditor() {
        // 优先返回临时设置的审计者
        Integer temp = tempAuditor.get();
        if (temp != null) {
            return Optional.of(temp);
        }

        // 原有审计逻辑
        Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
        if (authentication == null || !authentication.isAuthenticated() || authentication instanceof AnonymousAuthenticationToken) {
            return Optional.of(0);
        }
        User user = (User) authentication.getPrincipal();
        return Optional.of(user.getId());
    }

    // 设置临时审计者
    public void setTempAuditor(Integer auditorId) {
        if (auditorId == null) {
            tempAuditor.remove();
        } else {
            tempAuditor.set(auditorId);
        }
    }
}

在业务方法中使用 #

@Service
public class YourEntityService {
    @Autowired
    private ApplicationAuditAware auditAware;
    @Autowired
    private YourEntityRepository repository;

    public void saveWithManualCreator(YourEntity entity, Integer manualCreatorId) {
        // 保存原有审计上下文
        Optional<Integer> originalAuditor = auditAware.getCurrentAuditor();
        try {
            // 临时设置审计者为手动指定的ID
            auditAware.setTempAuditor(manualCreatorId);
            // 手动赋值（可选，因为临时上下文会自动填充）
            entity.setCreatedBy(manualCreatorId);
            repository.save(entity);
        } finally {
            // 恢复原有审计上下文
            auditAware.setTempAuditor(originalAuditor.orElse(0));
        }
    }
}

方案3：移除部分字段的审计注解（简化场景） #

如果不需要全局自动填充createdBy和createdDate，可以直接去掉这两个字段的@CreatedBy、@CreatedDate注解，改为在Service层手动控制赋值逻辑，同时保留@LastModifiedBy和@LastModifiedDate的自动更新。

修改后的BaseEntity：

@Data
@SuperBuilder
@EntityListeners({AuditingEntityListener.class})
public abstract class BaseEntity {
    private LocalDateTime createdDate;

    @LastModifiedDate
    @Column(nullable = false)
    private LocalDateTime modifiedDate;

    private Integer createdBy;

    @LastModifiedBy
    @Column(nullable = false)
    private Integer modifiedBy;

    @Column(nullable = false)
    private Boolean deleteFlag = false;
}

在Service层默认赋值：

public void saveDefault(YourEntity entity) {
    Integer currentUserId = auditAware.getCurrentAuditor().orElse(0);
    entity.setCreatedBy(currentUserId);
    entity.setCreatedDate(LocalDateTime.now());
    repository.save(entity);
}

// 需要手动赋值时直接覆盖
public void saveWithManualCreator(YourEntity entity, Integer manualCreatorId) {
    entity.setCreatedBy(manualCreatorId);
    entity.setCreatedDate(LocalDateTime.now());
    repository.save(entity);
}

内容的提问来源于stack exchange，提问作者deoso1ai