机构&&考点

zf

2023-09-14 8208e8137a3d950152d358ccd2ab0a8dc08cf4de

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