SpringBoot系列:Spring Boot多数据源,Mybatis方式

一、Mybatis配置多数据源

mybatis可谓是国内最火的ORM框架了,mybatis的多数据源当然在实际使用中也必不可少。mybatis多数据源一般有两种方式,一种是基于不同的包使用不同的数据源,这个类似于JPA配置多数据源,另一种则是注解+aop实现动态多数据源。

本章节就来看一看使用不同的包怎么做到多数据源。

首先是pom文件,除了SpringBoot的依赖外,主要就是mybatis及数据库的依赖了。

<!--mysql连接-->
<dependency>
    <groupId>mysql</groupId>
    <artifactId>mysql-connector-java</artifactId>
    <scope>runtime</scope>
</dependency>

<!--druid连接池-->
<dependency>
    <groupId>com.alibaba</groupId>
    <artifactId>druid-spring-boot-starter</artifactId>
    <version>1.1.10</version>
</dependency>

<!--mybatis依赖-->
<dependency>
    <groupId>org.mybatis.spring.boot</groupId>
    <artifactId>mybatis-spring-boot-starter</artifactId>
    <version>1.3.2</version>
</dependency>

然后是配置文件,我们需要配置多数据源。这里不需要对mybatis进行配置,因为配置都会移到配置类里。

server:
  port: 10900

spring:
  datasource:
    master:
      # 新版驱动从com.mysql.jdbc.Driver变更为com.mysql.cj.jdbc.Driver
      driver-class-name: com.mysql.cj.jdbc.Driver
      # 数据源需要添加时间标准和指定编码格式解决乱码 You must configure either the server or JDBC driver (via the serverTimezone configuration property) to use a more specifc time zone value if you want to utilize time zone support.
      url: jdbc:mysql://localhost:3306/springboot?useUnicode=true&characterEncoding=UTF-8&serverTimezone=UTC
      username: root
      password: 1234
    slave:
        driver-class-name: com.mysql.cj.jdbc.Driver
        url: jdbc:mysql://localhost:3306/springboot2?useUnicode=true&characterEncoding=UTF-8&serverTimezone=UTC
        username: root
        password: 1234

接下来是配置类,配置类除了要指明不同的数据源外,就是还要配置mybatis的核心组件Configuration和SqlSessionFactory。

先创建DataSourceConfig配置类。根据配置文件spring.datasource对应下的数据源配置masterDataSource和slaveDataSource的数据源对象,其中@Primary表示默认的数据源。在配置Configuration对象时,需要注意,因为与配置注解@Configuration重名,需要指明全限定名,请认准org.apache.ibatis.session包下的,这里的配置一般对应着平时使用单数据源时application.yml中的mybatis配置,演示项目并没有涉及过多的配置,主要就是打印sql到控制台。

@Configuration
public class DataSourceConfig {

    @Bean
    @Primary // 表示默认的数据源
    @ConfigurationProperties(prefix = "spring.datasource.master")
    DataSource masterDataSource(){
        // DataSourceBuilder.create().build();
        return DruidDataSourceBuilder.create().build();
    }

    @Bean
    @ConfigurationProperties(prefix = "spring.datasource.slave")
    DataSource slaveDataSource(){
        return DruidDataSourceBuilder.create().build();
    }

    /**
     * mybatis的configuration设置
     * @return
     */
    @Bean
    org.apache.ibatis.session.Configuration masterConfiguration(){
        // 添加sql打印,即配置文件中mybatis.configuration.log-impl: org.apache.ibatis.logging.stdout.StdOutImpl
        org.apache.ibatis.session.Configuration configuration = new org.apache.ibatis.session.Configuration();
        configuration.setLogImpl(StdOutImpl .class);
        return configuration;
    }

    // 必须得使用不同的Configuration对象,否则数据源加载会出现覆盖
    @Bean
    org.apache.ibatis.session.Configuration slaveConfiguration(){
        org.apache.ibatis.session.Configuration configuration = new org.apache.ibatis.session.Configuration();
        configuration.setLogImpl(StdOutImpl .class);
        return configuration;
    }

}

然后再创建MasterSqlSessionConfig类和SlaveSqlSessionConfig类,这两个类除了部分个性配置外大致相同。这里注入了之前配置的DataSource和Configuration,@MapperScan里的basePackages指明了当前配置类对应的SqlSessionFactory实例的作用范围,这里masterSqlSessionFactory配置中又关联数据源masterDataSource,也即com.yanger.mybatis.dao的mapper使用的是master数据源。在配置SqlSessionFactory时,如果是基于xml方式,还需要标注xml的位置。同样的,为了事务的支持,我们还需要配置事务管理器DataSourceTransactionManager,@Primary指明默认。

@Configuration
@MapperScan(
        basePackages = {"com.yanger.mybatis.dao"}, //mapper生效的包
        sqlSessionFactoryRef = "masterSqlSessionFactory" //mybatis依赖的SqlSessionFactory
)
public class MasterSqlSessionConfig {

    @Autowired
    @Qualifier("masterDataSource")
    private DataSource masterDataSource;

    @Autowired
    org.apache.ibatis.session.Configuration masterConfiguration;

    @Bean(name = "masterSqlSessionFactory")
    @Primary
    public SqlSessionFactory masterSqlSessionFactory(@Qualifier("masterDataSource") DataSource masterDataSource) throws Exception {
        final SqlSessionFactoryBean sessionFactory = new SqlSessionFactoryBean();
        sessionFactory.setDataSource(masterDataSource);
        // mapper映射的xml文件的位置,因为该工程mybatis是基于注解,所以不用设置
        //sessionFactory.setMapperLocations(new PathMatchingResourcePatternResolver().getResources(""));
        sessionFactory.setConfiguration(masterConfiguration);
        return sessionFactory.getObject();
    }

    @Bean(name = "masterTransactionManager")
    @Primary
    public DataSourceTransactionManager masterTransactionManager() {
        return new DataSourceTransactionManager(masterDataSource);
    }

}

SlaveSqlSessionConfig与MasterSqlSessionConfig大同小异,除了使用不同的DataSource和Configuration,作用的包不同,其他类似。

@Configuration
@MapperScan(
        basePackages = {"com.yanger.mybatis.sdao"}, //mapper生效的包
        sqlSessionFactoryRef = "slaveSqlSessionFactory" //mybatis依赖的SqlSessionFactory
)
public class SlaveSqlSessionConfig {

    @Autowired
    @Qualifier("slaveDataSource")
    private DataSource slaveDataSource;

    @Autowired
    private org.apache.ibatis.session.Configuration slaveConfiguration;

    @Bean(name = "slaveSqlSessionFactory")
    public SqlSessionFactory slaveSqlSessionFactory(@Qualifier("slaveDataSource") DataSource slaveDataSource) throws Exception {
        final SqlSessionFactoryBean sessionFactory = new SqlSessionFactoryBean();
        sessionFactory.setDataSource(slaveDataSource);
        // mapper映射的xml文件的位置,因为该工程mybatis是基于注解,所以不用设置
        //sessionFactory.setMapperLocations(new PathMatchingResourcePatternResolver().getResources(""));
        sessionFactory.setConfiguration(slaveConfiguration);
        return sessionFactory.getObject();
    }

    @Bean(name = "slaveTransactionManager")
    public DataSourceTransactionManager slaveTransactionManager() {
        return new DataSourceTransactionManager(slaveDataSource);
    }

}

在com.yanger.mybatis.dao下有IUserDao的Mapper类,在com.yanger.mybatis.sdao创建了ISlaveUserDao类,他们内容完全一样,这里就只贴出IUserDao了。

@Mapper // IUserDao注入名重复
public interface IUserDao {

    /**
     * 添加用户
     * @param user
     * @return
     */
    @Insert("insert into user(username, password) values(#{username}, #{password})")
    int add(User user);

    /**
     * 根据id删除用户
     * @param id
     * @return
     */
    @Delete("delete from user where id = #{id}")
    int del(@Param("id") int id);

    /**
     * 更新用户信息
     * @param user
     * @return
     */
    @Update("update user set username = #{username}, password = #{password} where id = #{id}")
    int update(User user);

    /**
     * 根据id查找用户
     * @param id
     * @return
     */
    @Select("select * from user where id = #{id}")
    User find(@Param("id") int id);

    /**
     * 获取全部用户
     * @return
     */
    @Select("select * from user")
    List<User> findAll();

    /**
     * 根据用户名和密码查找用户
     * @param username
     * @param password
     * @return
     */
    @Select("select * from user where username = #{username} and password = #{password}")
    User findByUsernameAndPassword(@Param("username") String username, @Param("password") String password);

}

在使用时呢,我们只需要引入不同的Mapper类,就可以做到不同数据源的操作了。

@Service
public class UserService {

    @Autowired
    private IUserDao userDao;

    @Autowired
    private ISlaveUserDao iSlaveUserDao;

    /**
     * 添加用户
     * @param user
     */
    @Transactional(value = "masterTransactionManager")
    public void addUser(User user){
        userDao.add(user);
        // 除零异常,测试事务
        int a = 1/0;
    }

    /**
     * 添加用户
     * @param user
     */
    @Transactional(value = "slaveTransactionManager")
    public void addUserSlave(User user){
        iSlaveUserDao.add(user);
        // 除零异常,测试事务
        int a = 1/0;
    }
}

源码地址:https://github.com/imyanger/springboot-project/tree/master/p14-springboot-muti-mybatis

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值