linwe
2024-08-09 8b7258c9427882bb1798f1502eaa35184c6e374e
src/main/java/org/springblade/modules/house/service/impl/HouseLabelServiceImpl.java
@@ -16,12 +16,15 @@
 */
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;
import org.springblade.modules.house.vo.UserHouseLabelVO;
import org.springblade.modules.house.mapper.HouseLabelMapper;
import org.springblade.modules.house.service.IHouseLabelService;
import org.springblade.core.mp.base.BaseServiceImpl;
import org.springblade.modules.label.entity.LabelEntity;
import org.springblade.modules.label.service.ILabelService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.baomidou.mybatisplus.core.metadata.IPage;
@@ -34,10 +37,35 @@
@Service
public class HouseLabelServiceImpl extends ServiceImpl<HouseLabelMapper, HouseLabelEntity> implements IHouseLabelService {
   @Autowired
   private ILabelService labelService;
   @Override
   public IPage<HouseLabelVO> selectHouseLabelPage(IPage<HouseLabelVO> page, HouseLabelVO houseLabel) {
   public IPage<UserHouseLabelVO> selectHouseLabelPage(IPage<UserHouseLabelVO> page, UserHouseLabelVO houseLabel) {
      return page.setRecords(baseMapper.selectHouseLabelPage(page, houseLabel));
   }
   /**
    * 房屋-标签 自定义新增或修改
    * @param houseLabel
    * @return
    */
   @Override
   public boolean saveOrUpdateHouseLabel(HouseLabelEntity houseLabel) {
      // 查询标签名称
      LabelEntity labelEntity = labelService.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){
         houseLabel.setId(one.getId());
         // 更新
         return updateById(houseLabel);
      }
      // 插入
      return save(houseLabel);
   }
}