ComplaintWorkOrderAssembler.java

package com.eksad.masterdata.domain.assembler;

import com.eksad.ddms.common.dto.CreationalSpecificationDTO;
import com.eksad.ddms.common.util.object.IObjectAssembler;
import com.eksad.masterdata.common.dto.ComplaintWorkOrderDTO;
import com.eksad.masterdata.domain.workorder.ComplaintWorkOrder;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

public class ComplaintWorkOrderAssembler implements IObjectAssembler<ComplaintWorkOrder, ComplaintWorkOrderDTO> {

    @Override
    public ComplaintWorkOrderDTO toDTO(ComplaintWorkOrder domainObject) {
        ComplaintWorkOrderDTO dto = new ComplaintWorkOrderDTO();
        dto.setComplaintWoID(domainObject.getComplaintWoID());
        dto.setComplaintID(domainObject.getComplaintID());
        dto.setComplaintName(domainObject.getComplaintName());
        dto.setAnalisa(domainObject.getAnalisa());
        dto.setWoID(domainObject.getWoID());
        dto.setComplaintWoCreationalDTO(domainObject.getComplaintWoCreational() == null ? new CreationalSpecificationDTO() : new CreationalSpecificationDTOAssembler().toDTO(domainObject.getComplaintWoCreational()));
        return dto;
    }

    @Override
    public ComplaintWorkOrder toDomain(ComplaintWorkOrderDTO dtoObject) {
        ComplaintWorkOrder data = new ComplaintWorkOrder();
        data.setComplaintWoID(dtoObject.getComplaintWoID());
        data.setComplaintID(dtoObject.getComplaintID());
        data.setComplaintName(dtoObject.getComplaintName());
        data.setAnalisa(dtoObject.getAnalisa());
        data.setWoID(dtoObject.getWoID());
        data.setComplaintWoCreational(dtoObject.getComplaintWoCreationalDTO() == null ? new CreationalSpecificationDTOAssembler().toDomain(new CreationalSpecificationDTO().getInstance()) : new CreationalSpecificationDTOAssembler().toDomain(dtoObject.getComplaintWoCreationalDTO()));
        return data;
    }

    public List<ComplaintWorkOrderDTO> toDTOs(Set<ComplaintWorkOrder> arg0) {
        List<ComplaintWorkOrderDTO> res = new ArrayList<>();
        arg0.stream().forEach((o) -> {
            res.add(toDTO(o));
        });
        return res;
    }

    public List<ComplaintWorkOrderDTO> toDTOs(List<ComplaintWorkOrder> arg0) {
        List<ComplaintWorkOrderDTO> res = new ArrayList<>();
        arg0.stream().forEach((o) -> {
            res.add(toDTO(o));
        });
        return res;
    }

    public Set<ComplaintWorkOrder> toDomain(List<ComplaintWorkOrderDTO> arg0) {
        Set<ComplaintWorkOrder> res = new HashSet<>();
        arg0.stream().forEach((o) -> {
            res.add(toDomain(o));
        });
        return res;
    }
}