org.javaweb.net.HttpResponse Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of javaweb-utils Show documentation
There is a newer version: 2.0.3
/*
 * Copyright yz 2016-01-14  Email:[email protected].
 *
 * Licensed 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.64
 */
package org.javaweb.net;

import org.javaweb.utils.HttpRequestUtils;

import java.net.MalformedURLException;
import java.net.URL;
import java.nio.charset.Charset;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

public class HttpResponse {

	/**
	 * 远程服务器的响应内容
	 */
	protected transient String data;

	private byte[] bodyBytes;

	/**
	 * 请求的URL地址
	 */
	private URL url;

	/**
	 * 发送HTTP请求后从服务器获取的响应状态码
	 */
	private int statusCode;

	/**
	 * 发送HTTP请求后从服务器获取的响应消息内容
	 */
	private String statusMessage;

	/**
	 * 发送HTTP请求后从服务器获取的响应类型
	 */
	private String contentType;

	/**
	 * 网站编码类型
	 */
	private String charset;

	/**
	 * 请求处理后发生的异常信息
	 */
	private Exception exception;

	/**
	 * 请求开始的时间
	 */
	private long requestTime;

	/**
	 * 远程服务器请求结束后响应时间
	 */
	private long responseTime;

	/**
	 * 请求的远程服务器文件的最后修改时间
	 */
	private long lastModified;

	/**
	 * 远程服务器的域名地址
	 */
	private String domain;

	/**
	 * Cookie Map
	 */
	private Map cookies = new LinkedHashMap();

	/**
	 * HTTP原始请求对象
	 */
	private HttpRequest request;

	/**
	 * 远程服务器响应的Header信息
	 */
	private Map> header;

	public HttpResponse(String url) throws MalformedURLException {
		this.url = new URL(url);
		this.domain = this.url.getHost();
	}

	public HttpResponse(URL url) {
		this.url = url;
		this.domain = url.getHost();
	}

	public URL getUrl() {
		return url;
	}

	public void setUrl(URL url) {
		this.url = url;
	}

	public byte[] getBodyBytes() {
		return bodyBytes;
	}

	public void setBodyBytes(byte[] bodyBytes) {
		this.bodyBytes = bodyBytes;
	}

	public int getStatusCode() {
		return statusCode;
	}

	public void setStatusCode(int statusCode) {
		this.statusCode = statusCode;
	}

	public String getStatusMessage() {
		return statusMessage;
	}

	public void setStatusMessage(String statusMessage) {
		this.statusMessage = statusMessage;
	}

	public String getContentType() {
		return contentType;
	}

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

	public String getCharset() {
		return charset;
	}

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

	public Exception getException() {
		return exception;
	}

	public void setException(Exception exception) {
		this.exception = exception;
	}

	public long getRequestTime() {
		return requestTime;
	}

	public void setRequestTime(long requestTime) {
		this.requestTime = requestTime;
	}

	public long getResponseTime() {
		return responseTime;
	}

	public void setResponseTime(long responseTime) {
		this.responseTime = responseTime;
	}

	public long getLastModified() {
		return lastModified;
	}

	public void setLastModified(long lastModified) {
		this.lastModified = lastModified;
	}

	public String getDomain() {
		return domain;
	}

	public void setDomain(String domain) {
		this.domain = domain;
	}

	public Map> getHeader() {
		return header;
	}

	public void setHeader(Map> header) {
		this.header = header;
	}

	protected String parseBody() {
		return parseBody("UTF-8");
	}

	/**
	 * 解析响应内容为字符串
	 *
	 * @param charset
	 * @return
	 */
	protected String parseBody(String charset) {
		if (bodyBytes != null) {
			String htmlCharset = null;

			try {
				if (charset != null && Charset.isSupported(charset)) {
					htmlCharset = charset;
				} else {
					htmlCharset = HttpRequestUtils.parseHTMLCharset(bodyBytes);
				}

				// 编码验证,如果是GB2312可能会出现乱码需转为GBK
				if (Charset.isSupported(htmlCharset)) {
					this.setCharset("GB2312".equalsIgnoreCase(htmlCharset) ? "GBK" : htmlCharset);
				} else {
					this.setCharset("UTF-8");
				}

				this.data = new String(bodyBytes, this.getCharset());
			} catch (Exception e) {
				// 忽略
			}
		}

		return this.data;
	}

	/**
	 * 获取HTTP原始的html编码后的字符串
	 *
	 * @return
	 */
	public String body() {
		return parseBody();
	}

	public String body(String charset) {
		return parseBody(charset);
	}

	/**
	 * 获取服务器设置的Cookie值
	 *
	 * @return
	 */
	public Map getCookies() {
		return cookies;
	}

	/**
	 * 设置元素的Cookie值
	 *
	 * @param cookies
	 */
	public void setCookies(Map cookies) {
		this.cookies = cookies;
	}

	/**
	 * 设置cookie
	 *
	 * @param cookieName
	 * @param cookieVal
	 */
	public void cookie(String cookieName, String cookieVal) {
		cookies.put(cookieName, cookieVal);
	}

	/**
	 * 移除Cookie值
	 *
	 * @param key
	 */
	public void removeCookie(String key) {
		cookies.remove(key);
	}

	/**
	 * 获取原始请求对象
	 *
	 * @return
	 */
	public HttpRequest getRequest() {
		return request;
	}

	/**
	 * 设置原始请求对象
	 *
	 * @param request
	 */
	public void setRequest(HttpRequest request) {
		this.request = request;
	}

}