Mybatis源碼分析(三)通過實例來看typeHandlers

一、案例分析

在日常開發中,我們肯定有對日期類型的操作。比如訂單時間、付款時間等,通常這一類數據在數據庫以datetime類型保存。如果需要在頁面上展示此值,在Java中以什么類型接收它呢?

在不執行任何二次操作的情況下:
java.util.Date接收,在頁面展示的就是Tue Oct 16 16:05:13 CST 2018
java.lang.String接收,在頁面展示的就是2018-10-16 16:10:47.0

顯然,我們不能顯示第一種。第二種似乎可行,但大部分情況下不能出現毫秒數。當然了,不管哪種方式,在顯示的時候format一下當然是可行的。有沒有更好的方式呢?

二、typeHandlers

無論是 MyBatis 在預處理語句(PreparedStatement)中設置一個參數時,還是從結果集中取出一個值時, 都會用類型處理器將獲取的值以合適的方式轉換成 Java 類型。
在數據庫中,datetime和timestamp類型含義是一樣的,不過timestamp存儲空間小, 所以它表示的時間范圍也更小。
下面來看幾個Mybatis默認的時間類型處理器。

JDBC 類型 Java 類型 類型處理器
DATE java.util.Date DateOnlyTypeHandler
DATE java.sql.Date SqlDateTypeHandler
DATE java.time.LocalDate LocalDateTypeHandler
DATE java.time.LocalTime LocalTimeTypeHandler
TIMESTAMP java.util.Date DateTypeHandler
TIMESTAMP java.time.Instant InstantTypeHandler
TIMESTAMP java.time.LocalDateTime LocalDateTimeTypeHandler
TIMESTAMP java.sql.Timestamp SqlTimestampTypeHandler

它是什么意思呢?如果數據庫字段類型為JDBC 類型,同時Java字段的類型為Java 類型,那么就調用類型處理器類型處理器

三、自定義處理器

基于上面這個邏輯,我們可以增加一種處理器來處理我們開頭所描述的問題。我們可以在Java中,以String類型接收數據庫的DateTime類型數據。因為現在的接口以restful風格居多,用String類型方便傳輸。
最后的毫秒數通過自定義的處理器統一截取去除即可。

JDBC 類型 Java 類型 類型處理器
TIMESTAMP java.lang.String CustomTypeHandler
<property name="typeHandlers">
    <array>
        <bean class="com.viewscenes.netsupervisor.util.CustomTypeHandler"></bean>
    </array>
</property>

@MappedJdbcTypes注解表示JDBC的類型,@MappedTypes表示Java屬性的類型。

@MappedJdbcTypes({ JdbcType.TIMESTAMP })
@MappedTypes({ String.class })
public class CustomTypeHandler extends BaseTypeHandler<String>{ 
    @Override
    public void setNonNullParameter(PreparedStatement ps, int i, String parameter, JdbcType jdbcType)
            throws SQLException {
        ps.setString(i, parameter);
    }
    @Override
    public String getNullableResult(ResultSet rs, String columnName) throws SQLException {
        return substring(rs.getString(columnName));
    }
    @Override
    public String getNullableResult(ResultSet rs, int columnIndex) throws SQLException {
        return rs.getString(columnIndex);
    }
    @Override
    public String getNullableResult(CallableStatement cs, int columnIndex) throws SQLException {
        return cs.getString(columnIndex);
    }
    private String substring(String value) {
        if (!"".endsWith(value) && value != null) {
            return value.substring(0, value.length() - 2);
        }
        return value;
    }
}

通過以上方式,我們就可以放心的在Java中以String接收數據庫的時間類型數據了。

四、源碼分析

1、注冊

public final class TypeHandlerRegistry {
    //typeHandler為當前自定義類型處理器
    public <T> void register(TypeHandler<T> typeHandler) {
        boolean mappedTypeFound = false;
        //mappedTypes即String
        MappedTypes mappedTypes = typeHandler.getClass().getAnnotation(MappedTypes.class);
        if (mappedTypes != null) {
            for (Class<?> handledType : mappedTypes.value()) {
                register(handledType, typeHandler);
            }
        }
    }
}
public final class TypeHandlerRegistry {
    private <T> void register(Type javaType, TypeHandler<? extends T> typeHandler) {
        //JDBC的類型,即TIMESTAMP
        MappedJdbcTypes mappedJdbcTypes = typeHandler.getClass().
                getAnnotation(MappedJdbcTypes.class);
        if (mappedJdbcTypes != null) {
            for (JdbcType handledJdbcType : mappedJdbcTypes.value()) {
                //TYPE_HANDLER_MAP是Java類型中的默認處理器。
                //以String為例,它默認可以處理VARCHAR、CHAR、NVARCHAR、CLOB、NCLOB、NULL
                Map<JdbcType, TypeHandler<?>> map = TYPE_HANDLER_MAP.get(javaType);
                //給String添加一種處理器為typeHandler
                map.put(jdbcType, typeHandler);
                //注冊處理器實例
                ALL_TYPE_HANDLERS_MAP.put(typeHandler.getClass(), typeHandler);
            }
        }
    }
}

2、調用

注冊完畢之后,它在什么地方生效呢?關鍵在于能否可以找到這個處理器。看完上面的注冊過程,查找其實很簡單。先從TYPE_HANDLER_MAP根據JavaType,獲取String類型的全部處理器,再從中過濾出JDBC類型為TIMESTAMP的即可。

private <T> TypeHandler<T> getTypeHandler(Type type, JdbcType jdbcType) {
    //根據JavaType獲取String類型的全部處理器
    Map<JdbcType, TypeHandler<?>> jdbcHandlerMap = getJdbcHandlerMap(type);
    TypeHandler<?> handler = null;
    if (jdbcHandlerMap != null) {
        //再根據jdbcType獲取到TIMESTAMP的處理器
        handler = jdbcHandlerMap.get(jdbcType);
    }
    return (TypeHandler<T>) handler;
}

拿到自定義的處理器,我們自己就隨便搞嘍~

不過,在Mybatis-3.2.7版本中,比較坑。在調用getTypeHandler方法時,它并沒有傳jdbcType這個參數,所以這個參數默認為NULL了。
那么,在執行jdbcHandlerMap.get(jdbcType)的時候,會找不到自定義的處理器,而是找到了NULL的處理器,即StringHandler。案發現場在下面:

public class ResultSetWrapper {
    public TypeHandler<?> getTypeHandler(Class<?> propertyType, String columnName) {
        //3.4.6
        JdbcType jdbcType = getJdbcType(columnName);
        handler = typeHandlerRegistry.getTypeHandler(propertyType, jdbcType);
        //3.2.7
        handler = typeHandlerRegistry.getTypeHandler(propertyType);
    }
}

五、總結

自定義處理器的應用場景很廣泛,比如對某些敏感字段加密、狀態值的轉換(正常、注銷、 已付款、未發貨)等。回顧一下你的項目中有哪些地方實現的不太理想,可以考慮用它來做。

六、后續

在筆者寫完這篇文章后,在另外一臺電腦做測試的時候,發現盡管沒有對時間類型做處理,但也不會出現.0的問題。這使我睡覺都沒安穩。。。難道自己認知有誤,文章寫錯了?筆者決定先拋開Mybatis,用最原始的JDBC做測試。

public static void main(String[] args) throws Exception {
    Connection conn = getConnection();
    Statement stat = conn.createStatement();
    String sql = "select * from user";
    ResultSet rs = stat.executeQuery(sql);
    while(rs.next()){
        String username = rs.getString("username");
        String createtime = rs.getString("createtime");
        System.out.print("姓名: " + username);
        System.out.print("  創建時間: " + createtime);
        System.out.print("\n");
    }
}

結果讓我很意外,用原始的JDBC查詢數據,并沒有任何其他操作,也沒有.0的問題。

姓名: 關小羽 創建時間: 2018-10-15 17:04:11
姓名: 小露娜 創建時間: 2018-10-15 17:10:46
姓名: 亞麻瑟 創建時間: 2018-10-15 17:10:46
姓名: 小魯班 創建時間: 2018-10-16 16:10:47

上面的代碼量很小,顯然問題出在ResultSet對象上。通過跟蹤源碼,最后筆者發現兩臺機器的mysql-connector-java版本不一樣。一個是5.1.31,一個是6.0.6。我們把版本換成5.1.31,執行上面的main方法再看結果。

姓名: 關小羽 創建時間: 2018-10-15 17:04:11.0
姓名: 小露娜 創建時間: 2018-10-15 17:10:46.0
姓名: 亞麻瑟 創建時間: 2018-10-15 17:10:46.0
姓名: 小魯班 創建時間: 2018-10-16 16:10:47.0

好了,讓我們看看它們的差別在哪里吧。其實就是因為5.1.31多做了一步操作,它針對時間類型的數據又處理了一次,導致問題產生。

5.1.31

package com.mysql.jdbc;
public class ResultSetImpl implements ResultSetInternalMethods {
    protected String getStringInternal(int columnIndex, boolean checkDateTypes)
        // JDBC is 1-based, Java is not !?
        int internalColumnIndex = columnIndex - 1;
        Field metadata = this.fields[internalColumnIndex];      
        String stringVal = null;    
        String encoding = metadata.getCharacterSet();
        //stringVal為已經從數據庫取到的值2018-10-16 16:10:47
        stringVal = this.thisRow.getString(internalColumnIndex, encoding, this.connection);
        
        // Handles timezone conversion and zero-date behavior
        //Mysql針對時間類型又做了一次處理
        if (checkDateTypes && !this.connection.getNoDatetimeStringSync()) {
            switch (metadata.getSQLType()) {
            case Types.TIME:
                ......略
            case Types.DATE:
                ......略
            case Types.TIMESTAMP:
                //數據庫的DateTime類型會走到這里
                //MySQL把它又轉成了Timestamp類型,  .0的問題從這里產生
                Timestamp ts = getTimestampFromString(columnIndex,
                        null, stringVal, this.getDefaultTimeZone(), false);
                return ts.toString();
            default:
                break;
            }
        }
        return stringVal;
    }
}

6.0.6

package com.mysql.cj.jdbc.result;

public class ResultSetImpl extends MysqlaResultset 
                implements ResultSetInternalMethods, WarningListener {
    
    public String getString(int columnIndex) throws SQLException {
        
        Field f = this.columnDefinition.getFields()[columnIndex - 1];
        ValueFactory<String> vf = new StringValueFactory(f.getEncoding());
        // return YEAR values as Dates if necessary
        if (f.getMysqlTypeId() == MysqlaConstants.FIELD_TYPE_YEAR && this.yearIsDateType) {
            vf = new YearToDateValueFactory<>(vf);
        }
        String stringVal = this.thisRow.getValue(columnIndex - 1, vf);

        return stringVal;
    }
}

如果大家項目里面有.0問題產生,可以通過升級mysql-java版本解決。如果不能動版本,再考慮自定義的類型處理器。

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 229,763評論 6 539
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 99,238評論 3 428
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 177,823評論 0 383
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 63,604評論 1 317
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 72,339評論 6 410
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 55,713評論 1 328
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,712評論 3 445
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,893評論 0 289
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 49,448評論 1 335
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 41,201評論 3 357
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 43,397評論 1 372
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,944評論 5 363
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,631評論 3 348
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 35,033評論 0 28
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 36,321評論 1 293
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 52,128評論 3 398
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 48,347評論 2 377

推薦閱讀更多精彩內容