当@ActiveProfiles("test") 时，如何忽略特定方法的 spring @Transactional 注释

Question

@Transactional在集成测试期间，我需要忽略以下注释。

@Service
public class MyClass {

    @Transactional(propagation = Propagation.NEVER)
    public void doSomething() {
        // do something that once in production can not be inside a transaction (reasons are omitted)
    }

}

问题是我所有的测试都是在默认回滚的事务中执行的。@Transactional(propagation = Propagation.NEVER)当此方法在测试 ( @ActiveProfiles("test"))范围内运行时，我怎么能忽略它的注释，允许它在事务内执行？

Answer 1

首先，您需要排除当前@EnableTransactionManagement注释才能在您的个人资料中处于活动状态test。您可以通过将@EnableTransactionManagement注释隔离到一个单独的配置类（该类不包括配置文件）来实现此目的，这样只有在配置文件不test活动时它才会被激活。test

@EnableTransactionManagement(mode=AdviceMode.PROXY)
@Profile("!test")
public class TransactionManagementConfig {}

准备就绪后，我们就可以开始为您的测试配置文件构建自定义事务管理配置。首先，我们定义一个将用于激活自定义事务管理的注释（为了示例的紧凑性而删除了 javadoc 注释，EnableTransactionManagement有关详细信息，请参阅 javadoc）。

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Import(CustomTransactionManagementConfigurationSelector.class)
public @interface EnableCustomTransactionManagement {
    boolean proxyTargetClass() default false;
    AdviceMode mode() default AdviceMode.PROXY;
    int order() default Ordered.LOWEST_PRECEDENCE;
}

然后我们需要一个导入选择器。请注意，由于您正在使用AdviceMode.PROXY，所以我跳过了该ASPECTJ部分的实现，但为了使用基于 AspectJ 的事务管理，应该类似地完成该操作。

public class CustomTransactionManagementConfigurationSelector extends
        AdviceModeImportSelector<EnableCustomTransactionManagement> {
    @Override
    protected String[] selectImports(AdviceMode adviceMode) {
        switch (adviceMode) {
        case PROXY:
            return new String[] { 
                AutoProxyRegistrar.class.getName(),
                CustomTransactionAttributeSourceConfig.class.getName()
            };
        case ASPECTJ:
        default:
            return null;
        }
    }
}

最后是您可以覆盖事务属性的部分。ProxyTransactionManagementConfiguration这是for的子类AdviceMode.PROXY，您需要一个基于AspectJTransactionManagementConfigurationfor 的类似实现AdviceMode.ASPECTJ。请随意实现您自己的逻辑，无论是像我的示例中那样不断覆盖原始版本AnnotationTransactionAttributeSource将确定的任何属性，还是通过为此目的引入和处理您自己的自定义注释来进行更详细的处理。

@Configuration
public class CustomTransactionAttributeSourceConfig
        extends ProxyTransactionManagementConfiguration {

    @Override
    public void setImportMetadata(AnnotationMetadata importMetadata) {
        this.enableTx = AnnotationAttributes
                .fromMap(importMetadata.getAnnotationAttributes(
                        EnableCustomTransactionManagement.class.getName(),
                        false));
        Assert.notNull(this.enableTx,
                "@EnableCustomTransactionManagement is not present on importing class "
                        + importMetadata.getClassName());
    }

    @Bean
    @Role(BeanDefinition.ROLE_INFRASTRUCTURE)
    @Override
    public TransactionAttributeSource transactionAttributeSource() {
        return new AnnotationTransactionAttributeSource() {

            private static final long serialVersionUID = 1L;

            @Override
            protected TransactionAttribute findTransactionAttribute(
                    Class<?> clazz) {
                TransactionAttribute transactionAttribute = 
                        super.findTransactionAttribute(clazz);
                if (transactionAttribute != null) {
                    // implement whatever logic to override transaction attributes 
                    // extracted from @Transactional annotation
                    transactionAttribute = new DefaultTransactionAttribute(
                            TransactionAttribute.PROPAGATION_REQUIRED);
                }
                return transactionAttribute;
            }

            @Override
            protected TransactionAttribute findTransactionAttribute(
                    Method method) {
                TransactionAttribute transactionAttribute = 
                        super.findTransactionAttribute(method);
                if (transactionAttribute != null) {
                    // implement whatever logic to override transaction attributes
                    // extracted from @Transactional annotation
                    transactionAttribute = new DefaultTransactionAttribute(
                            TransactionAttribute.PROPAGATION_REQUIRED);
                }
                return transactionAttribute;
            }
        };
    }
}

最后，您需要使用与配置test文件绑定的配置类来启用自定义事务管理配置。

@EnableCustomTransactionManagement(mode=AdviceMode.PROXY)
@Profile("test")
public class TransactionManagementTestConfig {}

我希望这有帮助。