android 多线程断点续传下载 二

文章出处
  在上一集中,我们简单介绍了如何创建多任务下载,但那种还不能拿来实用,这一集我们重点通过代码为大家展示如何创建多线程断点续传下载,这在实际项目中很常用.

main.xml:

  
  
      
      
        

String.xml

  
  
    Hello World, Main!  
    多线程断点续传下载  
  

AndroidManifest.xml:

  
  
      
          
              
                  
                  
              
          
       
      
      
      
      
      
      
      
   

activity程序:

package sms.multithreaddownload;

import java.io.File;

import sms.multithreaddownload.bean.DownloadListener;
import sms.multithreaddownload.service.DownloadService;
import android.app.Activity;
import android.os.Bundle;
import android.os.Environment;
import android.os.Handler;
import android.os.Message;
import android.view.View;
import android.view.View.OnClickListener;
import android.widget.Button;
import android.widget.EditText;
import android.widget.ProgressBar;
import android.widget.TextView;
import android.widget.Toast;

public class Main extends Activity {
    private EditText path;
    private TextView progress;
    private ProgressBar progressBar;
    private Handler handler = new UIHandler();
    private DownloadService servcie;
    private Button downButton;
    private Button pauseButton;

    private final class UIHandler extends Handler {
        @Override
        public void handleMessage(Message msg) {
            switch (msg.what) {
                case 1:
                    int downloaded_size = msg.getData().getInt("size");
                    progressBar.setProgress(downloaded_size);
                    int result = (int) ((float) downloaded_size / progressBar.getMax() * 100);
                    progress.setText(result + "%");
                    if (progressBar.getMax() == progressBar.getProgress()) {
                        Toast.makeText(getApplicationContext(), "下载完成", Toast.LENGTH_LONG).show();
                    }
            }
        }
    }

    @Override
    public void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.main);
        path = (EditText) this.findViewById(R.id.editText);
        progress = (TextView) this.findViewById(R.id.textView);
        progressBar = (ProgressBar) this.findViewById(R.id.progressBar);
        downButton = (Button) this.findViewById(R.id.downButton);
        pauseButton = (Button) this.findViewById(R.id.pauseButton);
        downButton.setOnClickListener(new DownloadButton());
        pauseButton.setOnClickListener(new PauseButton());
    }

    private final class DownloadButton implements View.OnClickListener {
        @Override
        public void onClick(View v) {
            DownloadTask task;
            try {
                task = new DownloadTask(path.getText().toString());
                servcie.isPause = false;
                v.setEnabled(false);
                pauseButton.setEnabled(true);
                new Thread(task).start();
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }

    public class PauseButton implements OnClickListener {
        @Override
        public void onClick(View v) {
            servcie.isPause = true;
            v.setEnabled(false);
            downButton.setEnabled(true);
        }
    }

    public void pause(View v) {
    }

    private final class DownloadTask implements Runnable {

        public DownloadTask(String target) throws Exception {
            if (Environment.getExternalStorageState().equals(Environment.MEDIA_MOUNTED)) {
                File destination = Environment.getExternalStorageDirectory();
                servcie = new DownloadService(target, destination, 3, getApplicationContext());
                progressBar.setMax(servcie.fileSize);
            } else {
                Toast.makeText(getApplicationContext(), "SD卡不存在或写保护!", Toast.LENGTH_LONG).show();
            }
        }

        @Override
        public void run() {
            try {
                servcie.download(new DownloadListener() {

                    @Override
                    public void onDownload(int downloaded_size) {
                        Message message = new Message();
                        message.what = 1;
                        message.getData().putInt("size", downloaded_size);
                        handler.sendMessage(message);
                    }

                });
            } catch (Exception e) {
                e.printStackTrace();
            }

        }
    }
}

工具类:

package sms.multithreaddownload.bean;

import android.content.Context;
import android.database.sqlite.SQLiteDatabase;
import android.database.sqlite.SQLiteOpenHelper;

public class DBHelper extends SQLiteOpenHelper {

    public DBHelper(Context context) {
        super(context, "MultiDownLoad.db", null, 1);
    }

    @Override
    public void onCreate(SQLiteDatabase db) {
        db.execSQL("CREATE TABLE fileDownloading(_id integer primary key autoincrement,downPath varchar(100),threadId INTEGER,downLength INTEGER)");
    }

    @Override
    public void onUpgrade(SQLiteDatabase db, int oldVersion, int newVersion) {
        // TODO Auto-generated method stub

    }

}
package sms.multithreaddownload.bean;

public interface DownloadListener {
    public void onDownload(int downloaded_size);
}
package sms.multithreaddownload.bean;  
  
import java.io.File;  
import java.io.InputStream;  
import java.io.RandomAccessFile;  
import java.net.HttpURLConnection;  
import java.net.URL;  
  
import sms.multithreaddownload.service.DownloadService;  
  
import android.util.Log;  
  
public final class MultiThreadDownload implements Runnable {  
    public int id;  
    private RandomAccessFile savedFile;  
    private String path;  
    /* 当前已下载量 */  
    public int currentDownloadSize = 0;  
    /* 下载状态 */  
    public boolean finished;  
    /* 用于监视下载状态 */  
    private final DownloadService downloadService;  
    /* 线程下载任务的起始点 */  
    public int start;  
    /* 线程下载任务的结束点 */  
    private int end;  
  
    public MultiThreadDownload(int id, File savedFile, int block, String path, Integer downlength, DownloadService downloadService) throws Exception {  
        this.id = id;  
        this.path = path;  
        if (downlength != null) this.currentDownloadSize = downlength;  
        this.savedFile = new RandomAccessFile(savedFile, "rwd");  
        this.downloadService = downloadService;  
        start = id * block + currentDownloadSize;  
        end = (id + 1) * block;  
    }  
  
    @Override  
    public void run() {  
        try {  
            HttpURLConnection conn = (HttpURLConnection) new URL(path).openConnection();  
            conn.setConnectTimeout(5000);  
            conn.setRequestMethod("GET");  
            conn.setRequestProperty("Range", "bytes=" + start + "-" + end); // 设置获取数据的范围  
  
            InputStream in = conn.getInputStream();  
            byte[] buffer = new byte[1024];  
            int len = 0;  
            savedFile.seek(start);  
            while (!downloadService.isPause && (len = in.read(buffer)) != -1) {  
                savedFile.write(buffer, 0, len);  
                currentDownloadSize += len;  
            }  
            savedFile.close();  
            in.close();  
            conn.disconnect();  
            if (!downloadService.isPause) Log.i(DownloadService.TAG, "Thread " + (this.id + 1) + "finished");  
            finished = true;  
        } catch (Exception e) {  
            e.printStackTrace();  
            throw new RuntimeException("File downloading error!");  
        }  
    }  
}  

service类:

package sms.multithreaddownload.service;  
  
import java.io.File;  
import java.io.RandomAccessFile;  
import java.net.HttpURLConnection;  
import java.net.URL;  
import java.util.HashMap;  
import java.util.List;  
import java.util.Map;  
import java.util.Map.Entry;  
import java.util.UUID;  
import java.util.concurrent.ConcurrentHashMap;  
import java.util.regex.Matcher;  
import java.util.regex.Pattern;  
  
import sms.multithreaddownload.bean.DBHelper;  
import sms.multithreaddownload.bean.DownloadListener;  
import sms.multithreaddownload.bean.MultiThreadDownload;  
  
import android.content.Context;  
import android.database.Cursor;  
import android.database.sqlite.SQLiteDatabase;  
  
  
public class DownloadService {  
    public static final String TAG = "tag";  
    /* 用于查询数据库 */  
    private DBHelper dbHelper;  
    /* 要下载的文件大小 */  
    public int fileSize;  
    /* 每条线程需要下载的数据量 */  
    private int block;  
    /* 保存文件地目录 */  
    private File savedFile;  
    /* 下载地址 */  
    private String path;  
    /* 是否停止下载 */  
    public boolean isPause;  
    /* 线程数 */  
    private MultiThreadDownload[] threads;  
    /* 各线程已经下载的数据量 */  
    private Map downloadedLength = new ConcurrentHashMap();  
  
    public DownloadService(String target, File destination, int thread_size, Context context) throws Exception {  
        dbHelper = new DBHelper(context);  
        this.threads = new MultiThreadDownload[thread_size];  
        this.path = target;  
        URL url = new URL(target);  
        HttpURLConnection conn = (HttpURLConnection) url.openConnection();  
        conn.setConnectTimeout(5000);  
        conn.setRequestMethod("GET");  
  
        if (conn.getResponseCode() != 200) {  
            throw new RuntimeException("server no response!");  
        }  
  
        fileSize = conn.getContentLength();  
        if (fileSize <= 0) {  
            throw new RuntimeException("file is incorrect!");  
        }  
        String fileName = getFileName(conn);  
        if (!destination.exists()) destination.mkdirs();  
        // 构建一个同样大小的文件  
        this.savedFile = new File(destination, fileName);  
        RandomAccessFile doOut = new RandomAccessFile(savedFile, "rwd");  
        doOut.setLength(fileSize);  
        doOut.close();  
        conn.disconnect();  
  
        // 计算每条线程需要下载的数据长度  
        this.block = fileSize % thread_size == 0 ? fileSize / thread_size : fileSize / thread_size + 1;  
        // 查询已经下载的记录  
        downloadedLength = this.getDownloadedLength(path);  
    }  
  
    private Map getDownloadedLength(String path) {  
        SQLiteDatabase db = dbHelper.getReadableDatabase();  
        String sql = "SELECT threadId,downLength FROM fileDownloading WHERE downPath=?";  
        Cursor cursor = db.rawQuery(sql, new String[] { path });  
        Map data = new HashMap();  
        while (cursor.moveToNext()) {  
            data.put(cursor.getInt(0), cursor.getInt(1));  
        }  
        db.close();  
        return data;  
    }  
  
    private String getFileName(HttpURLConnection conn) {  
        String fileName = path.substring(path.lastIndexOf("/") + 1, path.length());  
        if (fileName == null || "".equals(fileName.trim())) {  
            String content_disposition = null;  
            for (Entry> entry : conn.getHeaderFields().entrySet()) {  
                if ("content-disposition".equalsIgnoreCase(entry.getKey())) {  
                    content_disposition = entry.getValue().toString();  
                }  
            }  
            try {  
                Matcher matcher = Pattern.compile(".*filename=(.*)").matcher(content_disposition);  
                if (matcher.find()) fileName = matcher.group(1);  
            } catch (Exception e) {  
                fileName = UUID.randomUUID().toString() + ".tmp"; // 默认名  
            }  
        }  
        return fileName;  
    }  
  
    public void download(DownloadListener listener) throws Exception {  
        this.deleteDownloading(); // 先删除上次的记录,再重新添加  
        for (int i = 0; i < threads.length; i++) {  
            threads[i] = new MultiThreadDownload(i, savedFile, block, path, downloadedLength.get(i), this);  
            new Thread(threads[i]).start();  
        }  
        this.saveDownloading(threads);  
  
        while (!isFinish(threads)) {  
            Thread.sleep(900);  
            if (listener != null) listener.onDownload(getDownloadedSize(threads));  
            this.updateDownloading(threads);  
        }  
        if (!this.isPause) this.deleteDownloading();// 完成下载之后删除本次下载记录  
    }  
  
    private void saveDownloading(MultiThreadDownload[] threads) {  
        SQLiteDatabase db = dbHelper.getWritableDatabase();  
        try {  
            db.beginTransaction();  
            for (MultiThreadDownload thread : threads) {  
                String sql = "INSERT INTO fileDownloading(downPath,threadId,downLength) values(?,?,?)";  
                db.execSQL(sql, new Object[] { path, thread.id, 0 });  
            }  
            db.setTransactionSuccessful();  
        } finally {  
            db.endTransaction();  
            db.close();  
        }  
    }  
  
    private void deleteDownloading() {  
        SQLiteDatabase db = dbHelper.getWritableDatabase();  
        String sql = "DELETE FROM fileDownloading WHERE downPath=?";  
        db.execSQL(sql, new Object[] { path });  
        db.close();  
    }  
  
    private void updateDownloading(MultiThreadDownload[] threads) {  
        SQLiteDatabase db = dbHelper.getWritableDatabase();  
        try {  
            db.beginTransaction();  
            for (MultiThreadDownload thread : threads) {  
                String sql = "UPDATE fileDownloading SET downLength=? WHERE threadId=? AND downPath=?";  
                db.execSQL(sql, new String[] { thread.currentDownloadSize + "", thread.id + "", path });  
            }  
            db.setTransactionSuccessful();  
        } finally {  
            db.endTransaction();  
            db.close();  
        }  
    }  
  
    private int getDownloadedSize(MultiThreadDownload[] threads) {  
        int sum = 0;  
        for (int len = threads.length, i = 0; i < len; i++) {  
            sum += threads[i].currentDownloadSize;  
        }  
        return sum;  
    }  
  
    private boolean isFinish(MultiThreadDownload[] threads) {  
        try {  
            for (int len = threads.length, i = 0; i < len; i++) {  
                if (!threads[i].finished) {  
                    return false;  
                }  
            }  
            return true;  
        } catch (Exception e) {  
            return false;  
        }  
    }  
}  

运行效果:

android 多线程断点续传下载 二_第1张图片

源码下载地址

你可能感兴趣的:(android 多线程断点续传下载 二)