Mybatis-generator 注释插件,如何为?
摘要:插件地址(https:github.comsuyin58mybatis-generator-tddlblobmastergenerator-pluginsrcmainjavacomtoolplatgenerator
插件地址(https://github.com/suyin58/mybatis-generator-tddl/blob/master/generator-plugin/src/main/java/com/toolplat/generator/plugins/CommentGenerator.java)
可以修改类注释、属性注释、方法注释等
插件需要继承DefaultCommentGenerator
public class CommentGenerator extends DefaultCommentGenerator {
}
View Code
类注释需要重写addModelClassComment方法
/**
* set model comment from table remark
* @param topLevelClass
* @param introspectedTable
*/
@Override
public void addModelClassComment(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
String auth = properties.getProperty("author");
String tableDesc = introspectedTable.getRemarks();
topLevelClass.addJavaDocLine("/**");
topLevelClass.addJavaDocLine(" * " + tableDesc);
if (null != auth && auth.length() > 0) {
topLevelClass.addJavaDocLine(" * @author " + auth);
}
topLevelClass.addJavaDocLine(" */");
}
View Code
属性注释需要重写addFieldComment方法
/**
* set field comment from field remark
* @param field
* @param introspectedTable
* @param introspectedColumn
*/
@Override
public void addFieldComment(Field field, IntrospectedTable introspectedTable, IntrospectedColumn introspectedColumn) {
String remark = introspectedColumn.getRemarks();
field.addJavaDocLine("/**");
field.addJavaDocLine(" * " + remark);
field.addJavaDocLine(" */");
}
View Code
mapper方法注释需要重写addGeneralMethodComment方法
/**
* remove java mapping coment
*
* @param method
* @param introspectedTable
*/
@Override
public void addGeneralMethodComment(Method method, IntrospectedTable introspectedTable) {
// method.addJavaDocLine("/**");
// method.addJavaDocLine(" * auto method:" + method.getName());
// if (method.getParameters() != null && method.getParameters().size() > 0) {
// for (Parameter param : method.getParameters()) {
//
