package com.qianwen.mdc.service.workshop;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
import org.apache.tomcat.util.buf.StringUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.qianwen.mdc.domain.Section;
|
import com.qianwen.mdc.domain.workshop.Workshop;
|
import com.qianwen.mdc.domain.workshop.WorkshopAssembler;
|
import com.qianwen.mdc.dto.workshop.WorkshopDTO;
|
import com.qianwen.mdc.dto.workshop.WorkshopFullDTO;
|
import com.qianwen.mdc.dto.workshop.WorkshopQueryDTO;
|
import com.qianwen.mdc.mapper.SectionMapper;
|
import com.qianwen.mdc.mapper.WorkshopMapper;
|
|
/**
|
* @author y_ys79
|
* 车间服务，提供增删改查全部操作（逻辑比较简单）
|
*/
|
@Service
|
public class WorkshopService {
|
@Autowired
|
private WorkshopMapper workshopMapper;
|
@Autowired
|
private SectionMapper sectionMapper;
|
|
|
/**
|
* 获取车间列表，附带车间下的工段数据
|
* @return
|
*/
|
@Transactional(readOnly=true)
|
public List<WorkshopDTO> listWithSections() {
|
List<Workshop> list = workshopMapper.selectList(Wrappers.emptyWrapper());
|
return list.stream().map(ws -> {
|
List<Section> sectionList = sectionMapper.queryByWorkshopId(ws.getId());
|
WorkshopDTO wsDTO = WorkshopAssembler.toDTO(ws,sectionList);
|
|
return wsDTO;
|
}).collect(Collectors.toList());
|
}
|
|
@Transactional(readOnly=true)
|
public Page<WorkshopFullDTO> pageQuery(WorkshopQueryDTO dto) {
|
Page<WorkshopFullDTO> rowPage = new Page<>(dto.finalPageNo(),dto.finalPageSize());
|
rowPage = workshopMapper.queryPage(rowPage, dto);
|
|
rowPage.getRecords().forEach((ws) -> {
|
List<Section> sectionList = sectionMapper.queryByWorkshopId(ws.getId());
|
List<String> nameList = new ArrayList<>();
|
for(Section s : sectionList) {
|
nameList.add(s.getName());
|
}
|
|
ws.setSectionNames(StringUtils.join(nameList,','));
|
});
|
|
return rowPage;
|
}
|
}
|
