zhongrj
2023-10-31 85b366e87d9fc91b638ff592752b9f6f36364f3f
src/main/java/org/springblade/modules/house/service/impl/HouseLabelServiceImpl.java
@@ -16,6 +16,7 @@
 */
package org.springblade.modules.house.service.impl;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springblade.modules.house.entity.HouseLabelEntity;
import org.springblade.modules.house.vo.HouseLabelVO;
@@ -39,5 +40,26 @@
      return page.setRecords(baseMapper.selectHouseLabelPage(page, houseLabel));
   }
   /**
    * 房屋-标签 自定义新增或修改
    * @param houseLabel
    * @return
    */
   @Override
   public boolean saveOrUpdateHouseLabel(HouseLabelEntity houseLabel) {
      // 查询标签名称
      HouseLabelEntity labelEntity = getById(houseLabel.getLabelId());
      houseLabel.setLabelName(labelEntity.getLabelName());
      // 判断同一个房屋同一个标签是否已存在,已存在则更新,不存在则新增
      QueryWrapper<HouseLabelEntity> queryWrapper = new QueryWrapper<>();
      queryWrapper.eq("house_code",houseLabel.getHouseCode())
               .eq("label_id",houseLabel.getLabelId());
      HouseLabelEntity one = getOne(queryWrapper);
      if (null != one){
         // 更新
         return updateById(houseLabel);
      }
      // 插入
      return save(houseLabel);
   }
}