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这个子类AdviceMode.PROXY，您需要基于AspectJTransactionManagementConfiguration 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 {}

我希望这会有所帮助。

当@ActiveProfiles（＆＃34; test＆＃34;）时，如何忽略特定方法的spring @Transactional注释

1 个答案: