代码生成器

一、引入相关依赖

<dependency>
    <groupId>com.mybatis-flex</groupId>
    <artifactId>mybatis-flex-codegen</artifactId>
    <version>1.8.3</version>
</dependency>
<dependency>
    <groupId>com.zaxxer</groupId>
    <artifactId>HikariCP</artifactId>
    <version>4.0.3</version>
</dependency>

<dependency>
    <groupId>com.mysql</groupId>
    <artifactId>mysql-connector-j</artifactId>
    <version>8.0.32</version>
</dependency>

二、编写测试类/main函数

package com.tireba.wx.mall;

import com.mybatisflex.codegen.Generator;
import com.mybatisflex.codegen.config.GlobalConfig;
import com.zaxxer.hikari.HikariDataSource;
import org.junit.jupiter.api.Test;

/**
 * @Author Tireba
 * @Date 2024/4/7
 **/
public class CodeGenTest {

    @Test
    public void genCode() {
        //配置数据源
        HikariDataSource dataSource = new HikariDataSource();
        dataSource.setJdbcUrl("jdbc:mysql://localhost:3306/tireba");
        dataSource.setUsername("root");
        dataSource.setPassword("root");
        GlobalConfig globalConfig = createGlobalConfigUseStyle1();
        //通过 datasource 和 globalConfig 创建代码生成器
        Generator generator = new Generator(dataSource, globalConfig);
        //生成代码
        generator.generate();
    }

    public static GlobalConfig createGlobalConfigUseStyle1() {
        //创建配置内容
        GlobalConfig globalConfig = new GlobalConfig();
        globalConfig.setBasePackage("com.tireba.wx.mall");

        //设置根包
        globalConfig.setBasePackage("com.tireba.wx.mall");
        globalConfig.getJavadocConfig().setAuthor("Tireba");

        //设置表前缀和只生成哪些表
        globalConfig.setTablePrefix("t_");
        globalConfig.setGenerateTable("t_record");

        globalConfig.enableController().setOverwriteEnable(false);
        globalConfig.enableEntity().setOverwriteEnable(true).setWithLombok(true);
        globalConfig.enableService().setOverwriteEnable(false);
        globalConfig.enableServiceImpl().setOverwriteEnable(false);

        return globalConfig;
    }
}