Spring Boot 3.x 实战(1)

26、Spring Boot 3.x -Spring Data JPA多数据源-分包模式

多数据源：一个项目中注入多个自定义datasource。参见：自定义数据源配置。根据不同的业务场景多数据源的模式也不一样，比如一个项目中不同的业务表分布在不同的库，本节的例子：

按照如上的业务场景，如果存在多个数据源，数据表分布在不同的库，那么JPA中Entity Repository，需要在不同的包，方便扫描。

一、包结构

二、创建多个DataSource

@Configuration
public class DataSourceConfig {
   
     
    @Primary
    @Bean(name = "userDataSource")
    @ConfigurationProperties(prefix = "app.datasource.user")
    public DataSource userDataSource() {
   
     
        return DataSourceBuilder.create().build();
    }

    @Bean(name = "blogDataSource")
    @ConfigurationProperties(prefix = "app.datasource.blog")
    public DataSource secondDataSource() {
   
     
        return DataSourceBuilder.create().build();
    }
}

多datasource配置需要注意以下几点：
1、需要将其中一个数据源设置为@Primary，默认主数据源；
2、不同数据源注入不同的配置；

app:
  datasource:
    user:
     数据库驱动完整类名
      driver-class-name: com.mysql.jdbc.Driver
     数据库连接url
      jdbc-url: jdbc:mysql://127.0.0.1:3306/mulit-user
     数据库用户名
      username: root
     数据库密码
      password: 123456
    blog:
     数据库驱动完整类名
      driver-class-name: com.mysql.jdbc.Driver
     数据库连接url
      jdbc-url: jdbc:mysql://127.0.0.1:3306/mulit-blog
     数据库用户名
      username: root
     数据库密码
      password: 123456
debug: true
spring:
  jpa:
    hibernate:
      ddl-auto: update

三、创建EntityManagerFactory，TransactionManager

由于是注入了2个数据源，那么对应的数据源的EntityManagerFactory，和TransactionManager也许要自定义分开。

user相关

@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(basePackages = {
   
     "com.example.multidssplitpackage.user.repository"},
        entityManagerFactoryRef =
        "userEntityManagerFactory", transactionManagerRef = "userTransactionManager")
public class UserDataSourceConfig {
   
     

    @Autowired
    @Qualifier("userDataSource")
    private DataSource userDataSource;
    @Primary
    @Bean(name = "userEntityManager")
    public EntityManager userEntityManager(EntityManagerFactoryBuilder builder) {
   
     
        return userEntityManagerFactory(builder).getObject().createEntityManager();
    }

    @Primary
    @Bean(name = "userEntityManagerFactory")
    public LocalContainerEntityManagerFactoryBean userEntityManagerFactory(EntityManagerFactoryBuilder builder) {
   
     
        Map<String, String> map = new HashMap<>();
        map.put("hibernate.hbm2ddl.auto", "update");
        map.put("hibernate.show_sql", "true");
        return builder.dataSource(userDataSource).packages("com.example.multidssplitpackage.user").properties(map).persistenceUnit("userUnit").build();
    }

    @Primary
    @Bean(name = "userTransactionManager")
    public PlatformTransactionManager userTransactionManager(EntityManagerFactoryBuilder builder) {
   
     
        return new JpaTransactionManager(userEntityManagerFactory(builder).getObject());
    }

}

builder.dataSource(userDataSource).packages("com.example.multidssplitpackage.user").properties(map).persistenceUnit("userUnit").build();

针对user entity的管理，自定义userEntityManagerFactory，包只扫描user相关。blog以此类推。事物的管理，因为存在多个数据源，那么事物管理也是独立分开。

blog相关

@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(basePackages = {
   
     "com.example.multidssplitpackage.blog"},
        entityManagerFactoryRef =
        "blogEntityManagerFactory", transactionManagerRef = "blogTransactionManager")
public class BlogDataSourceConfig {
   
     

    @Autowired
    @Qualifier("blogDataSource")
    private DataSource blogDataSource;
    @Bean(name = "blogEntityManager")
    public EntityManager blogEntityManager(EntityManagerFactoryBuilder builder) {
   
     
        return blogEntityManagerFactory(builder).getObject().createEntityManager();
    }

    @Bean(name = "blogEntityManagerFactory")
    public LocalContainerEntityManagerFactoryBean blogEntityManagerFactory(EntityManagerFactoryBuilder builder) {
   
     
        Map<String, String> map = new HashMap<>();
        map.put("hibernate.hbm2ddl.auto", "update");
        map.put("hibernate.show_sql", "true");
        return builder.dataSource(blogDataSource).packages("com.example.multidssplitpackage.blog").properties(map).persistenceUnit("userUnit").build();
    }

    @Bean(name = "blogTransactionManager")
    public PlatformTransactionManager blogTransactionManager(EntityManagerFactoryBuilder builder) {
   
     
        return new JpaTransactionManager(blogEntityManagerFactory(builder).getObject());
    }

}

接下来主要是对repository的管理配置：

@EnableTransactionManagement
@EnableJpaRepositories(basePackages = {
   
     "com.example.multidssplitpackage.blog"},
        entityManagerFactoryRef =
        "blogEntityManagerFactory", transactionManagerRef = "blogTransactionManager")

不同包的repositories扫描设置成自己的包。

四、创建Entity，Repository

Entity和Respository跟正常单数据源配置一样。

@Data
@Entity
@ToString
public class User {
   
     
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;
    private String name;
    private Integer age;
    private String sex;
}
@Data
@Entity
public class Blog {
   
     
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;
    private String title;
    private String content;
    private Long authorId;
}

Respoistory针对事物注解，需要注入对应自己数据源的事物管理器。

@Transactional(value = "userTransactionManager", readOnly = true)
public interface UserRespoistory extends JpaRepository<User, Long> {
   
     
}

@Transactional(transactionManager = "blogTransactionManager", readOnly = true)
public interface BlogRepository extends JpaRepository<Blog, Long> {
   
     
}

五、测试

@SpringBootTest
class MultiDsSplitPackageApplicationTests {
   
     
    @Autowired
    private UserRespoistory userRespoistory;
    @Autowired
    private BlogRepository blogRepository;

    @Test
    void testMultiDS() {
   
     
        User user = new User();
        user.setName("才可夫斯基"+Math.random());
        user.setAge(100);
        user.setSex("男");
        User result = userRespoistory.save(user);
        System.out.println(result);

        Blog blog = new Blog();
        blog.setTitle("jpa 多数据源");
        blog.setContent("分包模式实现");
        blog.setAuthorId(user.getId());
        blogRepository.save(blog);
    }

}

运行测试后，发现一个问题，JPA naming strategy失效了，表名直接变成了类名，并没有将第一个字母转换成小写，驼峰的authorId，并没有转换成下划线格式。

六、命名策略失效问题

针对上述问题，需要手动设置命名策略，hibernate.implicit_naming_strategy（隐式），hibernate.physical_naming_strategy（物理显示）

 @Bean(name = "blogEntityManagerFactory")
    public LocalContainerEntityManagerFactoryBean blogEntityManagerFactory(EntityManagerFactoryBuilder builder) {
   
     
        Map<String, String> map = new HashMap<>();
        map.put("hibernate.hbm2ddl.auto", "update");
        map.put("hibernate.show_sql", "true");
        map.put("hibernate.physical_naming_strategy", CamelCaseToUnderscoresNamingStrategy.class.getName());
        map.put("hibernate.implicit_naming_strategy", SpringImplicitNamingStrategy.class.getName());
        return builder.dataSource(blogDataSource).packages("com.example.multidssplitpackage.blog").properties(map).persistenceUnit("userUnit").build();
    }

user设置同上。

再次运行程序正常：

总结

本周介绍的多数据场景，正常业务情况下，使用的非常少，针对目前微服务流行，肯定是需要分服务，而不是分不同的数据库。并且上述多数据源场景，事物是完全隔开的，因此局限性非常的大。一般真多多数据源情况，多租户的场景应该使用的非常多。下一章节介绍。