Spring Boot 动态重置数据源

使用 Spring Boot 动态重置数据源

java
public class DynamicDataSource extends AbstractRoutingDataSource {
    @Override
    protected Object determineCurrentLookupKey() {
        return DataSourceContextHolder.getDataSourceKey();
    }
}

yaml
spring:
  datasource:
    url: jdbc:mysql://localhost:3306/db1
    username: root
    password: root
    driver-class-name: com.mysql.cj.jdbc.Driver
  jpa:
    hibernate:
      ddl-auto: update
    show-sql: true
myapp:
  datasource:
    url: jdbc:mysql://localhost:3306/db2
    username: root
    password: root
    driver-class-name: com.mysql.cj.jdbc.Driver

java
public class DataSourceContextHolder {
    private static final ThreadLocal contextHolder = new ThreadLocal<>();
    public static void setDataSourceKey(String dataSourceKey) {
        contextHolder.set(dataSourceKey);
    }
    public static String getDataSourceKey() {
        return contextHolder.get();
    }
    public static void clearDataSourceKey() {
        contextHolder.remove();
    }
}

java
@RestController
@RequestMapping("/users")
public class UserController {
    @Autowired
    private UserService userService;
    @GetMapping
    public List getUsers() {
        DataSourceContextHolder.setDataSourceKey("myapp.datasource");
        List users = userService.getUsers();
        DataSourceContextHolder.clearDataSourceKey();
        return users;
    }
}