org.webpieces.httpparser.api.dto.Headers Maven / Gradle / Ivy

Go to download
package org.webpieces.httpparser.api.dto;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.webpieces.httpparser.api.common.Header;
import org.webpieces.httpparser.api.common.KnownHeaderName;

public class Headers {

	private Map> headers = new HashMap<>();
	
	public Header getHeader(KnownHeaderName name) {
		return getHeader(name.getHeaderName().toLowerCase());
	}
	public List getHeaders(KnownHeaderName name) {
		List headers2 = getHeaders(name.getHeaderName().toLowerCase());
		if(headers2 == null)
			return new ArrayList<>();
		return headers2;
	}
	public Header getLastInstanceOfHeader(KnownHeaderName name) {
		List headers = getHeaders(name.getHeaderName().toLowerCase());
		if(headers == null)
			return null;
		else if(headers.size() == 0)
			return null;
		return headers.get(headers.size()-1);
	}
	
	public Header getHeader(String key) {
		List list = headers.get(key.toLowerCase());
		if(list == null)
			return null;
		else if(list.size() > 1)
			throw new IllegalStateException("There are two headers in this http request with key="+key+". use getHeaders method instead");
		else if(list.size() == 0)
			return null;
		return list.get(0);
	}

	public List getHeaders(String key) {
		return headers.get(key.toLowerCase());
	}

	/**
	 * It is important that this is not exposed as the duplicate structure could get corrupted.
	 * Adding anything to this from a client does nothing so we don't want to allow this as the
	 * user would think that it would be marshalled out
	 */
	protected void addHeader(Header header) {
		addHeader(header.getName().toLowerCase(), header);
	}
	
	protected void addHeader(String name, Header header) {
		List list = headers.get(name);
		if(list == null) {
			list = new ArrayList<>(10);
			//Header names are not case sensitive while values are
			headers.put(name, list);
		}
		list.add(header);
	}
	
}