【Java实战】Mybatis plus根据数据库表自动生成代码

news2025/1/10 3:41:57

首先生成一个maven项目，不会生成的参考【Java实战】创建第一个Springboot项目Hello worldhttps://blog.csdn.net/forgetmiss/article/details/133889733

（写的过程中遇到过的问题：mybatis-plus 生成controller中的代码， RequestMapping除了双斜杠问题@RequestMapping("//article")。解决方法： pc.setModuleName(null) ）

1、pom.xml中添加依赖。

        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-boot-starter</artifactId>
            <version>3.3.2</version>
        </dependency>
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <version>8.0.23</version> <!-- 使用适合你的 MySQL 版本的驱动版本 -->
        </dependency>
        <dependency>
            <groupId>org.freemarker</groupId>
            <artifactId>freemarker</artifactId>
            <version>2.3.31</version> <!-- 使用适合你的 Freemarker 版本 -->
        </dependency>

        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-generator</artifactId>
            <version>3.3.2</version>
        </dependency>

2、新建一个CodeGenerator.java文件

package org.example.learningExample.generate;

import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;

import java.util.Scanner;

public class CodeGenerator {

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        System.out.println("请输入要生成代码的表名，多个表名使用逗号分隔：");
        String tablesInput = scanner.nextLine();
        String[] tables = tablesInput.split(",");
        AutoGenerator mpg = new AutoGenerator();
        mpg.setTemplateEngine(new FreemarkerTemplateEngine());

        GlobalConfig gc = new GlobalConfig();
        gc.setAuthor("LiangFeng"); // 作者信息：欢迎关注程序员良枫
        gc.setOutputDir(System.getProperty("user.dir") + "/src/main/java");//设置生成的路径
        gc.setServiceName("%sService");//去掉服务接口service前缀
        gc.setFileOverride(false);//是否覆盖文件。如果代码直接生成到生产代码里面，请谨慎启用，避免覆盖掉项目代码。
        gc.setOpen(false);//是否打开
        mpg.setGlobalConfig(gc);

        // 数据源配置
        DataSourceConfig dsc = new DataSourceConfig();
        //改成你自己的数据库
        dsc.setUrl("jdbc:mysql://localhost:3306/test?useUnicode=true&useSSL=false&characterEncoding=utf8");
        dsc.setDriverName("com.mysql.cj.jdbc.Driver");
        dsc.setUsername("root");//mysql用户名
        dsc.setPassword("123456");//mysql密码
        mpg.setDataSource(dsc);

        // 包配置
        PackageConfig pc = new PackageConfig();
        pc.setModuleName(null); //必须设置，不然requestMapping会出现双斜杠问题。
        pc.setParent("org.example.learningExample");//包配置
        pc.setController("controller"); //controller
        pc.setService("service"); //服务接口
        pc.setServiceImpl("service.impl"); //服务实现
        pc.setMapper("mapper"); //mapper
        pc.setEntity("entity"); //实体
        mpg.setPackageInfo(pc);

        // 策略配置
        StrategyConfig strategy = new StrategyConfig();
        strategy.setNaming(NamingStrategy.underline_to_camel);
        strategy.setColumnNaming(NamingStrategy.underline_to_camel);
        strategy.setEntityLombokModel(true);
        strategy.setRestControllerStyle(true);
        strategy.setInclude(tables); // 从命令行参数读取表名
        mpg.setStrategy(strategy);

        mpg.execute();
    }

}