Question

我正在使用Wildfly 8.2容器中的Java EE 7应用程序，该容器包含两个数据源中存在的一些entite。例如：

我有一个Setting实体的jar：

@Entity
public class Setting {
    @Id
    @GeneratedValue(strategy = GenerationType.AUTO)
    private Long id;

    @NotNull
    private String name;
    private String value;

    getters/setters...
}

和一个bean，它有几种方法可以通过条件查询从数据库中检索和保存这个实体：

@Stateless
public class SettingRepository {
    @Inject
    private Logger logger;

    @Inject
    private EntityManager entityManager;

    public Setting findByName(@NotNull String name) {
        logger.trace("Getting setting by name: name=" + name);
        CriteriaBuilder cb = entityManager.getCriteriaBuilder();
        CriteriaQuery<Setting> cq = cb.createQuery(Setting.class);
        Root<Setting> table = cq.from(Setting.class);
        cq.where(cb.equal(table.get(Setting_.name), name));

        TypedQuery<Setting> query = entityManager.createQuery(cq);
        List<Setting> results = query.getResultList();
        Setting setting = null;

        if (results.size() > 0)
            setting = results.get(0);

        logger.trace("Got setting: " + setting);
        return setting;
    }

    ...
}

我不想通过@Producer在一个应用程序中提供EntityManager和Logger实例，该应用程序在类路径中包含此jar：

@Produces @DataSource1 @PersistenceContext(unitName = "pu1")
private EntityManager entityManager1;

@Produces @DataSource2 @PersistenceContext(unitName = "pu2")
private EntityManager entityManager2;

@Produces
private Logger produceLogger(InjectionPoint injectionPoint) {
    return LogManager.getLogger(injectionPoint.getMember().getDeclaringClass().getName());
}

有没有办法在注入点配置SettingRepository并告诉它使用特定的实体管理器（@DataSource1或@DataSource2）？

与此类似：

@Inject @DataSource1
private SettingRepository settingRepository;

Answer 1

我选择的路线是这个：

我创建了一个限定符

@Qualifier
@Retention(RUNTIME)
@Target({ METHOD, FIELD, PARAMETER })
public @interface DataSource {
    @Nonbinding DataSourceName value() default DataSourceName.D1;
}

注意@Nonbinding注释告诉容器我不必在注释生成器方法时指定此参数（特别是这个，正则我可以实现泛型生成器，稍后会更多）或注入点。

DataSourceName枚举，它只列出了所有数据源：

public enum DataSourceName {
    D1, D2
}

我还更改了SettingRepository实现并添加了一个公共初始化方法。

@Dependent
public class SettingRepository {
    @Inject
    private Logger logger;

    private EntityManager entityManager;

    public void initialize(EntityManager entityManager) {
        this.entityManager = entityManager;
    }

    ...
}

请注意，EntityManager不再按容器注入。

当我不想注射SettingRepository时，我只是用这个限定符装饰它，如下：

@Inject @DataSource(DataSourceName.D1)
private SettingRepository settingRepository;

我现在需要的是为SettingRepository定义一个制作人：

@Stateless
public class TestResourcesForSettings {
    @PersistenceContext(unitName = "pu1")
    private EntityManager entityManager1;

    @PersistenceContext(unitName = "pu2")
    private EntityManager entityManager2;

    @Inject
    private SettingRepository settingRepository;

    @Produces @DataSource
    public SettingRepository produceSettingRepository(InjectionPoint ip) {
        DataSource annotation = ip.getAnnotated().getAnnotation(DataSource.class);

        if (annotation.value() == DataSourceName.D1)
            settingRepository.initialize(entityManager1);
        else if (annotation.value() == DataSourceName.D2)
            settingRepository.initialize(entityManager2);

        return settingRepository;
    }

瞧，我可以使用两个数据源和一个存储库实现（注意：数据源必须是XA）。 如果有人发现此选择有任何问题，请告诉我。

Answer 2

是的！

事实上，就生产者方法和限定符而言，你已经完成了所有艰苦的工作:-)你需要做的就是在注入的EntityManager实例上使用限定符（@DataSource1等）

@Inject @DataSource1
private EntityManager entityManager;

这将确保附加到持久性单元'pu1'的JPA实体管理器被容器注入。

将CDI限定符传输到注入的字段

2 个答案: