org.aliuge.crawler.page.Page.java Source code

Introduction

Here is the source code for org.aliuge.crawler.page.Page.java

Source

/**
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF 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 org.aliuge.crawler.page;

import java.io.Serializable;
import java.nio.charset.Charset;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import org.aliuge.crawler.url.WebURL;
import org.apache.http.Header;
import org.apache.http.HttpEntity;
import org.apache.http.entity.ContentType;
import org.apache.http.util.EntityUtils;

public class Page implements Serializable {

    public Page() {
        super();
    }

    private static final long serialVersionUID = -8815708487494541068L;

    protected Pattern p = Pattern.compile("(?<=charset=)(.+)(?=\")");
    /**
     * The URL of this page.
     */
    protected WebURL url;

    /**
     * The content of this page in binary format.
     */
    protected byte[] contentData;

    /**
     * The ContentType of this page. For example: "text/html; charset=UTF-8"
     */
    protected String contentType;

    /**
     * The encoding of the content. For example: "gzip"
     */
    protected String contentEncoding;

    /**
     * The charset of the content. For example: "UTF-8"
     */
    protected String contentCharset;

    /**
     * Headers which were present in the response of the fetch request
     */
    protected Header[] fetchResponseHeaders;

    /**
     * The parsed data populated by parsers
     */
    protected ParseData parseData;

    public Page(WebURL url) {
        this.url = url;
    }

    public WebURL getWebURL() {
        return url;
    }

    public void setWebURL(WebURL url) {
        this.url = url;
    }

    /**
     * Loads the content of this page from a fetched HttpEntity.
     */
    public void load(HttpEntity entity) throws Exception {

        contentType = null;
        Header type = entity.getContentType();
        if (type != null) {
            contentType = type.getValue();
        }

        contentEncoding = null;
        Header encoding = entity.getContentEncoding();
        if (encoding != null) {
            contentEncoding = encoding.getValue();
        }

        Charset charset = ContentType.getOrDefault(entity).getCharset();
        if (charset != null) {
            contentCharset = charset.displayName();
        }

        contentData = EntityUtils.toByteArray(entity);
        // headcharsetmeta?
        if (null == contentCharset || "" == contentCharset) {
            int length = contentData.length;
            if (length > 2048) {
                length = 2048;
            }
            String html = new String(contentData, 0, 2048);
            contentCharset = matchCharset(html.toLowerCase());
        }
    }

    /**
     * @param content
     * @return
     * @desc ??metacharset
     */
    public String matchCharset(String content) {
        Matcher m = p.matcher(content);
        if (m.find())
            return m.group();
        return "gb2312";
    }

    /**
     * Returns headers which were present in the response of the fetch request
     */
    public Header[] getFetchResponseHeaders() {
        return fetchResponseHeaders;
    }

    public void setFetchResponseHeaders(Header[] headers) {
        fetchResponseHeaders = headers;
    }

    /**
     * Returns the parsed data generated for this page by parsers
     */
    public ParseData getParseData() {
        return parseData;
    }

    public void setParseData(ParseData parseData) {
        this.parseData = parseData;
    }

    /**
     * Returns the content of this page in binary format.
     */
    public byte[] getContentData() {
        return contentData;
    }

    public void setContentData(byte[] contentData) {
        this.contentData = contentData;
    }

    /**
     * Returns the ContentType of this page. For example:
     * "text/html; charset=UTF-8"
     */
    public String getContentType() {
        return contentType;
    }

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

    /**
     * Returns the encoding of the content. For example: "gzip"
     */
    public String getContentEncoding() {
        return contentEncoding;
    }

    public void setContentEncoding(String contentEncoding) {
        this.contentEncoding = contentEncoding;
    }

    /**
     * Returns the charset of the content. For example: "UTF-8"
     */
    public String getContentCharset() {
        return contentCharset;
    }

    public void setContentCharset(String contentCharset) {
        this.contentCharset = contentCharset;
    }

}