linwe
2024-08-09 8b7258c9427882bb1798f1502eaa35184c6e374e
src/main/java/org/springblade/modules/police/service/impl/PoliceStationServiceImpl.java
@@ -1,12 +1,20 @@
package org.springblade.modules.police.service.impl;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springblade.common.node.TreeStringNode;
import org.springblade.common.utils.NodeTreeUtil;
import org.springblade.core.secure.utils.AuthUtil;
import org.springblade.modules.police.entity.PoliceStationEntity;
import org.springblade.modules.police.excel.PoliceStationExcel;
import org.springblade.modules.police.vo.PoliceStationVO;
import org.springblade.modules.police.mapper.PoliceStationMapper;
import org.springblade.modules.police.service.IPoliceStationService;
import org.springframework.stereotype.Service;
import com.baomidou.mybatisplus.core.metadata.IPage;
import java.util.Date;
import java.util.List;
/**
 * 派出所信息表 服务实现类
@@ -22,5 +30,45 @@
      return page.setRecords(baseMapper.selectPoliceStationPage(page, policeStation));
   }
   /**
    * 导入派出所数据
    * @param data
    * @param isCovered
    */
   @Override
   public void importPoliceStation(List<PoliceStationExcel> data, Boolean isCovered) {
      for (PoliceStationExcel policeStationExcel : data) {
         PoliceStationEntity policeStationEntity = new PoliceStationEntity();
         policeStationEntity.setCode(policeStationExcel.getCode());
         policeStationEntity.setName(policeStationExcel.getName());
         // 数据异常,无法导入
//         policeStationEntity.setGeom(policeStationExcel.getGeom());
         // 判断是否已录入
         QueryWrapper<PoliceStationEntity> queryWrapper = new QueryWrapper<>();
         queryWrapper.eq("is_deleted",0).eq("code",policeStationExcel.getCode());
         PoliceStationEntity one = getOne(queryWrapper);
         if (null!=one){
            // 更新
            policeStationEntity.setId(one.getId());
            updateById(policeStationEntity);
         }else {
            // 新增
            policeStationEntity.setCreateUser(AuthUtil.getUserId());
            policeStationEntity.setCreateTime(new Date());
            policeStationEntity.setUpdateUser(AuthUtil.getUserId());
            policeStationEntity.setUpdateTime(new Date());
            save(policeStationEntity);
         }
      }
   }
   /**
    * 派出所树查询
    * @param policeStation
    * @return
    */
   @Override
   public List<TreeStringNode> tree(PoliceStationEntity policeStation) {
      return NodeTreeUtil.getStringNodeTree(baseMapper.tree(policeStation));
   }
}