com.geccocrawler.gecco.response.HttpResponse Maven / Gradle / Ivy

Go to download
package com.geccocrawler.gecco.response;

import java.io.ByteArrayInputStream;
import java.io.InputStreamReader;

import com.google.common.io.CharStreams;

public class HttpResponse {
	private ByteArrayInputStream raw;

	private String content;

	private String contentType;
	
	private String charset;

	private int status;
	
	public static HttpResponse createSimple(String content) {
		HttpResponse response = new HttpResponse();
		response.setContent(content);
		return response;
	}

	public ByteArrayInputStream getRaw() {
		return raw;
	}

	public void setRaw(ByteArrayInputStream raw) {
		this.raw = raw;
	}

	public String getContent() {
		return content;
	}
	
	public String getContent(String charset) {
		if(charset == null) {
			return content;
		}
		try {
			return CharStreams.toString(new InputStreamReader(raw, charset));
		} catch (Exception e) {
			e.printStackTrace();
			return content;
		}
	}

	public void setContent(String content) {
		this.content = content;
	}

	public String getContentType() {
		return contentType;
	}

	public void setContentType(String contentType) {
		this.contentType = contentType;
	}

	public int getStatus() {
		return status;
	}

	public void setStatus(int status) {
		this.status = status;
	}

	public String getCharset() {
		return charset;
	}

	public void setCharset(String charset) {
		this.charset = charset;
	}

	public void close() {
		if(raw != null) {
			try{
				raw.close();
			} catch(Exception ex) {
				raw = null;
			}
		}
	}
}