From a1272d78844574b86645149a0d8ca257c2b8fe91 Mon Sep 17 00:00:00 2001
From: zhongrj <646384940@qq.com>
Date: Thu, 26 Dec 2024 09:43:14 +0800
Subject: [PATCH] 风险源统计等级空的设置为其他

---
 src/main/java/org/springblade/modules/yw/service/impl/ProTarServiceImpl.java |   72 ++++++++++++++++++++++++++++++++++++
 1 files changed, 72 insertions(+), 0 deletions(-)

diff --git a/src/main/java/org/springblade/modules/yw/service/impl/ProTarServiceImpl.java b/src/main/java/org/springblade/modules/yw/service/impl/ProTarServiceImpl.java
index 09a96a7..fdd9793 100644
--- a/src/main/java/org/springblade/modules/yw/service/impl/ProTarServiceImpl.java
+++ b/src/main/java/org/springblade/modules/yw/service/impl/ProTarServiceImpl.java
@@ -1,12 +1,20 @@
 package org.springblade.modules.yw.service.impl;
 
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import org.apache.logging.log4j.util.Strings;
+import org.springblade.common.utils.PositionUtil;
+import org.springblade.core.tool.utils.BeanUtil;
 import org.springblade.modules.yw.entity.ProTarEntity;
+import org.springblade.modules.yw.excel.ProTarExcel;
 import org.springblade.modules.yw.vo.ProTarVO;
 import org.springblade.modules.yw.mapper.ProTarMapper;
 import org.springblade.modules.yw.service.IProTarService;
 import org.springframework.stereotype.Service;
 import com.baomidou.mybatisplus.core.metadata.IPage;
+
+import java.util.List;
+import java.util.Objects;
 
 /**
  * 保护目标表 服务实现类
@@ -17,9 +25,73 @@
 @Service
 public class ProTarServiceImpl extends ServiceImpl<ProTarMapper, ProTarEntity> implements IProTarService {
 
+	/**
+	 * 自定义分页查询
+	 * @param page
+	 * @param proTar
+	 * @return
+	 */
 	@Override
 	public IPage<ProTarVO> selectProTarPage(IPage<ProTarVO> page, ProTarVO proTar) {
 		return page.setRecords(baseMapper.selectProTarPage(page, proTar));
 	}
 
+	/**
+	 * 导入保护目标信息
+	 * @param data
+	 * @param isCovered
+	 * @return
+	 */
+	@Override
+	public String importProTar(List<ProTarExcel> data, boolean isCovered) {
+		for (ProTarExcel proTarExcel : data) {
+			// 数据拷贝
+			ProTarEntity proTarEntity = Objects.requireNonNull(BeanUtil.copy(proTarExcel, ProTarEntity.class));
+			// 类目转换
+			if (!Strings.isBlank(proTarExcel.getCategory())){
+				proTarEntity.setCategory(Integer.parseInt(proTarExcel.getCategory()));
+			}
+			// 经纬度处理
+//			if (!Strings.isBlank(proTarExcel.getLng())){
+//				proTarEntity.setLng(PositionUtil.tranformPos(proTarExcel.getLng()).toString());
+//			}
+//			if (!Strings.isBlank(proTarExcel.getLat())){
+//				proTarEntity.setLat(PositionUtil.tranformPos(proTarExcel.getLat()).toString());
+//			}
+//			if (!Strings.isBlank(proTarExcel.getEndLng())){
+//				proTarEntity.setEndLng(PositionUtil.tranformPos(proTarExcel.getEndLng()).toString());
+//			}
+//			if (!Strings.isBlank(proTarExcel.getEndLat())){
+//				proTarEntity.setEndLat(PositionUtil.tranformPos(proTarExcel.getEndLat()).toString());
+//			}
+			// 判断是否已经入库
+			Long id = isSave(proTarExcel);
+			if (null!=id){
+				if (isCovered){
+					// 覆盖更新
+					proTarEntity.setId(id);
+					updateById(proTarEntity);
+					continue;
+				}
+			}
+			// 保存入库
+			save(proTarEntity);
+		}
+		return null;
+	}
+
+	/**
+	 * 判断保护目标是否已入库
+	 * @param proTarExcel
+	 * @return
+	 */
+	private Long isSave(ProTarExcel proTarExcel) {
+		QueryWrapper<ProTarEntity> wrapper = new QueryWrapper<>();
+		wrapper.eq("name",proTarExcel.getName()).eq("is_deleted",0);
+		ProTarEntity one = getOne(wrapper);
+		if (null!=one){
+			return one.getId();
+		}
+		return null;
+	}
 }

--
Gitblit v1.9.3