Question

我试图保存我的实体异步并返回结果的未来，但我遇到了以下问题：

import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.repository.Repository;
import org.springframework.scheduling.annotation.Async;
//import org.springframework.stereotype.Repository;

import java.util.concurrent.CompletableFuture;
import java.util.concurrent.Future;

// I must extend Repository because when I extended JpaRepository an argument 
// of save is ambigous when I try save(countryEntity) - compiler shows me 
// save (CountryEntity) in AsyncCountryRepository and save (S) in CrudRepository
public interface AsyncCountryRepository extends Repository<CountryEntity, Long> {

    // I don't know why I cannot return Future<CountryEntity>
    @Async
    <S extends CountryEntity> Future<S> save(CountryEntity countryEntity);

    // I cannot test it - test is below
    @Async
    CompletableFuture<CountryEntity> findByCode(String code);

}


@RunWith(SpringRunner.class)
@DataJpaTest
public class CountryRepositoryTest {

    @Autowired
    private TestEntityManager entityManager;

    @Autowired
    private AsyncCountryRepository countryRepository;

    @Test
    public void findByCodeTest() throws Exception {
        // given
        final CountryEntity countryEntity = new CountryEntity("EN");

        final CountryEntity persisted = entityManager.persist(countryEntity);
        entityManager.flush();

        // when
        final Future<CountryEntity> byCode = countryRepository.findByCode(persisted.getCode());

        // then
        assertThat(byCode.get())
                .isEqualTo(persisted);
    }

测试返回失败，我得到expected:<CountryEntity(id=1, code=EN)> but was:<null>

如何将我的存储库实现为异步存储库以及如何测试它？

哪种方法有更好的表现： 1.从存储库返回Futures 2.在ExecutorService中将存储库方法作为Callable运行？

Answer 1

如果使用@Async，存储库代码将在其自己的事务中的单独线程中运行。由于您的原始交易未提交，因此不会看到更改（即插入的实体），因此null结果。

因此，为了使其工作，您需要提交第一个事务。

关于性能的第二个问题（请在下次提出单独的问题）：多线程代码的性能取决于很多事情，这是回答它的唯一有用方法：尝试一下。但是，如果调度执行的方式对性能的影响比线程池和连接池大小更重要，我会感到惊讶。

Spring Data JPA - 返回Future

1 个答案: