AbstractRoutingDataSource AOP如何实现动态数据源切换

本篇内容介绍了"AbstractRoutingDataSource AOP如何实现动态数据源切换"的有关知识，在实际案例的操作过程中，不少人都会遇到这样的困境，接下来就让小编带领大家学习一下如何处理这些情况吧！希望大家仔细阅读，能够学有所成！

数据库配置：application.properties

## datasource master #spring.datasource.type=com.alibaba.druid.pool.DruidDataSourcespring.datasource.driver-class-name=com.mysql.jdbc.Driverspring.datasource.url=jdbc:mysql://localhost:3306/master?characterEncoding=UTF-8&serverTimezone=UTCspring.datasource.username=rootspring.datasource.password=123456## datasource slave #spring.datasource-slave.type=com.alibaba.druid.pool.DruidDataSourcespring.datasource-slave.driver-class-name=com.mysql.jdbc.Driverspring.datasource-slave.url=jdbc:mysql://localhost:3306/slave?characterEncoding=UTF-8&serverTimezone=UTCspring.datasource-slave.username=rootspring.datasource-slave.password=123456

编写数据库名注解

public interface Datasources {    String MASTER_DB = "masterDB";    String SLAVE_DB = "slaveDB";}

配置数据源

@Configurationpublic class DataSourceConfig {    //destroy-method="close"的作用是当数据库连接不使用的时候,就把该连接重新放到数据池中,方便下次使用调用.    @Bean(destroyMethod = "close", name = Datasources.MASTER_DB)    @ConfigurationProperties(prefix = "spring.datasource")    public DataSource dataSource() {        return DataSourceBuilder.create().type(DruidDataSource.class).build();    }    @Bean(destroyMethod = "close", name = Datasources.SLAVE_DB)    @ConfigurationProperties(prefix = "spring.datasource-slave")    public DataSource dataSourceSlave() {        return DataSourceBuilder.create().type(DruidDataSource.class).build();    }}

配置成动态数据源

@Configuration@MapperScan(basePackages = {"com.example.dao"})public class MybatisConfig {    @Autowired    @Qualifier(Datasources.MASTER_DB)    private DataSource masterDB;    @Autowired    @Qualifier(Datasources.SLAVE_DB)    private DataSource slaveDB;    /**     * 动态数据源     */    @Bean(name = "dynamicDataSource")    public DataSource dynamicDataSource() {        DynamicDataSource dynamicDataSource = new DynamicDataSource();        // 默认数据源        dynamicDataSource.setDefaultTargetDataSource(masterDB);        // 配置多数据源        Map dsMap = new HashMap<>();        dsMap.put(Datasources.MASTER_DB, masterDB);        dsMap.put(Datasources.SLAVE_DB, slaveDB);        dynamicDataSource.setTargetDataSources(dsMap);        return dynamicDataSource;    }    @Bean    @ConfigurationProperties(prefix = "mybatis")    public SqlSessionFactoryBean sqlSessionFactoryBean() {        SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();        // 配置数据源，此处配置为关键配置，如果没有将 dynamicDataSource 作为数据源则不能实现切换        sqlSessionFactoryBean.setDataSource(dynamicDataSource());        return sqlSessionFactoryBean;    }}

使用ThreadLocal安全的管理当前进程使用的数据源连接

public class DataSourceContextHolder {    /**     * 默认数据源     */    public static final String DEFAULT_DATASOURCE = Datasources.MASTER_DB;    private static final ThreadLocal contextHolder = new ThreadLocal<>();    // 设置数据源名    public static void setDB(String dbType) {        System.out.println("切换到{}数据源:" +  dbType);        contextHolder.set(dbType);    }    // 获取数据源名    public static String getDB() {        return (contextHolder.get());    }    // 清除数据源名    public static void clearDB() {        contextHolder.remove();    }}

通过编写切面，对所有我们自定义切库注解的方法进行拦截，动态的选择数据源

@Aspect@Componentpublic class DynamicDataSourceAspect {    @Before("@annotation(com.example.util.RoutingDataSource)")    public void beforeSwitchDS(JoinPoint joinPoint) {        MethodSignature methodSignature = (MethodSignature) joinPoint.getSignature();        Method method = methodSignature.getMethod();        String dataSource = DataSourceContextHolder.DEFAULT_DATASOURCE;        if (method.isAnnotationPresent(RoutingDataSource.class)) {            RoutingDataSource routingDataSource = method.getDeclaredAnnotation(RoutingDataSource.class);            dataSource = routingDataSource.value();        }        DataSourceContextHolder.setDB(dataSource);    }    @After("@annotation(com.example.util.RoutingDataSource)")    public void afterSwitchDS(JoinPoint point) {        DataSourceContextHolder.clearDB();    }}

动态的取出我们在切面里设置的数据源的字符串

public class DynamicDataSource extends AbstractRoutingDataSource{    @Override    protected Object determineCurrentLookupKey() {        System.out.println("数据源为{}:" + DataSourceContextHolder.getDB());        return DataSourceContextHolder.getDB();    }}

取消自动配置数据源，使用我们这里定义的数据源配置

@SpringBootApplication(exclude = {      DataSourceAutoConfiguration.class})public class CutDataBaseApplication {   public static void main(String[] args) {      SpringApplication.run(CutDataBaseApplication.class, args);   }}

使用

/** * @author aYong * @version 1.0 * @date 2019/7/24 */@RestController@RequestMapping("/route")public class SysUserController {    @Autowired    private SysUserService sysUserService;    @GetMapping("/test1")    public SysUser test1(long id) {        return sysUserService.test1(id);    }    @GetMapping("/test2")    public Integer test2(long id, String name) {        return sysUserService.test2(id, name);    }}

"AbstractRoutingDataSource AOP如何实现动态数据源切换"的内容就介绍到这里了，感谢大家的阅读。如果想了解更多行业相关的知识可以关注网站，小编将为大家输出更多高质量的实用文章！