Spring Boot 使用动态数据源实现多数据库连接

在实际开发中，我们常常需要在一个应用程序中连接多个数据库，比如MySQL和SQL Server。Spring Boot 提供了强大的数据源管理功能，而通过使用dynamic-datasource-spring-boot-starter，我们可以更方便地实现动态数据源切换。本文将详细介绍如何在Spring Boot中配置和使用多数据源。

环境准备

引入依赖

首先，在pom.xml中添加必要的依赖：

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-data-jdbc</artifactId>
</dependency>
<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>dynamic-datasource-spring-boot-starter</artifactId>
    <version>3.5.0</version>
</dependency>
<dependency>
    <groupId>mysql</groupId>
    <artifactId>mysql-connector-java</artifactId>
    <version>8.0.26</version>
</dependency>
<dependency>
    <groupId>com.microsoft.sqlserver</groupId>
    <artifactId>mssql-jdbc</artifactId>
    <version>9.2.1.jre8</version>
</dependency>

配置文件

在application.yml中配置多数据源信息：

spring:
  datasource:
    dynamic:
      primary: mysql # 设置默认的数据源或者数据源组，默认值即为mysql
      strict: false # 严格匹配数据源，默认为false。true时未匹配到指定数据源时抛异常，false时使用默认数据源
      datasource:
        mysql:
          driver-class-name: com.mysql.cj.jdbc.Driver
          url: jdbc:mysql://192.168.1.xx:3306/aaa?useSSL=false&serverTimezone=UTC&characterEncoding=utf8&useTimezone=true&serverTimezone=GMT%2b8&allowPublicKeyRetrieval=true&nullCatalogMeansCurrent=true
          username: xxx
          password: xxxx
        sqlserver:
          driver-class-name: com.microsoft.sqlserver.jdbc.SQLServerDriver
          url: jdbc:sqlserver://222.76.218.xx:1xx;databaseName=xxx;trustServerCertificate=true;
          username: xxx
          password: xxxx

配置数据源切换

通过@DS注解来实现数据源切换：

import com.baomidou.dynamic.datasource.annotation.DS;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

@Service
public class UserService {

    @Autowired
    private UserMapper userMapper;

    @DS("mysql")
    public List<User> getUsersFromMySQL() {
        return userMapper.getAllUsers();
    }

    @DS("sqlserver")
    public List<User> getUsersFromSQLServer() {
        return userMapper.getAllUsers();
    }
}

MyBatis Mapper 配置

配置你的MyBatis Mapper接口：

import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Select;
import java.util.List;

@Mapper
public interface UserMapper {

    @Select("SELECT * FROM users")
    List<User> getAllUsers();
}

测试多数据源切换

最后，在控制器中测试多数据源切换：

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;
import java.util.List;

@RestController
public class UserController {

    @Autowired
    private UserService userService;

    @GetMapping("/users/mysql")
    public List<User> getUsersFromMySQL() {
        return userService.getUsersFromMySQL();
    }

    @GetMapping("/users/sqlserver")
    public List<User> getUsersFromSQLServer() {
        return userService.getUsersFromSQLServer();
    }
}

总结

通过本文的步骤，我们实现了在Spring Boot中使用dynamic-datasource-spring-boot-starter配置多数据源，并实现了动态数据源的切换。这样，我们可以在一个应用程序中灵活地操作多个数据库，提高了开发效率和系统的扩展性。

希望这篇文章对你有所帮助，如果你有任何疑问或建议，欢迎在评论区留言讨论。