zhongrj
2025-02-11 b1e8a099eb3117bcbd5f2e0c9a2eb0accabd8cbf
src/main/java/org/springblade/modules/yw/service/impl/EmergencySpaceServiceImpl.java
@@ -1,12 +1,25 @@
package org.springblade.modules.yw.service.impl;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import lombok.extern.slf4j.Slf4j;
import org.apache.logging.log4j.util.Strings;
import org.springblade.common.cache.DictBizCache;
import org.springblade.core.tool.utils.BeanUtil;
import org.springblade.modules.yw.entity.EmergencySpaceEntity;
import org.springblade.modules.yw.entity.FirmInfo;
import org.springblade.modules.yw.excel.EmergencySpaceExcel;
import org.springblade.modules.yw.service.IFirmInfoService;
import org.springblade.modules.yw.vo.EmergencySpaceVO;
import org.springblade.modules.yw.mapper.EmergencySpaceMapper;
import org.springblade.modules.yw.service.IEmergencySpaceService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.baomidou.mybatisplus.core.metadata.IPage;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Objects;
/**
 * 应急空间表 服务实现类
@@ -14,12 +27,157 @@
 * @author BladeX
 * @since 2024-10-28
 */
@Slf4j
@Service
public class EmergencySpaceServiceImpl extends ServiceImpl<EmergencySpaceMapper, EmergencySpaceEntity> implements IEmergencySpaceService {
   @Autowired
   private IFirmInfoService firmInfoService;
   /**
    * 自定义分页查询
    * @param page
    * @param emergencySpace
    * @return
    */
   @Override
   public IPage<EmergencySpaceVO> selectEmergencySpacePage(IPage<EmergencySpaceVO> page, EmergencySpaceVO emergencySpace) {
      return page.setRecords(baseMapper.selectEmergencySpacePage(page, emergencySpace));
      List<EmergencySpaceVO> emergencySpaceVOS = baseMapper.selectEmergencySpacePage(page, emergencySpace);
      for (EmergencySpaceVO emergencySpaceVO : emergencySpaceVOS) {
         // 设置应急空间主要用途
         emergencySpaceVO.setMainFuncName(DictBizCache.getValues("emergency_space_use",emergencySpaceVO.getMainFunc()));
      }
      return page.setRecords(emergencySpaceVOS);
   }
   /**
    * 导入应急空间信息
    * @param data
    * @param isCovered
    * @return
    */
   @Override
   public String importEmergencySpace(List<EmergencySpaceExcel> data, boolean isCovered) {
      for (EmergencySpaceExcel emergencySpaceExcel : data) {
         // 数据拷贝
         EmergencySpaceEntity emergencySpaceEntity = Objects.requireNonNull(BeanUtil.copy(emergencySpaceExcel, EmergencySpaceEntity.class));
         // 类型转换
         if (!Strings.isBlank(emergencySpaceExcel.getType())){
            emergencySpaceEntity.setType(Integer.parseInt(emergencySpaceExcel.getType()));
         }
         // 防控级别转换
         if (!Strings.isBlank(emergencySpaceExcel.getPreLevel())){
            emergencySpaceEntity.setPreLevel(Integer.parseInt(emergencySpaceExcel.getPreLevel()));
         }
         // 经纬度转换
//         if (!Strings.isBlank(emergencySpaceExcel.getLng())) {
//            emergencySpaceEntity.setLng(PositionUtil.tranformPos(emergencySpaceExcel.getLng()).toString());
//         }
//         if (!Strings.isBlank(emergencySpaceExcel.getLat())) {
//            emergencySpaceEntity.setLat(PositionUtil.tranformPos(emergencySpaceExcel.getLat()).toString());
//         }
         // 设置企业id
         setFirm(emergencySpaceExcel,emergencySpaceEntity);
         log.info("应急空间名称:{},企业名称:{}",emergencySpaceEntity.getName(),emergencySpaceExcel.getFirmName());
         // 是否已入库
         Long id = isSave(emergencySpaceEntity);
         if (null!=id){
            if (isCovered){
               // 覆盖更新
               emergencySpaceEntity.setId(id);
               updateById(emergencySpaceEntity);
               continue;
            }
         }
         // 保存
         save(emergencySpaceEntity);
      }
      return null;
   }
   /**
    * 判断是否保存入库
    * @param emergencySpaceEntity
    * @return
    */
   private Long isSave(EmergencySpaceEntity emergencySpaceEntity) {
      QueryWrapper<EmergencySpaceEntity> wrapper = new QueryWrapper<>();
      wrapper.eq("name",emergencySpaceEntity.getName())
         .eq("is_deleted",0);
      if (null!=emergencySpaceEntity.getFirmId()){
         wrapper.eq("firm_id",emergencySpaceEntity.getFirmId());
      }
      EmergencySpaceEntity one = getOne(wrapper);
      if (null!=one){
         return one.getId();
      }
      return null;
   }
   /**
    * 设置企业信息
    * @param emergencySpaceExcel
    * @param emergencySpaceEntity
    */
   private void setFirm(EmergencySpaceExcel emergencySpaceExcel, EmergencySpaceEntity emergencySpaceEntity) {
      QueryWrapper<FirmInfo> wrapper = new QueryWrapper<>();
      wrapper.eq("name",emergencySpaceExcel.getFirmName())
         .eq("is_deleted",0);
      FirmInfo firmInfo = firmInfoService.getOne(wrapper);
      if (null!=firmInfo){
         emergencySpaceEntity.setFirmId(firmInfo.getId());
         emergencySpaceEntity.setFullName(firmInfo.getName() + emergencySpaceExcel.getName());
      }
   }
   /**
    * 获取应急空间统计数据
    * @param emergencySpace
    * @return
    */
   @Override
   public List<Map<String,Object>> getStatisticData(EmergencySpaceVO emergencySpace) {
      // 按应急空间类型进行统计
      List<Map<String,Object>> list = baseMapper.getStatisticData(emergencySpace);
      return list;
   }
   /**
    * 获取应急池容量
    * @return
    */
   @Override
   public String getEmePool() {
      return baseMapper.getEmePool();
   }
   /**
    * 通过空间分析点位,找到离风险源最近的防控点 风险源找1,1找2,2找3
    * @param emergencySpace pre_level: 1:一级防控点  2:二级防控点  3:三级防控点
    * @return
    */
   @Override
   public Object spatialAnalysisPoint(EmergencySpaceVO emergencySpace) {
      List<EmergencySpaceVO> list = new ArrayList<>();
      if (null==emergencySpace.getPreLevel()){
         emergencySpace.setPreLevel(1);
      }
      // 查询一级
      // 先查询当前风险源点对应的企业
      QueryWrapper<FirmInfo> wrapper = new QueryWrapper<>();
      wrapper.eq("lng",emergencySpace.getLng()).eq("lat",emergencySpace.getLat());
      FirmInfo firmInfo = firmInfoService.getOne(wrapper);
      emergencySpace.setFirmId(firmInfo.getId());
      list.add(baseMapper.spatialAnalysisPoint(emergencySpace));
      // 其他项查询设置为空
      emergencySpace.setFirmId(null);
      // 查询二级
      emergencySpace.setPreLevel(2);
      list.add(baseMapper.spatialAnalysisPointToLines(emergencySpace));
      // 查询三级
      emergencySpace.setPreLevel(3);
      list.add(baseMapper.spatialAnalysisPoint(emergencySpace));
      // 返回结果
      return list;
   }
}