DataSourceConfig.kt

DataSourceConfig.kt

@Configuration
@EntityScan(basePackages = ["com.beauty.domain"])
@EnableJpaRepositories(basePackages = ["com.beauty.repository"])
@EnableTransactionManagement
@EnableAutoConfiguration(exclude = [DataSourceAutoConfiguration::class])
class DataSourceConfig {

    companion object {
        const val MASTER = "master"
        const val SLAVE = "slave"
    }

    @Bean(name = ["masterDataSource"])
    @ConfigurationProperties(prefix = "spring.datasource.master")
    fun masterDataSource(): DataSource {
        return DataSourceBuilder.create().type(HikariDataSource::class.java).build()
    }

    @Bean(name = ["slaveDataSource"])
    @ConfigurationProperties(prefix = "spring.datasource.slave")
    fun slaveDataSource(): DataSource {
        return DataSourceBuilder.create().type(HikariDataSource::class.java).build()
    }

    @Bean(name = ["routingDataSource"])
    fun routingDataSource(@Qualifier("masterDataSource") masterDataSource: DataSource,
                          @Qualifier("slaveDataSource") slaveDataSource: DataSource): DataSource {
        val routingDataSource = ReplicationRoutingDataSource()
        val dataSourceMap = HashMap<String, Any>()
        dataSourceMap[MASTER] = masterDataSource
        dataSourceMap[SLAVE] = slaveDataSource
        routingDataSource.setTargetDataSources(dataSourceMap.toMap())
        routingDataSource.setDefaultTargetDataSource(masterDataSource)
        return routingDataSource
    }

    @Primary
    @Bean
    fun dataSource(@Qualifier("routingDataSource") routingDataSource: DataSource): DataSource {
        return LazyConnectionDataSourceProxy(routingDataSource)
    }

}