mybatis-plus整合多数据源（读写分离）

1.首先pom文件

<dependency>
    <groupId>mysql</groupId>
    <artifactId>mysql-connector-java</artifactId>
</dependency>
<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-boot-starter</artifactId>
</dependency>
<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>dynamic-datasource-spring-boot-starter</artifactId>
</dependency>

2.数据源公用类

import com.alibaba.druid.pool.DruidDataSource;
import lombok.Data;
import org.springframework.stereotype.Component;

/**
 * @author andy
 **/
@Component
@Data
public class DatasourceBuilder {
    private String driverClassName;
    private String url;
    private String username;
    private String password;

    /**
     * 创建 DruidDataSource连接
     *
     * @return DruidDataSource实例
     */
    public DruidDataSource createDruidDataSource() {
        DruidDataSource datasource = new DruidDataSource();
        datasource.setDriverClassName(driverClassName);
        datasource.setUrl(url);
        datasource.setUsername(username);
        datasource.setPassword(password);
        return datasource;
    }


}

3.

/**
 * @author andy
 **/
@Configuration
@ConfigurationProperties(prefix = "spring.read.one")
public class ReadOneDataSourceBuilder extends DatasourceBuilder {

}

/**
 * @author andy
 **/
@Configuration
@ConfigurationProperties(prefix = "spring.write.datasource")
public class WriteDataSourceBuilder extends DatasourceBuilder {
}

4.配置多数据源

/**
 * @author andy
 **/
@Configuration
public class DruidConfiguration {
    @Autowired
    private MyMybatisProperties mybatisProperties;

    /**
     * 写入数据源.
     */
    @Autowired
    private WriteDataSourceBuilder writeDataSourceBuilder;
    /**
     * 读库数据源.
     */
    @Autowired
    private ReadOneDataSourceBuilder readOneDataSourceBuilder;
    /**
     * 数据源配置.
     */
    @Autowired
    private DataSourceConfigBuilder dataSourceConfigBuilder;

    /**
     * 写库.
     *
     * @return 写库实例
     */
    @Bean(name = "writeDataSource")
    @Primary
    public DataSource writeDataSource() {
        DruidDataSource dataSource = writeDataSourceBuilder.createDruidDataSource();
        return dataSourceConfigBuilder.config(dataSource);
    }

    /**
     * 有多少个从库就要配置多少个.
     *
     * @return 读库实例
     */
    @Bean(name = "readDataSourceOne")
    public DataSource readDataSourceOne() {
        DruidDataSource dataSource = readOneDataSourceBuilder.createDruidDataSource();
        return dataSourceConfigBuilder.config(dataSource);
    }

    /**
     * @return 读库源
     */
    @Bean("readDataSources")
    public List<DataSource> readDataSources() {
        List<DataSource> dataSources = new ArrayList<DataSource>();
        dataSources.add(readDataSourceOne());
        return dataSources;
    }

    @Bean
    public RoutingDataSource dynamicDataSource() {
        return new RoutingDataSource(writeDataSource(),readDataSources());
    }

    /**
     * sql session 工厂
     * @return 工厂
     * @throws Exception include IOException
     */
    @Bean
    @ConditionalOnMissingBean
    public SqlSessionFactory sqlSessionFactory() throws Exception {
        Resource[] resources = new PathMatchingResourcePatternResolver().getResources(mybatisProperties.getMapperLocations());
        // 解决myBatis下 不能嵌套jar文件的问题
        VFS.addImplClass(SpringBootVFS.class);
        MybatisSqlSessionFactoryBean factory = new MybatisSqlSessionFactoryBean();
        factory.setDataSource(dynamicDataSource());
        factory.setTypeAliasesPackage(mybatisProperties.getTypeAliasesPackage());
        factory.setMapperLocations(resources);
        SqlSessionFactory sqlSessionFactory =  factory.getObject();
        if (sqlSessionFactory != null) {
            sqlSessionFactory.getConfiguration().setMapUnderscoreToCamelCase(true);
        }
        return sqlSessionFactory;
    }

    /**
     * 事务管理器
     * @param source 数据源
     * @return 事务管理器
     */
    @Bean
    public DataSourceTransactionManager transactionManager(AbstractRoutingDataSource source) {
        return new DataSourceTransactionManager(source);
    }

5.AOP加本地线程变量

/**
 * @author andy
 **/
@Aspect
@Component
@Slf4j
@Order(0)
public class DatasourceAop {
    /**
     * 读库一般前缀.
     */
    private final String[] readSuffix = {"read", "get", "find", "search"};

    /**
     * 事务注解配置.
     *
     * @param joinPoint 切入点
     */
    @Before("execution(* com.tvunetworks.invoice.service..*.*(..))")
    public void switchDataSourceType(final JoinPoint joinPoint) {
        //判断是否是Transactional方法
        if (joinPoint.getSignature() instanceof MethodSignature) {
            Method method = ((MethodSignature) joinPoint
                    .getSignature()).getMethod();
            if (method.isAnnotationPresent(Transactional.class)) {
                DynamicDataSourceContextHolder.push(DataSourceType.write.getType());
                return;
            }
        }
        for (String read : readSuffix) {
            String name = joinPoint.getSignature().getName().toLowerCase();
            if (name.startsWith(read)) {
                log.debug("dataSource switch to: Read");
                DynamicDataSourceContextHolder.push(DataSourceType.read.getType());
                return;
            }
        }
        log.debug("dataSource switch to: Write");
        DynamicDataSourceContextHolder.push(DataSourceType.write.getType());
    }

}


/**
 * @author andy
 **/

public class DataSourceContextHolder {

	private static final ThreadLocal<String> LOCAL = new ThreadLocal<String>();

	public static ThreadLocal<String> getLocal() {
		return LOCAL;
	}

	/**
	 * 读可能是多个库
	 */
	public static void read() {
		LOCAL.set(DataSourceType.read.getType());
	}

	/**
	 * 写只有一个库
	 */
	public static void write() {
		LOCAL.set(DataSourceType.write.getType());
	}

	/**
	 * 一次获取及时移除
	 * @return String
	 */
	public static String getJdbcType() {
		String s = LOCAL.get();
		LOCAL.remove();
		return s;
	}
}


切换数据源
/**
 * @author andy
 **/
public class RoutingDataSource extends AbstractRoutingDataSource {
    /**
     * 读库数量
     */
    private final int readSize;
    /**
     * 计数器
     */
    private final AtomicInteger count=new AtomicInteger(0);

    public RoutingDataSource(DataSource defaultTargetDataSource, List<DataSource> targetDataSources){
        Map<Object, Object> dataSourceMap = new HashMap<>(2);
        dataSourceMap.put(DataSourceType.write.getType(),defaultTargetDataSource);
        this.setDefaultTargetDataSource(defaultTargetDataSource);
        readSize= targetDataSources.size();
        for (int i = 0; i <readSize ; i++) {
            dataSourceMap.put(i,targetDataSources.get(i));
        }
        this.setTargetDataSources(dataSourceMap);
    }
    @Override
    protected Object determineCurrentLookupKey() {
        String typeKey = DataSourceContextHolder.getJdbcType();
        if (StringUtils.isBlank(typeKey) || typeKey.equals(DataSourceType.write.getType())) {
            return DataSourceType.write.getType();
        }
        // 读 简单负载均衡
        int number = count.getAndAdd(1);
        int lookupKey = number % readSize;
        return Integer.valueOf(lookupKey);
    }
}