org.openlmis.fulfillment.service.request.RequestHelper.java Source code

Introduction

Here is the source code for org.openlmis.fulfillment.service.request.RequestHelper.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.service.request;

import java.io.UnsupportedEncodingException;
import java.net.URI;
import java.nio.charset.StandardCharsets;
import java.util.Arrays;
import java.util.stream.Stream;
import org.apache.commons.lang3.tuple.Pair;
import org.springframework.http.HttpEntity;
import org.springframework.web.util.UriComponentsBuilder;
import org.springframework.web.util.UriUtils;

public final class RequestHelper {

    private RequestHelper() {
        throw new UnsupportedOperationException();
    }

    /**
     * Creates a {@link URI} from the given string representation without any parameters.
     */
    public static URI createUri(String url) {
        return createUri(url, null);
    }

    /**
     * Creates a {@link URI} from the given string representation and with the given parameters.
     */
    public static URI createUri(String url, RequestParameters parameters) {
        UriComponentsBuilder builder = UriComponentsBuilder.newInstance().uri(URI.create(url));

        RequestParameters.init().setAll(parameters).forEach(e -> e.getValue().forEach(one -> {
            try {
                builder.queryParam(e.getKey(),
                        UriUtils.encodeQueryParam(String.valueOf(one), StandardCharsets.UTF_8.name()));
            } catch (UnsupportedEncodingException ex) {
                throw new EncodingException(ex);
            }
        }));

        return builder.build(true).toUri();
    }

    /**
     * Creates an {@link HttpEntity} with the given payload as a body and headers.
     */
    public static <E> HttpEntity<E> createEntity(E payload, RequestHeaders headers) {
        return new HttpEntity<>(payload, headers.toHeaders());
    }

    /**
     * Creates an {@link HttpEntity} with the given headers.
     */
    public static <E> HttpEntity<E> createEntity(RequestHeaders headers) {
        return new HttpEntity<>(headers.toHeaders());
    }

    /**
     * Split the given {@link RequestParameters} into smaller chunks.
     */
    public static URI[] splitRequest(String url, RequestParameters queryParams, int maxUrlLength) {
        RequestParameters safeQueryParams = RequestParameters.init().setAll(queryParams);
        URI uri = createUri(url, safeQueryParams);

        if (uri.toString().length() > maxUrlLength) {
            Pair<RequestParameters, RequestParameters> split = safeQueryParams.split();

            if (null != split.getLeft() && null != split.getRight()) {
                URI[] left = splitRequest(url, split.getLeft(), maxUrlLength);
                URI[] right = splitRequest(url, split.getRight(), maxUrlLength);

                return Stream.concat(Arrays.stream(left), Arrays.stream(right)).distinct().toArray(URI[]::new);
            }
        }

        return new URI[] { uri };
    }

}