org.apache.droids.protocol.http.HttpClientContentLoader.java Source code

Introduction

Here is the source code for org.apache.droids.protocol.http.HttpClientContentLoader.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.apache.droids.protocol.http;

import java.io.IOException;
import java.io.InputStream;
import java.net.URI;

import org.apache.droids.norobots.ContentLoader;
import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;
import org.apache.http.HttpStatus;
import org.apache.http.StatusLine;
import org.apache.http.client.HttpClient;
import org.apache.http.client.HttpResponseException;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpHead;

/**
 * {@link ContentLoader} based on HttpClient 4.0.
 */
public class HttpClientContentLoader implements ContentLoader {

    private final HttpClient httpclient;

    public HttpClientContentLoader(HttpClient httpclient) {
        super();
        this.httpclient = httpclient;
    }

    public boolean exists(URI uri) throws IOException {
        HttpHead httphead = new HttpHead(uri);
        HttpResponse response = httpclient.execute(httphead);
        return response.getStatusLine().getStatusCode() == HttpStatus.SC_OK;
    }

    public InputStream load(URI uri) throws IOException {
        HttpGet httpget = new HttpGet(uri);
        HttpResponse response = httpclient.execute(httpget);
        StatusLine statusline = response.getStatusLine();
        if (statusline.getStatusCode() == HttpStatus.SC_NOT_FOUND) {
            return null;
        }
        if (statusline.getStatusCode() != HttpStatus.SC_OK) {
            throw new HttpResponseException(statusline.getStatusCode(), statusline.getReasonPhrase());
        }
        HttpEntity entity = response.getEntity();
        if (entity != null) {
            return entity.getContent();
        } else {
            return null;
        }
    }

}