org.openlmis.fulfillment.web.shipmentdraft.ShipmentDraftDtoBuilder.java Source code

Introduction

Here is the source code for org.openlmis.fulfillment.web.shipmentdraft.ShipmentDraftDtoBuilder.java

Source

/*
 * This program is part of the OpenLMIS logistics management information system platform software.
 * Copyright  2017 VillageReach
 *
 * This program is free software: you can redistribute it and/or modify it under the terms
 * of the GNU Affero General Public License as published by the Free Software Foundation, either
 * version 3 of the License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU Affero General Public License for more details. You should have received a copy of
 * the GNU Affero General Public License along with this program. If not, see
 * http://www.gnu.org/licenses. For additional information contact info@OpenLMIS.org.
 */

package org.openlmis.fulfillment.web.shipmentdraft;

import org.openlmis.fulfillment.domain.ShipmentDraft;
import org.openlmis.fulfillment.domain.ShipmentDraftLineItem;
import org.openlmis.fulfillment.web.shipment.ShipmentLineItemDto;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.stream.Collectors;

@Component
public class ShipmentDraftDtoBuilder {

    @Value("${service.url}")
    private String serviceUrl;

    /**
     * Create a new list of {@link ShipmentDraftDto} based on data
     * from {@link ShipmentDraft}.
     *
     * @param shipments collection used to create {@link ShipmentDraftDto} list (can be {@code null})
     * @return new list of {@link ShipmentDraftDto}. Empty list if passed argument is {@code null}.
     */
    public List<ShipmentDraftDto> build(Collection<ShipmentDraft> shipments) {
        if (null == shipments) {
            return Collections.emptyList();
        }
        return shipments.stream().map(this::export).collect(Collectors.toList());
    }

    /**
     * Create a new instance of {@link ShipmentDraftDto} based on data
     * from {@link ShipmentDraft}.
     *
     * @param draft instance used to create {@link ShipmentDraftDto} (can be {@code null})
     * @return new instance of {@link ShipmentDraftDto}.
     * {@code null} if passed argument is {@code null}.
     */
    public ShipmentDraftDto build(ShipmentDraft draft) {
        return draft != null ? export(draft) : null;
    }

    private ShipmentDraftDto export(ShipmentDraft draft) {
        ShipmentDraftDto dto = new ShipmentDraftDto();
        dto.setServiceUrl(serviceUrl);
        draft.export(dto);
        dto.setLineItems(exportToDtos(draft.viewLineItems()));

        return dto;
    }

    private List<ShipmentLineItemDto> exportToDtos(List<ShipmentDraftLineItem> lineItems) {
        List<ShipmentLineItemDto> lineItemDtos = new ArrayList<>(lineItems.size());
        lineItems.forEach(l -> lineItemDtos.add(exportToDto(l)));
        return lineItemDtos;
    }

    private ShipmentLineItemDto exportToDto(ShipmentDraftLineItem lineItem) {
        ShipmentLineItemDto lineItemDto = new ShipmentLineItemDto();
        lineItemDto.setServiceUrl(serviceUrl);
        lineItem.export(lineItemDto);
        return lineItemDto;
    }

}