.

src/main/java/me/mizhoux/mbgcomment/MySQLCommentGenerator.java

@@ -3,10 +3,10 @@ package me.mizhoux.mbgcomment;
 import org.mybatis.generator.api.IntrospectedColumn;
 import org.mybatis.generator.api.IntrospectedTable;
 import org.mybatis.generator.api.dom.java.Field;
-import org.mybatis.generator.api.dom.java.TopLevelClass;
+import org.mybatis.generator.api.dom.java.Method;
+import org.mybatis.generator.api.dom.xml.TextElement;
+import org.mybatis.generator.api.dom.xml.XmlElement;
 
-import java.text.SimpleDateFormat;
-import java.util.Date;
 import java.util.Properties;
 
 public class MySQLCommentGenerator extends EmptyCommentGenerator {
@@ -23,22 +23,6 @@ public class MySQLCommentGenerator extends EmptyCommentGenerator {
         this.properties.putAll(properties);
     }
 
-    @Override
-    public void addModelClassComment(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
-        String author = properties.getProperty("author");
-        String dateFormat = properties.getProperty("dateFormat", "yyyy-MM-dd");
-        SimpleDateFormat dateFormatter = new SimpleDateFormat(dateFormat);
-
-        // 获取表注释
-        String remarks = introspectedTable.getRemarks();
-
-        topLevelClass.addJavaDocLine("/**");
-        topLevelClass.addJavaDocLine(" * " + remarks);
-        topLevelClass.addJavaDocLine(" *");
-        topLevelClass.addJavaDocLine(" * @author " + author);
-        topLevelClass.addJavaDocLine(" * @date   " + dateFormatter.format(new Date()));
-        topLevelClass.addJavaDocLine(" */");
-    }
 
     @Override
     public void addFieldComment(Field field, IntrospectedTable introspectedTable, IntrospectedColumn introspectedColumn) {
@@ -46,6 +30,32 @@ public class MySQLCommentGenerator extends EmptyCommentGenerator {
         String remarks = introspectedColumn.getRemarks();
         field.addJavaDocLine("/**");
         field.addJavaDocLine(" * " + remarks);
+        field.addJavaDocLine(" * This field was generated by MyBatis Generator, do not modify.");
+        field.addJavaDocLine(" * @mbg.generated");
         field.addJavaDocLine(" */");
     }
+
+    @Override
+    public void addGetterComment(Method method, IntrospectedTable introspectedTable, IntrospectedColumn introspectedColumn) {
+        method.addJavaDocLine("/**");
+        method.addJavaDocLine(" * This field was generated by MyBatis Generator, do not modify.");
+        method.addJavaDocLine(" * @mbg.generated");
+        method.addJavaDocLine(" */");
+    }
+
+    @Override
+    public void addSetterComment(Method method, IntrospectedTable introspectedTable, IntrospectedColumn introspectedColumn) {
+        method.addJavaDocLine("/**");
+        method.addJavaDocLine(" * This field was generated by MyBatis Generator, do not modify.");
+        method.addJavaDocLine(" * @mbg.generated");
+        method.addJavaDocLine(" */");
+    }
+
+    @Override
+    public void addComment(XmlElement xmlElement) {
+        xmlElement.addElement(new TextElement("<!--\n" +
+                "      WARNING - @mbg.generated\n" +
+                "      This element is automatically generated by MyBatis Generator, do not modify.\n" +
+                "    -->"));
+    }
 }