简单的生产数据库重连策略优化

news2025/1/6 10:03:29

简单的druid生产数据库重连策略优化

1. 需求

我们生产环境有一次在大量请求拥堵后，好多数据库操作都超时了，在此之后数据库的连接池不能正常的获取数据库连接了，我们确认了数据库服务是没有问题的，那么就是连接池的配置有问题，所以需要优化连接池的配置信息，我们使用的是druid连接池，优化后服务便不会出现之前连接池获取不到连接情况，此处记录一下方便后续使用方便。

2. 具体优化配置

2.1 修改内容

我们的数据源是多数据源，从配置文件中读取，使用java代码实现，此处就不多赘述了，此处主要修改的breakAfterAcquireFailure这个参数。druidDataSource.setBreakAfterAcquireFailure(boolean flag)其具体参数作用如下所示：

当设置为 true 时，如果获取连接失败（例如数据库服务不可用），Druid 连接池将进入“断开”状态。一旦进入该状态，后续的连接请求都会立即失败，直到连接池被重置或重启。
当设置为 false 时，即使获取连接失败，连接池也不会进入“断开”状态，而是会继续尝试获取新的连接。

我们还添加了testOnBorrow参数设置，其具体作用如下：

当设置为 true 时，在从连接池中借用连接之前，会先测试连接的有效性。如果连接无效，则会从连接池中移除并尝试获取新的有效连接。
这种配置可以确保每次获取到的连接都是有效的，避免了使用失效连接导致的问题。

2.2 修改前后对比

修改前配置：setBreakAfterAcquireFailure(true):

优点: 在某些情况下可以快速失败，避免浪费资源在无效的连接尝试上。
缺点: 一旦连接池进入“断开”状态，除非手动重置或重启，否则无法恢复。这会导致在数据库服务短暂不可用后，应用无法自动恢复连接。

修改后配置：setBreakAfterAcquireFailure(false) + setTestOnBorrow(true):

优点: 即使多次连接失败，连接池也不会进入“断开”状态，而是继续尝试获取新连接。同时，setTestOnBorrow(true) 确保每次获取到的连接都是有效的，从而提高了系统的健壮性和自愈能力。
缺点: 可能会增加一些性能开销，因为每次借用连接前都需要进行有效性测试。

修改后具体代码如下：

/**
 * 动态数据源配置
 * @return DataSource 动态数据源
 */
@Primary
@Bean
public DataSource multipleDataSource() {
	// 从获取动态数据源配置信息
	List<BusinessModuleType.Module> moduleList = businessModuleType.getModules();
	DynamicDataSource dynamicDataSource = DynamicDataSource.getInstance();
	Map<Object, Object> targetDataSources = new HashMap<>(BaseCacheProperties.DEFAULT_MAP_SIZE);
	moduleList.forEach(module -> {
		DruidDataSource moduleDataSource = new DruidDataSource();
		// 设置密码用户错误重连次数
		moduleDataSource.setConnectionErrorRetryAttempts(BaseCacheProperties.ERROR_RETRY_ATTEMPTS);
		// 数据库连接池连接失败重连配置
		moduleDataSource.setBreakAfterAcquireFailure(false);
		moduleDataSource.setTestOnBorrow(true);
		moduleDataSource.setDriverClassName(module.getClassName());
		moduleDataSource.setUrl(module.getUrl());
		moduleDataSource.setUsername(module.getUsername());
		moduleDataSource.setPassword(module.getPassword());
		moduleDataSource.setInitialSize(module.getInitialSize());
		moduleDataSource.setMinIdle(module.getMinSize());
		moduleDataSource.setMaxActive(module.getMaxSize());
		moduleDataSource.setMaxWait(module.getWaitTime());
		moduleDataSource.setTimeBetweenEvictionRunsMillis(module.getTimeBetweenEvictionRunsMillis());
		moduleDataSource.setMinEvictableIdleTimeMillis(module.getMinEvictableIdleTimeMillis());
		moduleDataSource.setValidationQuery(module.getValidation());
		if (BaseCacheProperties.DEFAULT_FLAG.equals(module.getName())) {
			targetDataSources.put(module.getDb(), dataSource);
		} else {
			targetDataSources.put(module.getDb(), moduleDataSource);
		}
	});
	dynamicDataSource.setTargetDataSources(targetDataSources);
	dynamicDataSource.setDefaultTargetDataSource(dataSource);
	return dynamicDataSource;
}