给ShardingSphere提了个PR,不知道是不是嫌弃我?

2022-12-05 20:07:50 浏览数 (1)

说来惭愧,干了 10 来年程序员,还没有给开源做过任何贡献,以前只知道嘎嘎写,出了问题嘎嘎改,从来没想过提个 PR 去修复他,最近碰到个问题,发现挺简单的,就随手提了个 PR 过去。

问题

问题挺简单的,就是在使用 mybatis 和 ShardingSphere 的时候,有人在 model 类使用了 OffsetDateTime 这个时间类型,发现会报错。

代码语言:javascript复制
Caused by: java.lang.ClassCastException: class java.sql.Timestamp cannot be cast to class java.time.OffsetDateTime (java.sql.Timestamp is in module java.sql of loader 'platform'; java.time.OffsetDateTime is in module java.base of loader 'bootstrap')
 at org.apache.ibatis.type.OffsetDateTimeTypeHandler.getNullableResult(OffsetDateTimeTypeHandler.java:38)
 at org.apache.ibatis.type.OffsetDateTimeTypeHandler.getNullableResult(OffsetDateTimeTypeHandler.java:28)
 at org.apache.ibatis.type.BaseTypeHandler.getResult(BaseTypeHandler.java:85)
 ... 99 more

这就是一个简单的类型转换的异常,于是跟着源码看了下,先看到BaseTypeHandler#getResult这个方法,实际上就是根据列名返回查询结果。

根据调用关系,找到了OffsetDateTimeTypeHandler实现类。

发现最终会调用rs.getObject()这个方法,那么其实这个方法会最终走到由 ShardingSphere 实现的 getObject方法中。

看到这里的时候其实已经明白了为啥会报错了,Shardingsphere 只判断了几个LocalDateTime等类型,对于这个比较特殊的时间类型没有处理,最终会转换成 Timestamp ,然后强转就报错了。

最后调用到ResultSetUtil#convertTimestampValue方法,可以看到确实是这样哈。

那如果修改源码的话其实很简单了,getObject判断多加一个,convertTimestampValue再加一个,就这样很简单啊。

代码语言:javascript复制
@Override
    public <T> T getObject(final int columnIndex, final Class<T> type) throws SQLException {
        if (BigInteger.class.equals(type)) {
            return (T) BigInteger.valueOf(getLong(columnIndex));
        } else if (Blob.class.equals(type)) {
            return (T) getBlob(columnIndex);
        } else if (Clob.class.equals(type)) {
            return (T) getClob(columnIndex);
        } else if (LocalDateTime.class.equals(type) || LocalDate.class.equals(type) || LocalTime.class.equals(type) || OffsetDateTime.class.equals(type)) {
            return (T) ResultSetUtil.convertValue(mergeResultSet.getValue(columnIndex, Timestamp.class), type);
        } else {
            return (T) ResultSetUtil.convertValue(mergeResultSet.getValue(columnIndex, type), type);
        }
    }
    
private static Object convertTimestampValue(final Object value, final Class<?> convertType) {
        Timestamp timestamp = (Timestamp) value;
        if (LocalDateTime.class.equals(convertType)) {
            return timestamp.toInstant().atZone(ZoneId.systemDefault()).toLocalDateTime();
        }
        if (LocalDate.class.equals(convertType)) {
            return timestamp.toInstant().atZone(ZoneId.systemDefault()).toLocalDate();
        }
        if (LocalTime.class.equals(convertType)) {
            return timestamp.toInstant().atZone(ZoneId.systemDefault()).toLocalTime();
        }
        if (OffsetDateTime.class.equals(convertType)) {
            return timestamp.toInstant().atZone(ZoneId.systemDefault()).toOffsetDateTime();
        }
        return value;
}

修复

最开始我其实并不想改源码,我在想其他的实现方案,搜索后发现引入一个包就可以解决,也就是 mybatis 的 JSR310 规范。

代码语言:javascript复制
<dependency>
  <groupId>org.mybatis</groupId>
  <artifactId>mybatis-typehandlers-jsr310</artifactId>
  <version>1.0.1</version>
</dependency>

他为什么能解决这个问题?我看了下他的包里面的代码,这不就是加了个 TypeHandler 自己处理了嘛。

再去看了下 OffsetDateTimeTypeHandler的实现,其实就是自己就解决了,直接给返回OffsetDateTime ,根本不会走到 ShardingSphere 的逻辑里面去,这也就是他能解决这个问题的原因了。

当然,如果不想那么麻烦引入一个包,也可以单独把他拎出来自己去指定一下,这个很简单,就不多说了。

提PR

于是我想,这事情这么简单,我不如提个 PR 给官方吧,这里教下大家怎么提 PR 。

因为不是咱们的项目,是没法 push 代码的,所以进入到项目,然后fork,fork 好了以后,直接把项目 clone 下来,然后执行命令。

代码语言:javascript复制
git remote add upstream https://github.com/apache/shardingsphere.git

通过命令我们可以看到成功了,这样就 OK了,然后正常拉分支写代码吧。

写完之后,正常去我们的项目界面提交 PR,然后就可以了。

麻烦

当然,过程并没有这么顺利,虽然说只是很简单的修改。

首先,这个校验就给我提示错误了,第一点叫我不要用 *号去引用。

这个其实是 IDEA的锅,如果引用同一个包下类过多的话,会自动帮我们转成星号,这个我们可以在Editor-Code Style-Java,然后找到 Imports 下的这两个选项,把他们都改成 99 就可以了,防止他自动给我们改成星号。

还有一些其他的比如 if 后面没跟空格之类的,这是我忘记格式化了!

然后大佬回复觉得看不下去,这代码太恶心了,说我们是不是可以用java.time.temporal.TemporalAccessor来判断,不然这么多时间类型,搞个毛线呢。

然后我就翻译了一段英文,我也不知道大佬看没看懂,我告诉他,这个不好整啊,你看这个接口啊,很多乱七八糟的类实现了他,实际上我觉得我们覆盖常用的一些就行了,其他的特殊时间类型让他们自己用 TypeHandler 处理吧。

大佬说,嗯当然,没办法判断这个接口那我们也没辙了,我说那不可就是嘛。

其实,还有很多时间类型他都会报错的,最好的办法这个都抽象出来和Mybatis单独用实现类,不过那样的话就得大工作了,我太懒了,就这样。

。。。

不是嫌弃我?

0 人点赞