MyBatis-Plust + kotlin 自动生成代码（新）

最新推荐文章于 2024-03-20 15:38:32 发布

先前看

最新推荐文章于 2024-03-20 15:38:32 发布

阅读量586

点赞数

分类专栏： kotlin gradle 文章标签： mybatis kotlin java

本文链接：https://blog.csdn.net/wo_wo_wo1/article/details/127392811

版权

kotlin 同时被 2 个专栏收录

1 篇文章 0 订阅

订阅专栏

gradle

1 篇文章 0 订阅

订阅专栏

MyBatis-Plus 自动生成代码

概要

需要使用的依赖：主要是使用官网Generator包生成简单的代码，官网没有kotlin版本的示例
可以参考官网配置新增自己想要的属性（这里是新的代码生成器）
官网教程指南以及新（旧）
两版java生成器 https://baomidou.com/pages/779a6e/#%E5%BF%AB%E9%80%9F%E5%85%A5%E9%97%A8
官网代码生成器配置细节https://baomidou.com/pages/981406/#%E6%95%B0%E6%8D%AE%E5%BA%93%E9%85%8D%E7%BD%AE-datasourceconfig

代码

package com.duckstone.userapi.generator

import com.baomidou.mybatisplus.generator.FastAutoGenerator
import com.baomidou.mybatisplus.generator.config.OutputFile
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine
import java.util.*
import java.util.function.BiConsumer
import java.util.function.Consumer


fun main(args: Array<String>) {
    generator()
}

// kotlin 代码生成器
fun generator() {

    //快速构建 生成mybatis代码
    FastAutoGenerator.create("url", "username",
        "password")
        //全局配置
        .globalConfig(Consumer {
            it.author("zhangsan").enableKotlin()
                .disableOpenDir()
                //开启 swagger 模式: 需要引入swagger包
                //.enableSwagger()
                // 覆盖已生成文件 3.5.3 版本已删除 使用的是 3.5.2
                .fileOverride()
                .outputDir("E:\\new_pztWorkpace\\springboot-kotlin\\user-api\\src\\main\\kotlin")
        })
        //包配置
        .packageConfig(Consumer {
            //设置父包名
            it.parent("com.duckstone.userapi")
                //.moduleName("system") //设置父模块名称
                //mapperXml 生成路径 绝对路径
                .pathInfo(Collections.singletonMap(OutputFile.xml, "E:\\new_pztWorkpace\\springboot-kotlin\\user-api\\src\\main\\resources\\userapi"))
        })
        // 策略配置
        /*.strategyConfig(Consumer {
            // 生成表， 可以另写一个使用Scanner 接入一串输入的表
            it.addInclude("ts_user_role", "ts_menu", "ts_role", "ts_role_menu", "ts_user")
                //设置过滤表前缀
                //.addFieldPrefix("ts")
                // 设置实体策略
                .entityBuilder()
                //开启 lombok
                .enableLombok()

        })*/
            //交互手动输入
        .strategyConfig(BiConsumer { scanner, builder ->
            builder.addInclude(getTables(scanner.apply("请输入表名，多个英文逗号分隔？所有输入 all"))!!)
                .entityBuilder().enableLombok().build()
        })
        //使用Freemarker引擎模板，默认的是Velocity引擎模板
        .templateEngine(FreemarkerTemplateEngine())
        // 执行
        .execute()
}

// 处理 all 情况
fun getTables(tables: String): List<String?>? {
    return if ("all" == tables) Collections.emptyList() else tables.split(",")
}