Question

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

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

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

Answer 1

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

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”

Answer 2

配置所需的数据源＆amp; 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。因此，您可以相应地在运行时更改属性。

Answer 3

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

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

我的提供程序会覆盖称为Java EE容器的createContainerEntityManagerFactory(PersistenceUnitInfo,Map)方法（对于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获得相同的持久性单元名称）

Answer 4

我想说明

的用法

Persistence.createEntityManagerFactory(persistenceUnitName)

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

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

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

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

如何以编程方式将持久化上下文注入到不同的数据源中

4 个答案: