feats

src/main/java/com/yunzhi/shigongli/controller/TaHotelController.java

@@ -73,10 +73,11 @@ public class TaHotelController extends BaseController {
     @ApiOperation(value="保存", notes = "保存", httpMethod = "POST", response = ResponseBean.class)
     public ResponseBean taHotelAdd(@ApiParam("保存内容") @RequestBody TaHotel taHotel) throws Exception{
 
-        List<TaHotel> lst = iTaHotelService.getByName(taHotel.getHotelName());
-        if (null != lst && lst.size() > 0) {
-            return ResponseBean.error("民宿名称已存在", ResponseBean.ERROR_ILLEGAL_PARAMS);
-        }
+//        List<TaHotel> lst = iTaHotelService.getByName(taHotel.getHotelName());
+//        if (null != lst && lst.size() > 0) {
+//            return ResponseBean.error("民宿名称已存在", ResponseBean.ERROR_ILLEGAL_PARAMS);
+//        }
+        iTaHotelService.isExisted(TaHotel::getHotelName, taHotel.getHotelName());
 
         if (iTaHotelService.createNew(taHotel)){
             return ResponseBean.success(taHotel);
@@ -110,8 +111,11 @@ public class TaHotelController extends BaseController {
     public ResponseBean taHotelUpdate(@ApiParam("对象ID") @PathVariable String id,
                                         @ApiParam("更新内容") @RequestBody TaHotel taHotel) throws Exception{
 
-        List<TaHotel> lst = iTaHotelService.getByName(taHotel.getHotelName());
-        if (null != lst && lst.size() > 0) {
+        boolean exists = iTaHotelService.isExisted(TaHotel::getHotelName,
+                taHotel.getHotelName(),
+                TaHotel::getHotelId,
+                taHotel.getHotelId());
+        if (exists) {
             return ResponseBean.error("民宿名称已存在", ResponseBean.ERROR_ILLEGAL_PARAMS);
         }
 

src/main/java/com/yunzhi/shigongli/service/IBaseService.java

@@ -1,9 +1,42 @@
 package com.yunzhi.shigongli.service;
 
+import com.baomidou.mybatisplus.core.toolkit.support.SFunction;
 import com.baomidou.mybatisplus.extension.service.IService;
 
 import java.io.Serializable;
+import java.util.List;
 
 public interface IBaseService<T> extends IService<T> {
+    /**
+     * 逻辑删除
+     * @param id
+     * @return
+     */
     boolean removeLogicById(Serializable id);
+
+    /**
+     * 依据字段值获取列表
+     * @param column
+     * @param val
+     * @return
+     */
+    List<T> getBy(SFunction<T, ?> column, Serializable val);
+
+    /**
+     * 是否存在, 校验唯一
+     * @param column
+     * @param val
+     * @return
+     */
+    boolean isExisted(SFunction<T, ?> column, Serializable val);
+
+    /**
+     * 排除 notVal 的条件下是否存在
+     * @param column
+     * @param val
+     * @param notEq
+     * @param notVal
+     * @return
+     */
+    boolean isExisted(SFunction<T, ?> column, Serializable val, SFunction<T, ?>  notEq, Serializable notVal);
 }

src/main/java/com/yunzhi/shigongli/service/impl/BaseServiceImpl.java

@@ -1,13 +1,18 @@
 package com.yunzhi.shigongli.service.impl;
 
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
 import com.baomidou.mybatisplus.core.mapper.BaseMapper;
 import com.baomidou.mybatisplus.core.metadata.TableInfo;
+import com.baomidou.mybatisplus.core.toolkit.support.SFunction;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.baomidou.mybatisplus.extension.toolkit.SqlHelper;
 import com.yunzhi.shigongli.service.IBaseService;
 
 import java.io.Serializable;
+import java.util.List;
+import java.util.function.Supplier;
 
 public class BaseServiceImpl <M extends BaseMapper<T>, T> extends ServiceImpl <M, T> implements IBaseService<T> {
     @Override
@@ -20,4 +25,27 @@ public class BaseServiceImpl <M extends BaseMapper<T>, T> extends ServiceImpl <M
 
         return update(updateWrapper);
     }
+
+    @Override
+    public List<T> getBy(SFunction<T, ?> column, Serializable val) {
+        LambdaQueryWrapper<T> queryWrapper = new LambdaQueryWrapper<>();
+        queryWrapper.eq(column, val);
+
+        return list(queryWrapper);
+    }
+
+    @Override
+    public boolean isExisted(SFunction<T, ?> column, Serializable val) {
+        return isExisted(column, val, (SFunction<T, ?>) null, null);
+    }
+
+    @Override
+    public boolean isExisted(SFunction<T, ?> column, Serializable val, SFunction<T, ?> notEq, Serializable notVal) {
+        LambdaQueryWrapper<T> queryWrapper = new LambdaQueryWrapper<>();
+        queryWrapper.eq(column, val);
+        queryWrapper.ne(null != notVal, notEq, notVal);
+
+        int cnt = count(queryWrapper);
+        return cnt > 0;
+    }
 }