springboot mysql yml配置 springboot配置mybatis数据源

转载

编程小达人 2023-08-01 14:20:53

文章标签 SpringBoot 多数据源 mybatis spring 数据源 文章分类 MySQL 数据库

一、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:///imyanger/springboot-project/tree/master/p14-springboot-muti-mybatis

本文章为转载内容，我们尊重原作者对文章享有的著作权。如有内容错误或侵权问题，欢迎原作者联系我们进行内容更正或删除文章。