bug fix

src/main/java/com/huiju/estateagents/service/impl/TaBuildingServiceImpl.java

@@ -18,15 +18,13 @@ import com.huiju.estateagents.service.ITaBuildingService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.huiju.estateagents.service.TaPosterService;
 import com.huiju.estateagents.service.TaShareContentService;
+import javafx.scene.control.Tab;
 import org.apache.commons.collections.CollectionUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
 import java.time.LocalDateTime;
-import java.util.ArrayList;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 import java.util.stream.Collectors;
 
 /**
@@ -286,10 +284,17 @@ public class TaBuildingServiceImpl extends ServiceImpl<TaBuildingMapper, TaBuild
 
     @Override
     public ResponseBean buildingUpdate(String parameter, Integer orgId) {
+        ResponseBean responseBean = new ResponseBean();
         JSONObject object= JSONObject.parseObject(parameter);
 
         TaBuilding building = object.toJavaObject(TaBuilding.class);
 
+        boolean isMainMoreThanTenFlag = countIsMainBuilding(building);
+        if (!isMainMoreThanTenFlag){
+            responseBean.addError("每个城市只能推荐10个楼盘到首页，请先将已推荐楼盘取消推荐后再推荐当前楼盘。");
+            return responseBean;
+        }
+
         // 先删除视频
         UpdateWrapper<TaBuilding> updateWrapper = new UpdateWrapper<>();
         updateWrapper.eq("building_id", building.getBuildingId());
@@ -382,6 +387,8 @@ public class TaBuildingServiceImpl extends ServiceImpl<TaBuildingMapper, TaBuild
 
     @Override
     public ResponseBean buildingAdd(String parameter, Integer orgId) {
+        ResponseBean responseBean = new ResponseBean();
+
         JSONObject object= JSONObject.parseObject(parameter);
         // 根据orgId查询当前楼盘最大数
         TaOrg taOrg = taOrgMapper.selectById(orgId);
@@ -395,6 +402,13 @@ public class TaBuildingServiceImpl extends ServiceImpl<TaBuildingMapper, TaBuild
         }
 
         TaBuilding building = object.toJavaObject(TaBuilding.class);
+
+        boolean isMainMoreThanTenFlag = countIsMainBuilding(building);
+        if (!isMainMoreThanTenFlag){
+            responseBean.addError("每个城市只能推荐10个楼盘到首页，请先将已推荐楼盘取消推荐后再推荐当前楼盘。");
+            return responseBean;
+        }
+
         if(null!= object.getDate("openingDate")){
             building.setOpeningDate(DateUtils.date2LocalDateTime(object.getDate("openingDate")));
         }
@@ -818,6 +832,29 @@ public class TaBuildingServiceImpl extends ServiceImpl<TaBuildingMapper, TaBuild
         return taBuildingMapper.countNumByHeavy(heavy, dynamicId);
     }
 
+    /**
+     * 推荐到首页楼盘不能超过10个
+     * @param taBuilding
+     * @return
+     */
+    public boolean countIsMainBuilding(TaBuilding taBuilding) {
+        if (taBuilding.getIsMain() != 1){
+            return true;
+        }
+        QueryWrapper<TaBuilding> taBuildingQueryWrapper = new QueryWrapper<>();
+        taBuildingQueryWrapper.eq("city_id", taBuilding.getCityId());
+        taBuildingQueryWrapper.eq("is_main", 1);
+        taBuildingQueryWrapper.eq("status", 1);
+        if (taBuilding.getBuildingId() != null){
+            taBuildingQueryWrapper.ne("building_id", taBuilding.getBuildingId());
+        }
+        List<TaBuilding> taBuildings = taBuildingMapper.selectList(taBuildingQueryWrapper);
+        if (taBuildings != null && taBuildings.size() > 10){
+            return false;
+        }
+        return true;
+    }
+
 
     private List<TaBuildingImg> insertImgBatch(List<TaBuildingImg> imgs, String buildingId) {
         return imgs.stream().map(img -> {