mybatis-plus代码生成器扩展

最新推荐文章于 2024-06-10 18:54:35 发布
最新推荐文章于 2024-06-10 18:54:35 发布
阅读量560 收藏 1
点赞数
分类专栏： java 文章标签： java orm generator
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_39208453/article/details/120305417
版权

mybatis-plus代码生成器扩展

mybatis-plus代码生成器扩展

 根据服务名将生成实体类和其他类（mapper,service,controller....）分散到不同微服务项目包下

1. 概述

  • ORM方面我们选用mybatisplus框架进行开发，因为我们使用的是微服务架构，所以初期建库建表完成之后需要创建大量的实体类，mapper（没有使用xml文件），service等java文件。
  • 不管是mappe层还是service层，mybatisplus都为我们提供了很好的数据库操作函数，不用重复创建文件和方法了，所以可以通过代码生成器快速的生成实体类等文件。
  • Mybatisplus官方提供的代码生成器只能生成在同一个项目内，不满足我们想把生成的mapper,service等文件分散到不同的项目内的需求，所以对代码生成器进行了小小的改动

项目使用springcloud，根据业务模块划分不同为子模块。

2. 正文

2.1 导入依赖

	<dependency>
	   <groupId>com.baomidou</groupId>
	   <artifactId>mybatis-plus-generator</artifactId>
	   <version>3.1.1</version>
	</dependency>
	<!-- 模板引擎需要借助velocity依赖 -->
	<dependency>
	   <groupId>org.apache.velocity</groupId>
	   <artifactId>velocity-engine-core</artifactId>
	   <version>2.3</version>
	</dependency>

2.2 数据配置

2.2.1 CodeGenerator.java （主类-运行main即可）
package cn.com.agree.ait.generator;


import cn.com.agree.ait.generator.config.AitAutoGenerator;
import cn.com.agree.ait.generator.config.AitTemplateEngine;
import com.baomidou.mybatisplus.generator.InjectionConfig;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;

import java.io.File;

/**
 * 代码生成工具,运行本类main方法即可
 * @author lzx
 * @date 2021年8月26日15:36:44
 */
public class CodeGenerator {

   // 文件是否需要生成在不同的项目服务模块
    private static final boolean isMicroService = true;
    // 需要生成的表名，多表在数组内append
    private static final String[] tableNames = {
            "xxx",
            "xxxxx"
    };
    // 实体类生成在ait-commons服务 commons->实体类服务名
    private static final String CommonsServiceName = "commons"; 
     // mapper,service...生成在其他服务  xxx->需要生成mapper等文件的服务名
    private static final String OtherServiceName = "xxx"; 

    // 开发人员名称
    private static final String Author = "lzx";

    // jdbc配置 xxx代表填入自己的数据库环境配置 数据库环境为mysql，oracle环境未实验
    private static final String JDBC_URL = "xxx";
    private static final String JDBC_DRVERCLASS = "xxx";
    private static final String JDBC_USERNAME = "xxx";
    private static final String JDBC_PASSWORD = "xxx";
    
     // 指定commons服务(实体类服务) packagename为包名，xxx为服务名
    private static final String rootCommonsServicePackage = "/src/main/java/packagename.../xxx/pojo";根包路径
     // 指定省其他服务根包路径（mapper,service,serviceimpl,controller文件） packagename为包名，xxx为服务名
    private static final String rootOtherServicePackage = "/src/main/java/packagename/xxx/";   

    public static void main(String[] args) {
        aitAutoGeneratorExecute();
    }
    
    private static void aitAutoGeneratorExecute() {
        // 代码生成器
        AitAutoGenerator mpg = new AitAutoGenerator();

        GlobalConfig gc = popularGlobalConfig();
        mpg.setGlobalConfig(gc);

        DataSourceConfig dsc = popularDataSourceConfig();
        mpg.setDataSource(dsc);

        PackageConfig pc = popularPackageConfig();
        mpg.setPackageInfo(pc);

        InjectionConfig cfg = popularInjectionConfig();
        mpg.setCfg(cfg);

        TemplateConfig templateConfig = popularTemplateConfig();
        mpg.setTemplate(templateConfig);

        StrategyConfig strategy = popularStrategyConfig(pc);
        mpg.setStrategy(strategy);

        AitAutoGenerator.MicroServiceConfig microServiceConfig = popularMicroServiceConfig();
        mpg.setTemplateEngine(new AitTemplateEngine(microServiceConfig));

        mpg.execute();
    }

    private static AitAutoGenerator.MicroServiceConfig popularMicroServiceConfig() {
        AitAutoGenerator.MicroServiceConfig microServiceConfig = new AitAutoGenerator.MicroServiceConfig();
        //默认微服务模式，多服务分散生成
        microServiceConfig.setMicroService(isMicroService);

        microServiceConfig.setOtherServicePath(new File(System.getProperty("user.dir") + "/" + OtherServiceName + rootOtherServicePackage).getAbsolutePath());
        microServiceConfig.setCommonsServicePath(new File(System.getProperty("user.dir") + "/" + CommonsServiceName + rootCommonsServicePackage).getAbsolutePath());

        return microServiceConfig;
    }

    private static StrategyConfig popularStrategyConfig(PackageConfig pc) {
        // 策略配置
        StrategyConfig strategy = new StrategyConfig();
        strategy.setNaming(NamingStrategy.no_change);
        strategy.setColumnNaming(NamingStrategy.underline_to_camel);
        // strategy.setSuperEntityClass("你自己的父类实体,没有就不用设置!");
        strategy.setEntityLombokModel(true);
        strategy.setRestControllerStyle(true);
        // 公共父类
        // strategy.setSuperControllerClass("你自己的父类控制器,没有就不用设置!");
        // 写于父类中的公共字段
        // strategy.setSuperEntityColumns("id");


        strategy.setInclude(tableNames);
        strategy.setControllerMappingHyphenStyle(true);

        strategy.setTablePrefix(pc.getModuleName() + "_");

        return strategy;
    }

    private static TemplateConfig popularTemplateConfig() {
        // 配置模板
        TemplateConfig templateConfig = new TemplateConfig();
        templateConfig.setXml(null);
        return templateConfig;
    }

    private static InjectionConfig popularInjectionConfig() {
        // 自定义配置
        InjectionConfig cfg = new InjectionConfig() {
            @Override
            public void initMap() {
                // to do nothing
            }
        };
        return cfg;
    }

    private static PackageConfig popularPackageConfig() {
        // 包配置
        PackageConfig pc = new PackageConfig();
        // 不额外指定模块，如果指定为 test，则生成的xml会在 mapper/test/ 目录下
        pc.setModuleName("ait");
        pc.setParent("xxx");
        pc.setEntity("pojo");
        return pc;
    }

    private static DataSourceConfig popularDataSourceConfig() {
        // 数据源配置
        DataSourceConfig dsc = new DataSourceConfig();
        dsc.setUrl(JDBC_URL);
        dsc.setDriverName(JDBC_DRVERCLASS);
        dsc.setUsername(JDBC_USERNAME);
        dsc.setPassword(JDBC_PASSWORD);
        return dsc;
    }

    private static GlobalConfig popularGlobalConfig() {
        GlobalConfig gc = new GlobalConfig();
        // 全局配置
        String projectPath = System.getProperty("user.dir") + "/";
        gc.setOutputDir(projectPath);
        gc.setAuthor(Author);
        gc.setOpen(false);
        // 覆盖写
        gc.setFileOverride(true);
        gc.setControllerName(null);
        return gc;
    }
}
2.2.2 AitConstVal.java
package cn.com.agree.ait.generator.config;

import com.baomidou.mybatisplus.generator.config.ConstVal;

/**
 * @author lzx
 * @version 1.0
 * @date 2021/08/26 下午 01:51
 * @description:
 */

public class AitConstVal implements ConstVal {
    static String ENTITY_PATH = "entity_path";
    static String SERVICE_PATH = "service_path";
    static String SERVICE_IMPL_PATH = "service_impl_path";
    static String MAPPER_PATH = "mapper_path";
    static String XML_PATH = "xml_path";
    static String CONTROLLER_PATH = "controller_path";

    public static final String MAPPER_PACKAGE_PATH ="/mapper";
    public static final String SERVICE_PACKAGE_PATH ="/service";
    public static final String SERVICE_IMPL_PACKAGE_PATH ="/service/impl";
    public static final String CONTROLLER_PACKAGE_PATH ="/controller";

}
2.2.3 AitAutoGenerator.java
package cn.com.agree.ait.generator.config;

import com.baomidou.mybatisplus.annotation.TableLogic;
import com.baomidou.mybatisplus.annotation.TableName;
import com.baomidou.mybatisplus.annotation.Version;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.baomidou.mybatisplus.extension.activerecord.Model;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.InjectionConfig;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.config.builder.ConfigBuilder;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.baomidou.mybatisplus.generator.engine.AbstractTemplateEngine;
import com.baomidou.mybatisplus.generator.engine.VelocityTemplateEngine;
import lombok.*;
import lombok.experimental.Accessors;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.Serializable;
import java.util.List;

/**
 * @author lzx
 * @version 1.0
 * @date 2021/08/26 下午 01:09
 * @description:
 */

@Data
@Accessors(chain = true)
public class AitAutoGenerator extends AutoGenerator {
    private static final Logger logger = LoggerFactory.getLogger(AitAutoGenerator.class);

    /**
     * 配置信息
     */
    protected ConfigBuilder config;
    /**
     * 注入配置
     */
    @Getter(AccessLevel.NONE)
    @Setter(AccessLevel.NONE)
    protected InjectionConfig injectionConfig;
    /**
     * 数据源配置
     */
    private DataSourceConfig dataSource;
    /**
     * 数据库表配置
     */
    private StrategyConfig strategy;
    /**
     * 包 相关配置
     */
    private PackageConfig packageInfo;
    /**
     * 模板 相关配置
     */
    private TemplateConfig template;
    /**
     * 全局 相关配置
     */
    private GlobalConfig globalConfig;
    /**
     * 模板引擎
     */
    private AbstractTemplateEngine templateEngine;

    //微服务 相关配置
    private MicroServiceConfig microServiceConfig;

    /**
     * 生成代码
     */
    @Override
    public void execute() {
        logger.debug("==========================准备生成文件...==========================");
        // 初始化配置
        if (null == config) {
            config = new ConfigBuilder(packageInfo, dataSource, strategy, template, globalConfig);
            if (null != injectionConfig) {
                injectionConfig.setConfig(config);
            }
        }
        if (null == templateEngine) {
            // 为了兼容之前逻辑，采用 Velocity 引擎 【 默认 】
            templateEngine = new VelocityTemplateEngine();
        }
        // 模板引擎初始化执行文件输出
        templateEngine.setConfigBuilder(config);
        templateEngine.init(this.pretreatmentConfigBuilder(config)).mkdirs().batchOutput().open();
        logger.debug("==========================文件生成完成！！！==========================");
    }

    /**
     * 开放表信息、预留子类重写
     *
     * @param config 配置信息
     * @return ignore
     */

    @Override
    protected List<TableInfo> getAllTableInfoList(ConfigBuilder config) {
        return config.getTableInfoList();
    }

    /**
     * 预处理配置
     *
     * @param config 总配置信息
     * @return 解析数据结果集
     */
    @Override
    protected ConfigBuilder pretreatmentConfigBuilder(ConfigBuilder config) {
        /*
         * 注入自定义配置
         */
        if (null != injectionConfig) {
            injectionConfig.initMap();
            config.setInjectionConfig(injectionConfig);
        }
        /*
         * 表信息列表
         */
        List<TableInfo> tableList = this.getAllTableInfoList(config);
        for (TableInfo tableInfo : tableList) {
            /* ---------- 添加导入包 ---------- */
            if (config.getGlobalConfig().isActiveRecord()) {
                // 开启 ActiveRecord 模式
                tableInfo.setImportPackages(Model.class.getCanonicalName());
            }
            if (tableInfo.isConvert()) {
                // 表注解
                tableInfo.setImportPackages(TableName.class.getCanonicalName());
            }
            if (config.getStrategyConfig().getLogicDeleteFieldName() != null && tableInfo.isLogicDelete(config.getStrategyConfig().getLogicDeleteFieldName())) {
                // 逻辑删除注解
                tableInfo.setImportPackages(TableLogic.class.getCanonicalName());
            }
            if (StringUtils.isNotEmpty(config.getStrategyConfig().getVersionFieldName())) {
                // 乐观锁注解
                tableInfo.setImportPackages(Version.class.getCanonicalName());
            }
            boolean importSerializable = true;
            if (StringUtils.isNotEmpty(config.getSuperEntityClass())) {
                // 父实体
                tableInfo.setImportPackages(config.getSuperEntityClass());
                importSerializable = false;
            }
            if (config.getGlobalConfig().isActiveRecord()) {
                importSerializable = true;
            }
            if (importSerializable) {
                tableInfo.setImportPackages(Serializable.class.getCanonicalName());
            }
            // Boolean类型is前缀处理
            if (config.getStrategyConfig().isEntityBooleanColumnRemoveIsPrefix()) {
                tableInfo.getFields().stream().filter(field -> "boolean".equalsIgnoreCase(field.getPropertyType()))
                        .filter(field -> field.getPropertyName().startsWith("is"))
                        .forEach(field -> {
                            field.setConvert(true);
                            field.setPropertyName(StringUtils.removePrefixAfterPrefixToLower(field.getPropertyName(), 2));
                        });
            }
        }
        return config.setTableInfoList(tableList);
    }

    @Override
    public InjectionConfig getCfg() {
        return injectionConfig;
    }

    @Override
    public AitAutoGenerator setCfg(InjectionConfig injectionConfig) {
        this.injectionConfig = injectionConfig;
        return this;
    }

    @Data
    public static class MicroServiceConfig{

        //是否是微服务模块划分
        private boolean isMicroService;

        //ait-commons服务根路径(主要生成实体类)
        private String commonsServicePath;

        //其他服务路径（先支持单服务根路径）(主要生成mapper/service/seriveimpl/controller)
        private String otherServicePath;
    }
}
2.2.4 AitTemplateEngine.java
package cn.com.agree.ait.generator.config;

import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.baomidou.mybatisplus.core.toolkit.StringPool;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.baomidou.mybatisplus.generator.InjectionConfig;
import com.baomidou.mybatisplus.generator.config.ConstVal;
import com.baomidou.mybatisplus.generator.config.FileOutConfig;
import com.baomidou.mybatisplus.generator.config.TemplateConfig;
import com.baomidou.mybatisplus.generator.config.builder.ConfigBuilder;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.baomidou.mybatisplus.generator.config.rules.FileType;
import com.baomidou.mybatisplus.generator.engine.AbstractTemplateEngine;
import org.apache.velocity.Template;
import org.apache.velocity.VelocityContext;
import org.apache.velocity.app.Velocity;
import org.apache.velocity.app.VelocityEngine;

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileOutputStream;
import java.io.OutputStreamWriter;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.Properties;

/**
 * @author lzx
 * @version 1.0
 * @date 2021/08/26 下午 03:27
 * @description:
 */

public class AitTemplateEngine extends AbstractTemplateEngine {

    private static final String DOT_VM = ".vm";
    private VelocityEngine velocityEngine;
    private AitAutoGenerator.MicroServiceConfig microServiceConfig;
    private ConfigBuilder configBuilder;
    public AitTemplateEngine(AitAutoGenerator.MicroServiceConfig microServiceConfig) {
        this.microServiceConfig = microServiceConfig;
    }

    @Override
    public AitTemplateEngine init(ConfigBuilder configBuilder) {
        if (null == velocityEngine) {
            Properties p = new Properties();
            p.setProperty(ConstVal.VM_LOAD_PATH_KEY, ConstVal.VM_LOAD_PATH_VALUE);
            p.setProperty(Velocity.FILE_RESOURCE_LOADER_PATH, StringPool.EMPTY);
            p.setProperty(Velocity.ENCODING_DEFAULT, ConstVal.UTF8);
            p.setProperty(Velocity.INPUT_ENCODING, ConstVal.UTF8);
            p.setProperty("file.resource.loader.unicode", StringPool.TRUE);
            velocityEngine = new VelocityEngine(p);
        }
        this.configBuilder = configBuilder;
        super.setConfigBuilder(configBuilder);
        return this;
    }

    @Override
    public ConfigBuilder getConfigBuilder() {
        return super.getConfigBuilder();
    }
    /**
     * 处理输出目录
     */
    @Override
    public AitTemplateEngine mkdirs() {
        getConfigBuilder().getPathInfo().forEach((key, value) -> {
            if(microServiceConfig.isMicroService()){
                String servicePath;
                String info;
                if (AitConstVal.ENTITY.equals(key)){
                    servicePath = microServiceConfig.getCommonsServicePath();
                    info = "创建Ait-Commons-Service实体类目录： [";
                }else {
                    servicePath = microServiceConfig.getOtherServicePath();
                    info = "创建Ait-" + servicePath + "-Service实体类目录： [";
                }
                File dir = new File(servicePath);
                if (!dir.exists()) {
                    boolean result = dir.mkdirs();
                    if (result) {
                        logger.debug(info + servicePath + "]");
                    }
                }
            }else {
                File dir = new File(value);
                if (!dir.exists()) {
                    boolean result = dir.mkdirs();
                    if (result) {
                        logger.debug("创建目录： [" + value + "]");
                    }
                }
            }
        });
        return this;
    }

    @Override
    public void writer(Map<String, Object> objectMap, String templatePath, String outputFile) throws Exception {
        if (StringUtils.isEmpty(templatePath)) {
            return;
        }
        Template template = velocityEngine.getTemplate(templatePath, ConstVal.UTF8);
        try (FileOutputStream fos = new FileOutputStream(outputFile);
             OutputStreamWriter ow = new OutputStreamWriter(fos, ConstVal.UTF8);
             BufferedWriter writer = new BufferedWriter(ow)) {
            template.merge(new VelocityContext(objectMap), writer);
        }
        logger.debug("模板:" + templatePath + ";  文件:" + outputFile);
    }

    @Override
    public String templateFilePath(String filePath) {
        if (null == filePath || filePath.contains(DOT_VM)) {
            return filePath;
        }
        return filePath + DOT_VM;
    }

    @Override
    public AbstractTemplateEngine batchOutput() {
        try {
            List<TableInfo> tableInfoList = getConfigBuilder().getTableInfoList();
            for (TableInfo tableInfo : tableInfoList) {
                Map<String, Object> objectMap = getObjectMap(tableInfo);
                Map<String, String> pathInfo = getConfigBuilder().getPathInfo();
                TemplateConfig template = getConfigBuilder().getTemplate();
                // 自定义内容
                InjectionConfig injectionConfig = getConfigBuilder().getInjectionConfig();
                if (null != injectionConfig) {
                    injectionConfig.initMap();
                    objectMap.put("cfg", injectionConfig.getMap());
                    List<FileOutConfig> focList = injectionConfig.getFileOutConfigList();
                    if (CollectionUtils.isNotEmpty(focList)) {
                        for (FileOutConfig foc : focList) {
                            if (isCreate(FileType.OTHER, foc.outputFile(tableInfo))) {
                                writer(objectMap, foc.getTemplatePath(), foc.outputFile(tableInfo));
                            }
                        }
                    }
                }
                //覆盖原有目录并且优化文件名称
                coverOriDirectory(pathInfo,tableInfo);
                // Mp.java
                String entityName = tableInfo.getEntityName();
                if (null != entityName && null != pathInfo.get(ConstVal.ENTITY_PATH)) {
                    String entityFile = String.format((pathInfo.get(ConstVal.ENTITY_PATH) + File.separator + "%s" + suffixJavaOrKt()), entityName);
                    if (isCreate(FileType.ENTITY, entityFile)) {
                        writer(objectMap, templateFilePath(template.getEntity(getConfigBuilder().getGlobalConfig().isKotlin())), entityFile);
                    }
                }

                // MpMapper.java
                if (null != tableInfo.getMapperName() && null != pathInfo.get(ConstVal.MAPPER_PATH)) {
                    String mapperFile = String.format((pathInfo.get(ConstVal.MAPPER_PATH) + File.separator + tableInfo.getMapperName() + suffixJavaOrKt()), entityName);
                    if (isCreate(FileType.MAPPER, mapperFile)) {
                        writer(objectMap, templateFilePath(template.getMapper()), mapperFile);
                    }
                }
                // MpMapper.xml
                if (null != tableInfo.getXmlName() && null != pathInfo.get(ConstVal.XML_PATH)) {
                    String xmlFile = String.format((pathInfo.get(ConstVal.XML_PATH) + File.separator + tableInfo.getXmlName() + ConstVal.XML_SUFFIX), entityName);
                    if (isCreate(FileType.XML, xmlFile)) {
                        writer(objectMap, templateFilePath(template.getXml()), xmlFile);
                    }
                }
                // IMpService.java
                if (null != tableInfo.getServiceName() && null != pathInfo.get(ConstVal.SERVICE_PATH)) {
                    String serviceFile = String.format((pathInfo.get(ConstVal.SERVICE_PATH) + File.separator + tableInfo.getServiceName() + suffixJavaOrKt()), entityName);
                    if (isCreate(FileType.SERVICE, serviceFile)) {
                        writer(objectMap, templateFilePath(template.getService()), serviceFile);
                    }
                }
                // MpServiceImpl.java
                if (null != tableInfo.getServiceImplName() && null != pathInfo.get(ConstVal.SERVICE_IMPL_PATH)) {
                    String implFile = String.format((pathInfo.get(ConstVal.SERVICE_IMPL_PATH) + File.separator + tableInfo.getServiceImplName() + suffixJavaOrKt()), entityName);
                    if (isCreate(FileType.SERVICE_IMPL, implFile)) {
                        writer(objectMap, templateFilePath(template.getServiceImpl()), implFile);
                    }
                }
                // MpController.java
                if (null != tableInfo.getControllerName() && null != pathInfo.get(ConstVal.CONTROLLER_PATH)) {
                    String controllerFile = String.format((pathInfo.get(ConstVal.CONTROLLER_PATH) + File.separator + tableInfo.getControllerName() + suffixJavaOrKt()), entityName);
                    if (isCreate(FileType.CONTROLLER, controllerFile)) {
                        writer(objectMap, templateFilePath(template.getController()), controllerFile);
                    }
                }
            }
        } catch (Exception e) {
            logger.error("无法创建文件，请检查配置信息！", e);
        }
        return this;
    }

    private void coverOriDirectory(Map<String, String> pathInfo, TableInfo tableInfo) {
        //1.重新写入文件目录
        pathInfo.put(ConstVal.ENTITY_PATH,microServiceConfig.getCommonsServicePath());

        pathInfo.put(ConstVal.MAPPER_PATH,new File(microServiceConfig.getOtherServicePath() + AitConstVal.MAPPER_PACKAGE_PATH).getAbsolutePath());
        pathInfo.put(ConstVal.SERVICE_PATH,new File(microServiceConfig.getOtherServicePath() + AitConstVal.SERVICE_PACKAGE_PATH).getAbsolutePath());
        pathInfo.put(ConstVal.SERVICE_IMPL_PATH,new File(microServiceConfig.getOtherServicePath() + AitConstVal.SERVICE_IMPL_PACKAGE_PATH).getAbsolutePath());
        pathInfo.put(ConstVal.CONTROLLER_PATH,new File(microServiceConfig.getOtherServicePath() + AitConstVal.CONTROLLER_PACKAGE_PATH).getAbsolutePath());

        //2.优化文件名称
        tableInfo.setMapperName(optimizeFileName(tableInfo.getMapperName()));
        tableInfo.setServiceName(optimizeFileName(tableInfo.getServiceName()));
        tableInfo.setServiceImplName(optimizeFileName(tableInfo.getServiceImplName()));
        tableInfo.setControllerName(optimizeFileName(tableInfo.getControllerName()));
    }

    private static String optimizeFileName(String mapperName) {
        mapperName = mapperName.substring(mapperName.indexOf("_") + 1);
        mapperName = mapperName.replaceFirst(String.valueOf(mapperName.charAt(0)),String.valueOf(mapperName.charAt(0)).toUpperCase());
        StringBuilder tempStr = new StringBuilder(mapperName);
        if (mapperName.contains("_")){
            tempStr.setLength(0);
            Arrays.stream(mapperName.split("_")).forEach(c -> {
                tempStr.append(c.replace(String.valueOf(c.charAt(0)),String.valueOf(c.charAt(0)).toUpperCase()));
            });
        }
        return tempStr.toString();
    }
}

3. 小结

  1. 本文对于mybatisplus代码生成器的改造做了简单的介绍，粘贴代码即用。如果想深入了解这部分的知识，可以去了解一下官网 链接: link.
  2. 本次改动只是为了加快开发速度，提高效率，如果有什么疑问，欢迎留言
麦田杀手z
关注
  • 0
    点赞
  • 踩
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
MybatisPlus-Generator
王梦杰_MengJie
08-31 7980
本文将详细介绍mybatisplus-generator的相关使用，mybatisplus-generator是 MyBatis Plus 框架中的一个代码生成工具，它能够自动生成与数据库表对应的 Java 实体类、Mapper 接口以及 XML 配置文件。它的出现极大地提高了开发效率，让开发者能够将更多的精力集中在业务逻辑上，而不是繁琐的 CRUD 操作上。
argument type mismatch
qq_24981615的博客
05-06 694
com.baomidou.mybatisplus.core.exceptions.MybatisPlusException: java.lang.IllegalArgumentException: argument type mismatch
【微服务】第3节：MybatisPlus的扩展功能
最新发布
m0_46013789的博客
06-10 877
大家在日常开发中应该能发现，单表的CRUD功能代码重复度很高，也没有什么难度。而这部分代码量往往比较大，开发起来比较费时。因此，目前企业中都会使用一些组件来简化或省略单表的CRUD开发工作。目前在国内使用较多的一个组件就是MybatisPlus。
Mybatis plus 自动生成代码与自定义模板
学习学习学习学习
07-29 1万+
Mybatis-plus 生成代码，基于源码的自定义模板增强
mybatis-plus的包冲突问题
qq_44792548的博客
02-18 1134
mybatis-plus跟mybatis-plus-generator共用会报错的问题
MybatisPlus--个人笔记
laoliu的博客
03-08 1230
Mybatis-Plus 简介 MyBatis-Plus (opens new window)（简称 MP）是一个 MyBatis (opens new window)的增强工具，在 MyBatis 的基础上只做增强不做改变，为简化开发、提高效率而生。 愿景 我们的愿景是成为 MyBatis 最好的搭档，就像 魂斗罗 中的 1P、2P，基友搭配，效率翻倍。 特性 无侵入：只做增强不做改变，引入它不会对现有工程产生影响，如丝般顺滑 损耗小：启动即会自动注入基本 CURD，性能基本无损耗，直接面向对象操作
mybatis-plus代码生成器模版代码
08-17
本资源包含的是Mybatis-Plus的代码生成器模板代码，用于自动生成符合项目规范的Java实体类、Mapper接口及XML配置文件，提高开发效率。 代码生成器是开发中的利器，它能够根据数据库表结构快速生成对应的Java代码，...
非常好用的mybatis-plus代码生成器
05-20
Mybatis-Plus是一款基于Mybatis框架的扩展工具，它在Mybatis的基础上简化了开发工作，提供了许多便捷的功能，包括自动代码生成器。这个"非常好用的mybatis-plus代码生成器"能够帮助开发者快速生成常见的CRUD操作代码...
mybatis-plus 达梦 代码生成
06-10
1. **Mybatis-Plus**: Mybatis-Plus是一个优秀的Mybatis扩展，它简化了常见的数据库操作，如CRUD操作，动态SQL，以及一键生成代码等。它提供了一种便捷的方式管理数据库表和Java对象之间的映射关系。 2. **达梦...
mybatis-plus代码生成器代码.rar
07-14
而“mybatis-plus代码生成器”则是MyBatis-Plus库中的一个重要组成部分，用于自动生成常见的Java代码，包括Controller、DAO、Mapper、Service、ServiceImpl和Entity类，大大减少了手动编写这些重复性工作的负担。...
mybatis-plus 代码生成器
09-05
在本项目中，我们关注的是 "mybatis-plus 代码生成器"，这是一个强大的工具，能够自动生成基于 MyBatis-Plus 的实体类、Mapper 接口及实现、XML 映射文件，从而极大地提高了开发效率，减少了手动编写重复代码的时间...
Mybatis-Plus 3.5.1代码生成器生成多模块项目代码
aayygg1234的专栏
03-16 1457
MybatisPlus3.5.1版本将代码生成到不同模块下面，生产实战中方便使用。
使用 MyBatis-Plus 代码生成器生成代码
qq_38454776的博客
02-09 1682
在项目当中，如 Entity、Mapper、Mapper XML、Service、Controller 等各个模块的代码，重复去创建很繁琐，我们可以通过程序自动生成的，提升我们开发的效率。这篇文章记录如何使用 MyBatis-Plus 的代码生成器，生成代码，减少我饿们重复性劳动。下面给出一个可行的参考代码，具体的步骤可以参考官网文档 @SpringBootTest(classes = DistributedLockApp.class) @RunWith(SpringRunner.class) @Slf.
mybatis-plus自动生成代码
热门推荐
踏 浪的博客
01-12 3万+
一、介绍 本教程将介绍如何使用 mybatis-plus 工具自动给我们生成 Controller、Service、Entity、Mapper、Mapper.xml 层代码 要求: ① 生成的Controller类，需要继承 BaseController ② 生成的Entity 类，需要继承 BaseEntity ③ 生成的 Service，默认名称下是以 I 开头的接口, 在生成Se...
springboot项目使用mybatis-plus一键生成所有代码
lmsfv的博客
04-15 274
springboot项目使用mybatis-plus一键生成所有代码 pom.xml <?xml version="1.0" encoding="UTF-8"?> <project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" ...
解决mybatis-plus代码生成器在idea中的module生成代码，但是生成到外面的project
豌豆的博客
11-28 2万+
今天在用mybatis plus的代码生成器在idea里的module中生成代码时发现生成的代码都跑到外面的project中去了，打印了一下代码生成器中使用的 user.dir 属性 发现获取的是外面project的路径而不是里面module的路径 先解释一下user.dir 属性 user.dir属性就是获取当前工作的目录 当前工作目录就是执行这个class类的目录...
MyBatis Plus——自定义配置——自定义属性——注入时间
无限迭代中......
02-15 1927
解决方案 // 自定义配置 InjectionConfig cfg = new InjectionConfig() { @Override public void initMap() { // to do nothing Map<String, Object> map = new HashMap<>();
mybatis plus 简单使用及Generator使用记录笔记
大仕忆的博客
01-22 1693
pom.xml配置文件 <dependency> <groupId>com.baomidou</groupId> <artifactId>mybatis-plus-generator</artifactId> <version>3.4.1</version> </dependency>
MyBatis Plus Generator 配置文件介绍及使用
肖朋伟
04-07 1万+
MyBatis Plus Generator 配置文件介绍及使用 MyBatis Plus Generator 可以生成 Controller，Service，Mapper，Entity，可自写 SQL 的 mapper （resources 中） pom.xml 配置： <!-- MyBatis相关依赖 --> <dependency> ...
mybatis-plus3.5.2代码生成器
03-27
MyBatis-Plus是一个MyBatis的增强工具，在MyBatis的基础上进行了扩展，提供了更加便捷的CRUD操作、分页、乐观锁、逻辑删除等功能。而MyBatis-Plus的代码生成器则是MyBatis-Plus中的一个子模块，用于生成基于MyBatis-Plus的CRUD代码。 MyBatis-Plus的代码生成器是一个基于Velocity模板引擎的代码生成器，支持生成Java、XML、SQL脚本等文件。使用MyBatis-Plus的代码生成器可以快速生成CRUD代码，减少手动编写代码的工作量。 下面是使用MyBatis-Plus3.5.2代码生成器生成代码的步骤： 1. 添加MyBatis-Plus的依赖 在pom.xml文件中添加以下依赖： ```xml <dependency> <groupId>com.baomidou</groupId> <artifactId>mybatis-plus-generator</artifactId> <version>3.5.2</version> </dependency> ``` 2. 编写代码生成器配置文件 在src/main/resources目录下创建generator.properties文件，编写代码生成器的配置信息，例如： ```properties # 数据库配置 jdbc.driver=com.mysql.jdbc.Driver jdbc.url=jdbc:mysql://localhost:3306/mybatis_plus?useUnicode=true&characterEncoding=utf-8&serverTimezone=GMT%2B8 jdbc.username=root jdbc.password=root # 代码生成器配置 outputDir=D:/code-generator author=MyBatis-Plus fileOverride=true openDir=false ``` 其中，jdbc.driver、jdbc.url、jdbc.username、jdbc.password是数据库的配置信息，outputDir是生成文件的输出目录，author是代码的作者，fileOverride是是否覆盖已有文件，openDir是是否打开输出目录。 3. 编写代码生成器模板文件 在src/main/resources/templates目录下创建模板文件，例如entity.java.vm、mapper.xml.vm、service.java.vm、serviceImpl.java.vm等，编写模板文件的内容。 模板文件中使用Velocity的语法，可以使用变量、条件语句、循环语句等，例如： ```java package ${packageName}; import com.baomidou.mybatisplus.annotation.TableName; import lombok.Data; /** * ${tableComment} */ @Data @TableName("${tableName}") public class ${entityName} { #foreach($column in $table.columns) /** * ${column.columnComment} */ private ${column.javaType} ${column.fieldName}; #end } ``` 在模板文件中，$packageName、$tableComment、$tableName、$entityName、$table.columns等都是变量，会根据生成器配置文件和数据库表的信息动态替换为具体的值。 4. 运行代码生成器 编写好配置文件和模板文件后，就可以运行代码生成器了。在Java代码中使用CodeGenerator类，并传入配置文件和模板文件的路径，即可启动代码生成器，例如： ```java public class MybatisPlusGenerator { public static void main(String[] args) { String configFile = "generator.properties"; String templatePath = "/templates/%s.vm"; AutoGenerator generator = new AutoGenerator(); // 全局配置 GlobalConfig globalConfig = new GlobalConfig(); globalConfig.setAuthor(PropertyUtil.getProperty(configFile, "author")); globalConfig.setFileOverride(Boolean.parseBoolean(PropertyUtil.getProperty(configFile, "fileOverride"))); globalConfig.setOpen(Boolean.parseBoolean(PropertyUtil.getProperty(configFile, "openDir"))); globalConfig.setOutputDir(PropertyUtil.getProperty(configFile, "outputDir")); generator.setGlobalConfig(globalConfig); // 数据源配置 DataSourceConfig dataSourceConfig = new DataSourceConfig(); dataSourceConfig.setDbType(DbType.MYSQL); dataSourceConfig.setDriverName(PropertyUtil.getProperty(configFile, "jdbc.driver")); dataSourceConfig.setUrl(PropertyUtil.getProperty(configFile, "jdbc.url")); dataSourceConfig.setUsername(PropertyUtil.getProperty(configFile, "jdbc.username")); dataSourceConfig.setPassword(PropertyUtil.getProperty(configFile, "jdbc.password")); generator.setDataSource(dataSourceConfig); // 包配置 PackageConfig packageConfig = new PackageConfig(); packageConfig.setParent(PropertyUtil.getProperty(configFile, "package")); generator.setPackageInfo(packageConfig); // 自定义配置 InjectionConfig injectionConfig = new InjectionConfig() { @Override public void initMap() { } }; injectionConfig.setFileOutConfigList(getFileOutConfigList(templatePath)); generator.setCfg(injectionConfig); // 配置模板 TemplateConfig templateConfig = new TemplateConfig(); templateConfig.setXml(null); generator.setTemplate(templateConfig); // 策略配置 StrategyConfig strategyConfig = new StrategyConfig(); strategyConfig.setNaming(NamingStrategy.underline_to_camel); strategyConfig.setColumnNaming(NamingStrategy.underline_to_camel); strategyConfig.setEntityLombokModel(true); strategyConfig.setRestControllerStyle(true); strategyConfig.setInclude(PropertyUtil.getProperty(configFile, "tableNames").split(",")); generator.setStrategy(strategyConfig); generator.execute(); } private static List<FileOutConfig> getFileOutConfigList(String templatePath) { List<FileOutConfig> fileOutConfigList = new ArrayList<>(); fileOutConfigList.add(new FileOutConfig(templatePath) { @Override public String outputFile(TableInfo tableInfo) { return String.format("%s/%s%s", PropertyUtil.getProperty("generator.properties", "outputDir"), tableInfo.getEntityName(), this.getSuffix()); } }); return fileOutConfigList; } } ``` 在代码中，使用AutoGenerator类配置全局配置、数据源配置、包配置、自定义配置、模板配置、策略配置等，然后调用execute()方法即可生成代码。 总结：使用MyBatis-Plus的代码生成器可以快速生成基于MyBatis-Plus的CRUD代码，减少手动编写代码的工作量，提高开发效率。需要注意的是，生成的代码只是基础代码，需要根据具体业务进行修改和优化。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值