From 5385f7748062cc72c7b1d5c8651630751abe5a41 Mon Sep 17 00:00:00 2001
From: zhongrj <646384940@qq.com>
Date: Fri, 02 Feb 2024 20:05:42 +0800
Subject: [PATCH] 社区警格操作调整

---
 src/main/java/org/springblade/modules/police/service/impl/PoliceStationServiceImpl.java |   38 +++++++++++++++++++++++++++++++++++++-
 1 files changed, 37 insertions(+), 1 deletions(-)

diff --git a/src/main/java/org/springblade/modules/police/service/impl/PoliceStationServiceImpl.java b/src/main/java/org/springblade/modules/police/service/impl/PoliceStationServiceImpl.java
index ddad135..b4978ee 100644
--- a/src/main/java/org/springblade/modules/police/service/impl/PoliceStationServiceImpl.java
+++ b/src/main/java/org/springblade/modules/police/service/impl/PoliceStationServiceImpl.java
@@ -1,12 +1,18 @@
 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.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 +28,35 @@
 		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);
+			}
+		}
+	}
 }

--
Gitblit v1.9.3