跳至主要內容

26. Spring Data JPA 多数据源-分包模式

安图新大约 4 分钟

26. Spring Data JPA 多数据源-分包模式

简介

多数据源:一个项目中注入多个自定义datasource。参见:自定义数据源配置open in new window。根据不同的业务场景多数据源的模式也不一样,比如一个项目中不同的业务表分布在不同的库,本节的例子:
 
按照如上的业务场景,如果存在多个数据源,数据表分布在不同的库,那么JPAEntity 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

EntityRespository跟正常单数据源配置一样。

@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设置同上。

再次运行程序正常:
 


总结

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

上次编辑于:
贡献者: Andy