8

在标准 EJB 3 中，当注入实体管理器时，持久性单元（指数据源）被硬编码到注解中：（或者 xml 文件）

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

有没有办法使用实体管理器但在运行时按名称选择数据源？

4

4 回答 4

5

使用 EclipseLink，您可以设置在您的应用服务器中配置的数据源。

import org.eclipse.persistence.config.PersistenceUnitProperties;
...


....
Map props = new HashMap();  
props.put(PersistenceUnitProperties.JTA_DATASOURCE, "dataSource");  
EntityManagerFactory  emf = Persistence.createEntityManagerFactory("UNIT_NAME", props);
EntityManager em = emf.createEntityManager();

PU_NAME指在文件 persistence.xml 中使用的名称
dataSource 指在应用服务器中用于 jdbc 资源的名称为“jdbc/sample”

于 2012-02-16T16:16:10.967 回答
2

有可能的！我已经完成了，它可以在 JBoss AS 和 WebSphere 下工作。

我使用扩展的自定义持久性提供程序org.hibernate.ejb.HibernatePersistence（您需要修改一个private static final字段以将持久性提供程序名称设置为org.hibernate.ejb3.Ejb3Configuration.IMPLEMENTATION_NAME：这是一种黑魔法，但它有效）。确保您persistence.xml的持久性单元在<provider>标签中设置了自定义提供程序，并且您的自定义提供程序已在META-INF/services/javax.persistence.spi.PersistenceProvider.

我的提供者重写了createContainerEntityManagerFactory(PersistenceUnitInfo,Map)称为 Java EE 容器的方法（对于 JTA 数据源，但对于非 JTA 数据源也很容易）：

@Override
public EntityManagerFactory createContainerEntityManagerFactory(PersistenceUnitInfo info, Map map) {

    // load the DataSource
    String newDataSourceName = ...; // any name you want
    DataSource ds = (DataSource)(new InitialContext().lookup(newDataSourceName));

    // modify the datasource
    try {
        try {
            // JBoss implementation (any maybe other Java EE vendors except IBM WebSphere)
            Method m = info.getClass().getDeclaredMethod("setJtaDataSource", DataSource.class);
            m.setAccessible(true);
            m.invoke(info, ds);

        } catch (NoSuchMethodException e) {
            // method does not exist (WebSphere?) => try the WebSphere way

            // set the datasource name
            Method m = info.getClass().getDeclaredMethod("setJtaDataSource", String.class);
            m.setAccessible(true);
            m.invoke(info, newDataSourceName);

            // do the lookup
            Method m2 = info.getClass().getDeclaredMethod("lookupJtaDataSource", String.class);
            m2.setAccessible(true);
            m2.invoke(info);
        }
    } catch (Throwable e) {
        throw new RuntimeException("could not change DataSource for "+info.getClass().getName());
    }

    // delegate the EMF creation
    return new HibernatePersistence().createContainerEntityManaferFactory(info, map);
}

也被createEntityManagerFactory(String,Map)覆盖但更简单：

@Override
public EntityManagerFactory createEntityManagerFactory(String persistenceUnitInfo, Map map) {

    // change the datasource name
    String newDataSourceName = ...; // any name you want
    if (map==null) map = new HashMap();  
    map.put(HibernatePersistence.JTA_DATASOURCE, newDataSourceName);  

    // delegate the EMF creation
    return new HibernatePersistence().createEntityManaferFactory(persistenceUnitInfo, map);
}

注意我这里只写了核心代码。事实上，我的持久化提供程序还有很多其他功能：

  • 检查 DataSource 是否已启动并正在运行
  • 为 JBoss 或 WebSphere 设置事务管理器
  • 缓存 EMF 以降低内存使用率
  • 重新配置 Hibernate 查询计划缓存以减少内存使用量
  • 注册 JMX bean（允许多个 EAR 获得相同的持久化单元名称）
于 2013-11-07T08:29:04.447 回答
2
  • 在 persistence.xml 中配置所需的数据源和持久单元。
<persistence-unit name="UNIT_NAME" transaction-type="JTA">
      <provider>PERSISTENCE_PROVIDER</provider>
          <jta-data-source>java:DATA_SOURCE_NAME</jta-data-source>
</persistence-unit>

 -- other units

现在在运行时，您可以为所需的持久性单元构建实体管理器。为每个数据源创建单独的持久性单元。

//---
EntityManagerFactory emf = Persistence.createEntityManagerFactory(persistenceUnitName);
EntityManager em = emf.createEntityManager();
//---
  • 否则，您还可以通过提供 db-url、userName 等属性的映射来构建工厂。
createEntityManagerFactory(persistenceUnitName,propertiesMap);

这将使用给定的属性为命名的持久性单元创建并返回一个 EntityManagerFactory。因此，您可以在运行时相应地更改属性。

于 2011-03-12T04:43:39.470 回答
1

我想指出

Persistence.createEntityManagerFactory(persistenceUnitName)

Nayan 的回答中推荐的由 JPA 规范（JSR 317）分类如下（“9.2 Java SE 环境中的引导”中的脚注）：

“Java EE 容器可能支持使用这些 Java SE 引导 API；但是，不需要支持此类使用。”

所以这不是 EJB 的标准解决方案。无论如何，我可以确认这在 EclipseLink 中有效。

注意：我还不允许将此作为​​评论发布。

于 2012-12-09T21:21:31.520 回答