From 3aae8ddd5b9624626921abdcc33122dccdcd5c61 Mon Sep 17 00:00:00 2001
From: “djh” <“3298565835@qq.com”>
Date: 星期五, 27 十二月 2024 13:24:46 +0800
Subject: [PATCH] 修改学时证书接收接口

---
 exam-system/src/main/java/com/gkhy/exam/noncoalmine/service/impl/TrainingInstitutionServiceImpl.java |   31 +++++++++++++++++++++++++------
 1 files changed, 25 insertions(+), 6 deletions(-)

diff --git a/exam-system/src/main/java/com/gkhy/exam/noncoalmine/service/impl/TrainingInstitutionServiceImpl.java b/exam-system/src/main/java/com/gkhy/exam/noncoalmine/service/impl/TrainingInstitutionServiceImpl.java
index 33526c0..c67861a 100644
--- a/exam-system/src/main/java/com/gkhy/exam/noncoalmine/service/impl/TrainingInstitutionServiceImpl.java
+++ b/exam-system/src/main/java/com/gkhy/exam/noncoalmine/service/impl/TrainingInstitutionServiceImpl.java
@@ -1,5 +1,6 @@
 package com.gkhy.exam.noncoalmine.service.impl;
 
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.gkhy.exam.noncoalmine.entity.TrainingInstitution;
@@ -11,6 +12,7 @@
 import com.gkhy.exam.noncoalmine.model.vo.TrainingInstitutionVO;
 import com.gkhy.exam.noncoalmine.service.TrainingInstitutionService;
 import com.ruoyi.common.core.domain.entity.SysUser;
+import com.ruoyi.common.core.domain.model.LoginUser;
 import com.ruoyi.common.utils.SecurityUtils;
 import com.ruoyi.common.utils.StringUtils;
 import com.ruoyi.system.domain.SysDistrict;
@@ -19,15 +21,11 @@
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
+import java.util.ArrayList;
 import java.util.List;
+import java.util.Set;
 import java.util.stream.Collectors;
 
-/**
- * (TrainingInstitution)表服务实现类
- *
- * @author makejava
- * @since 2023-09-13 16:12:31
- */
 @Service("trainingInstitutionService")
 public class TrainingInstitutionServiceImpl extends ServiceImpl<TrainingInstitutionMapper, TrainingInstitution> implements TrainingInstitutionService {
 	@Autowired
@@ -80,5 +78,26 @@
 		this.update(updateWrapper);
 	}
 
+	@Override
+	public List<TrainingInstitutionVO> selectByPermissionList(TrainingInstitutionQuery query) {
+		LoginUser loginUser = SecurityUtils.getLoginUser();
+		List<TrainingInstitutionVO> institutionVOList = new ArrayList<>();
+		Set<String> permissions = loginUser.getPermissions();
+		if(permissions.contains("admin") || permissions.contains("common")){
+			//管理员
+			institutionVOList = trainingInstitutionMapper.selectAll(query);
+		}else {
+			query.setUserId(loginUser.getUserId());
+			institutionVOList = trainingInstitutionMapper.getUserInstitutions(query);
+		}
+
+		return institutionVOList;
+	}
+
+	@Override
+	public List<TrainingInstitutionVO> select(TrainingInstitutionQuery query) {
+		return trainingInstitutionMapper.selectAll(query);
+	}
+
 }
 

--
Gitblit v1.9.2