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

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

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

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

public abstract class HttpMessage extends HttpPayload {

	protected List headers = new ArrayList<>();
	//Convenience structure that further morphs the headers into a Map that can
	//be looked up by key.
	private transient Headers headersStruct = new Headers();
	
	/**
	 * Order of HTTP Headers matters for Headers with the same key
	 */
	public List getHeaders() {
		return Collections.unmodifiableList(headers);
	}

	public void addHeader(Header header) {
		headers.add(header);
		headersStruct.addHeader(header);
	}
	
	/** 
	 * 
	 * @return
	 */
	public Headers getHeaderLookupStruct() {
		return headersStruct;
	}
	
    public boolean isHasChunkedTransferHeader() {
		//need to account for a few Transfer Encoding headers
		Header header = headersStruct.getLastInstanceOfHeader(KnownHeaderName.TRANSFER_ENCODING);
        return header != null && "chunked".equals(header.getValue());
    }
    
    public boolean isHasNonZeroContentLength() {
    	Integer contentLength = getContentLength();
    	if(contentLength == null)
    		return false;
    	else if(contentLength.intValue() == 0)
    		return false;
    	return true;
    }
    
    public Integer getContentLength() {
		Header header = headersStruct.getLastInstanceOfHeader(KnownHeaderName.CONTENT_LENGTH);
		if(header == null)
			return null;
		String value = header.getValue();
		if(value == null)
			return null;
		
		int len = Integer.parseInt(value);
		return len;
    }
}