基于注解实现SpringBoot多数据源配置

1.功能介绍

在实际的开发中,同一个项目中使用多个数据源是很常见的场景。最近在学习的过程中使用注解的方式实现了一个Springboot项目多数据源的功能。具体实现方式如下。

2.在application.properties中添加多数据源配置

添加多个数据源和mapper文件路径配置,此配置用于基于java的配置数据源中使用。

#数据库配置
spring.datasource.demo.user.url=jdbc:mysql://xxx.xx.xx.xx:3306/demo-user
spring.datasource.demo.user.username=xxxx
spring.datasource.demo.user.password=xxxx
spring.datasource.demo.user.driver-class-name=com.mysql.jdbc.Driver

spring.datasource.demo.server.url=jdbc:mysql://xxx.xx.xx.xx:3306/springbootdemo
spring.datasource.demo.server.username=xxxx
spring.datasource.demo.server.password=xxxx
spring.datasource.demo.server.driver-class-name=com.mysql.jdbc.Driver

#mapper文件的路径
mybatis.demo.server.mapper-location=classpath*:mapper/demo-server/*.xml
mybatis.demo.user.mapper-location=classpath*:mapper/demo-user/*.xml

3.基于java的方式实现数据库配置

配置类图如下:

其中DemoUserDbConfig类源代码如下: 其中Configuration注解表识此类为Spring的配置类。 MapperScan注解中的basePackages、annotationClass、sqlSessionTemplateRef用于配置此数据库链接扫描com.example包中所有注解为DemoUserMapper的接口。

@Configuration
@MapperScan(basePackages = {"com.example"},annotationClass = DemoUserMapper.class,
       sqlSessionTemplateRef = "demoUserTemplate")
public class DemoUserDbConfig extends AbstractDbConfig {

    @Value("${spring.datasource.demo.user.url}")
    private String url;

    @Value("${spring.datasource.demo.user.username}")
    private String userName;

    @Value("${spring.datasource.demo.user.password}")
    private String password;

    @Value("${spring.datasource.demo.user.driver-class-name}")
    private String driveClassName;

    @Value(value = "${mybatis.demo.user.mapper-location}")
    private String mapperLocation;


    @Bean(name = "demoUser")
    public DataSource secondaryDataSource() {
        return dataSourceFactory(driveClassName, url, userName, password);
    }

    @Bean(name = "demoUserTemplate")
    public SqlSessionTemplate demoUserSqlTemplate() throws Exception {
        return new SqlSessionTemplate((sqlSessionFactory(secondaryDataSource(), mapperLocation)));
    }

    @Bean
    @Qualifier("demoUserTransaction")
    public PlatformTransactionManager demoUserTransaction() {
        return new DataSourceTransactionManager(secondaryDataSource());
    }
}

其中AbstractDatasource设置了通用的方法,源代码如下:

public abstract class AbstractDbConfig {

    protected SqlSessionFactory sqlSessionFactory(DataSource dataSource, String mapperLocation) throws Exception {
        SqlSessionFactoryBean factoryBean = new SqlSessionFactoryBean();
        factoryBean.setDataSource(dataSource);
        ResourcePatternResolver resourceResolver = new PathMatchingResourcePatternResolver();
        Resource[] resource= resourceResolver.getResources(mapperLocation);
        factoryBean.setMapperLocations(resource);
        return factoryBean.getObject();
    }

    protected DataSource dataSourceFactory(String driveClassName, String url, String userName, String password){
        DruidDataSource datasource = new DruidDataSource();
        datasource.setDriverClassName(driveClassName);
        datasource.setUrl(url);
        datasource.setUsername(userName);
        datasource.setPassword(password);
        datasource.setMaxActive(20);
        datasource.setInitialSize(20);
        return datasource;
    }
}

使用相同的方法定义其他数据源。

4.定义接口和mapper文件

分别定义连接demo-user数据库和springbootdemo数据库的mapper文件。连接demo-user数据库的接口如下,使用DemoUserMapper注解表识。

@DemoUserMapper
public interface UserDao {

    /**
     * 返回所有的dictionary列表
     *
     * @return 所有的dictionary列表
     */
    List<String> list();

}

mapper文件如下:

<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd" >
<mapper namespace="com.example.multidatasource.business.user.UserDao">

    <select id="list" resultType="string">
        SELECT `name` FROM `user`
    </select>
</mapper>

定义读取springbootdemo数据库的接口,代码如下。使用DemoServerMapper注解表识

@DemoServerMapper
public interface DictionaryDao {

    /**
     * 返回所有的dictionary列表
     *
     * @return 所有的dictionary列表
     */
    List<Dictionary> list();

    /**
     * 查询此key下的所有子节点
     *
     * @param key 数据字典key
     * @return 返回key所有的子节点列表
     */
    List<Dictionary> listChildrenByKey(String key);

    /**
     * 插入数据到数据库
     *
     * @param dictionary
     */
    void insert(Dictionary dictionary);

}

mapper文件代码如下:

<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd" >
<mapper namespace="com.example.multidatasource.business.dictionary.dao.DictionaryDao">

    <resultMap id="DictionaryResultMap" type="com.example.multidatasource.business.dictionary.Dictionary">
        <result property="id" column="id"></result>
        <result property="dictKey" column="dict_key"></result>
        <result property="dictValue" column="dict_value"></result>
        <result property="parentId" column="parent_id"></result>
        <result property="description" column="description"></result>
    </resultMap>

    <select id="list" resultMap="DictionaryResultMap">
        SELECT * FROM `dictionary`
    </select>

    <select id="listChildrenByKey" resultMap="DictionaryResultMap">
        SELECT * FROM dictionary where parent_id= (select id from dictionary where dict_key= #{key})
    </select>

    <delete id="delete" parameterType="int">
        delete from dictionary where id = #{id}
    </delete>

    <insert id="insert" parameterType="com.example.multidatasource.business.dictionary.Dictionary">
        INSERT INTO `dictionary`(`dict_key`,`dict_value`,`parent_id`,`description`)
        VALUES(#{dictKey}, #{dictValue}, #{parentId}, #{description})
    </insert>
</mapper>

5.定义注解

定义DemoUserMapper和DemoServerMapper注解,分别作为使用demo-user和springbootdemo数据库的表识。 定义代码如下:

@Documented
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.TYPE)
@Component
@Mapper
public @interface DemoServerMapper {

    /**
     * The value may indicate a suggestion for a logical component name,
     * to be turned into a Spring bean in case of an autodetected component.
     * @return the suggested component name, if any (or empty String otherwise)
     */
    String value() default "";
}

@Documented
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.TYPE)
@Component
@Mapper
public @interface DemoUserMapper {

    /**
     * The value may indicate a suggestion for a logical component name,
     * to be turned into a Spring bean in case of an autodetected component.
     * @return the suggested component name, if any (or empty String otherwise)
     */
    String value() default "";
}

6.使用单元测试验证配置

编写单元测试代码如下:

@RunWith(SpringRunner.class)
@SpringBootTest
public class MultiDatasourceApplicationTests {

    @Autowired
    private DictionaryDao dictionaryDao;

    @Autowired
    private UserDao userDao;

    @Test
    public void contextLoads() {
        System.out.println(dictionaryDao.list());
        System.out.println("===================");
        System.out.println(userDao.list());
    }

}

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏技术记录

JAVA-FTP批量大文件传输

FTP的具体使用      FTP是一种网络协议,用于进行不同服务器主机之间的文件传输,或者简单地说两台不同IP的机器之间的文件传输。在java中我们什么时候需...

5426
来自专栏Java成神之路

Spring_总结_03_装配Bean(一)之自动装配

(2)当必须要显示配置的时候,再使用类型安全并且比XML更强大的JavaConfig

852
来自专栏一个会写诗的程序员的博客

Springboot使用JPA操作数据库第七章 使用JPA操作数据库

本章主要介绍如何在Spring Boot的Web应用中使用Mysq数据库,也充分展示Spring Boot的优势(尽可能少的代码和配置).

723
来自专栏不想当开发的产品不是好测试

java 执行sql文件

# 背景 用例执行完毕,期望回滚数据,因此希望执行sql来回滚数据 # 步骤 直接show代码,借助的是mybatis的ScriptRunner /** ...

3228
来自专栏一个会写诗的程序员的博客

sqlSessionFactoryBean.setMapperLocations 通配符 在配置代码中怎样使用

sqlSessionFactoryBean.setMapperLocations 通配符 在配置代码中怎样使用

944
来自专栏Java帮帮-微信公众号-技术文章全总结

struts2+spring+hibernate整合步骤(1)

struts2、hibernate、spring所需jar包 struts-core-2.x.x.jar ----struts核心包 xwork-cor...

3244
来自专栏有刻

Java 小记 — Spring Boot 注解

33813
来自专栏菩提树下的杨过

JAVA CDI 学习(1) - @Inject基本用法

CDI(Contexts and Dependency Injection 上下文依赖注入),是JAVA官方提供的依赖注入实现,可用于Dynamic Web M...

852
来自专栏GIS讲堂

shape文件的生成与打包下载

632
来自专栏Java帮帮-微信公众号-技术文章全总结

Spring-拾遗

Junit集成 前面多次用到@RunWith与@ContextConfiguration,在测试类添加这两个注解,程序就会自动加载spring配置并初始化Spr...

2835

扫码关注云+社区