Merge pull request #105 from hocgin/develop

[新增功能](develop): 优化
hocgin · May 16, 2022 · 3ba5faa · 3ba5faa
2 parents 82c46b7 + ed34d29
commit 3ba5faa
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/...va/in/hocg/boot/mybatis/plus/autoconfiguration/core/struct/basic/AbstractServiceImpl.java b/...va/in/hocg/boot/mybatis/plus/autoconfiguration/core/struct/basic/AbstractServiceImpl.java
@@ -6,7 +6,7 @@
 import com.baomidou.mybatisplus.core.mapper.BaseMapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.core.toolkit.support.SFunction;
-import com.baomidou.mybatisplus.extension.conditions.query.QueryChainWrapper;
+import com.baomidou.mybatisplus.extension.conditions.query.ChainQuery;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import in.hocg.boot.mybatis.plus.autoconfiguration.core.enhance.convert.UseConvert;
@@ -16,7 +16,10 @@
 
 import java.io.Serializable;
 import java.lang.reflect.Method;
-import java.util.*;
+import java.util.Arrays;
+import java.util.List;
+import java.util.Objects;
+import java.util.Optional;
 import java.util.stream.Collectors;
 
 /**
@@ -72,8 +75,8 @@ public Optional<T> first(Wrapper<T> queryWrapper) {
     public List<T> limit(Wrapper<T> queryWrapper, Integer limit) {
         Page<T> limitPage = new Page<>(1, limit, false);
 
-        if (queryWrapper instanceof QueryChainWrapper) {
-            return ((QueryChainWrapper<T>) queryWrapper).page(limitPage).getRecords();
+        if (queryWrapper instanceof ChainQuery) {
+            return ((ChainQuery<T>) queryWrapper).page(limitPage).getRecords();
         }
         return page(limitPage, queryWrapper).getRecords();
     }