SpringBoot + MyBatisPlus 实战：代码生成器与多数据源配置

于 2025-03-04 09:05:02 发布
阅读量892 收藏 15
点赞数 13
分类专栏： Web 文章标签： spring boot mybatis 后端
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/ashyyyy/article/details/146005217
版权

🌟博主介绍：Java、Python、js全栈开发 “多面手”，精通多种编程语言和技术，痴迷于人工智能领域。秉持着对技术的热爱与执着，持续探索创新，愿在此分享交流和学习，与大家共进步。
📖全栈开发环境搭建运行攻略：多语言一站式指南(环境搭建+运行+调试+发布+保姆级详解)
👉感兴趣的可以先收藏起来，希望帮助更多的人
在这里插入图片描述

SpringBoot + MyBatisPlus 实战：代码生成器与多数据源配置

一、引言

在现代的 Java 开发中，Spring Boot 凭借其简化配置、快速开发的特性，成为了众多开发者的首选框架。而 MyBatis-Plus 作为 MyBatis 的增强工具，进一步提升了数据访问层的开发效率。本文将详细介绍如何使用 Spring Boot 和 MyBatis-Plus 实现代码生成器以及多数据源配置，帮助开发者更高效地完成项目开发。

二、环境准备

2.1 开发工具

推荐使用 IntelliJ IDEA 作为开发工具，它提供了丰富的插件和强大的代码编辑功能，能极大地提高开发效率。

2.2 依赖管理

创建一个 Spring Boot 项目，可以使用 Spring Initializr（https://start.spring.io/）来快速生成项目骨架。在 pom.xml 中添加以下依赖：

<dependencies>
    <!-- Spring Boot Starter Web -->
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-web</artifactId>
    </dependency>
    <!-- MyBatis-Plus Starter -->
    <dependency>
        <groupId>com.baomidou</groupId>
        <artifactId>mybatis-plus-boot-starter</artifactId>
        <version>3.4.3.4</version>
    </dependency>
    <!-- MySQL 驱动 -->
    <dependency>
        <groupId>mysql</groupId>
        <artifactId>mysql-connector-java</artifactId>
    </dependency>
    <!-- MyBatis-Plus 代码生成器 -->
    <dependency>
        <groupId>com.baomidou</groupId>
        <artifactId>mybatis-plus-generator</artifactId>
        <version>3.4.3.4</version>
    </dependency>
    <!-- Freemarker 模板引擎 -->
    <dependency>
        <groupId>org.freemarker</groupId>
        <artifactId>freemarker</artifactId>
    </dependency>
</dependencies>

三、MyBatis-Plus 代码生成器的使用

3.1 配置代码生成器

创建一个代码生成器配置类，以下是一个示例：

import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;

import java.util.Collections;

public class CodeGenerator {
    public static void main(String[] args) {
        FastAutoGenerator.create("jdbc:mysql://localhost:3306/your_database", "your_username", "your_password")
                .globalConfig(builder -> {
                    builder.author("your_name") // 设置作者
                            .outputDir("D://project/src/main/java"); // 指定输出目录
                })
                .packageConfig(builder -> {
                    builder.parent("com.example.demo") // 设置父包名
                            .moduleName("module") // 设置模块名
                            .pathInfo(Collections.singletonMap(OutputFile.xml, "D://project/src/main/resources/mapper")); // 设置mapperXml生成路径
                })
                .strategyConfig(builder -> {
                    builder.addInclude("your_table_name") // 设置需要生成的表名
                            .addTablePrefix("t_", "c_"); // 设置过滤表前缀
                })
                .templateEngine(new FreemarkerTemplateEngine()) // 使用Freemarker引擎模板，默认的是Velocity引擎模板
                .execute();
    }
}

3.2 代码生成器详解

  • 全局配置：设置作者信息、输出目录等。
  • 包配置：设置父包名、模块名以及 mapperXml 的生成路径。
  • 策略配置：指定需要生成的表名，可以通过 addTablePrefix 过滤表前缀。
  • 模板引擎：这里使用 Freemarker 模板引擎，也可以使用默认的 Velocity 引擎。

3.3 运行代码生成器

运行 CodeGenerator 类的 main 方法，代码生成器会根据配置自动生成实体类、Mapper 接口、Service 接口和实现类以及 Controller 类。

四、多数据源配置

4.1 配置数据源信息

在 application.properties 中配置多个数据源信息：

# 主数据源
spring.datasource.ds1.url=jdbc:mysql://localhost:3306/db1
spring.datasource.ds1.username=root
spring.datasource.ds1.password=root
spring.datasource.ds1.driver-class-name=com.mysql.cj.jdbc.Driver

# 从数据源
spring.datasource.ds2.url=jdbc:mysql://localhost:3306/db2
spring.datasource.ds2.username=root
spring.datasource.ds2.password=root
spring.datasource.ds2.driver-class-name=com.mysql.cj.jdbc.Driver

4.2 创建数据源配置类

创建一个数据源配置类，用于配置多个数据源：

import com.zaxxer.hikari.HikariDataSource;
import org.springframework.beans.factory.annotation.Qualifier;
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 DataSourceConfig {

    @Primary
    @Bean(name = "ds1")
    @ConfigurationProperties(prefix = "spring.datasource.ds1")
    public DataSource dataSource1() {
        return DataSourceBuilder.create().type(HikariDataSource.class).build();
    }

    @Bean(name = "ds2")
    @ConfigurationProperties(prefix = "spring.datasource.ds2")
    public DataSource dataSource2() {
        return DataSourceBuilder.create().type(HikariDataSource.class).build();
    }
}

4.3 创建动态数据源路由

创建一个动态数据源路由类，用于根据不同的数据源标识切换数据源：

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

import java.util.HashMap;
import java.util.Map;

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

4.4 配置动态数据源

在数据源配置类中配置动态数据源：

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

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

@Configuration
public class DataSourceConfig {

    @Bean(name = "ds1")
    @ConfigurationProperties(prefix = "spring.datasource.ds1")
    public DataSource dataSource1() {
        return DataSourceBuilder.create().type(HikariDataSource.class).build();
    }

    @Bean(name = "ds2")
    @ConfigurationProperties(prefix = "spring.datasource.ds2")
    public DataSource dataSource2() {
        return DataSourceBuilder.create().type(HikariDataSource.class).build();
    }

    @Bean
    public AbstractRoutingDataSource dynamicDataSource() {
        DynamicDataSource dynamicDataSource = new DynamicDataSource(dataSource1(), new HashMap<>());
        Map<Object, Object> targetDataSources = new HashMap<>();
        targetDataSources.put("ds1", dataSource1());
        targetDataSources.put("ds2", dataSource2());
        dynamicDataSource.setDefaultTargetDataSource(dataSource1());
        dynamicDataSource.setTargetDataSources(targetDataSources);
        return dynamicDataSource;
    }
}

4.5 使用动态数据源

在需要切换数据源的方法上使用 DynamicDataSource.setDataSource("ds2") 来切换到从数据源，使用完后调用 DynamicDataSource.clearDataSource() 清除数据源标识。

import org.springframework.stereotype.Service;

@Service
public class UserService {

    public void doSomethingWithDs2() {
        DynamicDataSource.setDataSource("ds2");
        try {
            // 执行从数据源的操作
        } finally {
            DynamicDataSource.clearDataSource();
        }
    }
}

五、总结

通过本文的介绍，我们学习了如何使用 Spring Boot 和 MyBatis-Plus 的代码生成器来快速生成项目代码，以及如何配置多数据源实现数据的多源访问。代码生成器可以帮助我们减少重复代码的编写，提高开发效率；多数据源配置则可以满足项目中对不同数据源的访问需求。

评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

fanxbl957

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值