springboot中怎么實現(xiàn)多數(shù)據(jù)源的動態(tài)切換

這篇文章給大家介紹springboot中怎么實現(xiàn)多數(shù)據(jù)源的動態(tài)切換，內(nèi)容非常詳細，感興趣的小伙伴們可以參考借鑒，希望對大家能有所幫助。

1、maven依賴

<!--數(shù)據(jù)庫連接--><dependency>      <groupId>com.oracle</groupId>      <artifactId>ojdbc6</artifactId>      <version>11.2.0.4</version>      <scope>runtime</scope>    </dependency><!--數(shù)據(jù)庫連接池-><dependency>      <groupId>com.alibaba</groupId>      <artifactId>druid-spring-boot-starter</artifactId>      <version>1.1.10</version>    </dependency>    <dependency>      <groupId>org.springframework.boot</groupId>      <artifactId>spring-boot-starter-jdbc</artifactId>    </dependency><!--aop-><dependency>      <groupId>org.springframework.boot</groupId>      <artifactId>spring-boot-starter-aop</artifactId>    </dependency>

2、多數(shù)據(jù)源信息配置

#多數(shù)據(jù)源測試spring: datasource:  druid:   master:    driver-class-name: oracle.jdbc.driver.OracleDriver    username: test    password: test    url: jdbc:oracle:thin:@//ip1:1521/orcl   slave:    driver-class-name: oracle.jdbc.driver.OracleDriver    username: test    password: test    url: jdbc:oracle:thin:@//ip2:1521/orcl

3、數(shù)據(jù)源配置信息轉(zhuǎn)換成實體類

@ConfigurationProperties(prefix = "spring.datasource.druid")@Data@Componentpublic class DataSourceProperties {  private Map<String,String>master;  private Map<String,String>slave;}

4、動態(tài)數(shù)據(jù)源切換類

public class DynamicDataSource extends AbstractRoutingDataSource {  private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();  public DynamicDataSource(DataSource defaultTargetDataSource, Map<Object, Object> targetDataSources) {    super.setDefaultTargetDataSource(defaultTargetDataSource);    super.setTargetDataSources(targetDataSources);    super.afterPropertiesSet();  }  @Override  protected Object determineCurrentLookupKey() {    return getDataSource();  }  public static void setDataSource(String dataSource) {    contextHolder.set(dataSource);  }  public static String getDataSource() {    return contextHolder.get();  }  public static void clearDataSource() {    contextHolder.remove();  }}

5、多數(shù)據(jù)源配置類

@Configurationpublic class DynamicDataSourceConfig {  @Bean  public DataSource master(@Autowired DataSourceProperties dataSourceProperties){    DruidDataSource druidDataSource = new DruidDataSource();    Map<String, String> master = dataSourceProperties.getMaster();    druidDataSource.setUsername(master.get("username"));    druidDataSource.setPassword(master.get("password"));    druidDataSource.setUrl(master.get("url"));    //其他參數(shù)配置 省略    return druidDataSource;  }  @Bean  public DataSource slave(@Autowired DataSourceProperties dataSourceProperties){    DruidDataSource druidDataSource = new DruidDataSource();    Map<String, String> slave = dataSourceProperties.getSlave();    druidDataSource.setUsername(slave.get("username"));    druidDataSource.setPassword(slave.get("password"));    druidDataSource.setUrl(slave.get("url"));    //其他參數(shù)配置 省略    return druidDataSource;  }  @Bean  @Primary  public DynamicDataSource dataSource(DataSource master,DataSource slave){    Map<Object,Object>map = new HashMap<>(4);    map.put("master",master);    map.put("slave",slave);    return new DynamicDataSource(master,map);  }}

6、自定義@DataSource注解

@Target(ElementType.METHOD)@Retention(RetentionPolicy.RUNTIME)public @interface DataSource {  String name() default "master";}

7、Aop切面類配置

@Component@Aspectpublic class DataSourceAspect {  @Pointcut("@annotation(com.zxgeo.sso.muiltDatasource.anons.DataSource)")  public void dataSourcePointCut(){}  @Around(value = "dataSourcePointCut()")  public Object around(ProceedingJoinPoint point) throws Throwable {    MethodSignature signature = (MethodSignature) point.getSignature();    Method method = signature.getMethod();    DataSource dataSource = method.getAnnotation(DataSource.class);    if(dataSource == null){      DynamicDataSource.setDataSource("master");    }else {      DynamicDataSource.setDataSource(dataSource.name());    }    try {      return point.proceed();    } finally {      DynamicDataSource.clearDataSource();    }  }}

8、啟動配置注解信息，重要（不然運行會報錯）

@SpringBootApplication(exclude= {DataSourceAutoConfiguration.class})

9、測試

（1）、service層（此處沒有使用mybatis）

@Servicepublic class TestService {  @Autowired  private javax.sql.DataSource dataSource;  @DataSource  public Map<String,Object> getMasterDataSource() throws SQLException {    Connection connection = dataSource.getConnection();    Map<String,Object> map;    try (PreparedStatement preparedStatement           = connection.prepareStatement("SELECT * FROM AA WHERE A=10001")) {      ResultSet resultSet = preparedStatement.executeQuery();      map = new HashMap<>();      while (resultSet.next()){        map.put("A",resultSet.getString("A"));        map.put("B",resultSet.getString("B"));        map.put("C",resultSet.getString("C"));      }    }    return map;  }  @DataSource(name = "slave")  public Map<String,Object> getSlaveDataSource() throws SQLException {    Connection connection = dataSource.getConnection();    Map<String,Object> map;    try (PreparedStatement preparedStatement           = connection.prepareStatement("SELECT * FROM AA WHERE A=10002")) {      ResultSet resultSet = preparedStatement.executeQuery();      map = new HashMap<>();      while (resultSet.next()){        map.put("A",resultSet.getString("A"));        map.put("B",resultSet.getString("B"));        map.put("C",resultSet.getString("C"));      }    }    return map;  }}

（2）、單元測試

@SpringBootTest@RunWith(SpringRunner.class)class SsoApplicationTests {  @Autowired  private TestService testService;  @Test  public void muliDatasorce() throws SQLException {    Map<String, Object> masterDataSourceUrl = testService.getMasterDataSource();    System.out.println(masterDataSourceUrl);    Map<String, Object> slaveDataSourceUrl = testService.getSlaveDataSource();    System.out.println(slaveDataSourceUrl);  }}

關(guān)于springboot中怎么實現(xiàn)多數(shù)據(jù)源的動態(tài)切換就分享到這里了，希望以上內(nèi)容可以對大家有一定的幫助，可以學(xué)到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到。