org.elasticsearch.rest.RestResponse Maven / Gradle / Ivy
/*
* Licensed to Elasticsearch under one or more contributor
* license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright
* ownership. Elasticsearch licenses this file to you under
* the Apache License, Version 2.0 (the "License"); you may
* not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.elasticsearch.rest;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import org.elasticsearch.common.Nullable;
import org.elasticsearch.common.bytes.BytesReference;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
*
*/
public abstract class RestResponse implements HasRestHeaders {
protected Map> customHeaders;
/**
* The response content type.
*/
public abstract String contentType();
/**
* The response content. Note, if the content is {@link org.elasticsearch.common.lease.Releasable} it
* should automatically be released when done by the channel sending it.
*/
public abstract BytesReference content();
/**
* The rest status code.
*/
public abstract RestStatus status();
public void addHeaders(Map> headers) {
if (customHeaders == null) {
customHeaders = new HashMap<>(headers.size());
}
for (Map.Entry> entry : headers.entrySet()) {
List values = customHeaders.get(entry.getKey());
if (values == null) {
values = Lists.newArrayList();
customHeaders.put(entry.getKey(), values);
}
values.addAll(entry.getValue());
}
}
/**
* Add a custom header.
*/
public void addHeader(String name, String value) {
if (customHeaders == null) {
customHeaders = new HashMap<>(2);
}
List header = customHeaders.get(name);
if (header == null) {
header = new ArrayList<>();
customHeaders.put(name, header);
}
header.add(value);
}
/**
* Returns custom headers that have been added, or null if none have been set.
*/
@Nullable
public Map> getHeaders() {
return customHeaders;
}
}