From 81e41215c522743d2d21b8011049347a523ef6ad Mon Sep 17 00:00:00 2001 From: 无关风月 <443237572@qq.com> Date: 星期三, 04 六月 2025 09:29:17 +0800 Subject: [PATCH] 管理后台代码 --- ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TLeaveServiceImpl.java | 55 +++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 55 insertions(+), 0 deletions(-) diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TLeaveServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TLeaveServiceImpl.java index e3d0f86..0d486e3 100644 --- a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TLeaveServiceImpl.java +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TLeaveServiceImpl.java @@ -1,10 +1,25 @@ package com.ruoyi.system.service.impl; +import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ruoyi.common.basic.PageInfo; +import com.ruoyi.system.mapper.TDeptMapper; import com.ruoyi.system.mapper.TLeaveMapper; +import com.ruoyi.system.mapper.TProjectDeptMapper; +import com.ruoyi.system.model.TDept; import com.ruoyi.system.model.TLeave; +import com.ruoyi.system.model.TProjectDept; +import com.ruoyi.system.query.LeaveListQuery; import com.ruoyi.system.service.TLeaveService; +import com.ruoyi.system.vo.system.DeptListVO; +import com.ruoyi.system.vo.system.LeaveListVO; +import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; +import org.springframework.util.StringUtils; + +import java.time.format.DateTimeFormatter; +import java.util.List; +import java.util.stream.Collectors; /** * <p> @@ -17,4 +32,44 @@ @Service public class TLeaveServiceImpl extends ServiceImpl<TLeaveMapper, TLeave> implements TLeaveService { + @Autowired + private TProjectDeptMapper projectDeptMapper; + @Autowired + private TDeptMapper deptMapper; + @Override + public PageInfo<LeaveListVO> pageList(LeaveListQuery query) { + if (StringUtils.hasLength(query.getDeptName())){ + List<String> collect = projectDeptMapper.selectList(new LambdaQueryWrapper<TProjectDept>().like(TProjectDept::getProjectName, query.getDeptName())) + .stream().map(TProjectDept::getId).collect(Collectors.toList()); + List<String> collect1 = deptMapper.selectList(new LambdaQueryWrapper<TDept>().like(TDept::getDeptName, query.getDeptName())) + .stream().map(TDept::getId).collect(Collectors.toList()); + collect.addAll(collect1); + if (collect.isEmpty()){ + collect.add("0"); + } + query.setDeptIds(collect); + } + + PageInfo<LeaveListVO> pageInfo = new PageInfo<>(query.getPageNum(), query.getPageSize()); + List<LeaveListVO> list = this.baseMapper.pageList(query,pageInfo); + for (LeaveListVO leaveListVO : list) { + String start = leaveListVO.getStartTime().format(DateTimeFormatter.ofPattern("yyyy-MM-dd")); + String end = leaveListVO.getEndTime().format(DateTimeFormatter.ofPattern("yyyy-MM-dd")); + leaveListVO.setLeaveTime(start+"至"+end); + if (leaveListVO.getDeptType() == 1){ + TProjectDept tProjectDept = projectDeptMapper.selectById(leaveListVO.getDeptId()); + if (!tProjectDept.getParentId().equals("0")){ + TProjectDept tProjectDept1 = projectDeptMapper.selectById(tProjectDept.getParentId()); + leaveListVO.setDeptName(tProjectDept1.getProjectName()+">"+tProjectDept.getProjectName()); + }else{ + leaveListVO.setDeptName(tProjectDept.getProjectName()); + } + }else{ + TDept tDept = deptMapper.selectById(leaveListVO.getDeptId()); + leaveListVO.setDeptName(tDept.getDeptName()); + } + } + pageInfo.setRecords(list); + return pageInfo; + } } -- Gitblit v1.7.1