Spring-boot MyBatis配置 (2)

多数据源配置

分包: 不同数据源的在不同的目录下;事务的回滚需要创建根据数据源创建
注解
AOP: aop注解切面需要在Service层进行数据源切换;事务可以将多个数据源放在一个事务中;

分包形式

不同的数据源的sql操作分布在不同的路径下

两个数据源的basepackage分别为：

com.yany.dao.multi.ads
com.yany.dao.multi.rds

在创建MapperScannerConfigurer时，对应不同的数据源扫描不同的basepackage路径

mapperScannerConfigurer.setBasePackage("xxxx");

对应的xml，即MAPPER_PATH

classpath:/com/yany/mapper/multi/ads/**.xml
classpath:/com/yany/mapper/multi/rds/**.xml

在创建SqlSessionFactoryBean时，MAPPER_PATH对应分别对应于ads和rds的sql路径

 sessionFactory.setMapperLocations(pathMatchingResourcePatternResolver.getResources(MAPPER_PATH));

在使用时，不同数据源的操作在分别在不同的路径创建即可。

注解形式

准备好两个注解类，分别对应于两个数据源：

public @interface RdsRepository {
}
public @interface AdsRepository {
}

同分包类似分别为Rds和Ads两个数据源创建两个SqlSessionFactoryBean和DataSourceTransactionManager，略微不同的是SqlSessionFactoryBean的setMapperLocations是允许相同路径。
在创建两个MapperScannerConfigurer

    /**
     * 以注解的方式 进行多数据源配置
     *
     * @return
     */
    @Bean
    public MapperScannerConfigurer createAnnotatationAdsMapperScannerConfigurer() {
        MapperScannerConfigurer mapperScannerConfigurer = new MapperScannerConfigurer();
        mapperScannerConfigurer.setBasePackage("com.yany.dao.multi.annotation");
        mapperScannerConfigurer.setSqlSessionFactoryBeanName("annotationAdsSqlSessionFactory");
        mapperScannerConfigurer.setAnnotationClass(AdsRepository.class);
        return mapperScannerConfigurer;
    }

    /**
     * 以注解的方式 进行多数据源配置
     *
     * @return
     */
    @Bean
    public MapperScannerConfigurer createAnnotatationRdsMapperScannerConfigurer() {
        MapperScannerConfigurer mapperScannerConfigurer = new MapperScannerConfigurer();
        mapperScannerConfigurer.setBasePackage("com.yany.dao.multi.annotation");
        mapperScannerConfigurer.setSqlSessionFactoryBeanName("annotationRdsSqlSessionFactory");
        mapperScannerConfigurer.setAnnotationClass(RdsRepository.class);
        return mapperScannerConfigurer;
    }

上述代码和以前的主要的区别在：setAnnotationClass设置对应不同的注解类

使用时，在不同数据源的dao接口上添加对应的注解

@RdsRepository
public interface AnnotationRdsDao {
    int selectCount();
}
@AdsRepository
public interface AnnotationAdsDao {
    int selectCount();
}

而sql对应的xml不变，对应好namespace即可

AOP形式

创建一个动态数据源

创建数据源类型的枚举类

public enum DatabaseType {
    Ads, Rds
}

创建一个线程安全的DatabaseType容器

public class DatabaseContextHolder {
    private final static ThreadLocal<DatabaseType> contextHolder = new ThreadLocal<>();

    public static DatabaseType getDatabaseType() {
        return contextHolder.get();
    }

    public static void setDatabaseType(DatabaseType type) {
        contextHolder.set(type);
    }

}

ThreadLocal类为每一个线程都维护了自己独有的变量拷贝，每个线程都拥有了自己独立的一个变量，避免并发问题。

创建动态数据源DynamicDataSource继承AbstractRoutingDataSource

public class DynamicDataSource extends AbstractRoutingDataSource {

    @Override
    protected Object determineCurrentLookupKey() {
        return DatabaseContextHolder.getDatabaseType();
    }
}

创建AOP对应的MyBatis配置
创建动态数据源的bean

    @Bean
    public DynamicDataSource setDataSource() {
        Map<Object, Object> targetDataSources = new HashMap<>();
        targetDataSources.put(DatabaseType.Ads, adsDataSource);
        targetDataSources.put(DatabaseType.Rds, rdsDataSource);

        DynamicDataSource dataSource = new DynamicDataSource();
        dataSource.setTargetDataSources(targetDataSources);
        dataSource.setDefaultTargetDataSource(rdsDataSource);// 默认的datasource设置为rdsDataSource
        return dataSource;
    }

创建SqlSessionFactoryBean和DataSourceTransactionManager以及这个和以前类似不再赘述，具体看github上代码

创建切边
扫描对应的Service层，在执行具体的服务代码前，根据调用的Service类进行数据源的切换。

@Aspect
@Component
public class DataSourceAspect {
    /**
     * 使用空方法定义切点表达式
     */
    @Pointcut("execution(* com.yany.service.**.*(..))")
    public void declareJointPointExpression() {
    }

    @Before("declareJointPointExpression()")
    public void setDataSourceKey(JoinPoint point) {
        if (point.getTarget() instanceof IAdsAopService ||
                point.getTarget() instanceof AdsAopServiceImpl) {
            //根据连接点所属的类实例，动态切换数据源
            System.out.println("IAdsAopService Aspect");
            DatabaseContextHolder.setDatabaseType(DatabaseType.Ads);
        } else {//连接点所属的类实例是（当然，这一步也可以不写，因为defaultTargertDataSource就是该类所用的rdsDataSource）
            System.out.println("IRdsAopService Aspect");
            DatabaseContextHolder.setDatabaseType(DatabaseType.Rds);
        }

    }
}

上述切换规则比较简单，具体可根据业务情况，包目录结构，或者是类名规则等进行解析切换。

具体代码将github：https://github.com/yany8060/SpringDemo.git
博客：http://yany8060.xyz