使用@Async方法

时间:2016-08-19 15:36:39

标签: java junit spring-transactions spring-test spring-async

我正在使用SpringJUnit4ClassRunner编写集成测试。 我有一个基类:

@RunWith(SpringJUnit4ClassRunner.class)
@WebAppConfiguration
@ContextConfiguration({ /*my XML files here*/}) 
@Ignore
public class BaseIntegrationWebappTestRunner {

@Autowired
protected WebApplicationContext wac; 

@Autowired
protected MockServletContext servletContext; 

@Autowired
protected MockHttpSession session;

@Autowired
protected MockHttpServletRequest request;

@Autowired
protected MockHttpServletResponse response;

@Autowired
protected ServletWebRequest webRequest;

@Autowired
private ResponseTypeFilter responseTypeFilter;

protected MockMvc mockMvc;

@BeforeClass
public static void setUpBeforeClass() {

}

@AfterClass
public static void tearDownAfterClass() {

}

@Before
public void setUp() {
    this.mockMvc = MockMvcBuilders.webAppContextSetup(this.wac).addFilter(responseTypeFilter).build();
}

@After
public void tearDown() {
    this.mockMvc = null;
}
}

然后我扩展它并使用mockMvc创建一个测试:

public class MyTestIT extends BaseMCTIntegrationWebappTestRunner {

@Test
@Transactional("jpaTransactionManager")
public void test() throws Exception {
    MvcResult result = mockMvc
            .perform(
                    post("/myUrl")
                            .contentType(MediaType.APPLICATION_XML)
                            .characterEncoding("UTF-8")
                            .content("content")
                            .headers(getHeaders())
            ).andExpect(status().isOk())
            .andExpect(content().contentType(MediaType.APPLICATION_XML))
            .andExpect(content().encoding("ISO-8859-1"))
            .andExpect(xpath("/*[local-name() ='myXPath']/")
                    .string("result"))
            .andReturn();
}

在流程结束时,实体将保存到DB中。但这里的要求是应该异步完成。所以考虑这个方法叫做:

@Component
public class AsyncWriter {

    @Autowired
    private HistoryWriter historyWriter;

    @Async
    public void saveHistoryAsync(final Context context) {
        History history = historyWriter.saveHistory(context);
    }
}

然后调用HistoryWriter

@Component
public class HistoryWriter {

    @Autowired
    private HistoryRepository historyRepository;

    @Transactional("jpaTransactionManager")
    public History saveHistory(final Context context) {
        History history = null;
        if (context != null) {
            try {
                history = historyRepository.saveAndFlush(getHistoryFromContext(context));
            } catch (Throwable e) {
                LOGGER.error(String.format("Cannot save history for context: [%s] ", context), e);
            }
        }
        return history;
    }
}

所有这一切的问题在于测试完成后,History对象留在数据库中。我需要让测试事务最终回滚所有更改。

现在,我到目前为止所做的一切:

  1. 删除@Async注释。显然,这不是解决方案,但是确保回滚将在没有它的情况下执行。的确如此。
  2. @Async注释移至HistoryWriter.saveHistory()方法,将其放在@Transactional的一个位置。这篇文章https://dzone.com/articles/spring-async-and-transaction表明它应该以这种方式工作,但对我来说,测试后没有回滚。
  3. 交换这两个注释的位置。它也没有给出期望的结果。
  4. 有没有人知道如何强制回滚在异步方法中进行的数据库更改?

    附注:

    交易配置:

    <tx:annotation-driven proxy-target-class="true" transaction- manager="jpaTransactionManager"/>

    异步配置:

    <task:executor id="executorWithPoolSizeRange" pool-size="50-75" queue-capacity="1000" /> <task:annotation-driven executor="executorWithPoolSizeRange" scheduler="taskScheduler"/>

1 个答案:

答案 0 :(得分:5)

  

有没有人知道如何强制回滚在异步方法中进行的数据库更改?

遗憾的是,这是不可能的。

Spring通过ThreadLocal变量管理事务状态。因此,在另一个线程(例如,为@Async方法调用创建的线程)中启动的事务可以参与为父线程管理的事务。

这意味着您的@Async方法使用的事务永远test-managed transaction相同,后者会被Spring TestContext Framework自动回滚。

因此,解决您问题的唯一方法是手动撤消对数据库的更改。您可以使用JdbcTestUtils@AfterTransaction方法中以编程方式执行SQL脚本,也可以通过Spring @Sql注释以声明方式配置SQL脚本(在执行阶段后使用)。对于后者,请参阅How to execute @Sql before a @Before method了解详细信息。

此致

Sam( Spring TestContext Framework的作者