詳解springboot+mybatis多數(shù)據(jù)源最簡解決方案

更新時間：2017年05月05日 09:37:33 作者：純潔的蟲子

本篇文章主要介紹了詳解springboot+mybatis多數(shù)據(jù)源最簡解決方案，具有一定的參考價值，感興趣的小伙伴們可以參考一下

說起多數(shù)據(jù)源，一般都來解決那些問題呢，主從模式或者業(yè)務比較復雜需要連接不同的分庫來支持業(yè)務。我們項目是后者的模式，網(wǎng)上找了很多，大都是根據(jù)jpa來做多數(shù)據(jù)源解決方案，要不就是老的spring多數(shù)據(jù)源解決方案，還有的是利用aop動態(tài)切換，感覺有點小復雜，其實我只是想找一個簡單的多數(shù)據(jù)支持而已，折騰了兩個小時整理出來，供大家參考。

廢話不多說直接上代碼吧

配置文件

pom包就不貼了比較簡單該依賴的就依賴，主要是數(shù)據(jù)庫這邊的配置：

mybatis.config-locations=classpath:mybatis/mybatis-config.xml

spring.datasource.test1.driverClassName = com.mysql.jdbc.Driver
spring.datasource.test1.url = jdbc:mysql://localhost:3306/test1?useUnicode=true&characterEncoding=utf-8
spring.datasource.test1.username = root
spring.datasource.test1.password = root

spring.datasource.test2.driverClassName = com.mysql.jdbc.Driver
spring.datasource.test2.url = jdbc:mysql://localhost:3306/test2?useUnicode=true&characterEncoding=utf-8
spring.datasource.test2.username = root
spring.datasource.test2.password = root

一個test1庫和一個test2庫，其中test1位主庫，在使用的過程中必須制定主庫，不然會報錯。

數(shù)據(jù)源配置

@Configuration
@MapperScan(basePackages = "com.neo.mapper.test1", sqlSessionTemplateRef = "test1SqlSessionTemplate")
public class DataSource1Config {

  @Bean(name = "test1DataSource")
  @ConfigurationProperties(prefix = "spring.datasource.test1")
  @Primary
  public DataSource testDataSource() {
    return DataSourceBuilder.create().build();
  }

  @Bean(name = "test1SqlSessionFactory")
  @Primary
  public SqlSessionFactory testSqlSessionFactory(@Qualifier("test1DataSource") DataSource dataSource) throws Exception {
    SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
    bean.setDataSource(dataSource);
    bean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath:mybatis/mapper/test1/*.xml"));
    return bean.getObject();
  }

  @Bean(name = "test1TransactionManager")
  @Primary
  public DataSourceTransactionManager testTransactionManager(@Qualifier("test1DataSource") DataSource dataSource) {
    return new DataSourceTransactionManager(dataSource);
  }

  @Bean(name = "test1SqlSessionTemplate")
  @Primary
  public SqlSessionTemplate testSqlSessionTemplate(@Qualifier("test1SqlSessionFactory") SqlSessionFactory sqlSessionFactory) throws Exception {
    return new SqlSessionTemplate(sqlSessionFactory);
  }

}

最關鍵的地方就是這塊了，一層一層注入,先創(chuàng)建DataSource，在創(chuàng)建SqlSessionFactory在創(chuàng)建事務，最后包裝到SqlSessionTemplate中。其中需要制定分庫的mapper文件地址，以及分庫到層代碼

復制代碼代碼如下:

@MapperScan(basePackages = "com.neo.mapper.test1", sqlSessionTemplateRef  = "test1SqlSessionTemplate")

這塊的注解就是指明了掃描dao層，并且給dao層注入指定的SqlSessionTemplate。所有@Bean都需要按照命名指定正確。

dao層和xml層

dao層和xml需要按照庫來分在不同的目錄，比如：test1庫dao層在com.neo.mapper.test1包下，test2庫在com.neo.mapper.test1

public interface User1Mapper {

  List<UserEntity> getAll();

  UserEntity getOne(Long id);

  void insert(UserEntity user);

  void update(UserEntity user);

  void delete(Long id);

}

xml層

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd" >
<mapper namespace="com.neo.mapper.test1.User1Mapper" >
  <resultMap id="BaseResultMap" type="com.neo.entity.UserEntity" >
    <id column="id" property="id" jdbcType="BIGINT" />
    <result column="userName" property="userName" jdbcType="VARCHAR" />
    <result column="passWord" property="passWord" jdbcType="VARCHAR" />
    <result column="user_sex" property="userSex" javaType="com.neo.enums.UserSexEnum"/>
    <result column="nick_name" property="nickName" jdbcType="VARCHAR" />
  </resultMap>

  <sql id="Base_Column_List" >
    id, userName, passWord, user_sex, nick_name
  </sql>

  <select id="getAll" resultMap="BaseResultMap" >
    SELECT 
    <include refid="Base_Column_List" />
    FROM users
  </select>

  <select id="getOne" parameterType="java.lang.Long" resultMap="BaseResultMap" >
    SELECT 
    <include refid="Base_Column_List" />
    FROM users
    WHERE id = #{id}
  </select>

  <insert id="insert" parameterType="com.neo.entity.UserEntity" >
    INSERT INTO 
      users
      (userName,passWord,user_sex) 
    VALUES
      (#{userName}, #{passWord}, #{userSex})
  </insert>

  <update id="update" parameterType="com.neo.entity.UserEntity" >
    UPDATE 
      users 
    SET 
    <if test="userName != null">userName = #{userName},</if>
    <if test="passWord != null">passWord = #{passWord},</if>
    nick_name = #{nickName}
    WHERE 
      id = #{id}
  </update>

  <delete id="delete" parameterType="java.lang.Long" >
    DELETE FROM
       users 
    WHERE 
       id =#{id}
  </delete>

</mapper>

測試

測試可以使用SpringBootTest,也可以放到Controller中，這里只貼Controller層的使用

@RestController
public class UserController {

  @Autowired
  private User1Mapper user1Mapper;

  @Autowired
  private User2Mapper user2Mapper;

  @RequestMapping("/getUsers")
  public List<UserEntity> getUsers() {
    List<UserEntity> users=user1Mapper.getAll();
    return users;
  }

  @RequestMapping("/getUser")
  public UserEntity getUser(Long id) {
    UserEntity user=user2Mapper.getOne(id);
    return user;
  }

  @RequestMapping("/add")
  public void save(UserEntity user) {
    user2Mapper.insert(user);
  }

  @RequestMapping(value="update")
  public void update(UserEntity user) {
    user2Mapper.update(user);
  }

  @RequestMapping(value="/delete/{id}")
  public void delete(@PathVariable("id") Long id) {
    user1Mapper.delete(id);
  }

}

最后源碼地址在這里spring-boot-mybatis-mulidatasource

以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: