com.liuguangqiang.download.core.DownloadTask.java Source code

Introduction

Here is the source code for com.liuguangqiang.download.core.DownloadTask.java

Source

/*
 * Copyright 2014 Eric Liu
 *
 * 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.
 */

package com.liuguangqiang.download.core;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;

import org.apache.http.HttpResponse;
import org.apache.http.HttpStatus;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpUriRequest;

import android.net.http.AndroidHttpClient;
import android.util.Log;

/**
 * Implemented Runnable for async downloading.
 * <p/>
 * Created by Eric on 2014-8-12
 *
 * @since 1.0
 */
public class DownloadTask implements Runnable {

    private boolean isCanceled = false;

    private AndroidHttpClient mHttpClient;
    private HttpUriRequest httpGet;

    private DownloadListener mListener;
    private DownloadParams mParams;

    public DownloadTask(AndroidHttpClient httpClient, DownloadParams params, DownloadListener listener) {
        this.mHttpClient = httpClient;
        this.mParams = params;
        this.mListener = listener;
        if (this.mListener != null)
            this.mListener.setDownloadParams(params);
    }

    public DownloadParams getDownloadParams() {
        return mParams;
    }

    @Override
    public void run() {
        if (isCanceled())
            return;
        download();
    }

    public boolean isCanceled() {
        return isCanceled;
    }

    public void cancel() {
        isCanceled = true;
        httpGet.abort();
        mListener.sendCancelMessage();
    }

    private boolean download() {
        if (mListener != null)
            mListener.sendStartMessage();

        httpGet = new HttpGet(mParams.getUrl());

        long totalSize;
        int progress = 0;

        InputStream io = null;
        BufferedInputStream bufferIo = null;
        OutputStream out = null;
        BufferedOutputStream bufferOut = null;

        try {
            HttpResponse httpResponse = mHttpClient.execute(httpGet);
            int statusCode = httpResponse.getStatusLine().getStatusCode();
            if (statusCode == HttpStatus.SC_OK) {
                totalSize = httpResponse.getEntity().getContentLength();
                File file = new File(mParams.getSavePath());
                io = httpResponse.getEntity().getContent();
                bufferIo = new BufferedInputStream(io);
                out = new FileOutputStream(file);
                bufferOut = new BufferedOutputStream(out);
                byte[] buf = new byte[1];
                int len;
                int percent;
                int lastPercent = -1;
                while ((len = bufferIo.read(buf)) > 0 && !isCanceled()) {
                    bufferOut.write(buf, 0, len);
                    progress++;
                    percent = (int) (progress * 100 / totalSize);
                    if (percent > lastPercent) {
                        lastPercent = percent;
                        if (mListener != null)
                            mListener.sendUpdateProgressMessage(percent);
                    }
                    if (percent == 100 && mListener != null) {
                        mListener.sendSuccessMessage();
                    }
                }
                return true;
            } else {
                Log.i("statusCode", "failure code :" + statusCode);
                if (mListener != null)
                    mListener.sendFailureMessage("status code:" + statusCode);
            }
            return false;
        } catch (Exception e) {
            if (mListener != null) {
                if (!isCanceled) {
                    mListener.sendFailureMessage(e.toString());
                }
            }
            return false;
        } finally {
            try {
                if (bufferOut != null) {
                    bufferOut.flush();
                    bufferOut.close();
                }
                if (out != null)
                    out.close();
                if (io != null)
                    io.close();
                if (bufferIo != null)
                    bufferIo.close();
            } catch (IOException ex) {
                ex.printStackTrace();
            }
        }
    }

}