spring-boot+dynamic-datasource實現真正的動態(tài)數據源

1、spring-boot+dynamic-datasource實現真正的動態(tài)數據源一、前言 現在有這樣一個連接表。數據庫連接信息是從連接表中獲取，想對連接表中的所有數據源進行維護只有用動態(tài)數據庫。在網上找了很多動態(tài)數據庫教程。都是用Ds注解選擇需要的數據源，意思是在編寫代碼時就要確定數據源，并不能實現需求。想要對對應的連接進行管理，只有通過連接id能創(chuàng)建一個數據源，并使其生效，下面進入正題。二、準備1.兩個不同的數據源以及連接表2.主要依賴 com.baomidou mybatis-plus-boot-starter mysql mysql-connector-java 8.0.25 com.b

2、aomidou dynamic-datasource-spring-boot-starter 3.4.1 com.alibaba druid-spring-boot-starter 1.2.6 org.springframework.boot spring-boot-starter-aop 3.實體類Data TableName(dm_connect) ApiModel(value = Connect對象, description = ) public class Connect extends Model ApiModelProperty(value = 主鍵) TableId(value

3、= id, type = IdType.AUTO) private Integer id; ApiModelProperty(value = ip地址) TableField private String ipAddress; ApiModelProperty(value = 端口) TableField private Integer port; ApiModelProperty(value = 賬號) TableField private String username; ApiModelProperty(value = 密碼) TableField private String pass

4、word; TableName(information_schema.SCHEMATA) Data public class Schemata TableField(CATALOG_NAME) private String catalogName; TableField(SCHEMA_NAME) private String schemaName; TableField(DEFAULT_CHARACTER_SET_NAME) private String defaultCharacterSetName; TableField(DEFAULT_COLLATION_NAME) private St

5、ring defaultCollationName; TableField(SQL_PATH) private String sqlPath; TableField(DEFAULT_ENCRYPTION) private String defaultEncryption; 三、出發(fā)1.默認數據源配置 默認數據源主要功能是保存連接信息ymlspring: datasource: url: jdbc:mysql:/*.*.*.227:3306/mysql_manager?serverTimezone=UTC&useUnicode=true&characterEncoding=utf-8&zeroD

6、ateTimeBehavior=convertToNull&useSSL=false&allowPublicKeyRetrieval=true username: root password: zxc4256101 driver-class-name: com.mysql.cj.jdbc.Driver type: com.alibaba.druid.pool.DruidDataSource # 指定數據源類型MybatisPlusConfigMapperScan(自己定義mapper掃描邏輯) Configuration EnableTransactionManagement public c

7、lass MybatisPlusConfig Value($spring.datasource.url) private String url; Value($spring.datasource.username) private String username; Value($spring.datasource.password) private String password; Value($spring.datasource.driver-class-name) private String driverClassName; /* * * return 數據源配置 * throws SQ

8、LException */ Bean Primary ConfigurationProperties(prefix = spring.datasource.master) public DataSource dataSourceMaster() throws SQLException DruidDataSource dataSource = new DruidDataSource(); dataSource.setUrl(url); dataSource.setDriver(new DruidDriver().createDriver(driverClassName); dataSource.

9、setUsername(username); dataSource.setPassword(password); dataSource.setUseGlobalDataSourceStat(true); dataSource.setPoolPreparedStatements(true); dataSource.setValidationQuery(select 1); dataSource.setTimeBetweenEvictionRunsMillis(6000); return dataSource; /* * 將默認數據源添加進動態(tài)數據源 * param dataSourceMaste

10、r * return */ Bean(dynamicDataSource) public DataSource dynamicDataSource(DataSource dataSourceMaster) DynamicRoutingDataSource dynamicRoutingDataSource = new DynamicRoutingDataSource(); dynamicRoutingDataSource.addDataSource(master, dataSourceMaster); return dynamicRoutingDataSource; Bean Primary C

11、onfigurationProperties(prefix = mybatis) public MybatisSqlSessionFactoryBean sqlSessionFactoryBean(Qualifier(dynamicDataSource) DataSource dataSource, Qualifier(MybatisPlusInterceptor) MybatisPlusInterceptor mybatisPlusInterceptor) MybatisSqlSessionFactoryBean sqlSessionFactoryBean = new MybatisSqlS

12、essionFactoryBean(); sqlSessionFactoryBean.setDataSource(dataSource); /分頁插件 Interceptor plugins = new Interceptor1; plugins0 = mybatisPlusInterceptor; sqlSessionFactoryBean.setPlugins(plugins); return sqlSessionFactoryBean; /* * myBatis-plus 分頁查詢配置 */ Bean(name=MybatisPlusInterceptor) public Mybatis

13、PlusInterceptor MpPaginationInterceptor() MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor(); interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.MYSQL); return interceptor; /在方法上使用 Target(ElementType.METHOD) Retention(RetentionPolicy.RUNTIME) Documented public inter

14、face Dynamic /* * 數據庫名稱 */ String database() default ; Aspect Component Slf4j RequiredArgsConstructor public class DynamicAspect public static final String ID = id; public static final String TABLE_SCHEMA=tableSchema; public static final String DS_NAME = dynamic; private final IConnectService connec

15、tService; private final DynamicRoutingDataSource dynamicRoutingDataSource; Pointcut(annotation(包名.annotation.Dynamic) public void pointcut() /* * 進入方法前通過連接id定義一個數據源并設為默認 * param joinPoint */ Before(pointcut() public void before(JoinPoint joinPoint) MethodSignature ms = (MethodSignature) joinPoint.ge

16、tSignature(); Method method = ms.getMethod(); Dynamic dynamic = method.getAnnotation(Dynamic.class); ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes(); if (null != attributes & null != attributes.getRequest().getParameter(ID) /獲取請求頭參數 Intege

17、r id = Integer.parseInt(attributes.getRequest().getParameter(ID); /數據庫名稱：information_schema請求注解 String database=information_schema; /從請求中獲取數據庫名 if(StringUtils.isNotBlank(attributes.getRequest().getParameter(TABLE_SCHEMA) database=attributes.getRequest().getParameter(TABLE_SCHEMA); /從注解上獲取數據庫名稱 if(St

18、ringUtils.isNotBlank(dynamic.database() database=dynamic.database(); /獲取連接參數 Connect connect = connectService.getById(id); if (null = connect) throw new RuntimeException(連接不存在！); /組裝datasource DruidDataSource dataSource = new DruidDataSource(); dataSource.setUrl(jdbc:mysql:/ + connect.getIpAddress()

19、 + : + connect.getPort() + / + database); dataSource.setUsername(connect.getUsername(); dataSource.setPassword(connect.getPassword(); /添加datasource到動態(tài)數據源并設為primary dynamicRoutingDataSource.addDataSource(dynamic, dataSource); dynamicRoutingDataSource.setPrimary(dynamic); Around(pointcut() public Object doAround(ProceedingJoinPoint joinPoint) throws Throwable try return joinPceed(); catch (Throwable throwable) /方法出錯時重置數據源 throwable.printStackTrace(); dynamicRoutingDataSource.setPrimary(master); dynamicRoutingDataSource.removeDataSource(dynamic); return null; /* * 在切入點return內容之后切入內容（可以用來對處理返