From 8208e8137a3d950152d358ccd2ab0a8dc08cf4de Mon Sep 17 00:00:00 2001
From: zf <1603559716@qq.com>
Date: 星期四, 14 九月 2023 17:13:20 +0800
Subject: [PATCH] 机构&&考点

---
 exam-system/src/main/java/com/gkhy/exam/noncoalmine/service/impl/ExamSiteServiceImpl.java |   63 +++++++++++++++++++++++++++++++
 1 files changed, 63 insertions(+), 0 deletions(-)

diff --git a/exam-system/src/main/java/com/gkhy/exam/noncoalmine/service/impl/ExamSiteServiceImpl.java b/exam-system/src/main/java/com/gkhy/exam/noncoalmine/service/impl/ExamSiteServiceImpl.java
index fa9899e..907d442 100644
--- a/exam-system/src/main/java/com/gkhy/exam/noncoalmine/service/impl/ExamSiteServiceImpl.java
+++ b/exam-system/src/main/java/com/gkhy/exam/noncoalmine/service/impl/ExamSiteServiceImpl.java
@@ -3,8 +3,22 @@
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.gkhy.exam.noncoalmine.entity.ExamSite;
 import com.gkhy.exam.noncoalmine.mapper.ExamSiteMapper;
+import com.gkhy.exam.noncoalmine.model.addForm.ExamSiteAddForm;
+import com.gkhy.exam.noncoalmine.model.modForm.ExamSiteModForm;
+import com.gkhy.exam.noncoalmine.model.query.ExamSiteQuery;
+import com.gkhy.exam.noncoalmine.model.vo.ExamSiteVO;
 import com.gkhy.exam.noncoalmine.service.ExamSiteService;
+import com.ruoyi.common.core.domain.entity.SysUser;
+import com.ruoyi.common.utils.SecurityUtils;
+import com.ruoyi.common.utils.StringUtils;
+import com.ruoyi.system.domain.SysDistrict;
+import com.ruoyi.system.service.SysDistrictService;
+import org.springframework.beans.BeanUtils;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+
+import java.util.List;
+import java.util.stream.Collectors;
 
 /**
  * (ExamSite)表服务实现类
@@ -14,6 +28,55 @@
  */
 @Service("examSiteService")
 public class ExamSiteServiceImpl extends ServiceImpl<ExamSiteMapper, ExamSite> implements ExamSiteService {
+	@Autowired
+	private SysDistrictService districtService;
+	@Autowired
+	private ExamSiteMapper examSiteMapper;
+	@Override
+	public List<ExamSiteVO> listByPage(ExamSiteQuery query) {
+		SysUser user = SecurityUtils.getLoginUser().getUser();
+		if(StringUtils.isEmpty(query.getDistrictCode())){
+			//传入区划为空
+			if(!user.isAdmin(user.getUserId())){
+				//非超管
+				List<SysDistrict> districtList = districtService
+						.selectAll()
+						.stream()
+						.filter(sysDistrict -> sysDistrict.getId().equals(user.getDistrictId()))
+						.collect(Collectors.toList());
+				if(districtList.size() > 0){
+					SysDistrict sysDistrict = districtList.get(0);
+					query.setDistrictCode(sysDistrict.getCode());
+				}
+			}
+		}
+		List<ExamSiteVO> trainingInstitutionVOS = examSiteMapper.listByPage(query);
+		return trainingInstitutionVOS;
+	}
 
+	/**
+	 * 新增
+	 * @param addForm
+	 * @return
+	 */
+	@Override
+	public int add(ExamSiteAddForm addForm) {
+		ExamSite examSite = new ExamSite();
+		BeanUtils.copyProperties(addForm,examSite);
+		examSite.setDelFlag((byte)0);
+		return baseMapper.insert(examSite);
+	}
+
+	/**
+	 * 修改
+	 * @param modForm
+	 * @return
+	 */
+	@Override
+	public int mod(ExamSiteModForm modForm) {
+		ExamSite examSite = new ExamSite();
+		BeanUtils.copyProperties(modForm,examSite);
+		return baseMapper.updateById(examSite);
+	}
 }
 

--
Gitblit v1.9.2