考试申报

zf

2024-02-19 1936cef71ca4ef95d52c951b4af5948c7885b893

 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);
   }

}