package com.chinaztt.mes.basic.service.impl;
|
|
import cn.hutool.core.collection.CollectionUtil;
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.chinaztt.mes.basic.dto.CrewDTO;
|
import com.chinaztt.mes.basic.entity.Crew;
|
import com.chinaztt.mes.basic.entity.JoinCrewStaff;
|
import com.chinaztt.mes.basic.entity.Part;
|
import com.chinaztt.mes.basic.entity.Staff;
|
import com.chinaztt.mes.basic.mapper.CrewMapper;
|
import com.chinaztt.mes.basic.mapper.JoinCrewStaffMapper;
|
import com.chinaztt.mes.basic.mapper.StaffMapper;
|
import com.chinaztt.mes.basic.service.CrewService;
|
import lombok.AllArgsConstructor;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
|
/**
|
* 人员班组维护表
|
*
|
* @author cxf
|
* @date 2020-12-01 09:58:28
|
*/
|
@Service
|
@AllArgsConstructor
|
@Transactional(rollbackFor = Exception.class)
|
public class CrewServiceImpl extends ServiceImpl<CrewMapper, Crew> implements CrewService {
|
|
private JoinCrewStaffMapper joinCrewStaffMapper;
|
private StaffMapper staffMapper;
|
|
|
@Override
|
public boolean updateDtoById(CrewDTO crewDTO) {
|
try {
|
baseMapper.updateById(crewDTO);
|
} catch (Exception e) {
|
throw new RuntimeException("班组修改失败");
|
}
|
return true;
|
}
|
|
@Override
|
public boolean deleteById(Long id) {
|
try {
|
baseMapper.deleteById(id);
|
joinCrewStaffMapper.delete(Wrappers.<JoinCrewStaff>lambdaQuery().eq(JoinCrewStaff::getCrewId, id));
|
} catch (Exception e) {
|
throw new RuntimeException("班组删除失败");
|
}
|
return true;
|
}
|
|
@Override
|
public IPage<List<Staff>> getStaffByCrewId(Page page, QueryWrapper<Staff> ew) {
|
return staffMapper.selectStaffByCrewId(page, ew);
|
}
|
|
@Override
|
public boolean saveDto(CrewDTO crewDTO) {
|
saveJoin(crewDTO);
|
return false;
|
}
|
|
@Override
|
public boolean deleteStaff(CrewDTO crewDTO) {
|
crewDTO.getStaffList().forEach(staff -> {
|
joinCrewStaffMapper.delete(Wrappers.<JoinCrewStaff>lambdaQuery().eq(JoinCrewStaff::getCrewId, crewDTO.getId()).eq(JoinCrewStaff::getStaffId, staff.getId()));
|
});
|
return false;
|
}
|
|
@Override
|
public List<Staff> getStaff(CrewDTO crewDTO) {
|
return staffMapper.selectStaffById(crewDTO.getId());
|
}
|
|
public void saveJoin(CrewDTO crewDTO) {
|
if (CollectionUtil.isNotEmpty(crewDTO.getStaffList())) {
|
List<Staff> staffList = joinCrewStaffMapper.selectStaffByCrewId(crewDTO.getId());
|
crewDTO.getStaffList().stream()
|
.filter(newStaff -> {
|
for (int i = 0; i < staffList.size(); i++) {
|
if (staffList.get(i).getId().equals(newStaff.getId())) {
|
return false;
|
}
|
}
|
return true;
|
}).forEach(newStaff -> {
|
JoinCrewStaff joinCrewStaff = new JoinCrewStaff();
|
joinCrewStaff.setCrewId(crewDTO.getId());
|
joinCrewStaff.setStaffId(newStaff.getId());
|
joinCrewStaffMapper.insert(joinCrewStaff);
|
});
|
}
|
}
|
}
|
