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

Introduction

Here is the source code for org.openlmis.fulfillment.service.request.RequestHeaders.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 static org.apache.commons.lang3.StringUtils.isNotBlank;

import com.google.common.collect.Maps;
import org.springframework.http.HttpHeaders;
import java.util.Map;
import java.util.function.Consumer;

public final class RequestHeaders {
    private Map<String, String> headers = Maps.newHashMap();

    private RequestHeaders() {
    }

    public static RequestHeaders init() {
        return new RequestHeaders();
    }

    public RequestHeaders setAuth(String token) {
        return isNotBlank(token) ? set(HttpHeaders.AUTHORIZATION, "Bearer " + token) : this;
    }

    public RequestHeaders setIfNoneMatch(String value) {
        return set(HttpHeaders.IF_NONE_MATCH, value);
    }

    /**
     * Set parameter (key argument) with the value only if the value is not null.
     */
    public RequestHeaders set(String key, String value) {
        if (isNotBlank(value)) {
            headers.put(key, value);
        }

        return this;
    }

    public RequestHeaders setAll(RequestHeaders headers) {
        headers.forEach(entry -> set(entry.getKey(), entry.getValue()));
        return this;
    }

    /**
     * Converts this instance to {@link HttpHeaders}.
     */
    public HttpHeaders toHeaders() {
        HttpHeaders httpHeaders = new HttpHeaders();
        forEach((entry -> httpHeaders.set(entry.getKey(), entry.getValue())));
        return httpHeaders;
    }

    void forEach(Consumer<Map.Entry<String, String>> action) {
        headers.entrySet().forEach(action);
    }

}