Spring Boot 2.x基础教程：MyBatis的多数据源配置

dyc87112 · dyc87112 · commit cc1771dd7b47 · 2020-06-24T12:30:07.000+08:00
diff --git a/2.1.x/chapter3-9/src/main/java/com/didispace/chapter39/Chapter39Application.java b/2.1.x/chapter3-9/src/main/java/com/didispace/chapter39/Chapter39Application.java
@@ -1,15 +1,13 @@
-package com.didispace.chapter36;
+package com.didispace.chapter39;
 
-import org.mybatis.spring.annotation.MapperScan;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
 
-@MapperScan("com.didispace.chapter36.mapper")
 @SpringBootApplication
-public class Chapter36Application {
+public class Chapter39Application {
 
 	public static void main(String[] args) {
-		SpringApplication.run(Chapter36Application.class, args);
+		SpringApplication.run(Chapter39Application.class, args);
 	}
 
 }
diff --git a/2.1.x/chapter3-9/src/main/java/com/didispace/chapter39/DataSourceConfiguration.java b/2.1.x/chapter3-9/src/main/java/com/didispace/chapter39/DataSourceConfiguration.java
@@ -0,0 +1,27 @@
+package com.didispace.chapter39;
+
+import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.boot.jdbc.DataSourceBuilder;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.context.annotation.Primary;
+
+import javax.sql.DataSource;
+
+@Configuration
+public class DataSourceConfiguration {
+
+    @Primary
+    @Bean
+    @ConfigurationProperties(prefix = "spring.datasource.primary")
+    public DataSource primaryDataSource() {
+        return DataSourceBuilder.create().build();
+    }
+
+    @Bean
+    @ConfigurationProperties(prefix = "spring.datasource.secondary")
+    public DataSource secondaryDataSource() {
+        return DataSourceBuilder.create().build();
+    }
+
+}
diff --git a/2.1.x/chapter3-9/src/main/java/com/didispace/chapter39/PrimaryConfig.java b/2.1.x/chapter3-9/src/main/java/com/didispace/chapter39/PrimaryConfig.java
@@ -0,0 +1,39 @@
+package com.didispace.chapter39;
+
+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.Qualifier;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+import javax.sql.DataSource;
+
+@Configuration
+@MapperScan(
+        basePackages = "com.didispace.chapter39.p",
+        sqlSessionFactoryRef = "sqlSessionFactoryPrimary",
+        sqlSessionTemplateRef = "sqlSessionTemplatePrimary")
+public class PrimaryConfig {
+
+    private DataSource primaryDataSource;
+
+    public PrimaryConfig(@Qualifier("primaryDataSource") DataSource primaryDataSource) {
+        this.primaryDataSource = primaryDataSource;
+    }
+
+    @Bean
+    public SqlSessionFactory sqlSessionFactoryPrimary() throws Exception {
+        SqlSessionFactory sessionFactory = null;
+        SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
+        bean.setDataSource(primaryDataSource);
+        return bean.getObject();
+    }
+
+    @Bean
+    public SqlSessionTemplate sqlSessionTemplatePrimary() throws Exception {
+        return new SqlSessionTemplate(sqlSessionFactoryPrimary());
+    }
+
+}
diff --git a/2.1.x/chapter3-9/src/main/java/com/didispace/chapter39/SecondaryConfig.java b/2.1.x/chapter3-9/src/main/java/com/didispace/chapter39/SecondaryConfig.java
@@ -0,0 +1,39 @@
+package com.didispace.chapter39;
+
+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.Qualifier;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+import javax.sql.DataSource;
+
+@Configuration
+@MapperScan(
+        basePackages = "com.didispace.chapter39.s",
+        sqlSessionFactoryRef = "sqlSessionFactorySecondary",
+        sqlSessionTemplateRef = "sqlSessionTemplateSecondary")
+public class SecondaryConfig {
+
+    private DataSource secondaryDataSource;
+
+    public SecondaryConfig(@Qualifier("secondaryDataSource") DataSource secondaryDataSource) {
+        this.secondaryDataSource = secondaryDataSource;
+    }
+
+    @Bean
+    public SqlSessionFactory sqlSessionFactorySecondary() throws Exception {
+        SqlSessionFactory sessionFactory = null;
+        SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
+        bean.setDataSource(secondaryDataSource);
+        return bean.getObject();
+    }
+
+    @Bean
+    public SqlSessionTemplate sqlSessionTemplateSecondary() throws Exception {
+        return new SqlSessionTemplate(sqlSessionFactorySecondary());
+    }
+
+}
diff --git a/2.1.x/chapter3-9/src/main/java/com/didispace/chapter39/mapper/UserMapper.java b/2.1.x/chapter3-9/src/main/java/com/didispace/chapter39/mapper/UserMapper.java
diff --git a/2.1.x/chapter3-9/src/main/java/com/didispace/chapter39/p/entity/UserPrimary.java b/2.1.x/chapter3-9/src/main/java/com/didispace/chapter39/p/entity/UserPrimary.java
@@ -1,19 +1,19 @@
-package com.didispace.chapter36.entity;
+package com.didispace.chapter39.p.entity;
 
 import lombok.Data;
 import lombok.NoArgsConstructor;
 
 
 @Data
 @NoArgsConstructor
-public class User {
+public class UserPrimary {
 
     private Long id;
 
     private String name;
     private Integer age;
 
-    public User(String name, Integer age) {
+    public UserPrimary(String name, Integer age) {
         this.name = name;
         this.age = age;
     }
diff --git a/2.1.x/chapter3-9/src/main/java/com/didispace/chapter39/p/mapper/UserMapperPrimary.java b/2.1.x/chapter3-9/src/main/java/com/didispace/chapter39/p/mapper/UserMapperPrimary.java
@@ -0,0 +1,26 @@
+package com.didispace.chapter39.p.mapper;
+
+import com.didispace.chapter39.p.entity.UserPrimary;
+import org.apache.ibatis.annotations.Delete;
+import org.apache.ibatis.annotations.Insert;
+import org.apache.ibatis.annotations.Param;
+import org.apache.ibatis.annotations.Select;
+
+/**
+ * Created by 程序猿DD/翟永超 on 2020/2/28.
+ * <p>
+ * Blog: http://blog.didispace.com/
+ * Github: https://github.com/dyc87112/
+ */
+public interface UserMapperPrimary {
+
+    @Select("SELECT * FROM USER WHERE NAME = #{name}")
+    UserPrimary findByName(@Param("name") String name);
+
+    @Insert("INSERT INTO USER(NAME, AGE) VALUES(#{name}, #{age})")
+    int insert(@Param("name") String name, @Param("age") Integer age);
+
+    @Delete("DELETE FROM USER")
+    int deleteAll();
+
+}
diff --git a/2.1.x/chapter3-9/src/main/java/com/didispace/chapter39/s/entity/UserSecondary.java b/2.1.x/chapter3-9/src/main/java/com/didispace/chapter39/s/entity/UserSecondary.java
@@ -0,0 +1,20 @@
+package com.didispace.chapter39.s.entity;
+
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+
+@Data
+@NoArgsConstructor
+public class UserSecondary {
+
+    private Long id;
+
+    private String name;
+    private Integer age;
+
+    public UserSecondary(String name, Integer age) {
+        this.name = name;
+        this.age = age;
+    }
+}
diff --git a/2.1.x/chapter3-9/src/main/java/com/didispace/chapter39/s/mapper/UserMapperSecondary.java b/2.1.x/chapter3-9/src/main/java/com/didispace/chapter39/s/mapper/UserMapperSecondary.java
@@ -0,0 +1,25 @@
+package com.didispace.chapter39.s.mapper;
+
+import com.didispace.chapter39.s.entity.UserSecondary;
+import org.apache.ibatis.annotations.Delete;
+import org.apache.ibatis.annotations.Insert;
+import org.apache.ibatis.annotations.Param;
+import org.apache.ibatis.annotations.Select;
+
+/**
+ * Created by 程序猿DD/翟永超 on 2020/2/28.
+ * <p>
+ * Blog: http://blog.didispace.com/
+ * Github: https://github.com/dyc87112/
+ */
+public interface UserMapperSecondary {
+
+    @Select("SELECT * FROM USER WHERE NAME = #{name}")
+    UserSecondary findByName(@Param("name") String name);
+
+    @Insert("INSERT INTO USER(NAME, AGE) VALUES(#{name}, #{age})")
+    int insert(@Param("name") String name, @Param("age") Integer age);
+
+    @Delete("DELETE FROM USER")
+    int deleteAll();
+}
diff --git a/2.1.x/chapter3-9/src/main/resources/application.properties b/2.1.x/chapter3-9/src/main/resources/application.properties
@@ -1,6 +1,13 @@
-spring.datasource.url=jdbc:mysql://localhost:3306/test
-spring.datasource.username=root
-spring.datasource.password=12345678
-spring.datasource.driver-class-name=com.mysql.cj.jdbc.Driver
+# pring boot 1.x的配置：spring.datasource.primary.url=jdbc:mysql://localhost:3306/test1
+spring.datasource.primary.jdbc-url=jdbc:mysql://localhost:3306/test1
+spring.datasource.primary.username=root
+spring.datasource.primary.password=12345678
+spring.datasource.primary.driver-class-name=com.mysql.cj.jdbc.Driver
 
-mybatis.mapper-locations=classpath:mapper/*.xml
+# spring boot 1.x的配置：spring.datasource.secondary.url=jdbc:mysql://localhost:3306/test2
+spring.datasource.secondary.jdbc-url=jdbc:mysql://localhost:3306/test2
+spring.datasource.secondary.username=root
+spring.datasource.secondary.password=12345678
+spring.datasource.secondary.driver-class-name=com.mysql.cj.jdbc.Driver
+
+#mybatis.mapper-locations=classpath:mapper/*.xml
diff --git a/2.1.x/chapter3-9/src/main/resources/mapper.primary/UserMapper.xml b/2.1.x/chapter3-9/src/main/resources/mapper.primary/UserMapper.xml
@@ -2,12 +2,14 @@
 <!DOCTYPE mapper
         PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
         "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
-<mapper namespace="com.didispace.chapter36.mapper.UserMapper">
-    <select id="findByName" resultType="com.didispace.chapter36.entity.User">
+
+<mapper namespace="com.didispace.chapter39.p.mapper.UserMapperPrimary">
+    <select id="findByName" resultType="com.didispace.chapter39.p.entity.UserPrimary">
         SELECT * FROM USER WHERE NAME = #{name}
     </select>
 
     <insert id="insert">
         INSERT INTO USER(NAME, AGE) VALUES(#{name}, #{age})
     </insert>
+
 </mapper>
diff --git a/2.1.x/chapter3-9/src/main/resources/mapper.secondary/UserMapper.xml b/2.1.x/chapter3-9/src/main/resources/mapper.secondary/UserMapper.xml
@@ -0,0 +1,16 @@
+<?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.didispace.chapter39.s.mapper.UserMapperSecondary">
+
+    <select id="findByName" resultType="com.didispace.chapter39.s.entity.UserSecondary">
+        SELECT * FROM USER WHERE NAME = #{name}
+    </select>
+
+    <insert id="insert">
+        INSERT INTO USER(NAME, AGE) VALUES(#{name}, #{age})
+    </insert>
+
+</mapper>
diff --git a/2.1.x/chapter3-9/src/test/java/com/didispace/chapter36/Chapter36ApplicationTests.java b/2.1.x/chapter3-9/src/test/java/com/didispace/chapter36/Chapter36ApplicationTests.java
diff --git a/2.1.x/chapter3-9/src/test/java/com/didispace/chapter39/Chapter39ApplicationTests.java b/2.1.x/chapter3-9/src/test/java/com/didispace/chapter39/Chapter39ApplicationTests.java
@@ -0,0 +1,60 @@
+package com.didispace.chapter39;
+
+import com.didispace.chapter39.p.entity.UserPrimary;
+import com.didispace.chapter39.p.mapper.UserMapperPrimary;
+import com.didispace.chapter39.s.entity.UserSecondary;
+import com.didispace.chapter39.s.mapper.UserMapperSecondary;
+import lombok.extern.slf4j.Slf4j;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.test.context.junit4.SpringRunner;
+import org.springframework.transaction.annotation.Transactional;
+
+@Slf4j
+@RunWith(SpringRunner.class)
+@SpringBootTest
+@Transactional
+public class Chapter39ApplicationTests {
+
+    @Autowired
+    private UserMapperPrimary userMapperPrimary;
+    @Autowired
+    private UserMapperSecondary userMapperSecondary;
+
+    @Before
+    public void setUp() {
+        // 清空测试表，保证每次结果一样
+        userMapperPrimary.deleteAll();
+        userMapperSecondary.deleteAll();
+    }
+
+    @Test
+    public void test() throws Exception {
+        // 往Primary数据源插入一条数据
+        userMapperPrimary.insert("AAA", 20);
+
+        // 从Primary数据源查询刚才插入的数据，配置正确就可以查询到
+        UserPrimary userPrimary = userMapperPrimary.findByName("AAA");
+        Assert.assertEquals(20, userPrimary.getAge().intValue());
+
+        // 从Secondary数据源查询刚才插入的数据，配置正确应该是查询不到的
+        UserSecondary userSecondary = userMapperSecondary.findByName("AAA");
+        Assert.assertNull(userSecondary);
+
+        // 往Secondary数据源插入一条数据
+        userMapperSecondary.insert("BBB", 20);
+
+        // 从Primary数据源查询刚才插入的数据，配置正确应该是查询不到的
+        userPrimary = userMapperPrimary.findByName("BBB");
+        Assert.assertNull(userPrimary);
+
+        // 从Secondary数据源查询刚才插入的数据，配置正确就可以查询到
+        userSecondary = userMapperSecondary.findByName("BBB");
+        Assert.assertEquals(20, userSecondary.getAge().intValue());
+    }
+
+}
