首页 > 代码库 > 第九章 springboot + mybatis + 多数据源 (AOP实现)

第九章 springboot + mybatis + 多数据源 (AOP实现)

在第八章 springboot + mybatis + 多数据源代码的基础上,做两点修改

1、ShopDao

技术分享
package com.xxx.firstboot.dao;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.stereotype.Repository;import com.xxx.firstboot.domain.Shop;import com.xxx.firstboot.mapper.ShopMapper;@Repositorypublic class ShopDao {    @Autowired    private ShopMapper mapper;    /**     * 获取shop     */    public Shop getShop(int id) {        return mapper.getShop(id);    }}
View Code

说明:只是去掉了设置数据源key的那一句代码

 

2、DataSourceAspect

技术分享
package com.xxx.firstboot.common.datasource;import org.aspectj.lang.JoinPoint;import org.aspectj.lang.annotation.Aspect;import org.aspectj.lang.annotation.Before;import org.springframework.stereotype.Component;import com.xxx.firstboot.dao.ShopDao;@Aspect@Componentpublic class DataSourceAspect {        @Before("execution(* com.xxx.firstboot.dao.*.*(..))")    public void setDataSourceKey(JoinPoint point){        //连接点所属的类实例是ShopDao        if(point.getTarget() instanceof ShopDao){            DatabaseContextHolder.setDatabaseType(DatabaseType.mytestdb2);        }else{//连接点所属的类实例是UserDao(当然,这一步也可以不写,因为defaultTargertDataSource就是该类所用的mytestdb)            DatabaseContextHolder.setDatabaseType(DatabaseType.mytestdb);        }    }    //    @Around("execution(* com.xxx.firstboot.dao.*.*(..))")//    public Object setDataSourceKeyByAround(ProceedingJoinPoint point) throws Throwable{//        if(point.getTarget() instanceof ShopDao){//            DatabaseContextHolder.setDatabaseType(DatabaseType.mytestdb2);//        }else{//连接点所属的类实例是UserDao(当然,这一步也可以不写,因为defaultTargertDataSource就是该类所用的mytestdb)//            DatabaseContextHolder.setDatabaseType(DatabaseType.mytestdb);//        }//        return point.proceed();//执行目标方法//    }    }
View Code

说明:列出了两种切面方法,在这里推荐使用前者,原因:

  • @Around:需要写执行目标方法的那一行代码,而这一行代码可能会抛异常,还需要抛出或捕获

对于切点表达式,可以抽取出来,进行重复利用。如上代码可以改为如下:

技术分享
package com.xxx.firstboot.common.datasource;import org.aspectj.lang.JoinPoint;import org.aspectj.lang.annotation.Aspect;import org.aspectj.lang.annotation.Before;import org.aspectj.lang.annotation.Pointcut;import org.springframework.stereotype.Component;import com.xxx.firstboot.dao.ShopDao;@Aspect@Componentpublic class DataSourceAspect {    /**     * 使用空方法定义切点表达式     */    @Pointcut("execution(* com.xxx.firstboot.dao.*.*(..))")    public void declareJointPointExpression() {    }    /**     * 使用定义切点表达式的方法进行切点表达式的引入     */    @Before("declareJointPointExpression()")    public void setDataSourceKey(JoinPoint point) {        // 连接点所属的类实例是ShopDao        if (point.getTarget() instanceof ShopDao) {            DatabaseContextHolder.setDatabaseType(DatabaseType.mytestdb2);        } else {// 连接点所属的类实例是UserDao(当然,这一步也可以不写,因为defaultTargertDataSource就是该类所用的mytestdb)            DatabaseContextHolder.setDatabaseType(DatabaseType.mytestdb);        }    }}
View Code

注意:该切点表达式也可以用在其他切面类中,引入的时候使用"全类名.切点方法名()",例:@Before("com.xxx.firstboot.common.datasource.DataSourceAspect.declareJointPointExpression()")

第九章 springboot + mybatis + 多数据源 (AOP实现)