• 支持生成xml,mapper,po,repository
• 用于替代MyBatis Generator

• 修改config/application.properties中的数据库配置
• 运行com.tz.code.generator.CodeGeneratorApplication#main方法即可生成代码

在使用MyBatis Generator生成代码时遇到了以下一些问题,所以自定义一个代码生成器来解决这些问题,从而替代MyBatis Generator用于代码的生成。

• 生成的代码文件中有一堆无用的注释,看着不简洁清晰

  /**
   *
   * This class was generated by MyBatis Generator.
   * This class corresponds to the database table t_config
   *
   * @mbg.generated do_not_delete_during_merge Sun Jan 08 15:31:44 CST 2023
   */
  public class ConfigPO {
   /**
   * Database Column Remarks:
   * 主键
   *
   * This field was generated by MyBatis Generator.
   * This field corresponds to the database column t_config.id
   *
   * @mbg.generated Sun Jan 08 15:31:44 CST 2023
   */
   private Long id;
   /**
   * Database Column Remarks:
   * 服务名
   *
   * This field was generated by MyBatis Generator.
   * This field corresponds to the database column t_config.service_name
   *
   * @mbg.generated Sun Jan 08 15:31:44 CST 2023
   */
   private String serviceName;
   
   //其他代码省略
  }

• 表中tinyint类型的字段生成的PO的属性对应的是Byte的类型,不方便和Integer类型的数据进行比较或者设置等操作

  CREATE TABLE `t_config` (
   `id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT '主键',
   `config_type` tinyint(2) NOT NULL COMMENT '类型[10=生效缓存配置,20=失效缓存配置]',
   PRIMARY KEY (`id`)
  ) ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin COMMENT='配置表';

  public class ConfigPO {
  	/**
   * Database Column Remarks:
   * 主键
   *
   * This field was generated by MyBatis Generator.
   * This field corresponds to the database column t_config_test_1.id
   *
   * @mbg.generated Sun Jan 08 15:50:31 CST 2023
   */
   private Long id;
   
   /**
   * Database Column Remarks:
   * 类型[10=生效缓存配置,20=失效缓存配置]
   *
   * This field was generated by MyBatis Generator.
   * This field corresponds to the database column t_config.config_type
   *
   * @mbg.generated Sun Jan 08 15:50:31 CST 2023
   */
   private Byte configType;
  }

• 生成的代码文件是通过在代码中进行拼接实现,修改起来不直观和方便

  例如生成selectByPrimaryKey方法的sql代码类:

  // org.mybatis.generator.codegen.mybatis3.xmlmapper.elements.SelectByPrimaryKeyElementGenerator
  public class SelectByPrimaryKeyElementGenerator extends
   AbstractXmlElementGenerator {
   public SelectByPrimaryKeyElementGenerator() {
   super();
   }
   @Override
   public void addElements(XmlElement parentElement) {
   XmlElement answer = new XmlElement("select"); //$NON-NLS-1$
   answer.addAttribute(new Attribute(
   "id", introspectedTable.getSelectByPrimaryKeyStatementId())); //$NON-NLS-1$
   if (introspectedTable.getRules().generateResultMapWithBLOBs()) {
   answer.addAttribute(new Attribute("resultMap", //$NON-NLS-1$
   introspectedTable.getResultMapWithBLOBsId()));
   } else {
   answer.addAttribute(new Attribute("resultMap", //$NON-NLS-1$
   introspectedTable.getBaseResultMapId()));
   }
   String parameterType;
   if (introspectedTable.getRules().generatePrimaryKeyClass()) {
   parameterType = introspectedTable.getPrimaryKeyType();
   } else {
   // PK fields are in the base class. If more than on PK
   // field, then they are coming in a map.
   if (introspectedTable.getPrimaryKeyColumns().size() > 1) {
   parameterType = "map"; //$NON-NLS-1$
   } else {
   parameterType = introspectedTable.getPrimaryKeyColumns().get(0)
   .getFullyQualifiedJavaType().toString();
   }
   }
   answer.addAttribute(new Attribute("parameterType", //$NON-NLS-1$
   parameterType));
   context.getCommentGenerator().addComment(answer);
   StringBuilder sb = new StringBuilder();
   sb.append("select "); //$NON-NLS-1$
   if (stringHasValue(introspectedTable
   .getSelectByPrimaryKeyQueryId())) {
   sb.append('\'');
   sb.append(introspectedTable.getSelectByPrimaryKeyQueryId());
   sb.append("' as QUERYID,"); //$NON-NLS-1$
   }
   answer.addElement(new TextElement(sb.toString()));
   answer.addElement(getBaseColumnListElement());
   if (introspectedTable.hasBLOBColumns()) {
   answer.addElement(new TextElement(",")); //$NON-NLS-1$
   answer.addElement(getBlobColumnListElement());
   }
   sb.setLength(0);
   sb.append("from "); //$NON-NLS-1$
   sb.append(introspectedTable
   .getAliasedFullyQualifiedTableNameAtRuntime());
   answer.addElement(new TextElement(sb.toString()));
   boolean and = false;
   for (IntrospectedColumn introspectedColumn : introspectedTable
   .getPrimaryKeyColumns()) {
   sb.setLength(0);
   if (and) {
   sb.append(" and "); //$NON-NLS-1$
   } else {
   sb.append("where "); //$NON-NLS-1$
   and = true;
   }
   sb.append(MyBatis3FormattingUtilities
   .getAliasedEscapedColumnName(introspectedColumn));
   sb.append(" = "); //$NON-NLS-1$
   sb.append(MyBatis3FormattingUtilities
   .getParameterClause(introspectedColumn));
   answer.addElement(new TextElement(sb.toString()));
   }
   if (context.getPlugins()
   .sqlMapSelectByPrimaryKeyElementGenerated(answer,
   introspectedTable)) {
   parentElement.addElement(answer);
   }
   }
  }