1. 程式人生 > >mybatis-generator資料庫註釋實體類生成以及generatorConfig檔案配置

mybatis-generator資料庫註釋實體類生成以及generatorConfig檔案配置

專案裡新建表時model,mapper以及mapper.xml基本都是用Mybatis Generator(以下簡稱為MBG)自動生成的,但是MBG自動生成的model的註釋實在有點非人類,至少中國人是完全接受不了的,在配置中禁用掉註釋吧,倒是簡單了,可是生成的model類光禿禿的,啥都沒有,欄位方法沒有註釋,使用很不方便,別人看也不知道這個欄位是啥含義,到最後還是要自己新增,一張表多點幾十個欄位,特麼新增累死了,不能這麼幹,得想法子,百度了一下網上,基本有兩種方法,第一種,就是直接修改MGB的原始碼,第二種就是自己寫一個類實現CommentGenerator介面,先說自己寫一個新類實現CommentGenerator介面的方法來使自動生成的model類含有中文註釋吧.

一:首先你得有一個maven專案,這個我就不多提

二:通過mybatis generator實現自動生產model,mapper以及mapper.xml,具體參考 Intellij IDEA 14中使用MyBatis-generator 自動生成MyBatis程式碼 

三:通過檢視你會發現,註解類是在mybatis-generator-core jar包中  org.mybatis.generator.internal包下的DefaultCommentGenerator類。那麼這裡有兩種方法,一種是重寫DefaultCommentGenerator類,另外一種中修改jar包中的原始碼,但很明顯第一種方法比較簡單,這裡也只介紹第一種方法的寫法。

3.1在原始碼中新建一個類MyCommentGenerator,實現CommentGenerator介面,類的程式碼如下:

package org.mybatis.generator;
 
/**
 * Created by 草帽boy on 2017/2/16.
 * mybatis generator 自定義comment生成器.
 * 基於MBG 1.3.2.
 * @author ZhangAY 2016-02-19
 */
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Properties;
 
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.XmlElement; import org.mybatis.generator.config.MergeConstants; import org.mybatis.generator.config.PropertyRegistry; import org.mybatis.generator.internal.util.StringUtility; public class MyCommentGenerator implements CommentGenerator { private Properties properties; private Properties systemPro; private boolean suppressDate; private boolean suppressAllComments; private String currentDateStr; public MyCommentGenerator() { super(); properties = new Properties(); systemPro = System.getProperties(); suppressDate = false; suppressAllComments = false; currentDateStr = (new SimpleDateFormat("yyyy-MM-dd")).format(new Date()); } public void addJavaFileComment(CompilationUnit compilationUnit) { // add no file level comments by default return; } /** * Adds a suitable comment to warn users that the element was generated, and * when it was generated. */ public void addComment(XmlElement xmlElement) { return; } public void addRootComment(XmlElement rootElement) { // add no document level comments by default return; } public void addConfigurationProperties(Properties properties) { this.properties.putAll(properties); suppressDate = StringUtility.isTrue(properties.getProperty(PropertyRegistry.COMMENT_GENERATOR_SUPPRESS_DATE)); suppressAllComments = StringUtility.isTrue(properties.getProperty(PropertyRegistry.COMMENT_GENERATOR_SUPPRESS_ALL_COMMENTS)); } /** * This method adds the custom javadoc tag for. You may do nothing if you do * not wish to include the Javadoc tag - however, if you do not include the * Javadoc tag then the Java merge capability of the eclipse plugin will * break. * * @param javaElement the java element */ protected void addJavadocTag(JavaElement javaElement, boolean markAsDoNotDelete) { javaElement.addJavaDocLine(" *"); StringBuilder sb = new StringBuilder(); sb.append(" * "); sb.append(MergeConstants.NEW_ELEMENT_TAG); if (markAsDoNotDelete) { sb.append(" do_not_delete_during_merge"); } String s = getDateString(); if (s != null) { sb.append(' '); sb.append(s); } javaElement.addJavaDocLine(sb.toString()); } /** * This method returns a formated date string to include in the Javadoc tag * and XML comments. You may return null if you do not want the date in * these documentation elements. * * @return a string representing the current timestamp, or null */ protected String getDateString() { String result = null; if (!suppressDate) { result = currentDateStr; } return result; } public void addClassComment(InnerClass innerClass, IntrospectedTable introspectedTable) { if (suppressAllComments) { return; } StringBuilder sb = new StringBuilder(); innerClass.addJavaDocLine("/**"); sb.append(" * "); sb.append(introspectedTable.getFullyQualifiedTable()); sb.append(" "); sb.append(getDateString()); innerClass.addJavaDocLine(sb.toString()); innerClass.addJavaDocLine(" */"); } public void addEnumComment(InnerEnum innerEnum, IntrospectedTable introspectedTable) { if (suppressAllComments) { return; } StringBuilder sb = new StringBuilder(); innerEnum.addJavaDocLine("/**"); // addJavadocTag(innerEnum, false); sb.append(" * "); sb.append(introspectedTable.getFullyQualifiedTable()); innerEnum.addJavaDocLine(sb.toString()); innerEnum.addJavaDocLine(" */"); } public void addFieldComment(Field field, IntrospectedTable introspectedTable, IntrospectedColumn introspectedColumn) { if (suppressAllComments) { return; } StringBuilder sb = new StringBuilder(); field.addJavaDocLine("/**"); sb.append(" * "); sb.append(introspectedColumn.getRemarks()); field.addJavaDocLine(sb.toString()); // addJavadocTag(field, false); field.addJavaDocLine(" */"); } public void addFieldComment(Field field, IntrospectedTable introspectedTable) { if (suppressAllComments) { return; } StringBuilder sb = new StringBuilder(); field.addJavaDocLine("/**"); sb.append(" * "); sb.append(introspectedTable.getFullyQualifiedTable()); field.addJavaDocLine(sb.toString()); field.addJavaDocLine(" */"); } public void addModelClassComment(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) { } public void addGeneralMethodComment(Method method, IntrospectedTable introspectedTable) { if (suppressAllComments) { return; } // method.addJavaDocLine("/**"); // addJavadocTag(method, false); // method.addJavaDocLine(" */"); } public void addGetterComment(Method method, IntrospectedTable introspectedTable, IntrospectedColumn introspectedColumn) { if (suppressAllComments) { return; } method.addJavaDocLine("/**"); StringBuilder sb = new StringBuilder(); sb.append(" * "); sb.append(introspectedColumn.getRemarks()); method.addJavaDocLine(sb.toString()); sb.setLength(0); sb.append(" * @return "); sb.append(introspectedColumn.getActualColumnName()); sb.append(" "); sb.append(introspectedColumn.getRemarks()); method.addJavaDocLine(sb.toString()); // addJavadocTag(method, false); method.addJavaDocLine(" */"); } public void addSetterComment(Method method, IntrospectedTable introspectedTable, IntrospectedColumn introspectedColumn) { if (suppressAllComments) { return; } method.addJavaDocLine("/**"); StringBuilder sb = new StringBuilder(); sb.append(" * "); sb.append(introspectedColumn.getRemarks()); method.addJavaDocLine(sb.toString()); Parameter parm = method.getParameters().get(0); sb.setLength(0); sb.append(" * @param "); sb.append(parm.getName()); sb.append(" "); sb.append(introspectedColumn.getRemarks()); method.addJavaDocLine(sb.toString()); // addJavadocTag(method, false); method.addJavaDocLine(" */"); } public void addClassComment(InnerClass innerClass, IntrospectedTable introspectedTable, boolean markAsDoNotDelete) { if (suppressAllComments) { return; } StringBuilder sb = new StringBuilder(); innerClass.addJavaDocLine("/**"); sb.append(" * "); sb.append(introspectedTable.getFullyQualifiedTable()); innerClass.addJavaDocLine(sb.toString()); sb.setLength(0); sb.append(" * @author "); sb.append(systemPro.getProperty("user.name")); sb.append(" "); sb.append(currentDateStr); // addJavadocTag(innerClass, markAsDoNotDelete); innerClass.addJavaDocLine(" */"); } }

 


3.2.再新建一個類StartUp,用於執行專案,(若是整合Eclipse,可直接執行操作,檢視mybatis與eclipse的整合,這裡不介紹了)也就是執行StartUp類 就會直接生成model,mapper以及mapper.xml,類的程式碼如下:

 

package org.mybatis.generator;
 
/**
 * Created by 草帽boy on 2017/2/16.
 * 啟動檔案,只需要點選執行就行
 */
import java.io.File;
import java.io.IOException;
import java.net.URISyntaxException;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
 
import org.mybatis.generator.api.MyBatisGenerator;
import org.mybatis.generator.config.Configuration;
import org.mybatis.generator.config.xml.ConfigurationParser;
import org.mybatis.generator.exception.InvalidConfigurationException;
import org.mybatis.generator.exception.XMLParserException;
import org.mybatis.generator.internal.DefaultShellCallback;
 
public class StartUp {
    public static void main(String[] args) throws URISyntaxException {
        try {
            List<String> warnings = new ArrayList<String>();
            boolean overwrite = true;
            //直接獲取generatorConfig.xml的檔案路徑 根據具體情況檢視
            File configFile = new File("E:\\IDEAWorkPlace\\GraduationDesign\\CourseDesignManageSystem\\20170122\\CourseDesignManage\\src\\main\\resources\\generatorConfig.xml");
            ConfigurationParser cp = new ConfigurationParser(warnings);
            Configuration config = cp.parseConfiguration(configFile);
            DefaultShellCallback callback = new DefaultShellCallback(overwrite);
            MyBatisGenerator myBatisGenerator = new MyBatisGenerator(config, callback, warnings);
            myBatisGenerator.generate(null);
        } catch (SQLException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        } catch (InterruptedException e) {
            e.printStackTrace();
        } catch (InvalidConfigurationException e) {
            e.printStackTrace();
        } catch (XMLParserException e) {
            e.printStackTrace();
        }
    }
}

 

3.3 另外需要簡單修改一下generatorConfig.xml中的註釋配置

  <commentGenerator type="org.mybatis.generator.MyCommentGenerator">
           <!-- <property name="suppressDate" value="true"/>
            &lt;!&ndash; 是否去除自動生成的註釋 true:是 : false:否 &ndash;&gt;
            <property name="suppressAllComments" value="false"/>-->
  </commentGenerator>

commentGenerator 的type是你剛剛重構的MyCommentGenerator類的位置


3.4其中我的檔案結構如下:

 

四:點選執行StartUp,你會發現model,mapper以及mapper.xml都已經產生,其中實體類的 執行的結果如下

 


注意:若是出現中文亂碼,在generatorConfig.xml檔案中增加以下配置

<property name="javaFileEncoding" value="UTF-8"/>

1.以下是generatorConfig.xml配置檔案

<?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 >
<!-- 資料庫驅動包位置 -->  
    <classPathEntry  
        location="D:\repository\mysql\mysql-connector-java\5.1.34\mysql-connector-java-5.1.34.jar" />  
    <context id="context1"> 
         <property name="javaFileEncoding" value="UTF-8"/> 
         <commentGenerator type="com.wareic.utils.MyCommentGenerator"></commentGenerator> 
        <!-- <commentGenerator>  
            是否去除自動生成的註釋 true:是 : false:否  
            <property name="suppressAllComments" value="true"/>  
        </commentGenerator>   -->
        <!-- 資料庫連結URL、使用者名稱、密碼 -->  
        <jdbcConnection driverClass="com.mysql.jdbc.Driver"  
            connectionURL="jdbc:mysql://17.16.32.2:3306/db_wareic" userId="root" password="123456" />  
        <!-- 生成模型的包名和位置  wareic為專案名稱 -->  
        <javaModelGenerator targetPackage="com.wareic.model" targetProject="wareic/src/main/java" />  
        <!-- 生成的對映檔案報名和位置 -->  
        <sqlMapGenerator targetPackage="com.wareic.mapper" targetProject="wareic/src/main/java" />  
        <!-- 生成DAO的包名和位置 -->  
        <javaClientGenerator targetPackage="com.wareic.mapper" targetProject="wareic/src/main/java" type="XMLMAPPER" />  
        <!-- 要生成的那些表(更改tableName 和domainObjectName 就可以了) -->  
        <table schema="db_wareic" tableName="t_product" domainObjectName="Product" enableCountByExample="false" enableUpdateByExample="false"  
            enableDeleteByExample="false" enableSelectByExample="false" selectByExampleQueryId="false">  
           <!--  <columnOverride column="FILMID" property="FILMID" />   -->
        </table>  
    </context>  
</generatorConfiguration>

2.執行startUp型別中的main函式

3.MyBatis Generator官網: http://www.mybatis.org/generator/

 

---------------------
作者:王衛東
來源:CSDN
原文:https://blog.csdn.net/wwd0501/article/details/76618363
版權宣告:本文為博主原創文章,轉載請附上博文連結!