MappingSolution

EntityManagerConfiguration.kt

@Configuration
class EntityManagerConfiguration(
    localContainerEntityManagerFactoryBean: LocalContainerEntityManagerFactoryBean
) {
    init {
        if (localContainerEntityManagerFactoryBean.jpaDialect is HibernateJpaDialect) {
            (localContainerEntityManagerFactoryBean.jpaDialect as HibernateJpaDialect)
                .setPrepareConnection(false)
        }
    }
}

TransactionRoutingConfiguration.kt

@Configuration
class TransactionRoutingConfiguration(
    private val dataSourceListProperties: DataSourceListProperties
) {

    @Bean
    @Primary
    fun dataSource(): DataSource {
        val transactionRoutingDataSource: AbstractRoutingDataSource = TransactionRoutingDataSource()
        val readDataSource = buildDataSourceFromProperties(dataSourceListProperties.readDatasource, true)
        val writeDataSource = buildDataSourceFromProperties(dataSourceListProperties.writeDatasource, false)
        val targetDataSources: MutableMap<Any, Any> = mutableMapOf(
            DataSourceType.READ_ONLY to readDataSource,
            DataSourceType.READ_WRITE to writeDataSource
        )

        transactionRoutingDataSource.setDefaultTargetDataSource(readDataSource)

        transactionRoutingDataSource.setTargetDataSources(targetDataSources)
        transactionRoutingDataSource.afterPropertiesSet()
        return transactionRoutingDataSource
    }

    private fun buildDataSourceFromProperties(dataSourceProperty: DataSourceProperties, readOnly: Boolean): DataSource {
        return connectionPoolDataSource(
            DataSourceBuilder.create()
                .url(dataSourceProperty.url)
                .username(dataSourceProperty.username)
                .password(dataSourceProperty.password)
                .driverClassName(dataSourceProperty.driverClassName)
                .build(),
            readOnly
        )
    }

    private fun hikariConfig(dataSource: DataSource, readOnly: Boolean): HikariConfig {
        val hikariConfig = HikariConfig()
        hikariConfig.dataSource = dataSource
        hikariConfig.isAutoCommit = false
        hikariConfig.isReadOnly = readOnly
        return hikariConfig
    }

    private fun connectionPoolDataSource(dataSource: DataSource, readOnly: Boolean): HikariDataSource {
        return HikariDataSource(hikariConfig(dataSource, readOnly))
    }
}

TransactionRoutingDataSource.kt

class TransactionRoutingDataSource : AbstractRoutingDataSource() {
    override fun determineCurrentLookupKey(): Any {
        return if (TransactionSynchronizationManager.isCurrentTransactionReadOnly()) {
            DataSourceType.READ_ONLY
        } else {
            DataSourceType.READ_WRITE
        }
    }
}