SpringBoot 整合多資料來源

手動目錄 ????

SpringBoot 整合多資料來源 （MyBatis + Druid）
一、專案環境準備
二、專案配置初始化
	2.0 專案結構
	2.1 座標依賴
	2.2 application.yml
	2.3 建立資料庫
三、MVC 三層編寫
	3.1 mapper 編寫
	3.2 entity 實體類編寫
	3.3 config 配置類
	3.4 控制層 controller
四、測試執行

大家在寫專案的時候有沒有遇到過這樣的情況，當你需要你的好基友電腦上，伺服器上的資料的時候，是不是就得依賴他給你的介面。這樣做是不是侷限性太大。假如有一種方式可以直接透過你的小夥伴的伺服器上的賬號密碼，就可以自己定製化的操作你的小夥伴的伺服器上的資料了，這樣是不是很酷呢？

一、專案環境準備

1. MySQL 5.5 版本以上
2. JDK 1.8 以上
3. 開發工具：idea 2020, maven 3.5.2 版本及以上
4. SpringBoot 版本 2.2.10

二、專案配置初始化

2.0 專案結構

2.1 座標依賴

 <dependencies>
        <!-- 這是一個 Web 專案 -->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        <dependency>
            <groupId>org.mybatis.spring.boot</groupId>
            <artifactId>mybatis-spring-boot-starter</artifactId>
            <version>2.1.3</version>
        </dependency>
		
		<!-- mysql 5.5 版本依然可以使用 8.0的驅動 -->
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <version>8.0.11</version>
        </dependency>

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

2.2 application.yml

因為配置了兩個資料來源，所以為了區分它們，我們會自定義配置，同時資料庫也會建立兩個

spring:
  datasource:
    druid:
      db1:
        url: jdbc:mysql://localhost:3306/dts1?useUnicode=true&characterEncoding=utf8&serverTimezone=UTC
        driverClassName: com.mysql.cj.jdbc.Driver
        username: root
        password: root
      db2:
        url: jdbc:mysql://localhost:3306/dst2?useUnicode=true&characterEncoding=utf8&serverTimezone=UTC
        driverClassName: com.mysql.cj.jdbc.Driver
        username: root
        password: root

2.3 建立資料庫

為了簡化操作，我只是保證兩個資料庫的名稱不一致，其它的資料庫名稱，欄位都是一致的

我們建立了一個名為 book 的表，並隨機插入幾條資料

DROP TABLE IF EXISTS `book`;

CREATE TABLE `book` (
  `id` tinyint(3) NOT NULL AUTO_INCREMENT,
  `name` varchar(20) COLLATE utf8_unicode_ci DEFAULT NULL,
  `author` varchar(20) COLLATE utf8_unicode_ci DEFAULT NULL,
  PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=5 DEFAULT CHARSET=utf8 COLLATE=utf8_unicode_ci;

/*Data for the table `book` */

insert  into `book`(`id`,`name`,`author`) values (1,'時間簡史','霍金'),(2,'放學後','東野圭吾'),(3,'白白夜行','東野圭吾'),(4,'紅樓夢','清 曹雪芹');

三、MVC 三層編寫

3.1 mapper 編寫

在 MyBatis 中，我沒有使用註解開發，而是使用的 xml 方式，所以我們還需要在 SpringBoot 的 Resources 目錄下建立 mapper 目錄 和 map1，map2 兩個子目錄

dao 層相關的專案結構如下，然後下面我只給出 map1 中的相關程式碼，map2 中只是相對應的修改

- mapper
	- map1
		- BookMapper1
	- map2
		- BookMapper2

- resources
	- mapper
		- map1
			- BookMapper1.xml
		- map2
			- BookMapper2.xml

BookMapper1程式碼如下：

package cn.gorit.mapper.map1;

import cn.gorit.entity.Book;
import org.springframework.stereotype.Repository;

import java.util.List;

@Repository
public interface BookMapper1 {

    List<Book> getAllBooks();
}

BookMapper1.xml 配置如下

<?xml version="1.0" encoding="utf-8" ?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
        "">
<mapper namespace="cn.gorit.mapper.map1.BookMapper1">

    <select id="getAllBooks" resultType="cn.gorit.entity.Book">
        select * from book
    </select>

</mapper>

3.2 entity 實體類編寫

package cn.gorit.entity;

public class Book {
    private Integer id;
    private String name;
    private String author;

	// getter 和 setter 、構造方法省略
}

3.3 config 配置類

DataSourceConfig 配置類編寫

package cn.gorit.config;

import com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceBuilder;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.sql.DataSource;

/**
 * DataSourceConfig 中提供了兩個資料來源：database1 和 database2，預設方法名即例項名
 * @ConfigurationProperties  使用不同的字首配置檔案
 */
@Configuration
public class DataSourceConfig {
	
    //  標識為第一個資料來源
    @Bean("dataSource1")
    @ConfigurationProperties(prefix = "spring.datasource.druid.db1")
    DataSource database1() {
        return DruidDataSourceBuilder.create().build();
    }
	
    // 標識為第二個資料來源
    @Bean("dataSource2")
    @ConfigurationProperties(prefix = "spring.datasource.druid.db2")
    DataSource database2() {
        return DruidDataSourceBuilder.create().build();
    }
}

MyBatisConfigOne 配置類 （第一個資料來源對應的配置類）

package cn.gorit.config;

import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.mybatis.spring.SqlSessionTemplate;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;

import javax.sql.DataSource;

/**
* 第二個資料來源同理，建立一個新的配置類，將 MapperScan 的路徑改為 map2，bean 的注入也要對那個的修改
*/

@Configuration
@MapperScan(value = "cn.gorit.mapper.map1",sqlSessionFactoryRef = "sqlSessionFactoryBean1")
public class MyBatisConfigOne {

    @Autowired
    @Qualifier("dataSource1")
    DataSource db1;

    @Bean
    public SqlSessionFactory sqlSessionFactoryBean1() throws Exception {
        SqlSessionFactoryBean factoryBean = new SqlSessionFactoryBean();
        factoryBean.setDataSource(db1);
        factoryBean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath:./mapper/map1/*.xml"));
        return factoryBean.getObject();
    }

    @Bean
    SqlSessionTemplate sqlSessionTemplate1() throws Exception {
        return new SqlSessionTemplate(sqlSessionFactoryBean1());
    }
}

3.4 控制層 controller

package cn.gorit.controller;

import cn.gorit.mapper.map1.BookMapper1;
import cn.gorit.mapper.map2.BookMapper2;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.HashMap;
import java.util.Map;

@RestController
public class BookController {

    @Autowired
    BookMapper1 bookMapper1;

    @Autowired
    BookMapper2 bookMapper2;

    @GetMapping("/book1")
    public Map<String,Object> test1() {
        Map<String,Object> map = new HashMap<>();
        map.put("code",200);
        map.put("msg","獲取成功");
        map.put("data",bookMapper1.getAllBooks());
        return map;
    }

    @GetMapping("/book2")
    public Map<String,Object> test2() {
        Map<String,Object> map = new HashMap<>();
        map.put("code",200);
        map.put("msg","獲取成功");
        map.put("data",bookMapper2.getAllBooks());
        return map;
    }
}

四、測試執行