国产AV无码专区亚洲AV毛网站,亚洲欧美熟妇综合久久久久 ,女bbbbxxxx另类亚洲

修改mybatis-generator-1.3.2源碼實現(xiàn)自定義代碼生成詳解（二）

  上一篇對mybatis-generator-core源碼的基礎(chǔ)修改作了說明，也可以說是最想改的地方，這一篇將在此基礎(chǔ)上再做進(jìn)一步的修改。
  首先說說上一篇最后提到的關(guān)于自定義注釋的問題，想實現(xiàn)這個功能就需要修改org.mybatis.generator.internal.DefaultCommentGenerator這個類。我將里面一些主要的方法講下。
  addJavaFileComment(CompilationUnit compilationUnit)
  給Java文件加注釋，這個注釋是在文件的頂部，也就是package上面。

  addComment(XmlElement xmlElement)
  給生成的XML文件加注釋。大象將這個方法清空了，不生成注釋。

  addClassComment(InnerClass innerClass,IntrospectedTable introspectedTable)
  Java類的類注釋。

請注意紅線的getRemarks()方法，這個remarks屬性在原來的FullyQualifiedTable里面是沒有的，這是大象自己加上去的，就是為了保存表的注釋信息。那是在哪里加進(jìn)去的呢？請看org.mybatis.generator.internal.db.DatabaseIntrospector這個類，大概瀏覽下就會發(fā)現(xiàn)，數(shù)據(jù)庫表以及列的信息讀取，類型設(shè)置都是由它來完成的，定位到608行，正好這里是個空行，插入幾行代碼。如下：

  這樣我們就取到了表的注釋信息，看到這里應(yīng)該就會明白了吧？
  另外我需要說明的是，想通過databaseMetaData.getTables()來獲得表注釋的童鞋，這個做法是沒用的，大象已經(jīng)試過了，這個結(jié)果集里面的rs.getString("REMARKS")得到的是一個空字符串，什么都木有。而且大象還想吐槽下，網(wǎng)上一大堆說返回的這個結(jié)果集是10列，呵呵，是的么？我debug了好久發(fā)現(xiàn)它還是只有5列，不知道這10列是從哪來的，請打印出10列的童鞋告之是怎么做的，我用的mysql驅(qū)動是5.1.29

  addFieldComment(Field field, IntrospectedTable introspectedTable,IntrospectedColumn introspectedColumn)
  Java屬性注釋。注釋為空就不給屬性添加。

  addGetterComment(Method method,IntrospectedTable introspectedTable,IntrospectedColumn introspectedColumn)
  給getter方法加注釋。這里添加注釋的方法和Field一樣，大象把這個方法以及addSetterComment方法都清空了，因為我比較習(xí)慣把注釋加到屬性上面，如果你習(xí)慣在getter方法上面加注釋可以自行改一下。
  上面這些修改做完后，記得要在org.mybatis.generator.codegen.mybatis3.model.BaseRecordGenerator的61行增加下面這行代碼：
  commentGenerator.addClassComment(topLevelClass, introspectedTable);
  因為mybatis-generator源碼中本來就是不加類注釋的。
  大家都在不同的公司，要求肯定都不一樣，大象沒法滿足所有人，只對該類做了一定程度的修改，姑且把它當(dāng)做一個示例模板吧，主要是弄明白怎么改就成了。

  mybatis-generator在1.3.2版里對生成xml的namespace作了修改，不再用表名當(dāng)為namespace的值，而是用包結(jié)構(gòu)+類名+Mapper后綴的形式設(shè)置，大象不喜歡這么復(fù)雜做法，所以這里需要改一改，只保留類名+Mapper的命名形式。去掉org.mybatis.generator.api.IntrospectedTable類的907行與908行代碼。

  接下來再來說說對大對象類型的處理，mybatis-generator默認(rèn)會把所有jdbcType為：BINARY、BLOB、CLOB、LONGVARBINARY、LONGVARCHAR、VARBINARY這些類型都作為大對象，反應(yīng)出來的效果就是生成的pojo類會多一個類名+WithBLOBs.java的文件(含有的大對象個數(shù)大于1時)，而在XML里面也會增加一個id為ResultMapWithBLOBs的resultMap，它繼承BaseResultMap，大象一向喜歡簡單的風(fēng)格，這看得太不爽了。如果你覺得無所謂，不需要修改，請?zhí)^這段往下看。
  表中的text或blob等類型，通過databaseMetaData.getColumns()取出來后，它的rs.getInt("DATA_TYPE")值是與java.sql.Types對應(yīng)的。
  text類型的值：Types.LONGVARCHAR
  blob類型是值：Types.LONGVARBINARY
  它用org.mybatis.generator.internal.types.JavaTypeResolverDefaultImpl解析java類型與jdbc類型，然后設(shè)置到org.mybatis.generator.api.IntrospectedColumn的jdbcTypeName屬性里，最關(guān)鍵的地方到了，IntrospectedColumn類第156行有個isBLOBColumn()方法，它就是用來判斷是否生成xxxWithBLOBs.java和ResultMapWithBLOBs的條件。當(dāng)然mybatis-generator不會這么直接用，它在里面定義了一個Rules接口，由它來統(tǒng)一進(jìn)行調(diào)用。所以我們只需要把isBLOBColumn()里面的代碼都去掉，直接返回false就可以了，再結(jié)合配置文件中的columnOverride屬性就能達(dá)到目的。

  jdbcType指定的值就是生成xml后result里面對應(yīng)的類型，javaType與生成的pojo里面屬性類型一致，這里其實可以不定義javaType，但是當(dāng)數(shù)據(jù)庫的表字段有的定義為tinyint時，如果不給它指定類型，那么默認(rèn)生成的java類型就是java.lang.Byte，假如你想轉(zhuǎn)換類型，而且保證不會超出字段類型的大小限制，那么你可以將它的javaType設(shè)為java.lang.Integer
  經(jīng)過這么一番修改之后，輸出結(jié)果已經(jīng)比較理想了，但是xml文件看上去還是有點問題，元素與元素之間沒空行，看起來好別扭。恩，這里要給它加上空行，所以請找到org.mybatis.generator.api.dom.xml.XmlElement類的getFormattedContent方法，這個方法有個參數(shù)：indentLevel，通過名字我們可以很直觀的理解它的作用就是來控制縮進(jìn)等級，它的初始值為0，這表示不縮進(jìn)，當(dāng)變成1時就是縮進(jìn)一級，也即縮進(jìn)四個空格，后面以此類推，我們應(yīng)該還注意到，方法內(nèi)部用到了遞歸，所以它是從最底元素開始，再一步步返回，所以我們就應(yīng)該在indentLevel為1的元素后面加上換行。

在生成xml的文件里面，我們看到有一個update元素，id為updateByPrimaryKey，這個基本上用不到，但是你還不能在table里加上enableUpdateByPrimaryKey="false"，如果你這么做，雖然updateByPrimaryKey沒有了，但updateByPrimaryKeySelective也會消失。所以這時請看看org.mybatis.generator.codegen.mybatis3.xmlmapper.XMLMapperGenerator這個類，getSqlMapElement()里面，有大量的addXXX方法，這些方法里面每個都用到了Rules規(guī)則來處理是否執(zhí)行，比如我們在table里面加了enableDeleteByExample="false"，它對應(yīng)的是addDeleteByExampleElement()，而它又調(diào)用了introspectedTable.getRules().generateDeleteByExample()方法，再進(jìn)入到generateDeleteByExample()里面，我們可以看到紅線部分，上一篇大象講過，在加載配置文件的時候，TableConfiguration會將table中的這些屬性設(shè)置到對應(yīng)的屬性里，所以說其實最后還是回到了判斷enableDeleteByExample的布爾值上面。

  上面啰嗦了一大堆，現(xiàn)在再來解決之前提到的問題，如何去掉id為updateByPrimaryKey的update元素，注釋掉addUpdateByPrimaryKeyWithoutBLOBsElement()或直接去掉這行代碼，跟蹤代碼你會發(fā)現(xiàn)，它和addUpdateByPrimaryKeySelectiveElement()的rules驗證里面都用到了enableUpdateByPrimaryKey的值。或者你不想采取我說的這個辦法，而是改BaseRules的generateUpdateByPrimaryKeyWithoutBLOBs()，讓它直接返回false
  最后大象再嘮叨一句，這些addXXX方法的順序決定了生成xml文件中的元素順序，所以有代碼潔癖的人可以作下調(diào)整。
  上一篇加這一篇都是講怎么少量的改動源碼以實現(xiàn)自定義文件輸出，寫的比較凌亂，不是很系統(tǒng)，完全是從實用的角度出發(fā)，關(guān)鍵還是大象水平有限，有什么錯誤還請大家?guī)臀抑赋鰜恚x謝！
  改了這么多，總要看下效果吧，下一篇大象將寫個測試看看生成的結(jié)果，然后用maven assembly將它打包生成一個zip，它將包含：

本文為菠蘿大象原創(chuàng)，如要轉(zhuǎn)載請注明出處。http://m.tkk7.com/bolo

posted on 2015-03-25 16:33 菠蘿大象閱讀(26589) 評論(0) 編輯收藏所屬分類: mybatis-generator

隨筆分類(67)

隨筆檔案(67)

搜索

積分與排名

最新隨筆

最新評論

閱讀排行榜

評論排行榜


只有注冊用戶登錄后才能發(fā)表評論。




網(wǎng)站導(dǎo)航: 博客園 IT新聞 Chat2DB C++博客博問管理
相關(guān)文章: 修改mybatis-generator-1.3.2源碼實現(xiàn)自定義代碼生成詳解（三）修改mybatis-generator-1.3.2源碼實現(xiàn)自定義代碼生成詳解（二）修改mybatis-generator-1.3.2源碼實現(xiàn)自定義代碼生成詳解（一）