springboot+mybat(多数据源) 通过aop实现分库的操作

1：pom.xml的引入相关jar

<?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>
    <parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>2.1.6.RELEASE</version>
        <relativePath/> <!-- lookup parent from repository -->
    </parent>
    <groupId>com.example</groupId>
    <artifactId>demo</artifactId>
    <version>0.0.1-SNAPSHOT</version>
    <name>demo</name>
    <description>Demo project for Spring Boot</description>

    <properties>
        <java.version>1.8</java.version>
    </properties>

    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</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-starter-aop</artifactId>
        </dependency>

        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <version>8.0.13</version>
        </dependency>

        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-jdbc</artifactId>
        </dependency>

        <dependency>
            <groupId>org.springframework</groupId>
            <artifactId>spring-beans</artifactId>
            <version>5.1.3.RELEASE</version>
        </dependency>

        <dependency>
            <groupId>org.mybatis</groupId>
            <artifactId>mybatis</artifactId>
            <version>3.4.6</version>
        </dependency>

        <dependency>
            <groupId>org.mybatis.spring.boot</groupId>
            <artifactId>mybatis-spring-boot-starter</artifactId>
            <version>2.0.0</version>
        </dependency>

        <dependency>
            <groupId>javax.servlet</groupId>
            <artifactId>jstl</artifactId>
            <version>1.2</version>
        </dependency>

        <dependency>
            <groupId>org.apache.tomcat.embed</groupId>
            <artifactId>tomcat-embed-jasper</artifactId>
        </dependency>


    </dependencies>

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

</project>

2 ：aplication.properties 配置

server.port=9999
#spring 数据源
#spring.datasource.url=jdbc:mysql://localhost:3306/test?useUnicode=true&characterEncoding=utf8&useSSL=false&allowPublicKeyRetrieval=true&serverTimezone=GMT%2B8
#spring.datasource.username=root
#spring.datasource.password=123456
#spring.datasource.driver-class-name=com.mysql.cj.jdbc.Driver
#spring 多数据源
#db1
spring.datasource.db1.jdbc-url=jdbc:mysql://localhost:3306/test?useUnicode=true&characterEncoding=utf8&useSSL=false&allowPublicKeyRetrieval=true&serverTimezone=GMT%2B8
spring.datasource.db1.username=root
spring.datasource.db1.password=123456
spring.datasource.db1.driver-class-name=com.mysql.cj.jdbc.Driver
#db2
spring.datasource.db2.jdbc-url=jdbc:mysql://localhost:3306/user_0?useUnicode=true&characterEncoding=utf8&useSSL=false&allowPublicKeyRetrieval=true&serverTimezone=GMT%2B8
spring.datasource.db2.username=root
spring.datasource.db2.password=123456
spring.datasource.db2.driver-class-name=com.mysql.cj.jdbc.Driver


mybatis.configuration.log-impl=org.apache.ibatis.logging.stdout.StdOutImpl  
mybatis.mapper-locations=classpath:mapper/*.xml
mybatis.type-aliases-package=com.example.demo.bean

3：启动类修改

package com.example.demo;

import org.mybatis.spring.annotation.MapperScan;
import org.mybatis.spring.annotation.MapperScans;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.boot.autoconfigure.jdbc.DataSourceAutoConfiguration;
/**

 * springboot入口类,此类需要在所有用到的package上层

 * exclude = {DataSourceAutoConfiguration.class}

 * 禁用springboot默认加载的application.properties单数据源配置

 */
@SpringBootApplication(exclude = {DataSourceAutoConfiguration.class})//多数据原配置
@MapperScan("com.example.demo.mapper.**") //指定要扫描的Mapper类的包的路径
public class DemoApplication {

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

}

4：多数据源配置类

package com.example.demo.common;


//import org.springframework.boot.autoconfigure.jdbc.DataSourceBuilder;

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 org.springframework.jdbc.datasource.DataSourceTransactionManager;

import org.springframework.transaction.PlatformTransactionManager;


import javax.sql.DataSource;

import java.util.HashMap;

import java.util.Map;

/**

 * 多数据源配置类

 * Created by wl

 */

@Configuration
public class DataSourceConfig {



        //数据源1
        @Bean(name = "datasource1")
        @ConfigurationProperties(prefix = "spring.datasource.db1") // application.properteis中对应属性的前缀
        public DataSource dataSource1() {
            return DataSourceBuilder.create().build();
        }

        //数据源2
        @Bean(name = "datasource2")
        @ConfigurationProperties(prefix = "spring.datasource.db2") // application.properteis中对应属性的前缀
        public DataSource dataSource2() {
            return DataSourceBuilder.create().build();
        }

        /**
         * 动态数据源: 通过AOP在不同数据源之间动态切换
         * @return
         */
        @Primary
        @Bean(name = "dynamicDataSource")
        public DataSource dynamicDataSource() {
            DynamicDataSource dynamicDataSource = new DynamicDataSource();
            // 默认数据源
            dynamicDataSource.setDefaultTargetDataSource(dataSource1());
            // 配置多数据源
            Map<Object, Object> dsMap = new HashMap();
            dsMap.put("datasource1", dataSource1());
            dsMap.put("user_O", dataSource2());
            dynamicDataSource.setTargetDataSources(dsMap);
            return dynamicDataSource;
        }



        /**
         * 配置@Transactional注解事物
         * @return
         */

        @Bean
        public PlatformTransactionManager transactionManager() {
            return new DataSourceTransactionManager(dynamicDataSource());
        }


}

5：保存切换数据源：

package com.example.demo.common;



/**

 * Created by wl

 */

public class DataSourceContextHolder {

    /**

     * 默认数据源

     */

    public static final String DEFAULT_DS = "datasource1";

    public static final String USER_0 = "user_O";



    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();

    }

}

6：当前数据源

package com.example.demo.common;

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

public class DynamicDataSource extends AbstractRoutingDataSource {
    @Override
    protected Object determineCurrentLookupKey() {

        System.out.println("数据源为"+DataSourceContextHolder.getDB());

        return DataSourceContextHolder.getDB();
    }
}

7：AOP动态切换

package com.example.demo.common;



import com.example.demo.mapper.Base;
import com.example.demo.mapper.BaseMapper;
import org.aspectj.lang.JoinPoint;

import org.aspectj.lang.annotation.After;

import org.aspectj.lang.annotation.Aspect;

import org.aspectj.lang.annotation.Before;

import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;

import org.springframework.stereotype.Component;

import java.lang.reflect.Method;

/**

 * 自定义注解 + AOP的方式实现数据源动态切换。

 * Created by pure on 2018-05-06.

 */

@Aspect
@Component
public class DynamicDataSourceAspect {

    @Pointcut("this(com.example.demo.mapper.Base)")  //根据类名贴换
    public void DynamicDataSourcePointCut(){

    }



    @Before("DynamicDataSourcePointCut()")
    public void beforeSwitchDS(JoinPoint point){
        //获得当前访问的class
        Object className = point.getThis();
        if(className instanceof BaseMapper) {
            String dataSource = DataSourceContextHolder.DEFAULT_DS;
            // 切换数据源
            DataSourceContextHolder.setDB(dataSource);
        }else{
            String dataSource = DataSourceContextHolder.USER_0;
            // 切换数据源
            DataSourceContextHolder.setDB(dataSource);
        }

    }



    @After("DynamicDataSourcePointCut()")
    public void afterSwitchDS(JoinPoint point){
        DataSourceContextHolder.clearDB();
    }

}

BaseMapper .class

package com.example.demo.mapper;

/**
 * test
 */
public interface BaseMapper extends Base{

}

Base .class

package com.example.demo.mapper;

/**
 * user_0
 */
public interface Base {
}

实现类Mapper

package com.example.demo.mapper;

import com.example.demo.bean.TStudent;
import java.util.List;
import java.util.Map;

public interface TStudentMapper extends Base{
    List<TStudent> findAll();
    int addStudent(TStudent student);
    List<TStudent> findStudent(Map<String,Object> map);
}

7：bean实现分表的操作

BaseBean.class

package com.example.demo.bean;

public class BaseBean {
    private String tableName;

    public String getTableName() {
        return tableName;
    }

    public void setTableName(String tableName) {
        this.tableName = tableName;
    }
}

TStudent .class

package com.example.demo.bean;

public class TStudent extends BaseBean{
    private String id;

    private String name;

    private Integer age;

    public String getId() {
        return id;
    }

    public void setId(String id) {
        this.id = id == null ? null : id.trim();
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name == null ? null : name.trim();
    }

    public Integer getAge() {
        return age;
    }

    public void setAge(Integer age) {
        this.age = age;
    }

    @Override
    public String toString() {
        return "name=" + name + "age=" + age;
    }
}

9:数据库表

DEMO

demo请点击