Distributed transaction managing in sharding environment - spring-boot

Question

I'm using two datasources named mainDB and ShardDB. ShardDB is consist of 4 databases, and I'm trying to create method that write something in mainDB and ShardDB_1 and ShardDB_2. This method should be run in atomically.
How can I make this atomic function?
What I tried was using chaindTransactionManager, but It only manages a single connections per datasource.(manages a mainDB connection and one of ShardDB connection)
#Bean("userHikariDataSource")
fun userDataSource(): DataSource? {
val shardSize = user.shards.size
val router = UserDataSourceRouter(
shardSize,
user.virtualShardSize,
user.virtualShards
)
val dataSourceMap: MutableMap<Any, Any> = LinkedHashMap()
// sharding
for (i in 0 until user.shards.size) {
val shard: UserShardingDataSourceProperty.Shard = user.shards[i]
val masterDataSourceName =
i.toString() + DataSourceKey.DATASOURCE_DELIMITER + MultiDataSource.USER + DataSourceKey.DATASOURCE_DELIMITER + shard.master.name
val masterDs: DataSource =
dataSource(shard.username, shard.password, shard.master.url, masterDataSourceName)
dataSourceMap[masterDataSourceName] = masterDs
for (slave in shard.slaves) {
val slaveDataSourceName =
i.toString() + DataSourceKey.DATASOURCE_DELIMITER + MultiDataSource.USER + DataSourceKey.DATASOURCE_DELIMITER + slave.name
val slaveDs: DataSource = dataSource(shard.username, shard.password, slave.url, slaveDataSourceName)
dataSourceMap[slaveDataSourceName] = slaveDs
}
}
router.setTargetDataSources(dataSourceMap)
router.afterPropertiesSet()
userDataSourceRouter = router
return LazyConnectionDataSourceProxy(router)
}
#Bean(name = ["userHikariEntityManager"])
fun entityManagerFactory(#Qualifier("userHikariDataSource") dataSource: DataSource): LocalContainerEntityManagerFactoryBean {
val entityManager = LocalContainerEntityManagerFactoryBean()
entityManager.dataSource = dataSource
val vendorAdapter = HibernateJpaVendorAdapter()
vendorAdapter.setDatabase(jpaVendorAdapterProperty.database)
vendorAdapter.setShowSql(jpaVendorAdapterProperty.showSql)
vendorAdapter.setDatabasePlatform(jpaVendorAdapterProperty.databasePlatform)
entityManager.jpaVendorAdapter = vendorAdapter
entityManager.setPackagesToScan("com.kakaoenterprise.kakaowork.gate.message")
entityManager.persistenceUnitName = "userHikariEntityManager"
return entityManager
}
#Bean(name = ["userHikariTransactionManager"])
fun userTransactionManager(
#Qualifier("userHikariEntityManager") entityManagerFactory: EntityManagerFactory
): PlatformTransactionManager {
val transactionManager = JpaTransactionManager()
transactionManager.entityManagerFactory = entityManagerFactory
return transactionManager
}
private fun dataSource(username: String, password: String, url: String, resourceName: String): DataSource {
val config = commonHikariConfig.hikariConfig()
config.jdbcUrl = url
config.username = username
config.password = password
return HikariDataSource(config)
}
I'm using above as datasource config.