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);
	}
}