Selaa lähdekoodia

[修复] 添加测试用例 修复BUG

Caratacus 6 vuotta sitten
vanhempi
commit
e7b4c8f283

+ 11 - 1
mybatis-plus-extension/src/main/java/com/baomidou/mybatisplus/extension/service/IService.java

@@ -27,6 +27,8 @@ import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.service.additional.query.impl.LambdaQueryChainWrapper;
 import com.baomidou.mybatisplus.extension.service.additional.query.impl.QueryChainWrapper;
+import com.baomidou.mybatisplus.extension.service.additional.update.impl.LambdaUpdateChainWrapper;
+import com.baomidou.mybatisplus.extension.service.additional.update.impl.UpdateChainWrapper;
 import com.baomidou.mybatisplus.extension.toolkit.SqlHelper;
 
 /**
@@ -365,7 +367,7 @@ public interface IService<T> {
      * @param queryWrapper 实体对象封装操作类 {@link com.baomidou.mybatisplus.core.conditions.query.QueryWrapper}
      */
     default List<Object> listObjs(Wrapper<T> queryWrapper) {
-        return listObjs(Wrappers.emptyWrapper(), Function.identity());
+        return listObjs(queryWrapper, Function.identity());
     }
 
     /**
@@ -409,4 +411,12 @@ public interface IService<T> {
     default LambdaQueryChainWrapper<T> lambdaQuery() {
         return new LambdaQueryChainWrapper<>(getBaseMapper());
     }
+
+    default UpdateChainWrapper<T> update() {
+        return new UpdateChainWrapper<>(getBaseMapper());
+    }
+
+    default LambdaUpdateChainWrapper<T> lambdaUpdate() {
+        return new LambdaUpdateChainWrapper<>(getBaseMapper());
+    }
 }

+ 4 - 0
mybatis-plus-extension/src/main/java/com/baomidou/mybatisplus/extension/service/additional/update/ChainUpdate.java

@@ -28,4 +28,8 @@ public interface ChainUpdate<T> extends ChainWrapper<T> {
     default int update(T entity) {
         return getBaseMapper().update(entity, getWrapper());
     }
+
+    default int update() {
+        return getBaseMapper().update(null, getWrapper());
+    }
 }

+ 14 - 12
mybatis-plus/src/test/java/com/baomidou/mybatisplus/test/h2/H2UserTest.java

@@ -1,11 +1,12 @@
 package com.baomidou.mybatisplus.test.h2;
 
-import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
-import com.baomidou.mybatisplus.core.exceptions.MybatisPlusException;
-import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
-import com.baomidou.mybatisplus.test.h2.entity.enums.AgeEnum;
-import com.baomidou.mybatisplus.test.h2.entity.persistent.H2User;
-import com.baomidou.mybatisplus.test.h2.service.IH2UserService;
+import java.math.BigDecimal;
+import java.math.RoundingMode;
+import java.util.Arrays;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
 import org.junit.Assert;
 import org.junit.FixMethodOrder;
 import org.junit.Test;
@@ -15,12 +16,12 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.test.context.ContextConfiguration;
 import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
 
-import java.math.BigDecimal;
-import java.math.RoundingMode;
-import java.util.Arrays;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.baomidou.mybatisplus.core.exceptions.MybatisPlusException;
+import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
+import com.baomidou.mybatisplus.test.h2.entity.enums.AgeEnum;
+import com.baomidou.mybatisplus.test.h2.entity.persistent.H2User;
+import com.baomidou.mybatisplus.test.h2.service.IH2UserService;
 
 /**
  * <p>
@@ -298,5 +299,6 @@ public class H2UserTest extends BaseTest {
         H2User tomcat = userService.query().eq("name", "Tomcat").one();
         Assert.assertNotNull("tomcat should not be null", tomcat);
         userService.query().nested(i -> i.eq("name", "Tomcat")).list();
+        userService.lambdaUpdate().set(H2User::getName, "Tom").eq(H2User::getName, "Tomcat").update();
     }
 }