在 Spring Boot 中获取 EntityManager 的句柄

Question

有什么方法可以获取给定实体对象的 EntityManager 句柄？我正在使用带有 JPA 启动器的 spring boot 1.2.3，并且我正在使用 @configuration 进一步明确配置多个数据源

我检查了[resolved]SPRING BOOT access to entityManager，它似乎没有回答这个问题。

谢谢。

编辑：我添加了关于如何定义我的数据源的说明：

@Component
@Configuration
public class DataSources {
    @Bean
    @Primary
    @ConfigurationProperties(prefix="first.datasource")
    public DataSource getPrimaryDataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean
    @ConfigurationProperties(prefix="second.datasource")
    public DataSource getSecondDataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean
    @ConfigurationProperties(prefix="third.final.datasource")
    public DataSource getThirdFinalDataSource() {
        return DataSourceBuilder.create().build();
    }
}

在我的 application.yml 我有以下部分

first.datasource:
  name: 'first_datasource',
  #other attributes...
second.datasource:
  name: 'second_datasource',
  #other attributes...
third.final.datasource:
  name: 'first_datasource',
  #other attributes...

到目前为止，我已经尝试了@Stephane 的两个建议，但我得到了NoSuchBeanDefinitionException

假设我的实体名为Customer，然后我尝试了

@Service
public class FooService {

    private final EntityManager entityManager;

    @Autowired
    public FooService(@Qualifier("customerEntityManager") EntityManager entityManager) {
        ...
    }

}

不过我也试过了

@PersistenceContext(unitName = "customer") // also tried "customers" and "first_datasource"
private EntityManager entityManager;

没有运气。

Answer 1

这取决于您如何配置它，但您是否尝试过注入 EntityManager 使用与创建它的工厂相对应的限定符？

这里是a sample project with two data sources。如果您想注入EntityManager 进行订购，只需在项目的任何Spring bean 中执行以下操作

@Service
public class FooService {

    private final EntityManager entityManager;

    @Autowired
    public FooService(@Qualifier("orderEntityManager") EntityManager entityManager) {
        ...
    }

}

对于客户，请使用customerEntityManager。

当然你也可以使用持久化单元名，也就是

@PersistenceContext(unitName = "customers")
private EntityManager entityManager;

@PersistenceContext(unitName = "orders")
private EntityManager entityManager;

查看项目配置了解更多详情。