springboot+mybatis+druid+mysql实现多数据源，通过注解的方式控制数据源

最新推荐文章于 2024-05-30 15:38:25 发布

于光yuguang

最新推荐文章于 2024-05-30 15:38:25 发布

阅读量628

点赞数 1

分类专栏： java springboot mybatis

本文链接：https://blog.csdn.net/xixiyuguang/article/details/105674694

版权

springboot 同时被 3 个专栏收录

33 篇文章 2 订阅

订阅专栏

java

30 篇文章 0 订阅

订阅专栏

mybatis

9 篇文章 0 订阅

订阅专栏

1、创建数据库和测试表

1.1、创建数据库

1.2、创建数据表

在两个数据库中都添加user表

CREATE TABLE `user` (
  `id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT '主键,其它表中的使用user_id',
  `name` varchar(100) NOT NULL DEFAULT '',
  PRIMARY KEY (`id`)
) ENGINE=InnoDB  DEFAULT CHARSET=utf8mb4 COMMENT='用户表';

2、创建项目

2.1、pom.xml配置

<?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 https://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.2.2.RELEASE</version>
        <relativePath/> <!-- lookup parent from repository -->
    </parent>
    <groupId>com.yg</groupId>
    <artifactId>springboot</artifactId>
    <version>0.0.1-SNAPSHOT</version>
    <packaging>war</packaging>
    <name>06-springboot-dynamic-datasource</name>
    <description>Demo project for Spring Boot</description>

    <properties>
        <java.version>1.8</java.version>
        <lombok.version>1.18.6</lombok.version>
        <mybatis.version>1.3.2</mybatis.version>
        <lombox.version>1.18.6</lombox.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-tomcat</artifactId>
            <scope>provided</scope>
        </dependency>

        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <version>${lombok.version}</version>
        </dependency>
 
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
        </dependency>

        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <version>${lombox.version}</version>
        </dependency>
        <dependency>
            <groupId>org.mybatis.spring.boot</groupId>
            <artifactId>mybatis-spring-boot-starter</artifactId>
            <version>${mybatis.version}</version>
        </dependency>

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

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

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

        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
            <scope>test</scope>
            <exclusions>
                <exclusion>
                    <groupId>org.junit.vintage</groupId>
                    <artifactId>junit-vintage-engine</artifactId>
                </exclusion>
            </exclusions>
        </dependency>
    </dependencies>

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

</project>

2.2、配置properties


spring.datasource.type=com.alibaba.druid.pool.DruidDataSource


#主表master
spring.datasource.druid.master.driverClassName=com.mysql.cj.jdbc.Driver
spring.datasource.druid.master.url=jdbc:mysql://localhost:3306/test1?serverTimezone=GMT%2B8&useUnicode=true&characterEncoding=utf8
spring.datasource.druid.master.username=root
spring.datasource.druid.master.password=root


#分表slave
spring.datasource.druid.slave.driverClassName=com.mysql.cj.jdbc.Driver
spring.datasource.druid.slave.url=jdbc:mysql://localhost:3306/test2?serverTimezone=GMT%2B8&useUnicode=true&characterEncoding=utf8
spring.datasource.druid.slave.username=root
spring.datasource.druid.slave.password=root

mybatis.mapper-locations=classpath:mapper/*.xml

2.3、注解

/**
 * 备注：自定义数据源选择注解
 **/
//@Target(ElementType.METHOD)
@Target({ElementType.TYPE,ElementType.METHOD})// //注解方法，注解类都可以
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface DataSource {
    String name() default "";
}

2.4、aop

package com.yg.springboot.common.aop;

import com.yg.springboot.common.DatabaseType;
import com.yg.springboot.common.annotation.DataSource;
import com.yg.springboot.common.config.DynamicDataSource;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.stereotype.Component;

import java.lang.reflect.Method;

/**
 * @author yuguang
 * @date 2020/4/4 10:11
 * @desc 使用注解的方式去选择指定得数据源
 */
@Aspect
@Component
@Slf4j
public class DataSourceAspect {

    @Pointcut("@annotation(com.yg.springboot.common.annotation.DataSource)")
    public void dataSourcePointCut() {

    }

    @Around("dataSourcePointCut()")
    public Object around(ProceedingJoinPoint point) throws Throwable {
        MethodSignature signature = (MethodSignature) point.getSignature();
        Method method = signature.getMethod();

        //todo 通过注解的方式去选择数据源
        DataSource dataSource = method.getAnnotation(DataSource.class);
        if (dataSource == null) {
            log.info("默认数据源 master");
            DynamicDataSource.setDataSource(DatabaseType.master.toString());
        } else {
            log.info("切换数据源： " + dataSource.name());
            DynamicDataSource.setDataSource(dataSource.name());
        }

        try {
            return point.proceed();
        } finally {
            //todo 选择完之后 清除当前数据源
            DynamicDataSource.clearDataSource();
        }
    }
}

2.5、多数据源配置

package com.yg.springboot.common.config;

import com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceBuilder;
import com.yg.springboot.common.DatabaseType;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.stereotype.Component;

import javax.sql.DataSource;
import java.util.HashMap;
import java.util.Map;

/**
 * @author yuguang
 * @date 2020/4/22 9:52
 * @desc 多数据源配置
 */
@Configuration
@Component
public class DynamicDataSourceConfig {

    @Bean(value = "master")
    @ConfigurationProperties(prefix = "spring.datasource.druid.master")
    public DataSource xiaobinMasterDataSource() {
        return DruidDataSourceBuilder.create().build();
    }

    @Bean(value = "slave")
    @ConfigurationProperties(prefix = "spring.datasource.druid.slave")
    public DataSource xiaobinSlaveDataSource() {
        return DruidDataSourceBuilder.create().build();
    }

    @Bean
    @Primary
    public DynamicDataSource dataSource(@Qualifier(value = "master") DataSource masterDataSource,
                                        @Qualifier(value = "slave")DataSource slaveDataSource) {
        Map<Object, Object> targetDataSources = new HashMap<>();
        targetDataSources.put(DatabaseType.master.toString(), masterDataSource);
        targetDataSources.put(DatabaseType.slave.toString(), slaveDataSource);

        return new DynamicDataSource(masterDataSource, targetDataSources);
    }
}

2.6动态选择数据源和配置默认数据源

package com.yg.springboot.common.config;

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

import javax.sql.DataSource;
import java.util.Map;

/**
 * @author yuguang
 * @date 2020/4/4 10:38
 * @desc 数据源动态选择
 */
public class DynamicDataSource extends AbstractRoutingDataSource {

    private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();

    public DynamicDataSource(DataSource defaultTargetDataSource, Map<Object, Object> targetDataSources) {
        super.setDefaultTargetDataSource(defaultTargetDataSource);
        super.setTargetDataSources(targetDataSources);
        super.afterPropertiesSet();
    }

    //返回当前数据源
    @Override
    protected Object determineCurrentLookupKey() {
        return getDataSource();
    }

    public static void setDataSource(String dataSource) {
        contextHolder.set(dataSource);
    }

    public static String getDataSource() {
        return contextHolder.get();
    }

    public static void clearDataSource() {
        contextHolder.remove();
    }
}

2.7、枚举

/**
 * @author yuguang
 * @date 2020/4/4 10:12
 * @desc 数据库类型
 */
public enum DatabaseType {
    master,
    slave
}

2.8、实体类

@Data
public class User {

    private Long id;
    private String name;
}

2.9、mapper


@Repository
public interface UserMapper {

    @DataSource(name = "master")
    List<User> queryAllWithMaster();

    @DataSource(name = "slave")
    List<User> queryAllWithSlave();
}

2.10、mapper.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.yg.springboot.mapper.UserMapper">

    <select id="queryAllWithMaster" resultType="com.yg.springboot.domain.User">
        select * from user;
    </select>

    <select id="queryAllWithSlave" resultType="com.yg.springboot.domain.User">
        select * from user;
    </select>

</mapper>

2.11接口

  /*
    //注解方法，注解类都可以
    http://127.0.0.1:8080/master1/list
    http://127.0.0.1:8080/master/list
    http://127.0.0.1:8080/slave/list
     */
    @GetMapping("/{name}/list")
    public List<User> list(@PathVariable("name") String name) {

        if (name.equals("master1")) {
            return user1Mapper.queryAllWithMaster();
        } else if (name.equals("master")) {
            return userMapper.queryAllWithMaster();
        } else {
            return userMapper.queryAllWithSlave();
        }
    }

2.12、启动类

@SpringBootApplication(exclude= {DataSourceAutoConfiguration.class})
@MapperScan(basePackages = "com.yg.springboot")
@Import({DynamicDataSourceConfig.class})
public class DataSourceApplication extends SpringBootServletInitializer {


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

    @Override
    protected SpringApplicationBuilder configure(SpringApplicationBuilder application) {
        return application.sources(DataSourceApplication.class);
    }

}

3、启动测试结果

1、master

2、slave

于光yuguang

关注

1
点赞
踩
2

收藏

觉得还不错? 一键收藏
0
评论
springboot+mybatis+druid+mysql实现多数据源，通过注解的方式控制数据源

1、创建数据库和测试表1.1、创建数据库1.2、创建数据表在两个数据库中都添加user表CREATE TABLE `user` ( `id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT '主键,其它表中的使用user_id', `name` varchar(100) NOT NULL DEFAULT '', PRIMARY...
复制链接

扫一扫

专栏目录