ch.mimo.netty.handler.codec.icap.AbstractIcapMessage Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of netty-icap Show documentation
ICAP Codec implementation for JBoss Netty
The newest version!
/*******************************************************************************
 * Copyright (c) 2012 Michael Mimo Moratti.
 *
 * Michael Mimo Moratti 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 ch.mimo.netty.handler.codec.icap;

import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;

import org.jboss.netty.handler.codec.http.HttpRequest;
import org.jboss.netty.handler.codec.http.HttpResponse;
import org.jboss.netty.util.internal.StringUtil;

/**
 * This is the main Icap message implementation where 
 * all common @see {@link DefaultIcapRequest} and @see {@link DefaultIcapResponse} member are present.
 * 
 * @author Michael Mimo Moratti ([email protected])
 *
 */
public abstract class AbstractIcapMessage implements IcapMessage {

	private IcapHeaders icapHeader;
	private IcapVersion version;
	private Encapsulated encapsulated;
	
	private HttpRequest httpRequest;
	private HttpResponse httpResponse;
	
	private IcapMessageElementEnum body;
	
	public AbstractIcapMessage(IcapVersion version) {
		this.version = version;
		icapHeader = new IcapHeaders();
	}

	@Override
	public String getHeader(String name) {
		return icapHeader.getHeader(name);
	}

	@Override
	public Set getHeaders(String name) {
		return icapHeader.getHeaders(name);
	}

	@Override
	public Set> getHeaders() {
		return icapHeader.getHeaders();
	}

	@Override
	public boolean containsHeader(String name) {
		return icapHeader.containsHeader(name);
	}

	@Override
	public Set getHeaderNames() {
		return icapHeader.getHeaderNames();
	}

	@Override
	public IcapMessage addHeader(String name, Object value) {
		icapHeader.addHeader(name,value);
		return this;
	}

	@Override
	public IcapMessage setHeader(String name, Object value) {
		icapHeader.setHeader(name,value);
		return this;
	}

	@Override
	public IcapMessage setHeader(String name, Iterable values) {
		icapHeader.setHeader(name,values);
		return this;
	}

	@Override
	public IcapMessage removeHeader(String name) {
		icapHeader.removeHeader(name);
		return this;
	}
	
	@Override
	public int getPreviewAmount() {
		return icapHeader.getPreviewHeaderValue();
	}

	@Override
	public IcapMessage clearHeaders() {
		icapHeader.clearHeaders();
		return this;
	}

	@Override
	public IcapVersion getProtocolVersion() {
		return version;
	}

	@Override
	public IcapMessage setProtocolVersion(IcapVersion version) {
		this.version = version;
		return this;
	}

	@Override
	public boolean containsHttpRequest() {
		return httpRequest != null;
	}

	@Override
	public HttpRequest getHttpRequest() {
		return httpRequest;
	}
	
	@Override
	public IcapMessage setHttpRequest(HttpRequest httpRequest) {
		this.httpRequest = httpRequest;
		return this;
	}

	@Override
	public boolean containsHttpResponse() {
		return httpResponse != null;
	}

	@Override
	public HttpResponse getHttpResponse() {
		return httpResponse;
	}
	
	public IcapMessage setHttpResponse(HttpResponse response) {
		this.httpResponse = response;
		return this;
	}

	@Override
	public IcapMessage setEncapsulatedHeader(Encapsulated encapsulated) {
		this.encapsulated = encapsulated;
		return this;
	}

	@Override
	public Encapsulated getEncapsulatedHeader() {
		return encapsulated;
	}
	
	@Override
	public boolean isPreviewMessage() {
		return icapHeader.getPreviewHeaderValue() > 0;
	}
	
	public IcapMessage setBody(IcapMessageElementEnum body) {
		this.body = body;
		return this;
	}

	public IcapMessageElementEnum getBodyType() {
		if(encapsulated != null) {
			return encapsulated.containsBodyEntry();
		} else {
			return body;
		}
	}
	
    @Override
    public String toString() {
        StringBuilder buf = new StringBuilder();
        buf.append(getClass().getSimpleName());
        buf.append("(version: ");
        buf.append(getProtocolVersion().getText());
        buf.append(')');
        buf.append(StringUtil.NEWLINE);
        appendHeaders(buf);

        if(httpRequest != null) {
        	buf.append("--- encapsulated HTTP Request ---").append(StringUtil.NEWLINE);
        	buf.append(httpRequest.toString());
        	if(httpRequest.getContent() != null && httpRequest.getContent().readableBytes() > 0) {
        		buf.append(StringUtil.NEWLINE).append("--> HTTP Request contains [" + httpRequest.getContent().readableBytes() + "] bytes of data").append(StringUtil.NEWLINE);
        	}
        }
        
        if(httpResponse != null) {
        	buf.append("--- encapsulated HTTP Response ---").append(StringUtil.NEWLINE);
        	buf.append(httpResponse.toString());
        	if(httpResponse.getContent() != null && httpResponse.getContent().readableBytes() > 0) {
        		buf.append(StringUtil.NEWLINE).append("--> HTTP Response contains [" + httpResponse.getContent().readableBytes() + "] bytes of data").append(StringUtil.NEWLINE);;
        	}
        }
        
        if(isPreviewMessage()) {
        	buf.append("--- Preview ---").append(StringUtil.NEWLINE);
        	buf.append("Preview size: " + icapHeader.getPreviewHeaderValue());
        }
        
        return buf.toString();
    }
    
    private void appendHeaders(StringBuilder buf) {
        for (Map.Entry e: getHeaders()) {
            buf.append(e.getKey());
            buf.append(": ");
            buf.append(e.getValue());
            buf.append(StringUtil.NEWLINE);
        }
    }
}