SpringBoot整合Mybatis多數據源 (AOP+注解)

SpringBoot整合Mybatis多數據源 (AOP+注解)

1、pom.xml文件(開發用的JDK 10)

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
    xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>

    <groupId>com.example</groupId>
    <artifactId>demo</artifactId>
    <version>0.0.1-SNAPSHOT</version>
    <packaging>jar</packaging>

    <name>demo</name>
    <description>Demo project for Spring Boot</description>

    <parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>2.0.3.RELEASE</version>
        <relativePath/> <!-- lookup parent from repository -->
    </parent>

    <properties>
        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
        <project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding>
        <java.version>10</java.version>
    </properties>

    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        
        <!-- alibaba的druid數據庫連接池 -->
        <dependency>
            <groupId>com.alibaba</groupId>
            <artifactId>druid</artifactId>
            <version>1.1.10</version>
        </dependency>
        
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-jdbc</artifactId>
        </dependency>
        
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-tomcat</artifactId>
        </dependency>
        
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-aop</artifactId>
        </dependency>
        
        <dependency>
            <groupId>org.aspectj</groupId>
            <artifactId>aspectjweaver</artifactId>
        </dependency>
        
        <dependency>
            <groupId>org.mybatis.spring.boot</groupId>
            <artifactId>mybatis-spring-boot-starter</artifactId>
            <version>1.3.0</version>
        </dependency>
        
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-thymeleaf</artifactId>
        </dependency>
        
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
        </dependency>
        
        <dependency>
            <groupId>com.github.pagehelper</groupId>
            <artifactId>pagehelper-spring-boot-starter</artifactId>
            <version>1.2.3</version>
        </dependency>
        
        <!-- 熱部署 -->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-devtools</artifactId>
            <optional>true</optional>
        </dependency>
        
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-data-jpa</artifactId>
        </dependency>

        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
            <scope>test</scope>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-configuration-processor</artifactId>
            <optional>true</optional>
        </dependency>
    </dependencies>

    <build>
        <plugins>
            <plugin>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-maven-plugin</artifactId>
            </plugin>
        </plugins>
    </build>


</project>

2、多數據源配置類

@Configuration
public class DataSourceConfig {
    
    @Bean(name = "readTestDb")
    @ConfigurationProperties(prefix = "spring.datasource.read-test-db") 
    public DataSource readTestDb() {
        return DataSourceBuilder.create().build();
    }


    @Bean(name = "writeTestDb")
    @ConfigurationProperties(prefix = "spring.datasource.write-test-db")
    public DataSource writeTestDb() {
        return DataSourceBuilder.create().build();
    }
    

    /**
     * 動態數據源: 通過AOP在不同數據源之間動態切換
     * @return
     */
    @Primary
    @Bean(name = "dynamicDataSource")
    public DataSource dynamicDataSource() {
        DynamicDataSource dynamicDataSource = new DynamicDataSource();
        // 默認數據源
        dynamicDataSource.setDefaultTargetDataSource(readTestDb());
        // 配置多數據源
        Map<Object, Object> dsMap = new HashMap<Object, Object>();
        dsMap.put("readTestDb", readTestDb());
        dsMap.put("writeTestDb", writeTestDb());
 
        dynamicDataSource.setTargetDataSources(dsMap);
        return dynamicDataSource;
    }
 
    /**
     * 配置@Transactional注解事物
     * @return
     */
    @Bean
    public PlatformTransactionManager transactionManager() {
        return new DataSourceTransactionManager(dynamicDataSource());
    }

3、數據源切換讀取類

public class DataSourceContextHolder {

    /**
     * 默認數據源
     */
    public static final String DEFAULT_DS = "readTestDb";
 
    private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();
 
    // 設置數據源名
    public static void setDB(String dbType) {
        System.out.println("切換到{"+dbType+"}數據源");
        contextHolder.set(dbType);
    }
 
    // 獲取數據源名
    public static String getDB() {
        return (contextHolder.get());
    }
 
    // 清除數據源名
    public static void clearDB() {
        contextHolder.remove();
    }

}

4、注解類(注意：要和配置類放在同一個包下，否則會拋出找不到注解異常)

@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.METHOD})
@Documented
public @interface DS {
    String value() default "readTestDb";
}

5、動態數據源(設置數據源自動調用該類方法)

public class DynamicDataSource extends AbstractRoutingDataSource  {

    @Override
    protected Object determineCurrentLookupKey() {
        System.out.println("數據源為"+DataSourceContextHolder.getDB());
        return DataSourceContextHolder.getDB();
    }

}

6、AOP

@Aspect
@Component
public class DynamicDataSourceAspect {
    
    @Before("@annotation(DS)")
    @SuppressWarnings("rawtypes")
    public void beforeSwitchDS(JoinPoint point){
        //獲得當前訪問的class
        Class<?> className = point.getTarget().getClass();
        //獲得訪問的方法名
        String methodName = point.getSignature().getName();
        //得到方法的參數的類型
        Class[] argClass = ((MethodSignature)point.getSignature()).getParameterTypes();
        String dataSource = DataSourceContextHolder.DEFAULT_DS;
        try {
            // 得到訪問的方法對象
            Method method = className.getMethod(methodName, argClass);
            // 判斷是否存在@DS注解
            if (method.isAnnotationPresent(DS.class)) {
                DS annotation = method.getAnnotation(DS.class);
                // 取出注解中的數據源名
                dataSource = annotation.value();
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
        // 切換數據源
        DataSourceContextHolder.setDB(dataSource);
    }
    
    @After("@annotation(DS)")
    public void afterSwitchDS(JoinPoint point){
        DataSourceContextHolder.clearDB();
    }
}

7、application.yml 配置文件 (和application.properties區別 可以百度一下 很詳細)

spring:
    datasource:
        read-test-db:
            jdbc-url: jdbc:mysql://127.0.0.1:3306/test?useUnicode=true&characterEncoding=utf-8
            username: root
            password: root
            # 使用druid數據源
            type: com.alibaba.druid.pool.DruidDataSource
            driver-class-name: com.mysql.jdbc.Driver
            
        write-test-db:
            jdbc-url: jdbc:mysql://127.0.0.1:3306/writetest?useUnicode=true&characterEncoding=utf-8
            username: root
            password: root
            # 使用druid數據源
            type: com.alibaba.druid.pool.DruidDataSource
            driver-class-name: com.mysql.jdbc.Driver  
    jpa:
      show-sql: true

mybatis:
  #mapper-locations: classpath:mapping/mapper/*.xml
  type-aliases-package: com.example.demo.model

 

8、啟動類( 注解：@MapperScan 掃描mapper接口類路徑 、多數據源 @SpringBootApplication這個注解是必須寫的 )

@MapperScan("com.example.demo.mapper")
@SpringBootApplication(exclude = {
        DataSourceAutoConfiguration.class
})
public class DemoApplication {

    public static void main(String[] args) {
        SpringApplication.run(DemoApplication.class, args);
    }
}

 

9、Controller( 注解：@RestController( 包含@controller、@responseBody等其他注解 所以用該注解就不用每個方法上@responseBody ) )

@RestController
public class DemoController {
    
    @Autowired
    private UserService userService;

    @RequestMapping("/getRequestDBObj")
    public User getRequestDBObj(HttpServletRequest request,HttpServletResponse response) {
        return userService.selectByPrimaryKey(1);
    }
    
    @RequestMapping("/getRequestDbList")
    public User getRequestDbList(HttpServletRequest request,HttpServletResponse response) {
        return userService.selectAllUser(1);
    }
}

10、Service

public interface UserService {
    
    public User selectByPrimaryKey(Integer userId);
    
    public User selectAllUser(Integer userId);
}

11、ServiceImpl

@Service
public class UserServiceImpl implements UserService {
    
    @Autowired
    private UserMapper userMapper;

    @Override
    @DS("writeTestDb")
    public User selectByPrimaryKey(Integer userId) {
        return userMapper.selectByPrimaryKey(userId);
    }

    @Override
    @DS("readTestDb")
    public User selectAllUser(Integer userId) {
        return userMapper.selectByPrimaryKey(userId);
    }
    
}

12、Mapper

public interface UserMapper {
    
    @Select("SELECT user_id as userId, user_name as userName, password, phone FROM t_user WHERE user_id = #{userId}")
    User selectByPrimaryKey(Integer userId);
    
}

13、本人 QQ 1036943655 本人萌新 有代碼不對的地方 大家指教一下 也可以加好友 大家一起互相學習、互相進步