java - TaskConfigurer 不考虑数据源的架构名称

Question

我有 2 个数据源。对于一个数据源，我想使用自定义模式名称。出于这个原因，我将我的数据源 url 设置为spring.datasource.url=jdbc:postgresql://192.168.33.10/analytics?currentSchema=bahmni_mart_scdf。但它在 public 架构中创建所有表，而不是 bahmni_mart_scdf 架构。

我已经覆盖了 TaskRepositoryInitializer bean 并实现了 TaskConfigurer。

这是我的实现

DatabaseConfiguration.class

@Configuration
public class DatabaseConfiguration {
    @Bean(name = "martDb")
    @ConfigurationProperties(prefix = "spring.ds_mart")
    public DataSource martDataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean(name = "martJdbcTemplate")
    public JdbcTemplate martJdbcTemplate(@Qualifier("martDb") DataSource dsMart) {
        return new JdbcTemplate(dsMart);
    }

    @Bean(name = "scdfDb")
    @ConfigurationProperties(prefix = "spring.datasource")
    public DataSource dataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean(name = "scdfJdbcTemplate")
    public JdbcTemplate scdfJdbcTemplate(@Qualifier("scdfDb") DataSource scdfDb) {
        return new JdbcTemplate(scdfDb);
    }
}

DataloadTaskConfigurer.class

@Component
public class DataloadTaskConfigurer implements TaskConfigurer {
    private DataSource dataSource;

    private TaskRepository taskRepository;

    private TaskExplorer taskExplorer;

    private PlatformTransactionManager transactionManager;

    @Autowired
    public DataloadTaskConfigurer(@Qualifier("scdfJdbcTemplate") JdbcTemplate jdbcTemplate) {

        this.dataSource = jdbcTemplate.getDataSource();
        TaskExecutionDaoFactoryBean taskExecutionDaoFactoryBean = new TaskExecutionDaoFactoryBean(dataSource);
        this.taskRepository = new SimpleTaskRepository(taskExecutionDaoFactoryBean);
        this.taskExplorer = new SimpleTaskExplorer(taskExecutionDaoFactoryBean);
    }

    @Override
    public TaskRepository getTaskRepository() {
        return this.taskRepository;
    }

    @Override
    public PlatformTransactionManager getTransactionManager() {
        if (this.transactionManager == null) {
            this.transactionManager = new DataSourceTransactionManager(this.dataSource);
        }
        return this.transactionManager;
    }

    @Override
    public TaskExplorer getTaskExplorer() {
        return this.taskExplorer;
    }

    public DataSource getTaskDataSource() {
        return this.dataSource;
    }
}

TaskConfiguration.class

@Configuration
public class TaskConfiguration {

    @Bean
    public TaskRepositoryInitializer taskRepositoryInitializerInDataMart(@Qualifier("scdfJdbcTemplate") JdbcTemplate jdbcTemplate) throws Exception {
        TaskRepositoryInitializer taskRepositoryInitializer = new TaskRepositoryInitializer();
        taskRepositoryInitializer.setDataSource(jdbcTemplate.getDataSource());
        taskRepositoryInitializer.afterPropertiesSet();
        return taskRepositoryInitializer;
    }
}

application.properties

spring.datasource.url=jdbc:postgresql://192.168.33.10/analytics?currentSchema=bahmni_mart_scdf
spring.datasource.username=analytics
spring.datasource.password=""
spring.datasource.driver-class-name=org.postgresql.Driver

spring.ds_mart.url=jdbc:postgresql://192.168.33.10/analytics
spring.ds_mart.username=analytics
spring.ds_mart.password=""
spring.ds_mart.driver-class-name=org.postgresql.Driver

Answer 1

您无需覆盖 TaskRepositoryInitializer，而是通过 spring.cloud.task.initialize.enable=false 将其全部关闭。从那里开始，使用 Spring Boot，您需要为每个数据源传递架构:

spring.datasource.schema=schema1.sql
spring.ds_mart.schema=schema2.sql