自定义mybatis-generator生成mapper代码

小灰灰 2022-12-05 01:36 296阅读 0赞

# 自定义mybatis-generator生成mapper代码 #

我是基于tk.mybatis.mapper.generator.MapperPlugin插件进行一个微小的拓展，大多数还是基于MapperPlugin逻辑进行处理，这里非常感谢tk.mybatis作者为我们提供源码基础。

tk.mybatis.mapper-generator源码地址https://repo1.maven.org/maven2/tk/mybatis/mapper-generator/1.1.5/

通过继承PluginAdapter类,开发自定义插件,生成自定义的中文注解的java实体类,dao接口和mapper映射文件

导入mybatis.generator包

<dependency>
        <groupId>org.mybatis.generator</groupId>
        <artifactId>mybatis-generator-core</artifactId>
        <version>1.3.7</version>
    </dependency>

引入tk.mybatis.mapper.generator.MapperPlugin相关类

import org.mybatis.generator.api.CommentGenerator;
    import org.mybatis.generator.api.IntrospectedColumn;
    import org.mybatis.generator.api.IntrospectedTable;
    import org.mybatis.generator.api.dom.java.*;
    import org.mybatis.generator.api.dom.xml.TextElement;
    import org.mybatis.generator.api.dom.xml.XmlElement;
    import org.mybatis.generator.config.MergeConstants;
    import org.mybatis.generator.internal.util.StringUtility;
    
    import java.text.MessageFormat;
    import java.util.Properties;
    import java.util.Set;
    
    public class MapperCommentGenerator implements CommentGenerator { 
        //开始的分隔符，例如mysql为`，sqlserver为[
        private String beginningDelimiter = "";
        //结束的分隔符，例如mysql为`，sqlserver为]
        private String endingDelimiter = "";
        //强制生成注解
        private boolean forceAnnotation;
        //是否生成swagger注解
        private boolean needsSwagger;
    
        public MapperCommentGenerator() { 
            super();
        }
    
        @Override
        public void addJavaFileComment(CompilationUnit compilationUnit) { 
            return;
        }
    
        /** * xml中的注释 * * @param xmlElement */
        @Override
        public void addComment(XmlElement xmlElement) { 
            xmlElement.addElement(new TextElement(""));
        }
    
        @Override
        public void addRootComment(XmlElement rootElement) { 
            return;
        }
    
        @Override
        public void addConfigurationProperties(Properties properties) { 
            String beginningDelimiter = properties.getProperty("beginningDelimiter");
            if (StringUtility.stringHasValue(beginningDelimiter)) { 
                this.beginningDelimiter = beginningDelimiter;
            }
            String endingDelimiter = properties.getProperty("endingDelimiter");
            if (StringUtility.stringHasValue(endingDelimiter)) { 
                this.endingDelimiter = endingDelimiter;
            }
            String forceAnnotation = properties.getProperty("forceAnnotation");
            if (StringUtility.stringHasValue(forceAnnotation)) { 
                this.forceAnnotation = "TRUE".equalsIgnoreCase(forceAnnotation);
            }
            String needsSwagger = properties.getProperty("needsSwagger");
            if (StringUtility.stringHasValue(needsSwagger)) { 
                this.needsSwagger = "TRUE".equalsIgnoreCase(needsSwagger);
            }
        }
    
        public String getDelimiterName(String name) { 
            StringBuilder nameBuilder = new StringBuilder();
            nameBuilder.append(beginningDelimiter);
            nameBuilder.append(name);
            nameBuilder.append(endingDelimiter);
            return nameBuilder.toString();
        }
    
        /** * 删除标记 * * @param javaElement * @param markAsDoNotDelete */
        protected void addJavadocTag(JavaElement javaElement, boolean markAsDoNotDelete) { 
            StringBuilder sb = new StringBuilder();
            sb.append(" * ");
            sb.append(MergeConstants.NEW_ELEMENT_TAG);
            if (markAsDoNotDelete) { 
                sb.append(" do_not_delete_during_merge");
            }
            javaElement.addJavaDocLine(sb.toString());
        }
    
        /** * Example使用 * * @param innerClass * @param introspectedTable */
        @Override
        public void addClassComment(InnerClass innerClass, IntrospectedTable introspectedTable) { 
        }
    
        @Override
        public void addEnumComment(InnerEnum innerEnum, IntrospectedTable introspectedTable) { 
        }
    
        /** * 给字段添加数据库备注 * * @param field * @param introspectedTable * @param introspectedColumn */
        @Override
        public void addFieldComment(Field field, IntrospectedTable introspectedTable, IntrospectedColumn introspectedColumn) { 
            if (StringUtility.stringHasValue(introspectedColumn.getRemarks())) { 
                field.addJavaDocLine("/**");
                StringBuilder sb = new StringBuilder();
                sb.append(" * ");
                sb.append(introspectedColumn.getRemarks());
                field.addJavaDocLine(sb.toString());
                field.addJavaDocLine(" */");
            }
            //添加注解
            if (field.isTransient()) { 
                //@Column
                field.addAnnotation("@Transient");
            }
            for (IntrospectedColumn column : introspectedTable.getPrimaryKeyColumns()) { 
                if (introspectedColumn == column) { 
                    field.addAnnotation("@Id");
                    break;
                }
            }
            String column = introspectedColumn.getActualColumnName();
            if (StringUtility.stringContainsSpace(column) || introspectedTable.getTableConfiguration().isAllColumnDelimitingEnabled()) { 
                column = introspectedColumn.getContext().getBeginningDelimiter()
                        + column
                        + introspectedColumn.getContext().getEndingDelimiter();
            }
            if (!column.equals(introspectedColumn.getJavaProperty())) { 
                //@Column
                field.addAnnotation("@Column(name = \"" + getDelimiterName(column) + "\")");
            } else if (StringUtility.stringHasValue(beginningDelimiter) || StringUtility.stringHasValue(endingDelimiter)) { 
                field.addAnnotation("@Column(name = \"" + getDelimiterName(column) + "\")");
            } else if (forceAnnotation) { 
                field.addAnnotation("@Column(name = \"" + getDelimiterName(column) + "\")");
            }
            if (introspectedColumn.isIdentity()) { 
                if ("JDBC".equals(introspectedTable.getTableConfiguration().getGeneratedKey().getRuntimeSqlStatement())) { 
                    field.addAnnotation("@GeneratedValue(generator = \"JDBC\")");
                } else { 
                    field.addAnnotation("@GeneratedValue(strategy = GenerationType.IDENTITY)");
                }
            } else if (introspectedColumn.isSequenceColumn()) { 
                //在 Oracle 中，如果需要是 SEQ_TABLENAME，那么可以配置为 select SEQ_{1} from dual
                String tableName = introspectedTable.getFullyQualifiedTableNameAtRuntime();
                String sql = MessageFormat.format(introspectedTable.getTableConfiguration().getGeneratedKey().getRuntimeSqlStatement(), tableName, tableName.toUpperCase());
                field.addAnnotation("@GeneratedValue(strategy = GenerationType.IDENTITY, generator = \"" + sql + "\")");
            }
            // region swagger注解
            if (this.needsSwagger) { 
                String remarks = introspectedColumn.getRemarks();
                if (remarks == null) { 
                    remarks = "";
                }
                String swaggerAnnotation = "@ApiModelProperty(value = \"%s\" da )";
                field.addAnnotation("@ApiModelProperty(\"" + remarks.replaceAll("\r", "").replaceAll("\n", "") + "\")");
            }
            // endregion
        }
    
        /** * Example使用 * * @param field * @param introspectedTable */
        @Override
        public void addFieldComment(Field field, IntrospectedTable introspectedTable) { 
        }
    
        @Override
        public void addModelClassComment(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) { 
    
        }
    
        /** * @param method * @param introspectedTable */
        @Override
        public void addGeneralMethodComment(Method method, IntrospectedTable introspectedTable) { 
        }
    
        /** * getter方法注释 * * @param method * @param introspectedTable * @param introspectedColumn */
        @Override
        public void addGetterComment(Method method, IntrospectedTable introspectedTable, IntrospectedColumn introspectedColumn) { 
            StringBuilder sb = new StringBuilder();
            method.addJavaDocLine("/**");
            if (StringUtility.stringHasValue(introspectedColumn.getRemarks())) { 
                sb.append(" * 获取");
                sb.append(introspectedColumn.getRemarks());
                method.addJavaDocLine(sb.toString());
                method.addJavaDocLine(" *");
            }
            sb.setLength(0);
            sb.append(" * @return ");
            sb.append(introspectedColumn.getActualColumnName());
            if (StringUtility.stringHasValue(introspectedColumn.getRemarks())) { 
                sb.append(" - ");
                sb.append(introspectedColumn.getRemarks());
            }
            method.addJavaDocLine(sb.toString());
            method.addJavaDocLine(" */");
        }
    
        /** * setter方法注释 * * @param method * @param introspectedTable * @param introspectedColumn */
        @Override
        public void addSetterComment(Method method, IntrospectedTable introspectedTable, IntrospectedColumn introspectedColumn) { 
            StringBuilder sb = new StringBuilder();
            method.addJavaDocLine("/**");
            if (StringUtility.stringHasValue(introspectedColumn.getRemarks())) { 
                sb.append(" * 设置");
                sb.append(introspectedColumn.getRemarks());
                method.addJavaDocLine(sb.toString());
                method.addJavaDocLine(" *");
            }
            Parameter parm = method.getParameters().get(0);
            sb.setLength(0);
            sb.append(" * @param ");
            sb.append(parm.getName());
            if (StringUtility.stringHasValue(introspectedColumn.getRemarks())) { 
                sb.append(" ");
                sb.append(introspectedColumn.getRemarks());
            }
            method.addJavaDocLine(sb.toString());
            method.addJavaDocLine(" */");
        }
    
        /** * Example使用 * * @param innerClass * @param introspectedTable * @param markAsDoNotDelete */
        @Override
        public void addClassComment(InnerClass innerClass, IntrospectedTable introspectedTable, boolean markAsDoNotDelete) { 
        }
    
        /** * @since mbg 1.3.6 */
        @Override
        public void addGeneralMethodAnnotation(Method method, IntrospectedTable introspectedTable, Set<FullyQualifiedJavaType> set) { 
    
        }
    
        /** * @since mbg 1.3.6 */
        @Override
        public void addGeneralMethodAnnotation(Method method, IntrospectedTable introspectedTable, IntrospectedColumn introspectedColumn, Set<FullyQualifiedJavaType> set) { 
    
        }
    
        /** * @since mbg 1.3.6 */
        @Override
        public void addFieldAnnotation(Field field, IntrospectedTable introspectedTable, Set<FullyQualifiedJavaType> set) { 
    
        }
    
        /** * @since mbg 1.3.6 */
        @Override
        public void addFieldAnnotation(Field field, IntrospectedTable introspectedTable, IntrospectedColumn introspectedColumn, Set<FullyQualifiedJavaType> set) { 
    
        }
    
        /** * @since mbg 1.3.6 */
        @Override
        public void addClassAnnotation(InnerClass innerClass, IntrospectedTable introspectedTable, Set<FullyQualifiedJavaType> set) { 
    
        }
    }

import org.mybatis.generator.api.IntrospectedTable;
    import org.mybatis.generator.api.PluginAdapter;
    import org.mybatis.generator.api.dom.java.Interface;
    import org.mybatis.generator.api.dom.java.Method;
    import org.mybatis.generator.api.dom.java.TopLevelClass;
    import org.mybatis.generator.api.dom.xml.XmlElement;
    
    import java.util.List;
    
    /** * 禁用大多数方法 * * @author liuzh */
    public class FalseMethodPlugin extends PluginAdapter { 
    
        @Override
        public boolean validate(List<String> warnings) { 
            return true;
        }
    
        //下面所有return false的方法都不生成。这些都是基础的CRUD方法，使用通用Mapper实现
        @Override
        public boolean clientDeleteByPrimaryKeyMethodGenerated(Method method, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean clientInsertMethodGenerated(Method method, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean clientInsertSelectiveMethodGenerated(Method method, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean clientSelectByPrimaryKeyMethodGenerated(Method method, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean clientUpdateByPrimaryKeySelectiveMethodGenerated(Method method, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean clientUpdateByPrimaryKeyWithBLOBsMethodGenerated(Method method, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean clientUpdateByPrimaryKeyWithoutBLOBsMethodGenerated(Method method, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean clientDeleteByPrimaryKeyMethodGenerated(Method method, Interface interfaze, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean clientInsertMethodGenerated(Method method, Interface interfaze, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean clientInsertSelectiveMethodGenerated(Method method, Interface interfaze, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean clientSelectAllMethodGenerated(Method method, Interface interfaze, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean clientSelectAllMethodGenerated(Method method, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean clientSelectByPrimaryKeyMethodGenerated(Method method, Interface interfaze, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean clientUpdateByPrimaryKeySelectiveMethodGenerated(Method method, Interface interfaze, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean clientUpdateByPrimaryKeyWithBLOBsMethodGenerated(Method method, Interface interfaze, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean clientUpdateByPrimaryKeyWithoutBLOBsMethodGenerated(Method method, Interface interfaze, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean sqlMapDeleteByPrimaryKeyElementGenerated(XmlElement element, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean sqlMapInsertElementGenerated(XmlElement element, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean sqlMapInsertSelectiveElementGenerated(XmlElement element, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean sqlMapSelectAllElementGenerated(XmlElement element, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean sqlMapSelectByPrimaryKeyElementGenerated(XmlElement element, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean sqlMapUpdateByPrimaryKeySelectiveElementGenerated(XmlElement element, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean sqlMapUpdateByPrimaryKeyWithBLOBsElementGenerated(XmlElement element, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean sqlMapUpdateByPrimaryKeyWithoutBLOBsElementGenerated(XmlElement element, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean providerGenerated(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean providerApplyWhereMethodGenerated(Method method, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean providerInsertSelectiveMethodGenerated(Method method, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) { 
            return false;
        }
    
        @Override
        public boolean providerUpdateByPrimaryKeySelectiveMethodGenerated(Method method, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) { 
            return false;
        }
    
    }

import org.mybatis.generator.api.IntrospectedColumn;
    import org.mybatis.generator.api.IntrospectedTable;
    import org.mybatis.generator.api.dom.java.*;
    import org.mybatis.generator.api.dom.xml.Element;
    import org.mybatis.generator.api.dom.xml.TextElement;
    import org.mybatis.generator.api.dom.xml.XmlElement;
    import org.mybatis.generator.config.CommentGeneratorConfiguration;
    import org.mybatis.generator.config.Context;
    import org.mybatis.generator.internal.util.StringUtility;
    
    import java.util.*;
    
    /** * @author JC.Lin * @date 2020-09-04 15:06 */
    public class MapperPlugin extends FalseMethodPlugin { 
        @Override
        public boolean validate(List<String> warnings) { 
            return true;
        }
    
        private Set<String> mappers = new HashSet<String>();
        private boolean caseSensitive = false;
        private boolean useMapperCommentGenerator = true;
        //开始的分隔符，例如mysql为`，sqlserver为[
        private String beginningDelimiter = "";
        //结束的分隔符，例如mysql为`，sqlserver为]
        private String endingDelimiter = "";
        //数据库模式
        private String schema;
        //注释生成器
        private CommentGeneratorConfiguration commentCfg;
        //强制生成注解
        private boolean forceAnnotation;
    
        //是否需要生成Data注解
        private boolean needsData = false;
        //是否需要生成Getter注解
        private boolean needsGetter = false;
        //是否需要生成Setter注解
        private boolean needsSetter = false;
        //是否需要生成ToString注解
        private boolean needsToString = false;
        //是否需要生成Accessors(chain = true)注解
        private boolean needsAccessors = false;
        //是否需要生成EqualsAndHashCode注解
        private boolean needsEqualsAndHashCode = false;
        //是否生成字段名常量
        private boolean generateColumnConsts = false;
        //是否生成默认的属性的静态方法
        private boolean generateDefaultInstanceMethod = false;
        //是否生成swagger注解,包括 @ApiModel和@ApiModelProperty
        private boolean needsSwagger = false;
    
        public String getDelimiterName(String name) { 
            StringBuilder nameBuilder = new StringBuilder();
            if (StringUtility.stringHasValue(schema)) { 
                nameBuilder.append(schema);
                nameBuilder.append(".");
            }
            nameBuilder.append(beginningDelimiter);
            nameBuilder.append(name);
            nameBuilder.append(endingDelimiter);
            return nameBuilder.toString();
        }
    
        /** * 生成的Mapper接口 * * @param interfaze * @param topLevelClass * @param introspectedTable * @return */
        @Override
        public boolean clientGenerated(Interface interfaze, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) { 
            // 接口添加注解 @Component
            interfaze.addAnnotation("@Component");
            //获取实体类
            FullyQualifiedJavaType entityType = new FullyQualifiedJavaType(introspectedTable.getBaseRecordType());
            //import接口
            for (String mapper : mappers) { 
                interfaze.addImportedType(new FullyQualifiedJavaType(mapper));
                interfaze.addSuperInterface(new FullyQualifiedJavaType(mapper + "<" + entityType.getShortName() + ">"));
            }
            //import实体类
            interfaze.addImportedType(entityType);
            return true;
        }
    
        /** * Base_Column_List 生成 * * @param element * @param introspectedTable 数据库表信息 创建后的结果： <sql id="Base_Column_List"> ${alias}.id id, ${alias}.exercise_id exerciseId, ${alias}.exercise_detail_id exerciseDetailId, ${alias}.exercise_right_flag exerciseRightFlag, ${alias}.exercise_user_answer exerciseUserAnswer, ${alias}.create_time createTime, ${alias}.update_time updateTime </sql> */
        @Override
        public boolean sqlMapBaseColumnListElementGenerated(XmlElement element, IntrospectedTable introspectedTable) { 
            List<Element> elements = element.getElements();
            elements.clear();
            int size = introspectedTable.getAllColumns().size();
            for (int i = 0; i < size; i++) { 
                IntrospectedColumn introspectedColumn = introspectedTable.getAllColumns().get(i);
                String text = "${alias}." + introspectedColumn.getActualColumnName() + " " + introspectedColumn.getJavaProperty();
                if (i < size - 1) { 
                    text += ",";
                }
                elements.add(new TextElement(text));
            }
            return true;
        }
    
        /** * 控制 BaseResultMap 生成 * @param element * @param introspectedTable * @return */
        @Override
        public boolean sqlMapResultMapWithoutBLOBsElementGenerated(
                XmlElement element, IntrospectedTable introspectedTable) { 
            return false; // 返回为false 不生成 BaseResultMap
        }
    
        /** * 处理实体类的包和@Table注解 * * @param topLevelClass * @param introspectedTable */
        private void processEntityClass(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) { 
            //引入JPA注解
            topLevelClass.addImportedType("javax.persistence.*");
            //lombok扩展开始
            //如果需要Data，引入包，代码增加注解
            if (this.needsData) { 
                topLevelClass.addImportedType("lombok.Data");
                topLevelClass.addAnnotation("@Data");
            }
            //如果需要Getter，引入包，代码增加注解
            if (this.needsGetter) { 
                topLevelClass.addImportedType("lombok.Getter");
                topLevelClass.addAnnotation("@Getter");
            }
            //如果需要Setter，引入包，代码增加注解
            if (this.needsSetter) { 
                topLevelClass.addImportedType("lombok.Setter");
                topLevelClass.addAnnotation("@Setter");
            }
            //如果需要ToString，引入包，代码增加注解
            if (this.needsToString) { 
                topLevelClass.addImportedType("lombok.ToString");
                topLevelClass.addAnnotation("@ToString");
            }
            //如果需要Getter，引入包，代码增加注解
            if (this.needsAccessors) { 
                topLevelClass.addImportedType("lombok.experimental.Accessors");
                topLevelClass.addAnnotation("@Accessors(chain = true)");
            }
            //如果需要Getter，引入包，代码增加注解
            if (this.needsEqualsAndHashCode) { 
                topLevelClass.addImportedType("lombok.EqualsAndHashCode");
                topLevelClass.addAnnotation("@EqualsAndHashCode");
            }
            //lombok扩展结束
            // region swagger扩展
            if (this.needsSwagger) { 
                //导包
                topLevelClass.addImportedType("io.swagger.annotations.ApiModel");
                topLevelClass.addImportedType("io.swagger.annotations.ApiModelProperty");
                //增加注解(去除注释中的转换符)
                String remarks = introspectedTable.getRemarks();
                if (remarks == null) { 
                    remarks = "";
                }
                topLevelClass.addAnnotation("@ApiModel(\"" + remarks.replaceAll("\r", "").replaceAll("\n", "") + "\")");
            }
            // endregion swagger扩展
            String tableName = introspectedTable.getFullyQualifiedTableNameAtRuntime();
            //如果包含空格，或者需要分隔符，需要完善
            if (StringUtility.stringContainsSpace(tableName)) { 
                tableName = context.getBeginningDelimiter()
                        + tableName
                        + context.getEndingDelimiter();
            }
            //是否忽略大小写，对于区分大小写的数据库，会有用
            if (caseSensitive && !topLevelClass.getType().getShortName().equals(tableName)) { 
                topLevelClass.addAnnotation("@Table(name = \"" + getDelimiterName(tableName) + "\")");
            } else if (!topLevelClass.getType().getShortName().equalsIgnoreCase(tableName)) { 
                topLevelClass.addAnnotation("@Table(name = \"" + getDelimiterName(tableName) + "\")");
            } else if (StringUtility.stringHasValue(schema)
                    || StringUtility.stringHasValue(beginningDelimiter)
                    || StringUtility.stringHasValue(endingDelimiter)) { 
                topLevelClass.addAnnotation("@Table(name = \"" + getDelimiterName(tableName) + "\")");
            } else if (forceAnnotation) { 
                topLevelClass.addAnnotation("@Table(name = \"" + getDelimiterName(tableName) + "\")");
            }
            if (generateColumnConsts) { 
                for (IntrospectedColumn introspectedColumn : introspectedTable.getAllColumns()) { 
                    Field field = new Field();
                    field.setVisibility(JavaVisibility.PUBLIC);
                    field.setStatic(true);
                    field.setFinal(true);
                    field.setName(introspectedColumn.getActualColumnName().toUpperCase()); //$NON-NLS-1$
                    field.setType(new FullyQualifiedJavaType(String.class.getName())); //$NON-NLS-1$
                    field.setInitializationString("\"" + introspectedColumn.getJavaProperty() + "\"");
                    context.getCommentGenerator().addClassComment(topLevelClass, introspectedTable);
                    topLevelClass.addField(field);
                    //增加字段名常量,用于pageHelper
                    Field columnField = new Field();
                    columnField.setVisibility(JavaVisibility.PUBLIC);
                    columnField.setStatic(true);
                    columnField.setFinal(true);
                    columnField.setName("DB_" + introspectedColumn.getActualColumnName().toUpperCase()); //$NON-NLS-1$
                    columnField.setType(new FullyQualifiedJavaType(String.class.getName())); //$NON-NLS-1$
                    columnField.setInitializationString("\"" + introspectedColumn.getActualColumnName() + "\"");
                    topLevelClass.addField(columnField);
                }
            }
            if (generateDefaultInstanceMethod) { 
                Method defaultMethod = new Method();
                defaultMethod.setStatic(true);
                defaultMethod.setName("defaultInstance");
                defaultMethod.setVisibility(JavaVisibility.PUBLIC);
                defaultMethod.setReturnType(topLevelClass.getType());
                defaultMethod.addBodyLine(String.format("%s instance = new %s();", topLevelClass.getType().getShortName(), topLevelClass.getType().getShortName()));
                for (IntrospectedColumn introspectedColumn : introspectedTable.getAllColumns()) { 
                    String shortName = introspectedColumn.getFullyQualifiedJavaType().getShortName();
                    List<String> supportType = Arrays.asList("Byte", "Short", "Character", "Integer", "Long", "Float", "Double", "String", "BigDecimal", "BigInteger");
                    if (!supportType.contains(shortName)) { 
                        continue;
                    }
                    if (introspectedColumn.getDefaultValue() != null) { 
                        String defaultValue = introspectedColumn.getDefaultValue();
                        //去除前后'',如 '123456' -> 123456
                        if (defaultValue.startsWith("'") && defaultValue.endsWith("'")) { 
                            if (defaultValue.length() == 2) { 
                                defaultValue = "";
                            } else { 
                                defaultValue = defaultValue.substring(1, defaultValue.length() - 1);
                            }
                        }
                        //暂不支持时间类型默认值识别,不同数据库表达式不同
                        if ("Boolean".equals(shortName)) { 
                            if ("0".equals(defaultValue)) { 
                                defaultValue = "false";
                            } else if ("1".equals(defaultValue)) { 
                                defaultValue = "true";
                            }
                        }
                        //通过 new 方法转换
                        defaultMethod.addBodyLine(String.format("instance.%s = new %s(\"%s\");", introspectedColumn.getJavaProperty(), shortName, defaultValue));
                    }
    
                }
                defaultMethod.addBodyLine("return instance;");
                topLevelClass.addMethod(defaultMethod);
            }
        }
    
        /** * 如果需要生成Getter注解，就不需要生成get相关代码了 */
        @Override
        public boolean modelGetterMethodGenerated(Method method,
                                                  TopLevelClass topLevelClass, IntrospectedColumn introspectedColumn,
                                                  IntrospectedTable introspectedTable,
                                                  ModelClassType modelClassType) { 
    
            return !(this.needsData || this.needsGetter);
        }
    
        /** * 如果需要生成Setter注解，就不需要生成set相关代码了 */
        @Override
        public boolean modelSetterMethodGenerated(Method method,
                                                  TopLevelClass topLevelClass, IntrospectedColumn introspectedColumn,
                                                  IntrospectedTable introspectedTable,
                                                  ModelClassType modelClassType) { 
            return !(this.needsData || this.needsSetter);
        }
    
        /** * 生成基础实体类 * * @param topLevelClass 目标实体类 * @param introspectedTable 数据库表信息 * @return */
        @Override
        public boolean modelBaseRecordClassGenerated(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) { 
            processEntityClass(topLevelClass, introspectedTable);
            return true;
        }
    
        /** * 生成实体类注解KEY对象 * * @param topLevelClass 目标实体类 * @param introspectedTable 数据库表信息 * @return */
        @Override
        public boolean modelPrimaryKeyClassGenerated(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) { 
            processEntityClass(topLevelClass, introspectedTable);
            return true;
        }
    
        /** * 生成带BLOB字段的对象 * * @param topLevelClass * @param introspectedTable * @return */
        @Override
        public boolean modelRecordWithBLOBsClassGenerated(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) { 
            processEntityClass(topLevelClass, introspectedTable);
            return false;
        }
    
        @Override
        public void setContext(Context context) { 
            super.setContext(context);
            //设置默认的注释生成器
            useMapperCommentGenerator = !"FALSE".equalsIgnoreCase(context.getProperty("useMapperCommentGenerator"));
            if (useMapperCommentGenerator) { 
                commentCfg = new CommentGeneratorConfiguration();
                commentCfg.setConfigurationType(MapperCommentGenerator.class.getCanonicalName());
                context.setCommentGeneratorConfiguration(commentCfg);
            }
            //支持oracle获取注释#114
            context.getJdbcConnectionConfiguration().addProperty("remarksReporting", "true");
            //支持mysql获取注释
            context.getJdbcConnectionConfiguration().addProperty("useInformationSchema", "true");
        }
    
        @Override
        public void setProperties(Properties properties) { 
            super.setProperties(properties);
            String mappers = getProperty("mappers");
            if (StringUtility.stringHasValue(mappers)) { 
                for (String mapper : mappers.split(",")) { 
                    this.mappers.add(mapper);
                }
            } else { 
                throw new RuntimeException("Mapper插件缺少必要的mappers属性!");
            }
            this.caseSensitive = Boolean.parseBoolean(this.properties.getProperty("caseSensitive"));
            this.forceAnnotation = getPropertyAsBoolean("forceAnnotation");
            this.beginningDelimiter = getProperty("beginningDelimiter", "");
            this.endingDelimiter = getProperty("endingDelimiter", "");
            this.schema = getProperty("schema");
            //lombok扩展
            String lombok = getProperty("lombok");
            if (lombok != null && !"".equals(lombok)) { 
                this.needsData = lombok.contains("Data");
                //@Data 优先级高于 @Getter @Setter @RequiredArgsConstructor @ToString @EqualsAndHashCode
                this.needsGetter = !this.needsData && lombok.contains("Getter");
                this.needsSetter = !this.needsData && lombok.contains("Setter");
                this.needsToString = !this.needsData && lombok.contains("ToString");
                this.needsEqualsAndHashCode = !this.needsData && lombok.contains("EqualsAndHashCode");
                this.needsAccessors = lombok.contains("Accessors");
            }
            //swagger扩展
            String swagger = getProperty("swagger", "false");
            if ("true".equalsIgnoreCase(swagger)) { 
                this.needsSwagger = true;
            }
            if (useMapperCommentGenerator) { 
                commentCfg.addProperty("beginningDelimiter", this.beginningDelimiter);
                commentCfg.addProperty("endingDelimiter", this.endingDelimiter);
                String forceAnnotation = getProperty("forceAnnotation");
                if (StringUtility.stringHasValue(forceAnnotation)) { 
                    commentCfg.addProperty("forceAnnotation", forceAnnotation);
                }
                commentCfg.addProperty("needsSwagger", this.needsSwagger + "");
            }
            this.generateColumnConsts = getPropertyAsBoolean("generateColumnConsts");
            this.generateDefaultInstanceMethod = getPropertyAsBoolean("generateDefaultInstanceMethod");
        }
    
        protected String getProperty(String key) { 
            return this.properties.getProperty(key);
        }
    
        protected String getProperty(String key, String defaultValue) { 
            return this.properties.getProperty(key, defaultValue);
        }
    
        protected Boolean getPropertyAsBoolean(String key) { 
            return Boolean.parseBoolean(getProperty(key));
        }
    }

启动类：

/** * @author JC.Lin * @date 2020-09-04 15:36 */
    public class Generator { 
        // 公共子项目
        private static String commonProject = "eduz-apis-common";
        private static String commonPackage = "com.eduz.model.exercise";
    
        private static List<TableMap> tableMapList = new ArrayList<>();
        private static List<TargetMap> targetMapList = new ArrayList<>();
    
    
        static { 
            TableMap tableMap = null;
    
            tableMap = new TableMap();
            tableMap.setTableName("exercise_classify_map");
            tableMap.setModelName("ExerciseClassifyMap");
            tableMapList.add(tableMap);
    
    
            // ===============================================================
    
            TargetMap targetMap = null;
            // B端子项目
            targetMap = new TargetMap();
            targetMap.setTargetProject("eduz-apis-business");
            targetMap.setTargetDaoPackage("com.eduz.business.exercise.dao");
            targetMap.setTargetMapperPackage("com.eduz.business.exercise.mapper");
            targetMapList.add(targetMap);
            // C端子项目
            targetMap = new TargetMap();
            targetMap.setTargetProject("eduz-apis-service");
            targetMap.setTargetDaoPackage("com.eduz.exercise.dao");
            targetMap.setTargetMapperPackage("com.eduz.exercise.mapper");
            targetMapList.add(targetMap);
            // 定时器子项目
    // targetMap = new TargetMap();
    // targetMap.setTargetProject("eduz-apis-quartz");
    // targetMap.setTargetDaoPackage("com.eduz.quartz.edu.dao");
    // targetMap.setTargetMapperPackage("com.eduz.quartz.edu.mapper");
    // targetMapList.add(targetMap);
    
        }
    
        public static void main(String[] args) throws Exception { 
            //MBG 执行过程中的警告信息
            List<String> warnings = new ArrayList<String>();
            //当生成的代码重复时，覆盖原代码
            boolean overwrite = true;
            //读取我们的 MBG 配置文件
            InputStream is = Generator.class.getResourceAsStream("/generatorConfig.xml");
            ConfigurationParser cp = new ConfigurationParser(warnings);
            Configuration config = cp.parseConfiguration(is);
            is.close();
    
            Context defaultContext = config.getContext("default");
    
            // Table
            if (tableMapList.isEmpty()) { 
                System.out.println("tableMapList is empty !!");
                return;
            }
    
            defaultContext.getTableConfigurations().clear();
            for (TableMap tableMap : tableMapList) { 
                TableConfiguration tableConfiguration = new TableConfiguration(defaultContext);
                tableConfiguration.setTableName(tableMap.getTableName());
                tableConfiguration.setDomainObjectName(tableMap.getModelName());
    
                tableConfiguration.setCountByExampleStatementEnabled(false);
                tableConfiguration.setUpdateByExampleStatementEnabled(false);
                tableConfiguration.setDeleteByExampleStatementEnabled(false);
                tableConfiguration.setSelectByExampleStatementEnabled(false);
                tableConfiguration.setSelectByExampleQueryId("false");
                defaultContext.getTableConfigurations().add(tableConfiguration);
            }
    
            // model
            defaultContext.getJavaModelGeneratorConfiguration().setTargetPackage(commonPackage);
            defaultContext.getJavaModelGeneratorConfiguration().setTargetProject(commonProject + "/src/main/java");
    
            for (TargetMap targetMap : targetMapList) { 
                // dao
                defaultContext.getJavaClientGeneratorConfiguration().setTargetPackage(targetMap.getTargetDaoPackage());
                defaultContext.getJavaClientGeneratorConfiguration().setTargetProject(targetMap.getTargetProject() + "/src/main/java");
                // mapper
                defaultContext.getSqlMapGeneratorConfiguration().setTargetPackage(targetMap.getTargetMapperPackage());
                defaultContext.getSqlMapGeneratorConfiguration().setTargetProject(targetMap.getTargetProject() + "/src/main/java");
    
                DefaultShellCallback callback = new DefaultShellCallback(overwrite);
                //创建 MBG
                MyBatisGenerator myBatisGenerator = new MyBatisGenerator(config, callback, warnings);
                //执行生成代码
                myBatisGenerator.generate(null);
                //输出警告信息
                for (String warning : warnings) { 
                    System.out.println(warning);
                }
            }
    
    
        }
    
    
        static class TableMap { 
            // DB 表名
            private String tableName;
            // java对象名
            private String modelName;
    
            public String getTableName() { 
                return tableName;
            }
    
            public void setTableName(String tableName) { 
                this.tableName = tableName;
            }
    
            public String getModelName() { 
                return modelName;
            }
    
            public void setModelName(String modelName) { 
                this.modelName = modelName;
            }
        }
    
        static class TargetMap { 
            // 项目名
            private String targetProject;
            // Dao
            private String targetDaoPackage;
            // Mapper
            private String targetMapperPackage;
    
            public String getTargetProject() { 
                return targetProject;
            }
    
            public void setTargetProject(String targetProject) { 
                this.targetProject = targetProject;
            }
    
            public String getTargetDaoPackage() { 
                return targetDaoPackage;
            }
    
            public void setTargetDaoPackage(String targetDaoPackage) { 
                this.targetDaoPackage = targetDaoPackage;
            }
    
            public String getTargetMapperPackage() { 
                return targetMapperPackage;
            }
    
            public void setTargetMapperPackage(String targetMapperPackage) { 
                this.targetMapperPackage = targetMapperPackage;
            }
        }
    }

generatorConfig.xml，存放src/main/resources

<?xml version="1.0" encoding="UTF-8"?>
    <!DOCTYPE generatorConfiguration
            PUBLIC "-//mybatis.org//DTD MyBatis Generator Configuration 1.0//EN"
            "http://mybatis.org/dtd/mybatis-generator-config_1_0.dtd">
    
    <generatorConfiguration>
    
    
        <context id="default" targetRuntime="MyBatis3">
    
            <property name="beginningDelimiter" value="`"/>
            <property name="endingDelimiter" value="`"/>
            <property name="javaFileEncoding" value="UTF-8"/>
    		
            <plugin type="org.mybatis.generator.plugins.ToStringPlugin"/>
    
            
            <plugin type="com.sendbp.eduz.generator.MapperPlugin">
                <property name="mappers" value="com.sendbp.eduz.util.common.MyMapper"/>
                <property name="swagger" value="true"/>
            </plugin>
    
            
            <commentGenerator>
                
                <property name="suppressAllComments" value="true"/>
                
                <property name="addRemarkComments" value="true"/>
                <property name="suppressDate" value="true"/>
            </commentGenerator>
    
            <jdbcConnection driverClass="com.mysql.jdbc.Driver"
                            connectionURL="jdbc:mysql://101.132.117.5:3306/mamaweike-beta"
                            userId="dev" password="XmxDLKwrR9M3wG3x">
            </jdbcConnection>
    
            
            <javaTypeResolver>
                <property name="forceBigDecimals" value="true"/>
            </javaTypeResolver>
    
            
            <javaModelGenerator targetPackage="com.**.model.exercise"
                                targetProject="${项目名}/src/main/java">
                
                <property name="enableSubPackages" value="false"/>
                
                <property name="trimStrings" value="true"/>
            </javaModelGenerator>
    
            
            <sqlMapGenerator targetPackage="com.**.exercise.mapper" targetProject="${项目名}/src/main/java">
                
                <property name="enableSubPackages" value="false"/>
            </sqlMapGenerator>
    
            
            <javaClientGenerator type="XMLMAPPER" targetPackage="com.**.exercise.dao"
                                 targetProject="${项目名}/src/main/java">
                
                <property name="enableSubPackages" value="false"/>
            </javaClientGenerator>
    
    
            <table tableName="exercise_test" domainObjectName="ExerciseTest"
                   enableCountByExample="false" enableUpdateByExample="false"
                   enableDeleteByExample="false" enableSelectByExample="false"
                   selectByExampleQueryId="false"></table>
    
        </context>
    </generatorConfiguration>