使用Spring的AbstractRoutingDataSource类来进行拓展多数据源

时间：2024-01-21 22:22:03

1.继承抽象类AbstractRoutingDataSource

通过重写这个抽象类的determineCurrentLookupKey()方法来决定返回哪个数据库。

2.配置多个数据库

<bean id="dataSource" class="com.bihang.util.DynamicDataSource">

    <property name="targetDataSources">

        <map key-type="java.lang.String">

            <entry key="ssm1DataSource" value-ref="ssm1DataSource"/>

            <entry key="ssm2DataSource" value-ref="ssm2DataSource"/>

        </map>

    </property>

    <!--默认数据源-->

    <property name="defaultTargetDataSource" ref="ssm1DataSource"/>

</bean>

因为com.bihang.util.DynamicDataSource这个类继承了AbstractRoutingDataSource这个抽象类，所以也继承了父类的成员，所以利用依赖注入将数据源名称和对应的数据库注入进去。

当Mybatis用到数据库就会找这个叫dataSource的数据源，而此时dataSource是有多个数据源的，具体用哪一个它会去找刚刚继承实现的determineCurrentLookupKey()这个方法返回的数据源名称，从而决定使用那个数据源。数据源名称是使用ThreadLocal来保存的。

总而言之，我们现在控制determineCurrentLookupKey()这个抽象方法返回的数据源名称就可以决定用哪个数据库。

但是这样每次想要调用数据库时要提供对应的数据源名称，很繁琐。

我们采用将不同数据库的业务方法存放在不同的包下，然后利用反射获取包名，再用aop动态的赋值数据源名称。

aop里面会利用JoinPoint对象来获取目标对象从而获取包名。

JoinPoint 对象

JoinPoint对象封装了SpringAop中切面方法的信息,在切面方法中添加JoinPoint参数,就可以获取到封装了该方法信息的JoinPoint对象.

String toString();         //连接点所在位置的相关信息

String toShortString();     //连接点所在位置的简短相关信息

String toLongString();     //连接点所在位置的全部相关信息

Object getThis();         //返回AOP代理对象

Object getTarget();       //返回目标对象

Object[] getArgs();       //返回被通知方法参数列表

Signature getSignature();  //返回当前连接点签名

SourceLocation getSourceLocation();//返回连接点方法所在类文件中的位置

String getKind();        //连接点类型

StaticPart getStaticPart(); //返回连接点静态部分

具体代码:

Logger logger = Logger.getLogger(this.getClass());

public void before(JoinPoint point){

    Class<?> targetClass  = point.getTarget().getClass();

    String targetPackageName =targetClass.getName();

    logger.debug(targetPackageName);

    if (targetPackageName.contains("ssmone")){

        DataSourceHolder.setDataSources(Constants.DATASOURCE_ONE);

    }else {

        DataSourceHolder.setDataSources(Constants.DATASOURCE_TWO);

    }

}

public void after() {

    DataSourceHolder.setDataSources(null);

}

相关文章