在 spring 引導中配置多個 azure sql 數據庫

Question

我正在嘗試配置兩個使用相同 azure sql 用戶名/密碼但網址/數據庫不同的數據源。 我正在使用@Primary 注釋並將數據訪問層拆分到它們自己的文件夾中，但只有沒有@Primary 的配置被配置。

數據源1：

@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(basePackages = Constants.BASE_PAKAGE,
entityManagerFactoryRef = Constants.ENTITY_MANAGER_FACTORY,
transactionManagerRef= Constants.TRANSACTION_MANAGER)
public class DataSourceConfiguration {

    private Logger log = LoggerFactory.getLogger(this.getClass());

    public DataSourceConfiguration(Environment environment) {
        this.environment = environment;
    }
    private Environment environment;

    @Primary
    @Bean
    public EntityManagerFactoryBuilder entityManagerFactoryBuilder() {
        return new EntityManagerFactoryBuilder(new HibernateJpaVendorAdapter(), new HashMap<>(), null);
    }
   
    @Primary
    @Bean(name = Constants.ENTITY_MANAGER_FACTORY)
    public LocalContainerEntityManagerFactoryBean sqlserverEntityManagerFactory(
            EntityManagerFactoryBuilder builder) throws IllegalAccessException, InvocationTargetException, InstantiationException {
    HashMap<String, Object> properties = new HashMap<>();
    properties.put("hibernate.dialect",environment.getProperty("sqlserver.datasource1.dialect"));
        return builder
                .dataSource(dataSource())
                .packages(Constants.ENTITY_PATH)
                .properties(properties)
                .build();
    }
    @Primary
    @Bean
    public PlatformTransactionManager sqlserverTransactionManager(
            final @Qualifier(Constants.ENTITY_MANAGER_FACTORY) LocalContainerEntityManagerFactoryBean sqlserverEntityManagerFactory) {
        return new JpaTransactionManager(sqlserverEntityManagerFactory.getObject());
    }

    @Primary
    @Bean
    protected HikariDataSource dataSource() throws IllegalAccessException, InvocationTargetException, InstantiationException {
        HikariConfig config = new HikariConfig();
        config.setDriverClassName(environment.getProperty("sqlserver.datasource1.driver"));
        config.setJdbcUrl(this.buildSnowflakeJDBCUrl());
        config.setUsername(environment.getProperty("sqlserver.datasource1.username"));
        config.setPassword(environment.getProperty("sqlserver.datasource1.password"));
        HikariDataSource dataSource = new HikariDataSource(config);
        return dataSource;
    }

    private String buildSnowflakeJDBCUrl() {
        StringBuilder url = new StringBuilder();
        url.append(environment.getProperty("sqlserver.datasource1.url"));
        return url.toString();
    }
   
}

和數據源2：

@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(basePackages = Constants.BASE_PACKAGE2,
        entityManagerFactoryRef = "sqlserver2EntityManagerFactory",
        transactionManagerRef= "sqlserver2TransactionManager")
public class DataSourceConfiguration2 {

    private Logger log = LoggerFactory.getLogger(this.getClass());

    public PhDataSourceConfiguration(Environment environment) {
        this.environment = environment;
    }
    private Environment environment;

    @Bean
    public EntityManagerFactoryBuilder entityManagerFactoryBuilder() {
        return new EntityManagerFactoryBuilder(new HibernateJpaVendorAdapter(), new HashMap<>(), null);
    }

    @Bean(name = "sqlserver2EntityManagerFactory")
    public LocalContainerEntityManagerFactoryBean sqlserverEntityManagerFactory(
            EntityManagerFactoryBuilder builder) throws IllegalAccessException, InvocationTargetException, InstantiationException {
        HashMap<String, Object> properties = new HashMap<>();
        properties.put("hibernate.dialect",environment.getProperty("sqlserver.datasource2.dialect"));
        return builder
                .dataSource(dataSource())
                .packages(Constants.ENTITY_PATH2)
                .properties(properties)
                .build();
    }

    @Bean(name = "sqlserver2TransactionManager")
    public PlatformTransactionManager sqlserverTransactionManager(
            final @Qualifier("sqlserver2EntityManagerFactory") LocalContainerEntityManagerFactoryBean sqlserverEntityManagerFactory) {
        return new JpaTransactionManager(sqlserverEntityManagerFactory.getObject());
    }

    @Bean
    protected HikariDataSource dataSource() throws IllegalAccessException, InvocationTargetException, InstantiationException {
        HikariConfig config = new HikariConfig();
        config.setDriverClassName(environment.getProperty("sqlserver.datasource2.driver"));
        config.setJdbcUrl(this.buildJDBCUrl());
        config.setUsername(environment.getProperty("sqlserver.datasource2.username"));
        config.setPassword(environment.getProperty("sqlserver.datasource2.password"));
        HikariDataSource dataSource = new HikariDataSource(config);
        return dataSource;
    }

    private String buildJDBCUrl() {
        StringBuilder url = new StringBuilder();
        url.append(environment.getProperty("sqlserver.datasource2.url"));
        return url.toString();
    }

}

這是一個問題，因為他們都從相同的用戶/密碼讀取 azure sql？ 我怎樣才能讓這兩個配置都工作？

Answer 1

謝謝安德魯·S 。 發布您的建議作為幫助其他社區成員的答案。

一些 bean 名稱重疊（如果未指定，則使用方法名稱），因此最后處理的 @Bean 將獲勝。 嘗試顯式命名 bean，然后在需要的地方注入正確的 bean 實例 @Qualifier。