java - 将 CDI 限定符转移到注入(inject)字段

Question

我正在 Wildfly 8.2 容器中开发 Java EE 7 应用程序，其中包含一些存在于两个数据源中的实体。示例:

我有一个带有 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;
    }

    ...
}

我不想在类路径中包含此 jar 的应用程序中通过 @Producer 提供 EntityManager 和 Logger 实例，如下所示:

@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());
}

有没有办法在注入(inject)点配置 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 注释，它告诉容器我在注释生产者方法时不必指定此参数(尤其是这个，因为我可以实现通用生产者，稍后会详细介绍)或注入(inject)点。

和DataSourceName枚举，它简单地列出了所有数据源:

public enum DataSourceName {
    D1, D2
}

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

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

    private EntityManager entityManager;

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

    ...
}

请注意，EntityManager 不再由容器注入(inject)。

当我不想注入(inject) 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)。 如果有人发现此选择有任何问题，请告诉我。