<dependency>
    <groupId>com.mybatis-flex</groupId>
    <artifactId>mybatis-flex-spring-boot3-starter</artifactId>
    <version>1.11.0</version>
</dependency>

<!-- 代碼生成模塊 -->
<dependency>
    <groupId>com.mybatis-flex</groupId>
    <artifactId>mybatis-flex-codegen</artifactId>
    <version>1.11.0</version>
</dependency>
<!-- 數(shù)據(jù)庫連接池 -->
<dependency>
    <groupId>com.zaxxer</groupId>
    <artifactId>HikariCP</artifactId>
</dependency>

public class MyBatisCodeGenerator {

    // 需要生成的表名
    private static final String[] TABLE_NAMES = {"user"};

    public static void main(String[] args) {
        // 獲取數(shù)據(jù)源信息
        Dict dict = YamlUtil.loadByPath("application.yml");
        Map<String, Object> dataSourceConfig = dict.getByPath("spring.datasource");
        String url = String.valueOf(dataSourceConfig.get("url"));
        String username = String.valueOf(dataSourceConfig.get("username"));
        String password = String.valueOf(dataSourceConfig.get("password"));
        // 配置數(shù)據(jù)源
        HikariDataSource dataSource = new HikariDataSource();
        dataSource.setJdbcUrl(url);
        dataSource.setUsername(username);
        dataSource.setPassword(password);

        // 創(chuàng)建配置內(nèi)容
        GlobalConfig globalConfig = createGlobalConfig();

        // 通過 datasource 和 globalConfig 創(chuàng)建代碼生成器
        Generator generator = new Generator(dataSource, globalConfig);

        // 生成代碼
        generator.generate();
    }

    // 詳細配置見：https://mybatis-flex.com/zh/others/codegen.html
    public static GlobalConfig createGlobalConfig() {
        // 創(chuàng)建配置內(nèi)容
        GlobalConfig globalConfig = new GlobalConfig();

        // 設(shè)置根包，建議先生成到一個臨時目錄下，生成代碼后，再移動到項目目錄下
        globalConfig.getPackageConfig()
                .setBasePackage("com.yupi.yuaicodemother.genresult");

        // 設(shè)置表前綴和只生成哪些表，setGenerateTable 未配置時，生成所有表
        globalConfig.getStrategyConfig()
                .setGenerateTable(TABLE_NAMES)
                // 設(shè)置邏輯刪除的默認字段名稱
                .setLogicDeleteColumn("isDelete");

        // 設(shè)置生成 entity 并啟用 Lombok
        globalConfig.enableEntity()
                .setWithLombok(true)
                .setJdkVersion(21);

        // 設(shè)置生成 mapper
        globalConfig.enableMapper();
        globalConfig.enableMapperXml();

        // 設(shè)置生成 service
        globalConfig.enableService();
        globalConfig.enableServiceImpl();

        // 設(shè)置生成 controller
        globalConfig.enableController();

        // 設(shè)置生成時間和字符串為空，避免多余的代碼改動
        globalConfig.getJavadocConfig()
                .setAuthor("<a href=\"https://github.com/liyupi\">程序員魚皮</a>")
                .setSince("");
        return globalConfig;
    }
}