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; /** * 应急空间表 服务实现类 * * @author BladeX * @since 2024-10-28 */ @Slf4j @Service public class EmergencySpaceServiceImpl extends ServiceImpl implements IEmergencySpaceService { @Autowired private IFirmInfoService firmInfoService; /** * 自定义分页查询 * @param page * @param emergencySpace * @return */ @Override public IPage selectEmergencySpacePage(IPage page, EmergencySpaceVO emergencySpace) { List 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 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 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 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> getStatisticData(EmergencySpaceVO emergencySpace) { // 按应急空间类型进行统计 List> 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 list = new ArrayList<>(); if (null==emergencySpace.getPreLevel()){ emergencySpace.setPreLevel(1); } // 查询一级 // 先查询当前风险源点对应的企业 QueryWrapper 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; } }