TaskConfigurer不考虑数据源的模式名称

时间:2018-04-26 05:15:34

标签: java spring postgresql spring-boot spring-cloud-task

我有2个数据源。对于一个数据源,我想使用自定义模式名称。出于这个原因,我正在设置我的数据源URL spring.datasource.url=jdbc:postgresql://192.168.33.10/analytics?currentSchema=bahmni_mart_scdf。 但它会在公开架构中创建所有表格,而不是 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

2 个答案:

答案 0 :(得分:2)

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

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

答案 1 :(得分:0)

问题在于postgresql驱动程序版本。一旦我更新到最新版本(42.2.2),一切都按预期正常工作