Spring Kotlin多租户+多数据源：TenantConnectionProvider无法注入DataSource

当你添加自定义的@EnableJpaRepositories配置类后，Spring Boot会自动禁用JPA的默认自动配置——因为它检测到你已经手动配置了JPA相关组件（比如指定repository扫描包、EntityManagerFactory引用等）。

在单数据源场景下，Spring Boot的自动配置会帮你创建一个默认的DataSource bean，所以TenantConnectionProvider能通过@Autowired顺利注入这个唯一的数据源实例。但自定义配置后，你没有在DataSourceConfigA中显式定义DataSource bean，导致Spring容器中没有可用的DataSource实例，自然无法完成lateinit var datasource的注入，最终抛出未初始化的错误。

1. 为每个数据源显式配置DataSource bean #

在每个数据源的配置类中，通过@Bean和@ConfigurationProperties加载对应数据源的配置参数，确保容器中存在独立的DataSource实例：

@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(
    transactionManagerRef = "transactionManagerA",
    entityManagerFactoryRef = "entityManagerFactoryA",
    basePackages = ["com.myapp.repository.a"]
)
class DataSourceConfigA {

    @Bean("dataSourceA")
    @ConfigurationProperties(prefix = "spring.datasource.a")
    fun dataSourceA(): DataSource {
        return DataSourceBuilder.create().build()
    }

    // 同理创建DataSourceConfigB，配置dataSourceB
}

2. 为每个数据源配置带多租户支持的EntityManagerFactory #

在每个数据源配置类中，创建LocalContainerEntityManagerFactoryBean时，绑定对应的DataSource，并配置多租户相关属性（策略、连接提供者、租户标识符解析器）：

@Bean("entityManagerFactoryA")
fun entityManagerFactoryA(
    @Qualifier("dataSourceA") dataSource: DataSource,
    hibernateProperties: HibernateProperties,
    tenantConnectionProvider: TenantConnectionProvider,
    tenantIdentifierResolver: TenantIdentifierResolver
): LocalContainerEntityManagerFactoryBean {
    val emfb = LocalContainerEntityManagerFactoryBean()
    emfb.dataSource = dataSource
    emfb.setPackagesToScan("com.myapp.entity.a")
    emfb.persistenceUnitName = "persistenceUnitA"
    emfb.jpaVendorAdapter = HibernateJpaVendorAdapter()

    // 组装多租户相关Hibernate属性
    val jpaProperties = hibernateProperties.determineHibernateProperties(
        emptyMap(),
        HibernateSettings()
    ).toMutableMap()
    jpaProperties[AvailableSettings.MULTI_TENANT] = MultiTenancyStrategy.SCHEMA.name // 根据你的租户策略调整
    jpaProperties[AvailableSettings.MULTI_TENANT_CONNECTION_PROVIDER] = tenantConnectionProvider
    jpaProperties[AvailableSettings.MULTI_TENANT_IDENTIFIER_RESOLVER] = tenantIdentifierResolver

    emfb.setJpaPropertyMap(jpaProperties)
    return emfb
}

@Bean("transactionManagerA")
fun transactionManagerA(
    @Qualifier("entityManagerFactoryA") emf: EntityManagerFactory
): PlatformTransactionManager {
    return JpaTransactionManager(emf)
}

3. 调整TenantConnectionProvider适配多数据源 #

不再注入单个DataSource，而是注入所有需要的数据源实例，通过上下文标识（比如ThreadLocal）动态切换当前使用的数据源：

@Component
class TenantConnectionProvider(
    @Qualifier("dataSourceA") private val dataSourceA: DataSource,
    @Qualifier("dataSourceB") private val dataSourceB: DataSource
) : MultiTenantConnectionProvider, HibernatePropertiesCustomizer {

    // 用ThreadLocal存储当前上下文对应的数据源标识
    private val currentDataSourceKey = ThreadLocal<String>()

    // 提供外部设置数据源的方法
    fun setCurrentDataSourceKey(key: String) {
        currentDataSourceKey.set(key)
    }

    // 根据标识获取当前数据源
    private fun getCurrentDataSource(): DataSource {
        return when(currentDataSourceKey.get()) {
            "A" -> dataSourceA
            "B" -> dataSourceB
            else -> throw IllegalArgumentException("未知数据源标识：${currentDataSourceKey.get()}")
        }
    }

    override fun getAnyConnection(): Connection = getCurrentDataSource().connection

    override fun releaseConnection(connection: Connection) {
        connection.close()
    }

    // 实现MultiTenantConnectionProvider的其他方法，均使用当前数据源
    override fun getConnection(tenantIdentifier: String?): Connection {
        val connection = getCurrentDataSource().connection
        // 根据租户标识切换schema/数据库（根据你的策略调整）
        tenantIdentifier?.let {
            connection.createStatement().execute("USE $it")
        }
        return connection
    }

    // 其他方法实现...

    override fun customize(hibernateProperties: MutableMap<String, Any>) {
        // 可在此统一设置多租户属性，也可在EntityManagerFactory配置中设置
    }
}

额外注意事项 #

• 每个EntityManagerFactory的persistenceUnitName必须唯一，避免容器内冲突。
• TenantIdentifierResolver需要和数据源切换逻辑配合，确保当前租户的数据源和租户标识匹配。
• 在业务代码中操作不同数据源时，要通过setCurrentDataSourceKey切换对应的数据源标识，确保多租户逻辑生效。

内容的提问来源于stack exchange，提问作者ojathelonius