mybatis-plus自动生成代码(整理版)

Springboot集成请参考上篇博文:Springboot集成mybatis-plus自动生成代码

一、CodeGenerator.java

整理版,添加了注释模版,常用基础方法

package com.hng.business;

import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.core.exceptions.MybatisPlusException;
import com.baomidou.mybatisplus.core.toolkit.StringPool;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.InjectionConfig;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.config.po.TableFill;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;

import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

/**
 * @Author 郝南过
 * @Date 2023/11/6 16:30
 * @Version 1.0
 * mybatis-plus代码生成器官网地址 https://baomidou.com/pages/d357af/
 */
public class CodeGenerator {

    private static final String Author = "郝南过";
    /**数据库连接字段配置**/
    private static final String JDBC_URL = "jdbc:mysql://139.198.179.225:3306/hng?useUnicode=true&useSSL=false&characterEncoding=utf8";
    private static final String JDBC_USER_NAME = "root";
    private static final String JDBC_PASSWORD = "@1Verizon";
    /**包名和模块名配置**/
    private static final String PACKAGE_NAME = "com.hng"; //包名
    private static final String MODULE_NAME = "business"; //模块名
    /**相关路径配置**/
    private static final String projectPath = System.getProperty("user.dir");//项目地址
    private static final String OUT_FILE_PATH =projectPath +"/"+MODULE_NAME+"/src/main/java";//生成文件存放路径(可根据需要指定生成到某个模块下)
    /**相关模式开关**/
    private static final boolean IF_RECOVER_ENABLE = true; //再次调用时是否覆盖原文件
    private static final boolean IF_OPEN_DIRECTORY = false; //是否打开输出目录 默认值:true
    private static final boolean IF_SWAGGER2 = true; //是否开启swagger2模式 实体属性 Swagger2 注解
    private static final boolean IF_BASE_COLUMN_LIST = false; //开启 baseColumnList (通用查询结果列),默认false,开启后mapper.xml文件中将生成对应代码
    private static final boolean IF_BASE_COLUMN_MAP = false; //开启 BaseResultMap (通用查询映射结果),默认false,开启后mapper.xml文件中将生成对应代码
    private static final boolean IF_ACTIVE_RECORD = true; //开启ActiveRecord模式,默认false,参考:https://blog.csdn.net/qq_31762741/article/details/120392656
    private static final boolean IF_ENABLE_CACHE = false; // 是否在xml中添加二级缓存配置,默认false
    /**表设置**/
    private static final String TABLE_PREFIX = "_";  //表名的前缀,从表生成代码时会去掉前缀,没有前缀就只写_
    /**模版设置**/
    private static final  String templatePath = "/templates/mapper.xml.ftl";// 如果模板引擎是 freemarker
    //private static final String templatePath = "/templates/mapper.xml.vm";// 如果模板引擎是 velocity
    private static final  String Controller_template = "templates/vm/controller.java";
    private static final  String Service_template = "templates/vm/service.java";
    private static final  String ServiceImpl_template = "templates/vm/serviceImpl.java";

    /**
     * <p>
     * 读取控制台内容 控制台输入模块表名回车自动生成对应项目目录中
     * </p>
     */
    public static String scanner(String tip) {
        Scanner scanner = new Scanner(System.in);
        StringBuilder help = new StringBuilder();
        help.append("请输入" + tip + ":");
        System.out.println(help.toString());
        if (scanner.hasNext()) {
            String ipt = scanner.next();
            if (StringUtils.isNotBlank(ipt)) {
                return ipt;
            }
        }
        throw new MybatisPlusException("请输入正确的" + tip + "!");
    }

    /**
     * 主方法,包括(1.全局配置、2.数据源配置、3.包配置、4.自定义配置、5.模版配置、6.策略配置)
     * @param args
     */
    public static void main(String[] args) {
        //代码生成器
        AutoGenerator mpg = new AutoGenerator();
        //全局配置
        GlobalConfig gc = new GlobalConfig();
        GlobalConfig(gc,mpg);
        //数据源配置
        DataSourceConfig dsc = new DataSourceConfig();
        DataSourceConfig(dsc,mpg);
        //包配置
        PackageConfig pc = new PackageConfig();
        PackageConfig(pc,mpg);
        //自定义配置
        InjectionConfig cfg = new InjectionConfig() {
            @Override
            public void initMap() {
                // to do nothing
            }
        };
        InjectionConfig(cfg,mpg);
        //模版配置
        TemplateConfig templateConfig = new TemplateConfig();
        TemplateConfig(templateConfig,mpg);
        //策略配置
        StrategyConfig strategy = new StrategyConfig();
        StrategyConfig(strategy,pc,mpg);
        mpg.setTemplateEngine(new FreemarkerTemplateEngine());
        mpg.execute();
    }

    // 全局配置
    private static AutoGenerator GlobalConfig(GlobalConfig gc,AutoGenerator mpg){
        gc.setFileOverride(IF_RECOVER_ENABLE);//再次调用时是否覆盖原文件
        gc.setOutputDir(OUT_FILE_PATH);
        gc.setAuthor(Author);
        gc.setOpen(IF_OPEN_DIRECTORY); //是否打开输出目录 默认值:true
        gc.setSwagger2(IF_SWAGGER2); //是否开启swagger2模式 实体属性 Swagger2 注解
        gc.setBaseColumnList(IF_BASE_COLUMN_LIST); //开启 baseColumnList (通用查询结果列),默认false
        gc.setBaseResultMap(IF_BASE_COLUMN_MAP); //开启 BaseResultMap (通用查询映射结果),默认false
        gc.setActiveRecord(IF_ACTIVE_RECORD); //开启ActiveRecord模式,默认false
        gc.setEnableCache(IF_ENABLE_CACHE); // 是否在xml中添加二级缓存配置,默认false
        gc.setIdType(IdType.AUTO);// AUTO(0)使用数据库id,NONE(1)不设置id生成策略,INPUT(2)用户手工输入id,ASSIGN_ID(3)雪花算法生成id(可兼容数值型与字符串型),ASSIGN_UUID(4)以UUID生成算法作为id生成策略
        // 自定义文件命名,注意 %s 会自动填充表实体属性(各层文件名称方式,例如:%sAction生成UserAction, %s为占位符)
        gc.setControllerName("%sController");
        gc.setServiceName("%sService");
        gc.setServiceImplName("%sServiceImpl");
        gc.setMapperName("%sMapper");
        gc.setXmlName("%sMapper");
        //gc.setEntityName("%sEntity");
        mpg.setGlobalConfig(gc);
        return mpg;
    }
    //数据源配置
    private static AutoGenerator DataSourceConfig(DataSourceConfig dsc,AutoGenerator mpg){
        dsc.setUrl(JDBC_URL);
        //dsc.setSchemaName("public");
        dsc.setDriverName("com.mysql.cj.jdbc.Driver");
        dsc.setUsername(JDBC_USER_NAME);
        dsc.setPassword(JDBC_PASSWORD);
        mpg.setDataSource(dsc);
        return mpg;
    }
    //包配置
    private static AutoGenerator PackageConfig(PackageConfig pc,AutoGenerator mpg){
        //pc.setModuleName(scanner("模块名"));
        pc.setModuleName(MODULE_NAME);
        //你的项目包结构名
        pc.setParent(PACKAGE_NAME);
        mpg.setPackageInfo(pc);
        return mpg;
    }
    //自定义配置
    private static AutoGenerator InjectionConfig(InjectionConfig cfg ,AutoGenerator mpg){
        // 自定义输出配置
        List<FileOutConfig> focList = new ArrayList<>();
        // 自定义配置会被优先输出
        focList.add(new FileOutConfig(templatePath) {
            @Override
            public String outputFile(TableInfo tableInfo) {
                // 自定义输出文件名 , 如果你 Entity 设置了前后缀、此处注意 xml 的名称会跟着发生变化
                // 注意此处要想生效,需要在模版设置中将XML设置为null,使默认的xml目录不再生成
                return projectPath +"/"+ MODULE_NAME + "/src/main/resources/mapper/"
                         + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
            }
        });
        /*
        cfg.setFileCreate(new IFileCreate() {
            @Override
            public boolean isCreate(ConfigBuilder configBuilder, FileType fileType, String filePath) {
                // 判断自定义文件夹是否需要创建
                checkDir("调用默认方法创建的目录,自定义目录用");
                if (fileType == FileType.MAPPER) {
                    // 已经生成 mapper 文件判断存在,不想重新生成返回 false
                    return !new File(filePath).exists();
                }
                // 允许生成模板文件
                return true;
            }
        });
        */
        cfg.setFileOutConfigList(focList);
        mpg.setCfg(cfg);
        return mpg;
    }
    //模版配置
    private static AutoGenerator TemplateConfig(TemplateConfig templateConfig,AutoGenerator mpg){
        // 配置自定义输出模板
        //指定自定义模板路径,注意不要带上.ftl/.vm, 会根据使用的模板引擎自动识别
        templateConfig.setController(Controller_template);
        templateConfig.setService(Service_template);
        templateConfig.setServiceImpl(ServiceImpl_template);
//        templateConfig.setMapper("templates/vm/mapper.java");
//        templateConfig.setEntity("templates/vm/entity.java");
//        templateConfig.setXml("templates/vm/mapper.xml");
        //使默认的xml目录不再生成,使用自定义的resource/mapper输出目录
        templateConfig.setXml(null);
        mpg.setTemplate(templateConfig);
        return mpg;
    }
    //策略配置
    private static AutoGenerator StrategyConfig(StrategyConfig strategy,PackageConfig pc,AutoGenerator mpg){
        //数据库表映射到实体的命名策略
        strategy.setNaming(NamingStrategy.underline_to_camel);
        //数据库表字段映射到实体的命名策略
        strategy.setColumnNaming(NamingStrategy.underline_to_camel);
        //strategy.setSuperEntityClass("你自己的父类实体,没有就不用设置!");
        strategy.setEntityLombokModel(true);
        strategy.setRestControllerStyle(true);//restful api风格控制器
        // 公共父类
        //strategy.setSuperControllerClass("你自己的父类控制器,没有就不用设置!");
        // 自动填充
        TableFill createTime = new TableFill("create_Time", FieldFill.INSERT);
        TableFill updateTime = new TableFill("update_Time",FieldFill.INSERT_UPDATE);
        ArrayList<TableFill> tableFills = new ArrayList<TableFill>();
        tableFills.add(createTime);
        tableFills.add(updateTime);
        strategy.setTableFillList(tableFills);

        // 写于父类中的公共字段
        // strategy.setSuperEntityColumns("id");
        strategy.setInclude(scanner("表名,多个英文逗号分割").split(","));
        strategy.setControllerMappingHyphenStyle(true);
        //生成实体类等时去掉表前缀
        strategy.setTablePrefix(TABLE_PREFIX);
        strategy.setTablePrefix(pc.getModuleName() + "_");
        mpg.setStrategy(strategy);
        return mpg;
    }

}

二、controller.java.ftl

仅提供基础方法,可根据具体需求自行改造

package ${package.Controller};
import com.hng.config.exception.error.ErrorEnum;
import org.springframework.beans.factory.annotation.Autowired;
import com.hng.config.response.Result;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import io.swagger.annotations.Api;
import lombok.extern.slf4j.Slf4j;


import ${package.Entity}.${entity};
import ${package.Service}.${table.serviceName};


/**
* <p>
   * ${table.comment!} 前端控制器
   * </p>
*/
@Slf4j
@Api(value = "${entity}", tags = "${entity}")
@RestController
@RequestMapping("<#if package.ModuleName?? && package.ModuleName != "">/${package.ModuleName}</#if>/<#if controllerMappingHyphenStyle>${controllerMappingHyphen}<#else>${table.entityPath}</#if>")
public class ${table.controllerName} {

   @Autowired
   private ${table.serviceName} ${table.serviceName?uncap_first};

   /**
   * 新增数据
   *
   * @param ${entity?uncap_first} 实体对象
   * @return Result
   */
   @PostMapping("add")
   @ResponseBody
   public Result<?> add(@Validated @RequestBody ${entity} ${entity?uncap_first}) {
      return ${table.serviceName?uncap_first}.add${entity}(${entity?uncap_first});
   }

   /**
   * 根据ID查询数据
   *
   * @param id ID
   * @return Result
   */
   @GetMapping("getById/{id}")
   @ResponseBody
   public Result<?> getById(@PathVariable Long id) {
      if(id!=null){
         return ${table.serviceName?uncap_first}.get${entity}ById(id);
      }
      return Result.Result(ErrorEnum.REQUEST_FAIL, "id为空");
   }

   /**
   * 更新数据
   *
   * @param ${entity?uncap_first} 实体对象
   * @return Result
   */
   @PostMapping("update")
   @ResponseBody
   public Result<?> update(@Validated @RequestBody ${entity} ${entity?uncap_first}) {
      if(${entity?uncap_first}.getId()!=null){
         return ${table.serviceName?uncap_first}.update${entity}(${entity?uncap_first});
      }
      return Result.Result(ErrorEnum.REQUEST_FAIL, "id为空");
   }

   /**
   * 删除数据
   *
   * @param id ID
   * @return Result
   */
   @PostMapping("delete/{id}")
   @ResponseBody
   public Result<?> delete(@PathVariable Long id) {
      if(id!=null){
         return ${table.serviceName?uncap_first}.delete${entity}(id);
      }
      return Result.Result(ErrorEnum.REQUEST_FAIL, "id为空");
   }

}

三、service.java.ftl

仅提供基础方法,可根据具体需求自行改造

package ${package.Service};
import com.hng.config.response.Result;
import ${package.Entity}.${entity};
import ${superServiceClassPackage};

/**
* <p>
    * ${table.comment!} 服务类
    * </p>
*/
public interface ${table.serviceName} extends ${superServiceClass}<${entity}> {

    /**
    * 新增数据
    *
    * @param ${"${entity}"} 实体对象
    * @return Result
    */
    Result<?> ${"add${entity}"}(${"${entity}"} ${"${entity?uncap_first}"});

    /**
    * 根据ID查询数据
    *
    * @param id ID
    * @return Result
    */
    Result<?> ${"get${entity}ById(Long id)"};

    /**
    * 更新数据
    *
    * @param ${entity?uncap_first} 实体对象
    * @return Result
    */
    Result<?> ${"update${entity}"}(${entity} ${entity?uncap_first});

    /**
    * 删除数据
    *
    * @param id ID
    * @return Result
    */
    Result<?> ${"delete${entity}"}(Long id);

}

四、serviceImpl.java.ftl

仅提供基础方法,可根据具体需求自行改造

package ${package.ServiceImpl};

import ${package.Entity}.${entity};
import ${package.Mapper}.${table.mapperName};
import ${package.Service}.${table.serviceName};
import com.hng.config.exception.error.ErrorEnum;
import com.hng.config.response.Result;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.bean.copier.CopyOptions;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.io.Serializable;

/**
* <p>
    * ${table.comment!} 服务实现类
    * </p>
*/
@Service
public class ${table.serviceImplName} extends ${superServiceImplClass}<${table.mapperName}, ${entity}> implements ${table.serviceName} {

    @Autowired
    private ${table.mapperName} ${table.mapperName?uncap_first};

    /**
    * 新增数据
    *
    * @param ${entity?uncap_first} 实体对象
    * @return Result
    */
    @Override
    public Result<?> add${entity}(${entity} ${entity?uncap_first}) {
<#--        //忽略不需要手动填入的属性插入,预防误填出错-->
<#--        CopyOptions options = CopyOptions.create()-->
<#--        .setIgnoreProperties("createTime").setIgnoreProperties("updateTime")-->
<#--        .setIgnoreProperties("createBy").setIgnoreProperties("updateBy");-->
<#--        BeanUtil.copyProperties(${entity?uncap_first}, ${entity?uncap_first},options);-->
        BeanUtil.copyProperties(${entity?uncap_first}, ${entity?uncap_first});
        return Result.Result(${table.mapperName?uncap_first}.insert(${entity?uncap_first}) > 0 ?
        ErrorEnum.SUCCESS:ErrorEnum.REQUEST_FAIL , null);
    }

    /**
    * 根据ID查询数据
    *
    * @param id ID
    * @return responseResult
    */
    @Override
    public Result<?> get${entity}ById(Long id) {
        ${entity} ${entity?uncap_first} = ${table.mapperName?uncap_first}.selectById(id);
        if(${entity?uncap_first} != null){
            return Result.Result(ErrorEnum.SUCCESS, ${entity?uncap_first});
        }else{
            return Result.Result(ErrorEnum.DATA_NOT_FOUND, null);
        }
    }

    /**
    * 更新数据
    *
    * @param ${entity?uncap_first} 实体对象
    * @return Result
    */
    @Override
    public Result<?> update${entity}(${entity} ${entity?uncap_first}) {
        if(${table.mapperName?uncap_first}.updateById(${entity?uncap_first}) > 0){
            return Result.Result(ErrorEnum.SUCCESS, "更新成功");
        }else{
            return Result.Result(ErrorEnum.DATA_NOT_FOUND, null);
        }
    }

    /**
    * 删除数据
    *
    * @param id ID
    * @return Result
    */
    @Override
    public Result<?> delete${entity}(Long id) {
        if(${table.mapperName?uncap_first}.deleteById(id) > 0 ){
            return Result.Result(ErrorEnum.SUCCESS, "删除成功");
        }else{
            return Result.Result(ErrorEnum.DATA_NOT_FOUND,null);
        }
    }
}

五、其他注意事项

代码中涉及的ErrorEnum.java 和 Result.java 为之前博文【Springboot项目返回数据统一封装】所写,如需要可参考: Springboot项目返回数据统一封装。也可直接替换成公共的或自己代码中自定义的。

相关推荐
计算机-秋大田1 小时前
基于Spring Boot的船舶监造系统的设计与实现,LW+源码+讲解
java·论文阅读·spring boot·后端·vue
沐雪架构师1 小时前
mybatis连接PGSQL中对于json和jsonb的处理
json·mybatis
代码之光_19802 小时前
保障性住房管理:SpringBoot技术优势分析
java·spring boot·后端
鹿屿二向箔3 小时前
基于SSM(Spring + Spring MVC + MyBatis)框架的咖啡馆管理系统
spring·mvc·mybatis
戴眼镜的猴4 小时前
Spring Boot的过滤器与拦截器的区别
spring boot
尘浮生4 小时前
Java项目实战II基于Spring Boot的光影视频平台(开发文档+数据库+源码)
java·开发语言·数据库·spring boot·后端·maven·intellij-idea
尚学教辅学习资料4 小时前
基于SpringBoot的医药管理系统+LW示例参考
java·spring boot·后端·java毕业设计·医药管理
morris1315 小时前
【SpringBoot】Xss的常见攻击方式与防御手段
java·spring boot·xss·csp
阿伟*rui8 小时前
配置管理,雪崩问题分析,sentinel的使用
java·spring boot·sentinel