package org.springblade.modules.yw.service.impl;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import org.apache.logging.log4j.util.Strings;
|
import org.springblade.common.cache.DictBizCache;
|
import org.springblade.common.utils.PositionUtil;
|
import org.springblade.core.tool.utils.BeanUtil;
|
import org.springblade.modules.yw.entity.FirmInfo;
|
import org.springblade.modules.yw.excel.FirmInfoExcel;
|
import org.springblade.modules.yw.mapper.FirmInfoMapper;
|
import org.springblade.modules.yw.service.IFirmInfoService;
|
import org.springblade.modules.yw.vo.FirmInfoVO;
|
import org.springblade.modules.yw.vo.SearchVO;
|
import org.springframework.stereotype.Service;
|
|
import java.util.List;
|
import java.util.Map;
|
import java.util.Objects;
|
|
/**
|
* 企业信息-接口服务层
|
* @author zhongrj
|
* @date 2024-10-24
|
*/
|
@Service
|
public class FirmInfoServiceImpl extends ServiceImpl<FirmInfoMapper, FirmInfo> implements IFirmInfoService {
|
|
/**
|
* 自定义分页查询
|
* @param page
|
* @param firmInfo
|
* @return
|
*/
|
@Override
|
public IPage<FirmInfoVO> selectFirmInfoPage(IPage<FirmInfoVO> page, FirmInfoVO firmInfo) {
|
List<FirmInfoVO> firmInfoVOS = baseMapper.selectFirmInfoPage(page, firmInfo);
|
for (FirmInfoVO firmInfoVO : firmInfoVOS) {
|
if (!Strings.isBlank(firmInfoVO.getCategory())){
|
firmInfoVO.setCategoryName(
|
DictBizCache.getValues("industry_category",firmInfoVO.getCategory())
|
);
|
}
|
}
|
return page.setRecords(firmInfoVOS);
|
}
|
|
/**
|
* 企业信息导入
|
* @param data
|
* @param isCovered
|
*/
|
@Override
|
public String importFirmInfo(List<FirmInfoExcel> data, Boolean isCovered) {
|
StringBuilder builder = new StringBuilder();
|
int total = data.size();
|
int successNum = 0;
|
int errorNum = 0;
|
// 遍历数据
|
for (FirmInfoExcel firmInfoExcel : data) {
|
// 数据转换
|
FirmInfo firmInfo = Objects.requireNonNull(BeanUtil.copy(firmInfoExcel, FirmInfo.class));
|
// 经纬度转换
|
// positionConvert(firmInfo);
|
// 判断信息是否已录入
|
FirmInfo oldFirmInfo = isSave(firmInfo);
|
// 已存在并且不更新
|
if (null!=oldFirmInfo && !isCovered){
|
successNum++;
|
continue;
|
}
|
// 已存在，需要覆盖更新
|
if (null!=oldFirmInfo && isCovered){
|
firmInfo.setId(oldFirmInfo.getId());
|
// 更新
|
boolean update = updateById(firmInfo);
|
if (update){
|
successNum++;
|
}else {
|
errorNum++;
|
}
|
continue;
|
}
|
// 保存
|
boolean save = save(firmInfo);
|
if (save) {
|
successNum++;
|
} else {
|
errorNum++;
|
}
|
}
|
// 拼接消息
|
builder.append("本次导入数据：").append(total).append(" 条")
|
.append("其中导入成功：").append(successNum).append(" 条")
|
.append("导入失败：").append(errorNum).append(" 条");
|
// 返回
|
return builder.toString();
|
}
|
|
/**
|
* 经纬度转换
|
* @param firmInfo
|
*/
|
private void positionConvert(FirmInfo firmInfo) {
|
if (!Strings.isBlank(firmInfo.getLng())){
|
firmInfo.setLng(PositionUtil.tranformPos(firmInfo.getLng()).toString());
|
}
|
if (!Strings.isBlank(firmInfo.getLat())){
|
firmInfo.setLat(PositionUtil.tranformPos(firmInfo.getLat()).toString());
|
}
|
}
|
|
/**
|
* 判断企业信息是否已录入
|
* @param firmInfo
|
* @return
|
*/
|
public FirmInfo isSave(FirmInfo firmInfo){
|
// 查询是否已存在
|
QueryWrapper<FirmInfo> wrapper = new QueryWrapper<>();
|
wrapper.eq("name",firmInfo.getName()).eq("is_deleted",0);
|
// 返回
|
return getOne(wrapper);
|
}
|
|
/**
|
* 自定义详情查询
|
* @param firmInfo
|
* @return
|
*/
|
@Override
|
public FirmInfoVO getDetail(FirmInfoVO firmInfo) {
|
FirmInfoVO detail = baseMapper.getDetail(firmInfo);
|
// 设置类型解析
|
if (!Strings.isBlank(detail.getCategory())){
|
detail.setCategoryName(
|
DictBizCache.getValues("industry_category",detail.getCategory())
|
);
|
}
|
// 返回
|
return detail;
|
}
|
|
/**
|
* 模糊查询企业名称、应急空间名称、风险源名称
|
* @param firmInfo
|
* @return
|
*/
|
@Override
|
public List<SearchVO> fuzzyQuery(FirmInfoVO firmInfo) {
|
return baseMapper.fuzzyQuery(firmInfo);
|
}
|
}
|
