Spring Boot中1000+单库租户的高效动态多数据源连接池配置问询

针对1000+单租户单数据库模式下的动态数据源管理需求，可通过以下方案替代静态Map实现，兼顾扩展性与性能：

1. 基于Hibernate多租户机制的动态数据源管理 #

利用Hibernate原生多租户API结合缓存实现动态数据源的创建、回收与切换，解决静态Map无法动态增删、资源占用过高的问题。

1.1 实现带自动回收的数据源缓存 #

使用Caffeine缓存存储租户数据源，配置闲置过期与容量上限，自动回收闲置租户的连接池资源：

@Component
public class TenantDataSourceCache {
    private final Cache<String, DataSource> dataSourceCache;
    private final DataSourceProperties dataSourceProperties;

    public TenantDataSourceCache(DataSourceProperties dataSourceProperties) {
        this.dataSourceProperties = dataSourceProperties;
        // 配置缓存：闲置1小时自动回收，预留50%冗余容量适配1000+租户
        this.dataSourceCache = Caffeine.newBuilder()
                .expireAfterAccess(1, TimeUnit.HOURS)
                .maximumSize(1500)
                .removalListener((tenantId, dataSource, cause) -> {
                    // 移除时关闭连接池释放资源
                    if (dataSource instanceof HikariDataSource hikariDS) {
                        hikariDS.close();
                    }
                })
                .build();
    }

    public DataSource getDataSource(String tenantId) {
        return dataSourceCache.get(tenantId, this::createTenantDataSource);
    }

    public void removeDataSource(String tenantId) {
        dataSourceCache.invalidate(tenantId);
    }

    private DataSource createTenantDataSource(String tenantId) {
        // 从配置中心/租户配置表拉取对应租户的数据库信息
        HikariDataSource ds = new HikariDataSource();
        ds.setJdbcUrl(getTenantJdbcUrl(tenantId));
        ds.setUsername(dataSourceProperties.getUsername());
        ds.setPassword(dataSourceProperties.getPassword());
        // 单租户连接池参数调优：避免资源浪费
        ds.setMinimumIdle(2);
        ds.setMaximumPoolSize(10);
        ds.setConnectionTimeout(30000);
        return ds;
    }

    private String getTenantJdbcUrl(String tenantId) {
        // 示例：按租户ID拼接独立数据库地址
        return "jdbc:mysql://localhost:3306/tenant_db_" + tenantId + "?useSSL=false&serverTimezone=UTC";
    }
}

1.2 实现Hibernate多租户连接提供者 #

将缓存整合到Hibernate连接提供者中，动态获取租户专属数据源：

public class DynamicTenantConnectionProvider implements MultiTenantConnectionProvider {
    private final TenantDataSourceCache dataSourceCache;

    public DynamicTenantConnectionProvider(TenantDataSourceCache dataSourceCache) {
        this.dataSourceCache = dataSourceCache;
    }

    @Override
    public Connection getConnection(String tenantIdentifier) throws SQLException {
        return dataSourceCache.getDataSource(tenantIdentifier).getConnection();
    }

    @Override
    public void releaseConnection(String tenantIdentifier, Connection connection) throws SQLException {
        connection.close();
    }

    // 实现其他必要接口方法（如isUnwrappableAs、unwrap等，可参考Hibernate官方实现）
}

1.3 实现租户标识符解析器 #

从业务上下文（如ThreadLocal、请求头）获取当前租户ID：

@Component
public class CurrentTenantIdentifierResolverImpl implements CurrentTenantIdentifierResolver {
    @Override
    public String resolveCurrentTenantIdentifier() {
        // 从ThreadLocal获取当前请求绑定的租户ID
        String tenantId = TenantContext.getTenantId();
        return tenantId != null ? tenantId : "default_tenant"; // 配置默认租户兜底
    }

    @Override
    public boolean validateExistingCurrentSessions() {
        return true;
    }
}

// 租户上下文工具类
public class TenantContext {
    private static final ThreadLocal<String> TENANT_ID = new ThreadLocal<>();

    public static void setTenantId(String tenantId) {
        TENANT_ID.set(tenantId);
    }

    public static String getTenantId() {
        return TENANT_ID.get();
    }

    public static void clear() {
        TENANT_ID.remove();
    }
}

1.4 Spring Boot配置整合 #

在application.yml中配置Hibernate多租户模式：

spring:
  jpa:
    properties:
      hibernate:
        multi_tenant: DATABASE # 单租户单数据库模式选DATABASE，单库多Schema选SCHEMA
        multi_tenant_connection_provider: com.example.DynamicTenantConnectionProvider
        tenant_identifier_resolver: com.example.CurrentTenantIdentifierResolverImpl
        hbm2ddl.auto: none # 禁用自动建表，租户数据库需提前初始化

2. 动态租户的增删管理 #

提供接口实现租户的动态注册与移除，同步管理数据源缓存：

2.1 添加租户接口 #

@RestController
@RequestMapping("/tenants")
public class TenantController {
    private final TenantDataSourceCache dataSourceCache;
    private final TenantConfigRepository tenantConfigRepository;

    public TenantController(TenantDataSourceCache dataSourceCache, TenantConfigRepository tenantConfigRepository) {
        this.dataSourceCache = dataSourceCache;
        this.tenantConfigRepository = tenantConfigRepository;
    }

    @PostMapping
    public ResponseEntity<Void> addTenant(@RequestBody TenantConfig tenantConfig) {
        // 持久化租户配置
        tenantConfigRepository.save(tenantConfig);
        // 预初始化数据源，避免首次请求延迟
        dataSourceCache.getDataSource(tenantConfig.getTenantId());
        return ResponseEntity.ok().build();
    }
}

2.2 移除租户接口 #

@DeleteMapping("/{tenantId}")
public ResponseEntity<Void> removeTenant(@PathVariable String tenantId) {
    tenantConfigRepository.deleteById(tenantId);
    // 清理缓存并关闭连接池
    dataSourceCache.removeDataSource(tenantId);
    return ResponseEntity.ok().build();
}

3. 性能优化建议 #

• 连接池参数调优：每个租户连接池最大连接数控制在10-20，最小空闲设为2-3，避免整体资源耗尽；
• 异步初始化：添加租户时通过异步任务初始化数据源，避免接口阻塞；
• 监控告警：监控缓存容量、各租户连接池活跃数，设置资源阈值告警；
• 预热机制：针对高频访问租户，定时触发数据源预热，减少冷启动延迟。

4. 轻量替代方案：Spring AbstractRoutingDataSource #

若无需依赖Hibernate，可扩展Spring原生的AbstractRoutingDataSource实现动态路由：

public class DynamicRoutingDataSource extends AbstractRoutingDataSource {
    private final TenantDataSourceCache dataSourceCache;

    public DynamicRoutingDataSource(TenantDataSourceCache dataSourceCache) {
        this.dataSourceCache = dataSourceCache;
        setDefaultTargetDataSource(dataSourceCache.getDataSource("default_tenant"));
    }

    @Override
    protected Object determineCurrentLookupKey() {
        return TenantContext.getTenantId();
    }

    @Override
    protected DataSource determineTargetDataSource() {
        String tenantId = (String) determineCurrentLookupKey();
        if (tenantId == null) {
            return super.determineTargetDataSource();
        }
        return dataSourceCache.getDataSource(tenantId);
    }
}

配置类注册该数据源：

@Configuration
public class DataSourceConfig {
    @Bean
    public DataSource dynamicDataSource(TenantDataSourceCache dataSourceCache) {
        return new DynamicRoutingDataSource(dataSourceCache);
    }
}

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