package com.qianwen.mdc.domain.workshop;
|
|
import java.util.List;
|
import java.util.Optional;
|
import java.util.stream.Collectors;
|
|
import com.qianwen.mdc.domain.Section;
|
import com.qianwen.mdc.dto.SectionDTO;
|
import com.qianwen.mdc.dto.workshop.WorkshopDTO;
|
|
/**
|
* 负责类的组装转换工作
|
*/
|
public class WorkshopAssembler {
|
|
/**
|
* 实体转换为DTO
|
* @param workshop
|
* @return
|
*/
|
public static WorkshopDTO toDTO(Workshop workshop,List<Section> sections) {
|
WorkshopDTO dto = new WorkshopDTO();
|
|
dto.setId(workshop.getId());
|
dto.setName(workshop.getName());
|
|
Optional.ofNullable(sections).ifPresent(slist -> {
|
dto.setSections(slist.stream().map(s -> {
|
|
SectionDTO sdto = new SectionDTO();
|
sdto.setId(s.getId());
|
sdto.setName(s.getName());
|
sdto.setWorkshopId(workshop.getId());
|
return sdto;
|
}).collect(Collectors.toList()));
|
});
|
return dto;
|
}
|
|
public static WorkshopDTO toDTO(Workshop workshop) {
|
|
return toDTO(workshop,null);
|
}
|
}
|