修改

heheng

2025-05-19 3a762add44449332d6d379e361698850f6066e16

 exam-system/src/main/java/com/gkhy/exam/noncoalmine/service/impl/TrainingInstitutionServiceImpl.java

@@ -1,7 +1,10 @@
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;
import com.gkhy.exam.noncoalmine.entity.ViolationRegistration;
import com.gkhy.exam.noncoalmine.mapper.TrainingInstitutionMapper;
import com.gkhy.exam.noncoalmine.model.addForm.TrainingInstitutionAddForm;
import com.gkhy.exam.noncoalmine.model.modForm.TrainingInstitutionModForm;
@@ -9,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;
@@ -17,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
@@ -36,7 +36,7 @@
   @Override
   public List<TrainingInstitutionVO> listByPage(TrainingInstitutionQuery query) {
      SysUser user = SecurityUtils.getLoginUser().getUser();
      if(StringUtils.isEmpty(query.getRegionCode())){
      if(StringUtils.isEmpty(query.getDistrictCode())){
         //传入区划为空
         if(!user.isAdmin(user.getUserId())){
            //非超管
@@ -47,7 +47,7 @@
                  .collect(Collectors.toList());
            if(districtList.size() > 0){
               SysDistrict sysDistrict = districtList.get(0);
               query.setRegionCode(sysDistrict.getCode());
               query.setDistrictCode(sysDistrict.getCode());
            }
         }
      }
@@ -69,5 +69,35 @@
      trainingInstitution.setDelFlag((byte)0);
      return baseMapper.insert(trainingInstitution);
   }

   @Override
   public void delBatch(List<Long> institutionIds) {
      UpdateWrapper<TrainingInstitution> updateWrapper = new UpdateWrapper<>();
      updateWrapper.in("institution_id",institutionIds)
            .set("del_flag",(byte)2);
      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);
   }

}