文章目录
需求
上一节我们分析了如何使用mybatis-generator来生成一些文件,但是对于生成的文件自带了一些系统注释,显得杂乱无章。现在有更简洁的方法生成清爽的代码。
1、原始样式
按照上一节的方法,可以生成如下实体类,但是注释信息过于杂乱。
/**
*
* This class was generated by MyBatis Generator.
* This class corresponds to the database table ws_beneficiary_info
*
* @mbg.generated do_not_delete_during_merge Thu May 28 18:54:46 CST 2020
*/
public class WsBeneficiaryInfo {
/**
* 受益方式
*
* This field was generated by MyBatis Generator.
* This field corresponds to the database column ws_beneficiary_info.benf_mode
*
* @mbg.generated Thu May 28 18:54:46 CST 2020
*/
private Integer benfMode;
//setter,getter
}
但是我们想生成的是这样的
/**
* 受益人信息
* ws_beneficiary_info
*/
public class WsBeneficiaryInfo {
/**
* 受益方式
*/
private Integer benfMode;
//setter,getter
}
2、引入jar包
<dependency>
<groupId>org.mybatis.generator</groupId>
<artifactId>mybatis-generator-core</artifactId>
<version>1.3.6</version>
</dependency>
3、自定义实现CommentGenerator接口
import org.mybatis.generator.api.CommentGenerator;
import org.mybatis.generator.api.IntrospectedColumn;
import org.mybatis.generator.api.IntrospectedTable;
import org.mybatis.generator.api.MyBatisGenerator;
import org.mybatis.generator.api.dom.java.*;
import org.mybatis.generator.api.dom.xml.XmlElement;
import org.mybatis.generator.config.Configuration;
import org.mybatis.generator.config.PropertyRegistry;
import org.mybatis.generator.config.xml.ConfigurationParser;
import org.mybatis.generator.internal.DefaultShellCallback;
import org.mybatis.generator.internal.util.StringUtility;
import java.io.File;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.List;
import java.util.Properties;
import java.util.Set;
import static org.mybatis.generator.internal.util.StringUtility.isTrue;
/**
* @author
* @date 2020/5/28 19:48
* @description
*/
public class MyCommentGenerator implements CommentGenerator {
public static void main(String[] args) {
List<String> warnings = new ArrayList<>();
boolean overwrite = true;
// 给出generatorConfig.xml文件的位置,绝对地址或者类路径下
//File configFile = new File("E:\\xxxxxxxxx\\generatorConfig.xml");
File configFile = new File(MyCommentGenerator.class.getClassLoader().getResource("generatorConfi