如何在Quarkus中根据环境注入特定EntityManager？

要实现开发模式注入指定持久化单元的EntityManager，测试模式注入默认实例，你可以通过CDI生产者+Build Profile注解的方式正确实现，核心是统一注入点并让生产者根据环境提供对应实例。

步骤1：创建EntityManager生产者类 #

创建一个CDI管理的生产者类，分别为dev和test两种环境提供对应的EntityManager实例：

import jakarta.enterprise.context.ApplicationScoped;
import jakarta.enterprise.inject.Produces;
import jakarta.persistence.EntityManager;
import jakarta.persistence.PersistenceUnit;
import io.quarkus.arc.profile.IfBuildProfile;

@ApplicationScoped
public class EntityManagerProducer {

    // 开发模式（dev）：提供指定"result-db"持久化单元的EntityManager
    @Produces
    @IfBuildProfile("dev")
    @PersistenceUnit("result-db")
    EntityManager devEntityManager;

    // 测试模式（test）：提供默认持久化单元的EntityManager（适配单数据源场景）
    @Produces
    @IfBuildProfile("test")
    @PersistenceUnit // 不指定名称，自动绑定到唯一的数据源
    EntityManager testEntityManager;
}

如果需要额外初始化逻辑，也可以用方法式生产者：

@ApplicationScoped
public class EntityManagerProducer {

    @PersistenceUnit("result-db")
    EntityManager devEntityManager;

    @PersistenceUnit
    EntityManager testEntityManager;

    @Produces
    @IfBuildProfile("dev")
    public EntityManager produceDevEntityManager() {
        // 可添加自定义初始化逻辑
        return devEntityManager;
    }

    @Produces
    @IfBuildProfile("test")
    public EntityManager produceTestEntityManager() {
        // 可添加自定义初始化逻辑
        return testEntityManager;
    }
}

步骤2：统一业务类的注入方式 #

在所有需要使用EntityManager的业务类中，统一使用@Inject注入，无需再添加@PersistenceUnit注解：

import jakarta.inject.Inject;
import jakarta.persistence.EntityManager;

@ApplicationScoped
public class YourBusinessService {

    @Inject
    EntityManager entityManager; // 生产者会根据当前环境自动提供对应实例

    // 业务方法示例
    public void someBusinessLogic() {
        // 使用entityManager操作数据库
    }
}

关键注意点 #

1. Profile名称匹配：Quarkus默认开发模式为dev，测试模式为test，确保@IfBuildProfile的参数与实际环境一致，自定义Profile需调整参数。
2. 测试环境单数据源：测试环境需保证仅配置一个数据源，这样@PersistenceUnit无参数时会自动绑定到该数据源的持久化单元。
3. 注入点统一：不要在业务类的注入点同时使用@Inject和@PersistenceUnit，完全由生产者接管实例提供逻辑。
4. 生产者作用域：生产者类必须添加CDI作用域注解（如@ApplicationScoped），确保被Quarkus容器识别并管理。

内容的提问来源于stack exchange，提问作者Sergio Agüero