27. Spring Data JDBC&JPA 多数据源(AbstractRoutingDataSource)

前言

Spring Boot 3.x-Spring Data JPA 多数据源-分包模式此文中介绍的是多数据源，每个数据源的表不同，接下来需要解决的是多数据源，每个数据源的表结构一致，这种情况最常见的是数据库主从，查询读从库，新增删除修改走主库。

上述方案，需要一个动态切换的datasource根据不同的条件，切换不同的数据源。Spring 从 2.0 版本开始就提供方案AbstractRoutingDataSource。

AbstractRoutingDataSource：抽象DataSource实现，基于查找键(determineCurrentLookupKey())将getConnection()调用路由到目标数据源之一。

 

数据库准备：
主库：mulit-ds1
从库：mulit-ds2

DROP TABLE IF EXISTS user;
CREATE TABLE user (
  id bigint(20) NOT NULL AUTO_INCREMENT,
  age int(11) DEFAULT NULL,
  name varchar(255) DEFAULT NULL,
  sex varchar(255) DEFAULT NULL,
  PRIMARY KEY (id)
) ENGINE=InnoDB AUTO_INCREMENT=2 DEFAULT CHARSET=utf8;

---

一、数据源定义

1、 数据源类型定义：；

first->主库(mulit-ds1)
second->从库(mulit-ds2)

public class DataSourceType {


    public final static String FIRST = "first";
    public final static String SECOND = "second";

}

2、 自定义动态数据源继承AbstractRoutingDataSource；

public class DynamicDataSource extends AbstractRoutingDataSource {


    /**
     * ThreadLocal 用于提供线程局部变量，在多线程环境可以保证各个线程里的变量独立于其它线程里的变量。
     */
    private static final ThreadLocal<String> CONTEXT_HOLDER = new ThreadLocal<>();

    public DynamicDataSource(DataSource defaultTargetDataSource, Map<Object, Object> targetDataSources) {


        //默认目标数据源
        super.setDefaultTargetDataSource(defaultTargetDataSource);
        //目标数据源集合。数据源切换时从此列表选择
        super.setTargetDataSources(targetDataSources);
        //属性设置
        super.afterPropertiesSet();
    }

    @Override
    protected Object determineCurrentLookupKey() {


        //关键：更具数据源key。获取选择的数据源。
        return getDataSource();
    }

    public static void setDataSource(String dataSource) {


        CONTEXT_HOLDER.set(dataSource);
    }

    public static String getDataSource() {


        return CONTEXT_HOLDER.get();
    }

    public static void clearDataSource() {


        CONTEXT_HOLDER.remove();
    }

}

3、 数据源注入；

@Configuration
public class DynamicDataSourceConfig {



    //创建第一个主库数据源
    @Bean(name = "firstDataSource")
    @ConfigurationProperties(prefix = "app.datasource.first")
    public DataSource userDataSource() {


        return DataSourceBuilder.create().build();
    }

    //创建第二个从库数据源
    @Bean(name = "secondDataSource")
    @ConfigurationProperties(prefix = "app.datasource.second")
    public DataSource secondDataSource() {


        return DataSourceBuilder.create().build();
    }
    //动态数据源，实际使用的数据源时数据源路由根据key 选择的。默认数据源为第一个数据源(主库)
    @Bean(name = "dynamicDataSource")
    //主数据源，使用的是此数据源
    @Primary
    public DynamicDataSource dataSource(DataSource firstDataSource, DataSource secondDataSource) {


        Map<Object, Object> targetDataSources = new HashMap<>(2);
        targetDataSources.put(DataSourceType.FIRST, firstDataSource);
        targetDataSources.put(DataSourceType.SECOND, secondDataSource);
        //默认返回的也是一个datasource
        return new DynamicDataSource(firstDataSource, targetDataSources);
    }
}

4、 数据连接属性配置；

app:
  datasource:
    first:
      #数据库驱动完整类名
      driver-class-name: com.mysql.jdbc.Driver
      #数据库连接url
      jdbc-url: jdbc:mysql://127.0.0.1:3306/multi-ds1
      #数据库用户名
      username: root
      #数据库密码
      password: 123456
    second:
      #数据库驱动完整类名
      driver-class-name: com.mysql.jdbc.Driver
      #数据库连接url
      jdbc-url: jdbc:mysql://127.0.0.1:3306/multi-ds2
      #数据库用户名
      username: root
      #数据库密码
      password: 123456

二、Spring Data JPA 使用

引入依赖:

   <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-data-jpa</artifactId>
        </dependency>

1.创建 entity

@Data
@ToString
@Entity
public class User {


    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;
    private String firstName;
    private String secondName;
}

2.创建 repository

@Transactional(readOnly = true)
public interface UserRepository extends JpaRepository<User, Long> {



    public List<User> findByFirstName(String firstName);
}

3.测试

public class MultiDsRouteApplication {


    @Autowired
    private UserRepository userRepository;
    @Autowired
    private DynamicDataSource dynamicDataSource;
    @Autowired
    private IUserService userService;

    public static void main(String[] args) {


        SpringApplication.run(MultiDsRouteApplication.class, args);
    }

    @Bean
    CommandLineRunner commandLineRunner() {


        return new CommandLineRunner() {


            @Override
            public void run(String... args) throws Exception {


                //查询使用从库
                DynamicDataSource.setDataSource(DataSourceType.SECOND);
                userRepository.findByFirstName("张三");
                log.info("当前使用的数据源是:"+dynamicDataSource.getConnection().getCatalog());
                //新增使用主库
                DynamicDataSource.setDataSource(DataSourceType.FIRST);
                log.info("当前使用的数据源是:"+dynamicDataSource.getConnection().getCatalog());
                User user = new User();
                user.setFirstName("hello");
                user.setSecondName("word");
                userRepository.save(user);
            }
        };
    }
}

控制台输出：

4.AOP 自动选择数据源

按照上述的方法，正式业务开发过程中，肯定是有某个方式触发数据源，不应该每次手动设置：

DynamicDataSource.setDataSource(DataSourceType.SECOND);

因为我们模拟的是数据库主从结构，查询走从库，写走主库因此我们从Repository方法命名规则建立切面，自动选择数据源。

@Slf4j
@Aspect
@Component
public class DataSourceAspect {


    //此处为了打印数据源引入。
    @Autowired
    private DynamicDataSource dynamicDataSource;

    @Pointcut("execution(* org.springframework.data.repository.CrudRepository.*(..))||execution(* com" +
            ".example.multidsroute.repository.*.*(" +
            "..))")
    private void aspect() {


    }

    @Around("aspect()")
    public Object around(ProceedingJoinPoint joinPoint) throws Throwable {


        String method = joinPoint.getSignature().getName();

        if (method.startsWith("find") || method.startsWith("select") || method.startsWith("query") || method.startsWith("search")) {


            DynamicDataSource.setDataSource("second");
        } else {


            DynamicDataSource.setDataSource("first");
            log.info("switch to first datasource...");
        }
        log.info("aop当前使用的数据源是:" + dynamicDataSource.getConnection().getCatalog());
        try {


            return joinPoint.proceed();
        } finally {


            log.info("清除 datasource router...");
            DynamicDataSource.clearDataSource();
        }
    }
}

@Pointcut(“execution(* org.springframework.data.repository.CrudRepository.(…))||execution( com” +
“.example.multidsroute.repository..(” +
“…))”) 切面一定要把 CrudRepository 方法加入，这样才能处理所有的Repository方法。

测试：

@Bean
    CommandLineRunner commandLineRunner() {


        return new CommandLineRunner() {


            @Override
            public void run(String... args) throws Exception {


                //查询使用从库
                DynamicDataSource.setDataSource(DataSourceType.SECOND);
                userRepository.findByFirstName("张三");
                //新增使用主库
                DynamicDataSource.setDataSource(DataSourceType.FIRST);
                User user = new User();
                user.setFirstName("hello");
                user.setSecondName("word");
                userRepository.save(user);

            }
        };
    }

 

二、Spring Data JDBC 使用

Spring Data JDBC 只需要把依赖换成如下即可：

       <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-data-jdbc</artifactId>
        </dependency>

---