Spring Boot多数据源配置

在生产环境中,可能存在同一个项目访问多个数据源的情况,本文通过Spring Boot实现多数据源数据操作。为开发示例简单,采用JPA进行数据库操作;配置文件采用yml进行配置,数据源为两个不同名的MySQL数据库。同时为示例简单,配置信息我们只添加必要的配置,其他配置将省略。

一、单数据库操作示例

通过Spring Boot实现单数据源操作比较简单,只需要按照Spring Boot的约定进行配置即可,具体配置如下:

spring:
  datasource:
    hikari:
      driver-class-name: com.mysql.cj.jdbc.Driver
      username: root
      password: root
    url: jdbc:mysql://127.0.0.1:3306/test1?serverTimezone=UTC

这里只配置数据库连接的必备信息,通过默认的配置方式,Spring Boot在启动时会直接加载这些必要信息。

其他一些辅助测试的代码如下:

  • User
/**
 * 用户信息
 *
 * @author zhang peng
 * @since 2019/3/6 10:12
 */
@Entity
@Table(name = "user")
public class User {

    @Id
    @GeneratedValue
    private Integer id;

    private String name;

    private int age;

    //省略getter setter
}

  • Repository
/**
 * User Repository
 *
 * @author zhang peng
 * @since 2019/3/6 10:13
 */
public interface UserRepository extends JpaRepository<User, Integer> {
}
  • Service 、Service Impl
/**
 * User Service Interface
 *
 * @author zhang peng
 * @since 2019/3/6 10:15
 */
public interface IUserService {

    User save(User user);

    User save(String name, Integer age);
}
/**
 * User Service Interface Impl
 *
 * @author zhang peng
 * @since 2019/3/6 10:16
 */
@Service
public class UserServiceImpl implements IUserService {

    @Autowired
    private UserRepository userRepository;

    @Override
    public User save(User user) {
        return userRepository.save(user);
    }

    @Override
    @Transactional
    public User save(String name, Integer age) {
        User user = new User();
        user.setName(name);
        user.setAge(age);
        return save(user);
    }
}

  • Controller
/**
 * User Controller
 *
 * @author zhang peng
 * @since 2019/3/6 10:18
 */
@RestController
@RequestMapping("/user")
public class UserController {

    @Autowired
    private IUserService userService;

    @RequestMapping(value = "/save")
    public Object saveUser(String name, Integer age) {
        User user = userService.save(name, age);
        return user.getId();
    }
}

以上就是整个从控制层到数据库存储的所有代码,在仅仅考虑简单数据库操作的情况下,利用Spring Boot实现起来非常简单,这也是它作为一个快速开发框架的优势所在。

二、多数据源配置

1、自定义配置信息读取

Spring Boot以契约模式简化了项目的配置。如果我们需要配置应用参数,可以用过指定的配置文件application.yml或者application.properties进行配置。而在这里配置的信息我们可以使用spring提供的默认key进行配置,这样spring在启动的时候会帮我们自动加载这些配置信息。同时,除了这些约定好的配置key,我们也可以自定义自己的key进行配置,并通过配置类进行配置信息的加载。

在多数据源配置类中,我们需要配置DataSourceEntityManagerFactoryPlatformTransactionManager等信息。在单个数据源中,虽然没有明确编码配置,实际上是Spring Boot已经做了默认处理,而在多数据源中,需要根据不同的数据源进行配置。本文数据源均采用MySQL,只是数据库名称不一样,所以在两个数据源配置上基本一样,只是实例名称不同而已,而不同的数据库可能存在一些细节的处理,这些是要注意的。

在从单数据源转到多数据源之前,我们需要了解一些Spring Boot的注解相关特性,以方便我们更好的理解和运用这些特性(此处指说明注解用到的特性,其他特性参考官方文档)。

@Configuration

类注解,被该注解修饰的类Spring在加载的时候认为该类是一个配置类。

@Bean

方法注解,被该注解修饰的方法返回的bean对象被Spring接收管理。本示例会用到name属性,该属性用于定义bean别名。

ConfigurationProperties

方法注解,被该注解修饰的方法可以通过注解过去配置属性。本示例会用到prefix属性,该属性用于描述配置文件中自定义key的前缀信息。

@EnableJpaRepositories

类注解,该注解用于描述Jpa数据库操作信息。本示例会用到entityManagerFactoryReftransactionManagerRefbasePackages属性,属性详细说明见下文代码注释。

此外,如果用到事务,还需要关注一个注解:@Transactional ,在多数据源配置中,由于每个数据库都会配置自己的事务管理器,因为Spring管理这多个事务管理器,在操作数据库需要事务时,Spring并不知道需要使用哪个事务管理器进行事务管理,所以在使用事务时,我们需要通过该注解的transactionManager属性进行事务管理器的指定。由于本文主要是讲多数据源配置,并不关心事务,所以在此简单提示一下,避免按照此示例测试时出现事务配置问题。

2、示例代码:

  • application.yml
data:
  source:
    first:
      jdbc-url: jdbc:mysql://127.0.0.1:3306/test1?serverTimezone=UTC
      driver-class-name: com.mysql.cj.jdbc.Driver
      username: root
      password: root

    second:
      jdbc-url: jdbc:mysql://127.0.0.1:3306/test2?serverTimezone=UTC
      driver-class-name: com.mysql.cj.jdbc.Driver
      username: root
      password: root

配置文件配置了两个数据源,分别指向不同的MySQL数据库。其中需要注意,在自定义配置信息key的时候应该保持key的规律,方便后期在获取值的时候有规律可循。上述配置中,我们把data.source作为前缀。

配置中由于存在两个数据源(也可以更多),所以在定义配置类时,需要多个配置类和数据源信息做对应。本示例中会有两个配置类针对数据源信息,同时为了代码清晰,还会多出一个配置类专门处理数据源共有的配置信息。

  • 数据源一配置
/**
 * First DataSource Configuration
 *
 * @author zhang peng
 * @since 2019/3/6 13:36
 */
@Configuration
@EnableJpaRepositories(
        entityManagerFactoryRef = "firstEntityManagerFactory",  //EntityManagerFactory信息配置
        transactionManagerRef = "firstTransactionManager", //事务管理器
        basePackages = {"com.whucke.multi.datasource.repository.first"}) //扫描包定义。同样,为方便使用,在定义包名时也需要遵循一定规律
public class FirstConfig {

    @Autowired
    private DataSource firstDataSource;
    @Autowired
    JpaVendorAdapter jpaVendorAdapter;

    /**
     * firstDataSource
     *
     * @return
     */
    @Bean(name = "firstDataSource")
    @ConfigurationProperties(prefix = "data.source.first") //定义配置信息前缀,这也是前面为什么说key的定义需要遵循一定的规律
    public DataSource firstDataSource() {
        //通过该方法调用,Spring会自动装配配置信息
        DataSource dataSource = DataSourceBuilder.create().build();
        return dataSource;
    }

    /**
     * 配置 EntityManagerFactory
     *
     * @return
     */
    @Bean("firstEntityManagerFactory")
    public EntityManagerFactory entityManagerFactory() {
        LocalContainerEntityManagerFactoryBean factoryBean = new LocalContainerEntityManagerFactoryBean();
        factoryBean.setDataSource(firstDataSource());
        factoryBean.setJpaVendorAdapter(jpaVendorAdapter);
        factoryBean.setPackagesToScan("com.whucke.multi.datasource.domain");
        //持久层名称定义,在配置多数据源时必须配置该信息
        factoryBean.setPersistenceUnitName("first");
        factoryBean.afterPropertiesSet();
        return factoryBean.getObject();
    }

     /**
     * 事务管理器
     *
     * @return
     */
    @Bean("firstTransactionManager")
    public PlatformTransactionManager transactionManager() {
        return new JpaTransactionManager(entityManagerFactory());
    }

}

第二个数据源配置与第一个类似,唯一区别在于bean名称的定义和引用,在配置时需要注意,不要把bean名称配置混淆了。代码如下:

  • 数据源二配置
/**
 * Second DataSource Configuration
 *
 * @author zhang peng
 * @since 2019/3/6 13:36
 */
@Configuration
@EnableJpaRepositories(
        entityManagerFactoryRef = "secondEntityManagerFactory",
        transactionManagerRef = "secondTransactionManager",
        basePackages = {"com.whucke.multi.datasource.repository.second"})
public class SecondConfig {

    @Autowired
    private DataSource secondDataSource;
    @Autowired
    JpaVendorAdapter jpaVendorAdapter;

    /**
     * secondDataSource
     *
     * @return
     */
    @Bean(name = "secondDataSource")
    @ConfigurationProperties(prefix = "data.source.second")
    public DataSource secondDataSource() {
        DataSource dataSource = DataSourceBuilder.create().build();
        return dataSource;
    }

    /**
     * 配置 EntityManagerFactory
     *
     * @return
     */
    @Bean("secondEntityManagerFactory")
    public EntityManagerFactory entityManagerFactory() {
        LocalContainerEntityManagerFactoryBean factoryBean = new LocalContainerEntityManagerFactoryBean();
        factoryBean.setDataSource(secondDataSource());
        factoryBean.setJpaVendorAdapter(jpaVendorAdapter);
        factoryBean.setPackagesToScan("com.whucke.multi.datasource.domain");
        factoryBean.setPersistenceUnitName("second");
        factoryBean.afterPropertiesSet();
        return factoryBean.getObject();
    }

    @Bean("secondTransactionManager")
    public PlatformTransactionManager transactionManager() {
        return new JpaTransactionManager(entityManagerFactory());
    }

}

前面我们说到会有一个针对数据源的公共配置信息类存在,在本示例中公共配置信息并不多,可能在不同的项目场景下配置信息还会有增加或修改,这需要根据项目的实际情况做处理。

  • 数据源公共配置信息
/**
 * Data Source Configuration
 *
 * @author zhang peng
 * @since 2019/3/6 10:26
 */
@Configuration
public class DataSourceConfig {

    @Bean
    public JpaVendorAdapter jpaVendorAdapter() {
        HibernateJpaVendorAdapter jpaVendorAdapter = new HibernateJpaVendorAdapter();
        jpaVendorAdapter.setShowSql(true);
        jpaVendorAdapter.setGenerateDdl(true);
        jpaVendorAdapter.setDatabase(Database.MYSQL);
        return jpaVendorAdapter;
    }
}

此处对JpaVendorAdapter的配置是直接通过硬编码实现的。如果参数较多、不同数据源等情况,这段代码也可以根据不同的数据源进行配置加载。此处只为说明问题,所以不再引申更多内容。

三、事务

最后简单提说一下上文提到的事务管理配置信息。由于真实开发环境中多数据源事务很少直接通过上文配置的事务管理器去处理,所以这里只做简单的说明,需要实现真正的跨数据库事务,还需要用到其他技术才行。

  • 示例:
@Transactional(transactionManager = "firstTransactionManager")
public User save(String name, Integer age) {
    User user = new User();
    user.setName(name);
    user.setAge(age);
    return save(user);
}

从上述示例中可以发现,我们在添加事务注解@Transactional的时候会额外配置transactionManager属性。原因是Spring容器中存在多个事务管理器配置(前面配置文件中进行配置的),在处理事务过程中,框架不知道使用哪一个,所以需要开发者明确指定。
在配置中也可以看出,指定一个事务管理器,而该事务管理器只会在对应数据源起作用,所以跨数据库操作事务是不会同时生效的。

以上就是Spring Boot多数据源配置的基础参考方案,需要注意的是该示例是整合的JPA。如果要整合mybatis,需要做相应的修改。

代码地址:https://gitee.com/umeone/multi-datasource.git

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 213,099评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,828评论 3 387
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 158,540评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,848评论 1 285
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,971评论 6 385
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,132评论 1 291
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,193评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,934评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,376评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,687评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,846评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,537评论 4 335
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,175评论 3 317
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,887评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,134评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,674评论 2 362
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,741评论 2 351

推荐阅读更多精彩内容