Quarkus 的自定义配置源

Question

我现在正在尝试在我的 Quarkus 应用程序中配置自定义 ConfigSource。与许多其他手册一样，我创建了自己的 DatabaseSourceConfig 并实现了 org.eclipse.microprofile.config.spi.ConfigSource interface.I 在以下位置注册了我的 ConfigSource：

/META-INF/services/org.eclipse.microprofile.config.spi.ConfigSource

这是我的配置源：

public class DatabaseConfigSource implements ConfigSource {
private DataSource dataSource;

public DatabaseConfigSource() {
    try {
        dataSource = (DataSource) new InitialContext().lookup("openejb:Resource/config-source-database");
    } catch (final NamingException e) {
        throw new IllegalStateException(e);
    }
}

@Override
public Map<String, String> getProperties() {
    // Implementing Method
}

@Override
public String getValue(final String propertyName) {
    // Implementing Method
}

@Override
public String getName() {
    return DatabaseConfigSource.class.getSimpleName();
}

}

但是由于 JNDI 名称，这不适用于 Quarkus。我需要使用 CDI。我试图使用这样的东西：

@Inject
@io.quarkus.agroal.DataSource("my_connection")
AgroalDataSource usersDataSource;

并在 application.properties 中声明此连接，但它对我没有帮助。我一直收到 NULL 异常。也许有人有想法，我如何在不使用 JNDI 命名空间的情况下获得数据库连接？

Answer 1

您可以通过

获取数据源

AgroalDataSource dataSource = Arc.container()
    .instance(AgroalDataSource.class, new DataSource.DataSourceLiteral("my_connection"))
    .get();

尽管我认为您需要在构造函数之外的其他地方执行此操作，因为 ConfigSource 实例是在 CDI 完全启动之前创建的。可以缓存获取到的数据源实例，避免多次执行。

Answer 2

我自己找到了一些答案，也许对其他人也有用。正如@Janmartiška 所说，CDI 比ConfigSource 启动得晚，这就是为什么我看不到任何通过CDI 注入连接的原因。我创建了一些 HibernateUtil Class:

package org.myproject.config;

import java.util.Properties;

import org.hibernate.SessionFactory;
import org.hibernate.boot.registry.StandardServiceRegistryBuilder;
import org.hibernate.cfg.Configuration;
import org.hibernate.service.ServiceRegistry;
import org.myproject.entities.ConfigurationsEntity;

public class HibernateUtil {
    private static SessionFactory sessionFactory;

    private static SessionFactory buildSessionFactory() {
        try {
            Properties props = new Properties();
            props.setProperty("hibernate.connection.url", "jdbc:mysql://[db-host]:[db-port]/db_name");
            props.setProperty("hibernate.connection.driver_class", "com.mysql.cj.jdbc.Driver");
            props.setProperty("hibernate.connection.username", "username");
            props.setProperty("hibernate.connection.password", "password");

            props.setProperty("hibernate.current_session_context_class", "thread");
            props.setProperty("hibernate.dialect", "org.hibernate.dialect.MySQL8Dialect");

            Configuration configuration = new Configuration();
            configuration.addProperties(props);
            configuration.addAnnotatedClass(ConfigurationsEntity.class);
            System.out.println("Hibernate Configuration loaded");

            ServiceRegistry serviceRegistry = new StandardServiceRegistryBuilder().applySettings(configuration.getProperties()).build();
            System.out.println("Hibernate serviceRegistry created");

            SessionFactory sessionFactory = configuration.buildSessionFactory(serviceRegistry);

            return sessionFactory;
        }
        catch (Throwable ex) {
            // Make sure you log the exception, as it might be swallowed
            System.err.println("Initial SessionFactory creation failed." + ex);
            throw new ExceptionInInitializerError(ex);
        }
    }

    public static SessionFactory getSessionFactory() {
        if(sessionFactory == null) sessionFactory = buildSessionFactory();
        return sessionFactory;
    }

}

比我在我的 SourceConfig 中使用的还要多：

package org.myproject.config;

import io.quarkus.runtime.annotations.RegisterForReflection;
import org.eclipse.microprofile.config.spi.ConfigSource;
import org.hibernate.Session;
import org.hibernate.SessionFactory;
import org.myproject.entities.ConfigurationsEntity;

import javax.persistence.NoResultException;
import javax.persistence.Query;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

@RegisterForReflection
public class DatabaseSourceConfig implements ConfigSource {

    public SessionFactory sessionFactory;
    public Session currentSession;

    public DatabaseSourceConfig() {

        sessionFactory = HibernateUtil.getSessionFactory();
        this.checkFactoryConnection();

    }

    public void checkFactoryConnection() {

        if (currentSession == null || (currentSession != null && !currentSession.isOpen())) {
            try {
                currentSession = sessionFactory.getCurrentSession();
            } catch (NullPointerException e) {
                currentSession = sessionFactory.openSession();
            }
        }

    }

    @Override
    public Map<String, String> getProperties() {
        // Implementing Method
    }

    @Override
    public String getValue(String propertyName) {
            this.checkFactoryConnection();
            ConfigurationsEntity conf = new ConfigurationsEntity();
            currentSession.beginTransaction();

            try {
                Query query = currentSession.createNamedQuery("Configuration.selectOne", ConfigurationsEntity.class);
                query.setParameter("name", propertyName);
                conf = (ConfigurationsEntity) query.getSingleResult();
                currentSession.getTransaction().commit();

            } catch (Exception ex) {
                currentSession.getTransaction().rollback();
            }

            return conf.getValue();

    }

    @Override
    public String getName() {
        return DatabaseSourceConfig.class.getSimpleName();
    }
}

现在我可以在其他类中使用我的 ConfigSource，例如：

@Inject
@ConfigProperty(name = "[property-name-like-in-db]")
public String someProperty;

经过我的进一步研究，发现 ConfigSource 无法访问 CDi 和 application.properties。这就是为什么除了以上述方式建立与数据库的连接之外别无他法。但是，我对该示例进行了一些编辑。我缓存了数据库中的属性并创建了一个 @ApplicationScoped Bean，它每 5 分钟查看一次数据库，以查看其中一个属性 "updated_at" 的时间戳是否晚于 Bean 加载属性中的其他属性。

但是，我不得不说，根据 Quarkus 和 Apache 开发人员的说法——这违反了“不可变部署”，并且不打算在运行时更改应用程序设置。所以写不写在app里就看你了

Quarkus 的自定义配置源

Custom ConfigSource for Quarkus

java

datasource

cdi

quarkus

microprofile