Spring Data JPA - 返回 Future

Question

我尝试异步保存我的实体并返回结果的 future ，但遇到以下问题:

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。

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

关于关于性能的第二个问题(请下次单独提出问题):多线程代码的性能取决于很多因素，因此唯一有用的回答方法是:尝试一下。但如果安排执行的方式比线程池和连接池大小等对性能的影响更大，我会感到惊讶。