解决mybatis-plus动态数据源切换不生效的问题

目录

• 一、问题描述
• 二、环境
• 1.依赖
• 2.配置
• 三、解决方法
• 四、测试
• 五、问题分析

一、问题描述

在我们项目中，既要连接mysql，又要连接TDEngine（taos），正确配置后也无法动态切换数据源执行sql

二、环境

1.依赖

        
            com.taosdata.jdbc
            taos-jdbcdriver

            3.0.0
        
        
   		
            com.baomidou
            dynamic-datasource-spring-boot-starter
            3.3.2
        

2.配置

spring:
  datasource:
    dynamic:
      strict: true
      primary: mysql
      datasource:
        mysql:
          driver-class-name: com.mysql.cj.jdbc.Driver
          url: jdbc:mysql://192.xxx.xxx.xxx:3306/db_iot?useUnicode=true&characterEncoding=UTF-8&useSSL=false&serverTimezone=Asia/Shanghai
          username: root
          password: 123456
        tdengine:
          driver-class-name: com.taosdata.jdbc.TSDBDriver
          url: jdbc:TAOS://192.xxx.xxx.xxx:6030/iot_data?timezone=UTC-8&charset=UTF-8&locale=en_US.UTF-8
          username: root
          password: taosdata
        mysql1:
          driver-class-name: com.mysql.cj.jdbc.Driver
          url: jdbc:mysql://192.xxx.xxx.xxx:3306/db_portal?useUnicode=true&characterEncoding=UTF-8&useSSL=false&serverTimezone=Asia/Shanghai
          username: root
          password: 123456

三、解决方法

直接上代码，最后会有问题分析，因为涉及到源码，这里暂时不讲
这里的dynamicRoutingDataSource()不能有DynamicDataSourceAutoConfiguration里的那样命名和返回值，那样在注入时就会因为其他数据源的注入导致无法注入动态数据源，需要细化到具体的类型

package com.xxx.project.iotconf.configs;

import com.baomidou.dynamic.datasource.DynamicRoutingDataSource;
import com.baomidou.dynamic.datasource.provider.DynamicDataSourceProvider;
import com.baomidou.dynamic.datasource.provider.YmlDynamicDataSourceProvider;
import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DataSourceProperty;
import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DynamicDataSourceProperties;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;

import javax.sql.DataSource;
import java.util.Map;


@Configuration
public class TDEngineConfig {

    @Autowired
    private DynamicDataSourceProperties properties;

    @Bean
    @ConditionalOnMissingBean
    @Primary
    public DynamicDataSourceProvider dynamicDataSourceProvider() {
        Map datasourceMap = this.properties.getDatasource();
        return new YmlDynamicDataSourceProvider(datasourceMap);
    }

    @Bean
    @ConditionalOnMissingBean
    public DynamicRoutingDataSource dynamicRoutingDataSource(DynamicDataSourceProvider dynamicDataSourceProvider) {
        DynamicRoutingDataSource dynamicRoutingDataSource = new DynamicRoutingDataSource();
        dynamicRoutingDataSource.setPrimary(this.properties.getPrimary());
        dynamicRoutingDataSource.setStrict(this.properties.getStrict());
        dynamicRoutingDataSource.setStrategy(this.properties.getStrategy());
        dynamicRoutingDataSource.setProvider(dynamicDataSourceProvider);
        dynamicRoutingDataSource.setP6spy(this.properties.getP6spy());
        dynamicRoutingDataSource.setSeata(this.properties.getSeata());
        Map dataSourceMap = dynamicDataSourceProvider.loadDataSources();
        for (String key : dataSourceMap.keySet()) {
            dynamicRoutingDataSource.addDataSource(key, dataSourceMap.get(key));
        }
        return dynamicRoutingDataSource;
    }

}

四、测试

放入TAOS创建超表的SQL，由JdbcTemplate去执行，执行成功

package com.xxx.project.iot.pulsar.handler;

import com.baomidou.dynamic.datasource.annotation.DS;
import com.shandy.project.api.basic.dto.SDevice;
import com.shandy.project.api.basic.dto.Tuple2;
import com.shandy.project.iot.pulsar.utils.ReflectUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.stereotype.Service;

import javax.annotation.PostConstruct;
import java.util.List;
import java.util.Map;

@Service
@Slf4j
public class DataHandler implements DsHandler{

    @Autowired
    private JdbcTemplate jdbcTemplate;

    @PostConstruct
    public void init() {

    }

    /**
     * 执行单条DDL sql
     */
    @DS(value = "tdengine")
    public void execute(String sql) {
        jdbcTemplate.execute(sql);
        log.info(sql);
    }

    /**
     * 执行单条DML sql
     */
    @DS(value = "tdengine")
    public List> query(String sql, Object[] args) {
        log.info(String.format("%s，params is {%s}", sql, args));
        return jdbcTemplate.queryForList(sql, args);
    }

    /**
     * 保存设备数据(批量，多表多条)
     */
    @DS(value = "tdengine")
    public void batchInsertDevice(List devices, String table) {
        String[] sqls = null;
        jdbcTemplate.batchUpdate(sqls);
        log.info(sqls.toString());
    }

    /**
     * 单条插入
     *
     * @param device  消息结构体
     * @param mqttObj 模型对象，IMqttR或IMqttS对象
     */
    @DS(value = "tdengine")
    public void insertDevice(SDevice device, Object mqttObj) {
        Tuple2> t2 = ReflectUtils.getInsSql(mqttObj.getClass(), device);
        List