Skip to content

Commit

Permalink
Merge pull request #21 from anxinJ/master
Browse files Browse the repository at this point in the history
Master
  • Loading branch information
Mr-ChangK authored Mar 28, 2017
2 parents 0634fd8 + 3c16b0b commit 3971921
Show file tree
Hide file tree
Showing 15 changed files with 856 additions and 14 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,31 @@
package com.coderising.download;

import java.io.IOException;
import java.io.RandomAccessFile;

import com.coderising.download.api.Connection;

public class DownloadThread extends Thread{

Connection conn;
int startPos;
int endPos;

public DownloadThread( Connection conn, int startPos, int endPos){

this.conn = conn;
this.startPos = startPos;
this.endPos = endPos;
}
public void run(){
try {
byte[] bytes = conn.read(startPos, endPos);
RandomAccessFile file = new RandomAccessFile("/Users/jie/Desktop/test.png", "rw");
file.seek(startPos);
file.write(bytes);
file.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,104 @@
package com.coderising.download;

import com.coderising.download.api.Connection;
import com.coderising.download.api.ConnectionException;
import com.coderising.download.api.ConnectionManager;
import com.coderising.download.api.DownloadListener;


public class FileDownloader {

String url;

DownloadListener listener;

ConnectionManager cm;

Thread[] threads = new Thread[3];


public FileDownloader(String _url) {
this.url = _url;

}

public void execute(){
// 在这里实现你的代码, 注意: 需要用多线程实现下载
// 这个类依赖于其他几个接口, 你需要写这几个接口的实现代码
// (1) ConnectionManager , 可以打开一个连接,通过Connection可以读取其中的一段(用startPos, endPos来指定)
// (2) DownloadListener, 由于是多线程下载, 调用这个类的客户端不知道什么时候结束,所以你需要实现当所有
// 线程都执行完以后, 调用listener的notifiedFinished方法, 这样客户端就能收到通知。
// 具体的实现思路:
// 1. 需要调用ConnectionManager的open方法打开连接, 然后通过Connection.getContentLength方法获得文件的长度
// 2. 至少启动3个线程下载, 注意每个线程需要先调用ConnectionManager的open方法
// 然后调用read方法, read方法中有读取文件的开始位置和结束位置的参数, 返回值是byte[]数组
// 3. 把byte数组写入到文件中
// 4. 所有的线程都下载完成以后, 需要调用listener的notifiedFinished方法

// 下面的代码是示例代码, 也就是说只有一个线程, 你需要改造成多线程的。
Connection conn = null;
try {

conn = cm.open(this.url);

int length = conn.getContentLength();

int size = length / 3;
int remain = length - size;

for(int i=0;i<3;i++){
conn = cm.open(this.url);
int startPos = i*size;
int endPos = (i+1) * size - 1;
if(i == 2){
endPos = (i+1) * size - 1 + remain;
}
threads[i] = new DownloadThread(conn,startPos,endPos);
threads[i].start();
}

new Thread(new Runnable(){
public void run(){
boolean finish;
do{
finish = true;
for(int i=0;i<threads.length;i++){
// System.out.println("线程" + i + "状态:" + threads[i].getState());
if(!"TERMINATED".equals(threads[i].getState().toString())){
finish = false;
continue;
}
}
}while(!finish);
getListener().notifyFinished();
}
}).start();;

} catch (ConnectionException e) {
e.printStackTrace();
}finally{
if(conn != null){
conn.close();
}
}




}

public void setListener(DownloadListener listener) {
this.listener = listener;
}



public void setConnectionManager(ConnectionManager ucm){
this.cm = ucm;
}

public DownloadListener getListener(){
return this.listener;
}

}
Original file line number Diff line number Diff line change
@@ -0,0 +1,59 @@
package com.coderising.download;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import com.coderising.download.api.ConnectionManager;
import com.coderising.download.api.DownloadListener;
import com.coderising.download.impl.ConnectionManagerImpl;

public class FileDownloaderTest {
boolean downloadFinished = false;
@Before
public void setUp() throws Exception {
}

@After
public void tearDown() throws Exception {
}

@Test
public void testDownload() {

String url = "http://pic6.huitu.com/res/20130116/84481_20130116142820494200_1.jpg";

FileDownloader downloader = new FileDownloader(url);


ConnectionManager cm = new ConnectionManagerImpl();
downloader.setConnectionManager(cm);

downloader.setListener(new DownloadListener() {
@Override
public void notifyFinished() {
downloadFinished = true;
}

});


downloader.execute();

// 等待多线程下载程序执行完毕
while (!downloadFinished) {
try {
System.out.println("还没有下载完成,休眠五秒");
//休眠5秒
Thread.sleep(5000);
} catch (InterruptedException e) {
e.printStackTrace();
}
}
System.out.println("下载完成!");



}

}
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
package com.coderising.download.api;

import java.io.IOException;
import java.net.HttpURLConnection;

public interface Connection {
/**
* 给定开始和结束位置, 读取数据, 返回值是字节数组
* @param startPos 开始位置, 从0开始
* @param endPos 结束位置
* @return
*/
public byte[] read(int startPos,int endPos) throws IOException;
/**
* 得到数据内容的长度
* @return
*/
public int getContentLength();

/**
* 关闭连接
*/
public void close();

public void setConnection(HttpURLConnection conn);
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
package com.coderising.download.api;

public class ConnectionException extends Exception {

}
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
package com.coderising.download.api;

public interface ConnectionManager {
/**
* 给定一个url , 打开一个连接
* @param url
* @return
*/
public Connection open(String url) throws ConnectionException;
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
package com.coderising.download.api;

public interface DownloadListener {
public void notifyFinished();
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,62 @@
package com.coderising.download.impl;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.HttpURLConnection;

import com.coderising.download.api.Connection;

public class ConnectionImpl implements Connection{

private HttpURLConnection conn;

private InputStream inputStream;

private ByteArrayOutputStream outputStream;
@Override
public byte[] read(int startPos, int endPos) throws IOException {
byte[] bytes = new byte[1024];
conn.setRequestProperty("Range", "bytes=" + startPos + "-" + endPos);
inputStream = conn.getInputStream();
outputStream = new ByteArrayOutputStream();
int n = 0;
while((n = inputStream.read(bytes)) != -1){
outputStream.write(bytes, 0, n);
}
return outputStream.toByteArray();
}

@Override
public int getContentLength() {
return conn.getContentLength();
}

@Override
public void close() {
conn.disconnect();
if(inputStream != null){
try {
inputStream.close();
} catch (IOException e) {
e.printStackTrace();
}
}
if(outputStream != null){
try {
outputStream.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}

public HttpURLConnection getConnection() {
return conn;
}

public void setConnection(HttpURLConnection conn) {
this.conn = conn;
}

}
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
package com.coderising.download.impl;

import java.io.IOException;
import java.net.HttpURLConnection;
import java.net.URL;

import com.coderising.download.api.Connection;
import com.coderising.download.api.ConnectionException;
import com.coderising.download.api.ConnectionManager;

public class ConnectionManagerImpl implements ConnectionManager {

@Override
public Connection open(String url) throws ConnectionException {
Connection connection = new ConnectionImpl();
try {
URL u = new URL(url);
HttpURLConnection conn = (HttpURLConnection)u.openConnection();
connection.setConnection(conn);
} catch (IOException e) {
e.printStackTrace();
}
return connection;
}

}
Loading

0 comments on commit 3971921

Please sign in to comment.