linwe
2023-11-12 8d8c7da4e91028fc877f101dcd244bef3e18c197
src/main/java/org/springblade/modules/house/service/impl/HouseholdServiceImpl.java
@@ -18,14 +18,19 @@
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springblade.common.node.TreeNode;
import org.springblade.core.secure.utils.AuthUtil;
import org.springblade.modules.house.entity.HouseholdEntity;
import org.springblade.modules.house.vo.HouseholdVO;
import org.springblade.modules.house.mapper.HouseholdMapper;
import org.springblade.modules.house.service.IHouseholdService;
import org.springblade.core.mp.base.BaseServiceImpl;
import org.springblade.modules.system.entity.DictBiz;
import org.springblade.modules.system.service.IDictBizService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.baomidou.mybatisplus.core.metadata.IPage;
import java.util.Date;
import java.util.List;
/**
@@ -36,6 +41,9 @@
 */
@Service
public class HouseholdServiceImpl extends ServiceImpl<HouseholdMapper, HouseholdEntity> implements IHouseholdService {
   @Autowired
   private IDictBizService dictBizService;
   @Override
   public IPage<HouseholdVO> selectHouseholdPage(IPage<HouseholdVO> page, HouseholdVO household) {
@@ -51,4 +59,74 @@
   public List<TreeNode> selectHouseNodeList(Long userId) {
      return baseMapper.selectHouseNodeList(userId);
   }
   /**
    * 查询房屋人员情况
    * @param code
    * @return
    */
   @Override
   public List<HouseholdVO> getHouseholdListByCode(String code) {
      // 查询
      List<HouseholdVO> householdList = baseMapper.getHouseholdListByCode(code);
      // 处理字典
      handleDictBiz(householdList);
      // 返回
      return householdList;
   }
   /**
    * 处理字典
    * @param householdList
    */
   private void handleDictBiz(List<HouseholdVO> householdList) {
      if (householdList.size()>0){
         // 查询角色关系字典
         List<DictBiz> dictBizList = dictBizService.getList("roleRelation");
         if (dictBizList.size()>0) {
            // 遍历
            for (HouseholdVO householdVO : householdList) {
               if (null != householdVO.getRelationship()) {
                  for (DictBiz dictBiz : dictBizList) {
                     if (householdVO.getRelationship().toString().equals(dictBiz.getDictKey())) {
                        householdVO.setRoleRelationName(dictBiz.getDictValue());
                     }
                  }
               }
            }
         }
      }
   }
   /**
    * 住户 自定义新增或修改
    * @param household
    * @return
    */
   @Override
   public boolean saveOrUpdateHousehold(HouseholdEntity household) {
      boolean flag = false;
      household.setUpdateTime(new Date());
      household.setUpdateUser(AuthUtil.getUserId());
      if (null!=household.getId()){
         // 更新
         flag = updateById(household);
      }else {
         // 新增
         household.setCreateTime(new Date());
         household.setCreateUser(AuthUtil.getUserId());
         flag = save(household);
      }
      return flag;
   }
   /**
    * 住户 自定义查询详情
    * @param household
    * @return
    */
   @Override
   public Object getDetail(HouseholdEntity household) {
      return baseMapper.getHouseholdListById(household);
   }
}