package org.jeecg.modules.mdc.service.impl;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import org.apache.commons.lang3.StringUtils;
|
import org.jeecg.modules.mdc.entity.MdcDeviceCalendar;
|
import org.jeecg.modules.mdc.entity.MdcShiftSub;
|
import org.jeecg.modules.mdc.mapper.MdcDeviceCalendarMapper;
|
import org.jeecg.modules.mdc.service.IMdcDeviceCalendarService;
|
import org.jeecg.modules.mdc.service.IMdcEquipmentService;
|
import org.jeecg.modules.mdc.service.IMdcShiftSubService;
|
import org.jeecg.modules.mdc.vo.EquipmentCalendarVo;
|
import org.jeecg.modules.mdc.vo.MdcDeviceCalendarQueryVo;
|
import org.jeecg.modules.mdc.vo.MdcDeviceCalendarVo;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import javax.annotation.Resource;
|
import javax.servlet.http.HttpServletRequest;
|
import java.util.*;
|
|
/**
|
* @Description: 设备工作日历表
|
* @Author: jeecg-boot
|
* @Date: 2023-04-10
|
* @Version: V1.0
|
*/
|
@Service
|
public class MdcDeviceCalendarServiceImpl extends ServiceImpl<MdcDeviceCalendarMapper, MdcDeviceCalendar> implements IMdcDeviceCalendarService {
|
|
@Resource
|
private IMdcEquipmentService mdcEquipmentService;
|
@Resource
|
private IMdcShiftSubService mdcShiftSubService;
|
|
/**
|
* 分页列表查询
|
*/
|
@Override
|
public IPage<MdcDeviceCalendarVo> pageList(String userId, MdcDeviceCalendarQueryVo mdcDeviceCalendarQueryVo, Integer pageNo, Integer pageSize, HttpServletRequest req) {
|
IPage<MdcDeviceCalendarVo> pageData = new Page<>(pageNo, pageSize);
|
List<String> equipmentIds = new ArrayList<>();
|
if (StringUtils.isNotEmpty(mdcDeviceCalendarQueryVo.getParentId()) && StringUtils.isEmpty(mdcDeviceCalendarQueryVo.getEquipmentId())) {
|
if ("2".equals(mdcDeviceCalendarQueryVo.getTypeTree())) {
|
// 部门层级
|
equipmentIds = mdcEquipmentService.getEquipmentIdsByDepart(userId, mdcDeviceCalendarQueryVo.getParentId());
|
} else {
|
// 产线层级
|
equipmentIds = mdcEquipmentService.getEquipmentIdsProduction(userId, mdcDeviceCalendarQueryVo.getParentId());
|
}
|
} else if (StringUtils.isNotEmpty(mdcDeviceCalendarQueryVo.getEquipmentId())) {
|
// 单台设备信息
|
mdcDeviceCalendarQueryVo.setEquipmentIdList(Collections.singletonList(mdcDeviceCalendarQueryVo.getEquipmentId()));
|
} else {
|
// 查询用户拥有的所有设备信息
|
if ("2".equals(mdcDeviceCalendarQueryVo.getTypeTree())) {
|
//部门层级
|
equipmentIds = mdcEquipmentService.getEquipmentIdsByDepart(userId, null);
|
} else {
|
//产线层级
|
equipmentIds = mdcEquipmentService.getEquipmentIdsProduction(userId, null);
|
}
|
}
|
if (mdcDeviceCalendarQueryVo.getEquipmentIdList() == null || mdcDeviceCalendarQueryVo.getEquipmentIdList().isEmpty()) {
|
mdcDeviceCalendarQueryVo.setEquipmentIdList(equipmentIds);
|
}
|
|
if (mdcDeviceCalendarQueryVo.getEquipmentIdList() == null || mdcDeviceCalendarQueryVo.getEquipmentIdList().isEmpty()) {
|
return null;
|
}
|
|
return this.baseMapper.pageList(pageData, mdcDeviceCalendarQueryVo);
|
}
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public boolean saveCalendar(EquipmentCalendarVo calendarVo) {
|
String[] equipmentIdList = calendarVo.getEquipmentId().split(",");
|
List<String> dateList = calendarVo.getDateList();
|
List<MdcShiftSub> mdcShiftSubList = mdcShiftSubService.list(new LambdaQueryWrapper<MdcShiftSub>().eq(MdcShiftSub::getShiftId, calendarVo.getShiftId()).eq(MdcShiftSub::getShiftSubStatus, "1"));
|
List<MdcDeviceCalendar> calendarList = new ArrayList<>();
|
for (String equipmentId : equipmentIdList) {
|
for (String date : dateList) {
|
for (MdcShiftSub mdcShiftSub : mdcShiftSubList) {
|
this.baseMapper.delete(new LambdaQueryWrapper<MdcDeviceCalendar>().eq(MdcDeviceCalendar::getEquipmentId, equipmentId).eq(MdcDeviceCalendar::getEffectiveDate, date));
|
MdcDeviceCalendar mdcDeviceCalendar = new MdcDeviceCalendar();
|
mdcDeviceCalendar.setEffectiveDate(date);
|
mdcDeviceCalendar.setEquipmentId(equipmentId);
|
mdcDeviceCalendar.setShiftId(mdcShiftSub.getShiftId());
|
mdcDeviceCalendar.setShiftSubId(mdcShiftSub.getId());
|
calendarList.add(mdcDeviceCalendar);
|
}
|
}
|
}
|
this.saveBatch(calendarList);
|
return true;
|
}
|
|
/**
|
* 查询班制数据
|
*
|
* @param equipmentId
|
* @param stringDates
|
* @return
|
*/
|
@Override
|
public List<MdcDeviceCalendarVo> listByEquipmentAndDate(String equipmentId, List<String> stringDates) {
|
List<MdcDeviceCalendarVo> result = new ArrayList<>();
|
try {
|
//查询默认班制
|
List<MdcDeviceCalendarVo> acquiesceShift = this.baseMapper.findAcquiesceShift();
|
|
for (String stringDate : stringDates) {
|
List<MdcDeviceCalendarVo> mdcDeviceCalendarVos = this.baseMapper.listByEquipmentAndDate(equipmentId, stringDate);
|
if (mdcDeviceCalendarVos != null && !mdcDeviceCalendarVos.isEmpty()) {
|
result.addAll(mdcDeviceCalendarVos);
|
} else {
|
acquiesceShift.forEach(mdcDeviceCalendarVo -> {
|
mdcDeviceCalendarVo.setEquipmentId(equipmentId);
|
mdcDeviceCalendarVo.setEffectiveDate(stringDate);
|
});
|
// 设置默认班制
|
result.addAll(acquiesceShift);
|
}
|
}
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
|
return result;
|
}
|
|
}
|
