修复 IService 的实现类间接继承 ServiceImpl 时用 Kt 系 Wrapper 会出错的问题；完善一下文档；Kt 系的链式查询获取 Class 时增加缓存

mybatis-plus-extension/src/main/kotlin/com/baomidou/mybatisplus/extension/kotlin/IServiceEx.kt

@@ -18,29 +18,63 @@ package com.baomidou.mybatisplus.extension.kotlin
 import com.baomidou.mybatisplus.extension.kotlin.chain.KtQueryChainWrapper
 import com.baomidou.mybatisplus.extension.kotlin.chain.KtUpdateChainWrapper
 import com.baomidou.mybatisplus.extension.service.IService
+import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl
 import kotlin.reflect.jvm.javaType
 
+/**
+ * 每个 IService 对于实现类 ServiceImpl 上的第二个范型参数的 Class 的缓存
+ *
+ * 一般而言 Service 的实现类不会太多，也不会出现类卸载的情况，因此无需考虑清除的问题
+ */
+private val entityClassCache: MutableMap<IService<*>, Class<*>> = mutableMapOf()
+
 /**
  * 获取 IService 的实现类 ServiceImpl 上的第二个范型参数的 Class
  *
  * @return 获取到的 Class
  */
 @Suppress("UNCHECKED_CAST")
-private fun <T : Any> IService<T>.entityClass(): Class<T> =
-        // TODO cache
-        // TODO ServiceImpl 不一定是直接超类
-        (this::class.supertypes[0].arguments[1].type?.javaType) as Class<T>
+private fun <T : Any> IService<T>.entityClass() =
+        (entityClassCache.getOrPut(this) {
+            // 理论上不可能找不到，除非用户的代码直接就没继承 ServiceImpl
+            // 理论上 as 转换时不可能出现异常，除非用的不知道什么鬼家的 JVM？
+            (this::class.supertypes.first { it == ServiceImpl::class }.arguments[1].type?.javaType) as Class<*>
+        }) as Class<T>
 
 /**
  * Kotlin 用的 Lambda 式链式查询
  *
- * **如实现类并未直接继承 ServiceImpl，本方法会报错，这个问题将在近期解决**
+ * 使用示例：
+ *
+ * ```kotlin
+ *
+ * @Service
+ * class UserServiceImpl : ServiceImpl<UserMapper, User>(), IUserService {
+ *     override fun selectByUsername(username: String): User? {
+ *         return this.ktQuery()
+ *                 .eq(User::username, username)
+ *                 .one()
+ *     }
+ * }
+ * ```
  */
 fun <T : Any> IService<T>.ktQuery() = KtQueryChainWrapper(this.baseMapper, this.entityClass())
 
 /**
  * Kotlin 用的 Lambda 式链式更改
  *
- * **如实现类并未直接继承 ServiceImpl，本方法会报错，这个问题将在近期解决**
+ * 使用示例：
+ *
+ * ```kotlin
+ *
+ * @Service
+ * class UserServiceImpl : ServiceImpl<UserMapper, User>(), IUserService {
+ *     override fun removeByUsername(username: String): User? {
+ *         return this.ktUpdate()
+ *                 .eq(User::username, username)
+ *                 .remove()
+ *     }
+ * }
+ * ```
  */
 fun <T : Any> IService<T>.ktUpdate() = KtUpdateChainWrapper(this.baseMapper, this.entityClass())