add updatewrapper

mybatis-plus-core/src/main/java/com/baomidou/mybatisplus/core/conditions/UpdateWrapper.java

@@ -0,0 +1,87 @@
+/*
+ * Copyright (c) 2011-2014, hubin (jobob@qq.com).
+ * <p>
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not
+ * use this file except in compliance with the License. You may obtain a copy of
+ * the License at
+ * <p>
+ * http://www.apache.org/licenses/LICENSE-2.0
+ * <p>
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+ * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+ * License for the specific language governing permissions and limitations under
+ * the License.
+ */
+package com.baomidou.mybatisplus.core.conditions;
+
+
+import com.baomidou.mybatisplus.core.toolkit.StringUtils;
+
+/**
+ * <p>
+ * Entity 对象封装操作类，定义T-SQL语法
+ * </p>
+ *
+ * @author hubin , yanghu , Dyang , Caratacus
+ * @Date 2016-11-7
+ */
+@SuppressWarnings("serial")
+public class UpdateWrapper<T> extends Wrapper<T> {
+
+    /**
+     * 数据库表映射实体类
+     */
+    protected T entity = null;
+
+    public UpdateWrapper() {
+        /* 注意，传入查询参数 */
+    }
+
+    public UpdateWrapper(T entity) {
+        this.entity = entity;
+    }
+
+    public UpdateWrapper(T entity, String sqlSelect) {
+        this.entity = entity;
+        this.sqlSelect = sqlSelect;
+    }
+
+    @Override
+    public T getEntity() {
+        return entity;
+    }
+
+    public void setEntity(T entity) {
+        this.entity = entity;
+    }
+
+    /**
+     * SQL SET 字段
+     */
+    @Override
+    public String getSqlSet() {
+
+        return null;
+    }
+
+    /**
+     * SQL 片段
+     */
+    @Override
+    public String getSqlSegment() {
+        /*
+         * 无条件
+         */
+        String sqlWhere = sql.toString();
+        if (StringUtils.isEmpty(sqlWhere)) {
+            return null;
+        }
+
+        /*
+         * 根据当前实体判断是否需要将WHERE替换成 AND 增加实体不为空但所有属性为空的情况
+         */
+        return isWhere != null ? (isWhere ? sqlWhere : sqlWhere.replaceFirst("WHERE", AND_OR)) : sqlWhere.replaceFirst("WHERE", AND_OR);
+    }
+
+}

mybatis-plus-core/src/main/java/com/baomidou/mybatisplus/core/conditions/Wrapper.java

@@ -185,12 +185,18 @@ public abstract class Wrapper<T> implements Serializable {
     }
 
     /**
-     * <p>
      * SQL 片段 (子类实现)
-     * </p>
      */
     public abstract String getSqlSegment();
 
+    /**
+     * SQL SET 字段
+     */
+    public String getSqlSet() {
+        // to do nothing
+        return null;
+    }
+
     @Override
     public String toString() {
         StringBuilder sb = new StringBuilder("Wrapper<T>:");

mybatis-plus-extension/src/test/java/com/baomidou/mybatisplus/extension/test/h2/H2MetaObjectHandler.java

@@ -35,6 +35,7 @@ public class H2MetaObjectHandler extends MetaObjectHandler {
     /**
      * 测试 user 表 name 字段为空自动填充
      */
+    @Override
     public void insertFill(MetaObject metaObject) {
         System.out.println("*************************");
         System.out.println("insert fill");

mybatis-plus-extension/src/test/java/com/baomidou/mybatisplus/extension/test/mysql/MyMetaObjectHandler.java

@@ -33,6 +33,7 @@ public class MyMetaObjectHandler extends MetaObjectHandler {
     /**
      * 测试 user 表 name 字段为空自动填充
      */
+    @Override
     public void insertFill(MetaObject metaObject) {
         // 测试下划线
         Object testType = getFieldValByName("testType", metaObject);