如何临时禁用Spring Data中的AuditEntityListener？

你遇到的这个场景确实挺常见的——既要保留Spring Data Audit的自动维护能力，又要在批量迁移时手动指定审计信息，还要兼容Envers的需求。完全没必要复制实体类，给你几个实用的方案：

1. 给AuditorAware加上下文感知逻辑 #

Spring Data Audit的核心是AuditorAware接口，我们可以改造它，让它能识别当前是否需要使用手动传入的审计用户：

@Component
public class CustomAuditorAware implements AuditorAware<String> {
    private static final ThreadLocal<String> MANUAL_AUDITOR = new ThreadLocal<>();

    // 供Batch任务手动设置审计用户
    public static void setManualAuditor(String auditor) {
        MANUAL_AUDITOR.set(auditor);
    }

    // 清理ThreadLocal，避免内存泄漏
    public static void clearManualAuditor() {
        MANUAL_AUDITOR.remove();
    }

    @Override
    public Optional<String> getCurrentAuditor() {
        // 优先取手动设置的用户
        String manualAuditor = MANUAL_AUDITOR.get();
        if (manualAuditor != null) {
            return Optional.of(manualAuditor);
        }

        // 常规场景下从SecurityContext获取当前用户
        Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
        if (authentication != null && authentication.isAuthenticated() && !(authentication instanceof AnonymousAuthenticationToken)) {
            return Optional.of(authentication.getName());
        }

        return Optional.empty();
    }
}

然后在Spring Batch的ItemWriter里，保存实体前先设置手动审计用户：

public class LegacyDataItemWriter implements ItemWriter<LegacyEntity> {
    @Autowired
    private EntityManager entityManager;

    @Override
    public void write(List<? extends LegacyEntity> items) throws Exception {
        for (LegacyEntity item : items) {
            // 从旧数据中取出创建/修改用户，设置到AuditorAware的ThreadLocal里
            CustomAuditorAware.setManualAuditor(item.getCreatedBy());
            // 转换为目标实体
            TargetEntity target = convertToTargetEntity(item);
            // 手动设置创建/修改日期
            target.setCreatedDate(item.getCreatedDate());
            target.setLastModifiedDate(item.getLastModifiedDate());
            entityManager.persist(target);
            // 清理ThreadLocal
            CustomAuditorAware.clearManualAuditor();
        }
    }
}

这样AuditEntityListener在处理时，会优先使用我们手动设置的用户，既保留了自动审计的功能，又满足了批量迁移的需求，还不用改动实体类。

2. 利用Spring Profiles条件化控制监听器 #

如果你的Batch作业是单独的Profile运行（比如batch profile），可以给实体类的监听器加条件化逻辑：

首先创建一个自定义监听器，根据Profile决定是否委托给原生的AuditEntityListener：

public class ConditionalAuditEntityListener {
    @Autowired(required = false)
    private AuditEntityListener auditEntityListener;

    @Value("${spring.profiles.active}")
    private String activeProfile;

    @PrePersist
    public void onPrePersist(Object entity) {
        // 非Batch环境下，正常触发原生审计逻辑
        if (!"batch".equals(activeProfile) && auditEntityListener != null) {
            auditEntityListener.onPrePersist(entity);
        }
        // Batch环境下，留空让代码手动设置审计字段
    }

    @PreUpdate
    public void onPreUpdate(Object entity) {
        if (!"batch".equals(activeProfile) && auditEntityListener != null) {
            auditEntityListener.onPreUpdate(entity);
        }
    }
}

然后修改实体类的监听器配置：

@Entity
@EntityListeners(ConditionalAuditEntityListener.class)
@Audited // 保留Envers注解
public class TargetEntity {
    // 实体字段...
}

这样在非Batch环境下还是用原生逻辑自动处理；Batch环境下监听器不会自动干预，你可以完全手动控制审计字段，同时不影响Envers的功能。

3. 谨慎使用：临时绕过监听器（不推荐优先使用） #

如果你只是在Batch迁移时需要完全手动控制，也可以尝试通过Hibernate的原生API绕过JPA监听器，但这种方式风险较高，可能影响其他逻辑：

// 在Batch的ItemWriter中
Session session = entityManager.unwrap(Session.class);
session.save(targetEntity); // Hibernate的save方法不会触发JPA实体监听器

不过这个方案需要确认Envers的兼容性，所以仅作为备选方案。

总之，最推荐的是第一个方案——通过ThreadLocal让AuditorAware支持手动传入用户，既不用修改实体类，又能完美兼容Envers和常规自动审计场景。

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