Spring Boot应用中如何动态指定数据库，实现不同用户不同数据库的场景

￡神魔★判官ぃ 2024-05-24 12:59 6阅读 0赞

当在 Spring Boot 应用程序中使用[Spring Data JPA][] 进行数据库操作时，配置Schema名称是一种常见的做法。然而，在某些情况下，模式名称需要是动态的，可能会在应用程序运行时发生变化。比如：需要做数据隔离的SaaS应用。

所以，这篇博文将帮助您解决了在 Spring Boot 应用程序中如何设置动态 Schema。

### 问题场景 ###

假设，您的应用程序是一个SaaS软件，需要为多个租户提供服务，每个租户都需要一个单独的数据库架构。

在这种情况下，在应用程序属性中对Shema名称进行硬编码是不太可能的，这样有一个用户新增，就要去写代码更新。

所以，为了应对这一挑战，我们将探索一种允许在运行时动态配置模式名称的解决方案。

我们创建了一个高质量的技术交流群，与优秀的人在一起，自己也会优秀起来，赶紧[点击加群][Link 1]，享受一起成长的快乐。

### 代码案例 ###

让我们创建一个 Spring Boot 项目 首先设置一个具有必要依赖项的新 Spring Boot 项目。在项目配置中包括 Spring Web、Spring Data JPA 和关于数据库的依赖项。

定义Spring Data JPA的实体类，例如：

import javax.persistence.Entity;
    import javax.persistence.Id;
    import javax.persistence.Table;
    
    @Entity
    @Table(name = "product")
    public class Product {
        @Id
        private Long id;
        private String name;
        private double price;
    
    }

创建数据访问接口，以便您的实体提供 CRUD 操作：

import org.springframework.data.jpa.repository.JpaRepository;
    import org.springframework.stereotype.Repository;
    
    @Repository
    public interface ProductRepository extends JpaRepository<Product, Long> {
    
    }

创建一个用来处理业务逻辑，包括与数据库交互的方法：

import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.stereotype.Service;
    
    import java.util.List;
    
    @Service
    public class ProductService {
        private final ProductRepository productRepository;
    
        @Autowired
        public ProductService(ProductRepository productRepository) {
            this.productRepository = productRepository;
        }
    
        public List<Product> getAllProducts() {
            return productRepository.findAll();
        }
    
    }

实现API接口：

import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.web.bind.annotation.GetMapping;
    import org.springframework.web.bind.annotation.RequestMapping;
    import org.springframework.web.bind.annotation.RestController;
    
    import java.util.List;
    
    @RestController
    @RequestMapping("/api/products")
    public class ProductController {
        private final ProductService productService;
    
        @Autowired
        public ProductController(ProductService productService) {
            this.productService = productService;
        }
    
        @GetMapping
        public List<Product> getAllProducts() {
            return productService.getAllProducts();
        }
    
    }

重点：配置动态Schema

import org.springframework.beans.factory.annotation.Value;
    import org.springframework.boot.jdbc.DataSourceBuilder;
    import org.springframework.context.annotation.Bean;
    import org.springframework.context.annotation.Configuration;
    
    import javax.sql.DataSource;
    
    @Configuration
    public class DynamicSchemaConfig {
    
        @Value("${custom.schema.name}")
        private String customSchemaName;
    
        @Bean
        public DataSource dataSource() {
            String dataSourceUrl = "jdbc:mysql://localhost:3306/" + customSchemaName;
            return DataSourceBuilder.create().url(dataSourceUrl).build();
        }
    }

重新打包该Spring Boot应用，然后当我们要为不同用户使用完全隔离的数据库、完全隔离的应用的时候，只需要通过下面的启动命令，就能轻松实现了：

java -jar -Dcustom.schema.name=my_dynamic_schema your-application.jar

这里，通过启动命令中的`custom.schema.name`参数，就能去指定不同的数据库Schema，而应用程序端都是同一套代码，由于启动了新的Spring Boot应用，所以应用端进程也是完全隔离的。这种方法，对于使用Spring Boot构建需要一定资源隔离SaaS软件来说，是个不错的实现方案。

> 欢迎关注我的公众号：程序猿DD。前沿技术早知道，弯道超车有希望！积累超车资本，从关注DD开始！

[Spring Data JPA]: https://www.didispace.com/spring-boot-2/4-4-spring-data-jpa.html
[Link 1]: https://www.didispace.com/jiaqun.html