laoyuyu
5 years ago
parent
7ee1347ab4
commit
2a92321785
@ -0,0 +1,44 @@ |
||||
/* |
||||
* 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 com.arialyy.aria.core.command; |
||||
|
||||
import com.arialyy.aria.core.download.AbsGroupTaskWrapper; |
||||
import com.arialyy.aria.core.task.ITask; |
||||
import com.arialyy.aria.core.wrapper.AbsTaskWrapper; |
||||
|
||||
public class CmdHelper { |
||||
/** |
||||
* 创建任务命令 |
||||
* |
||||
* @param taskType {@link ITask#DOWNLOAD}、{@link ITask#DOWNLOAD_GROUP}、{@link ITask#UPLOAD} |
||||
*/ |
||||
public static <T extends AbsTaskWrapper> AbsNormalCmd createNormalCmd(T entity, int cmd, |
||||
int taskType) { |
||||
return NormalCmdFactory.getInstance().createCmd(entity, cmd, taskType); |
||||
} |
||||
|
||||
/** |
||||
* 创建任务组命令 |
||||
* |
||||
* @param childUrl 子任务url |
||||
*/ |
||||
public static <T extends AbsGroupTaskWrapper> AbsGroupCmd createGroupCmd(T entity, int cmd, |
||||
String childUrl) { |
||||
return GroupCmdFactory.getInstance().createCmd(entity, cmd, childUrl); |
||||
} |
||||
} |
||||
@ -1,457 +0,0 @@ |
||||
/* |
||||
* Copyright (C) 2016 AriaLyy(https://github.com/AriaLyy/Aria)
|
||||
* |
||||
* 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.arialyy.aria.core.common; |
||||
|
||||
import com.arialyy.aria.core.config.Configuration; |
||||
import com.arialyy.aria.core.download.DTaskWrapper; |
||||
import com.arialyy.aria.core.download.DownloadEntity; |
||||
import com.arialyy.aria.core.download.m3u8.BaseM3U8Loader; |
||||
import com.arialyy.aria.core.inf.AbsNormalEntity; |
||||
import com.arialyy.aria.core.inf.AbsTaskWrapper; |
||||
import com.arialyy.aria.core.inf.ITaskWrapper; |
||||
import com.arialyy.aria.core.upload.UploadEntity; |
||||
import com.arialyy.aria.orm.DbEntity; |
||||
import com.arialyy.aria.util.ALog; |
||||
import com.arialyy.aria.util.CommonUtil; |
||||
import com.arialyy.aria.util.DbDataHelper; |
||||
import com.arialyy.aria.util.FileUtil; |
||||
import com.arialyy.aria.util.RecordUtil; |
||||
import java.io.File; |
||||
import java.util.ArrayList; |
||||
import java.util.HashSet; |
||||
import java.util.List; |
||||
import java.util.Properties; |
||||
import java.util.Set; |
||||
|
||||
/** |
||||
* 处理任务记录，分配线程区间 |
||||
*/ |
||||
public class RecordHandler { |
||||
private final String TAG = "RecordHandler"; |
||||
|
||||
public static final int TYPE_DOWNLOAD = 1; |
||||
public static final int TYPE_UPLOAD = 2; |
||||
public static final int TYPE_M3U8_VOD = 3; |
||||
public static final int TYPE_M3U8_LIVE = 4; |
||||
|
||||
private static final String STATE = "_state_"; |
||||
private static final String RECORD = "_record_"; |
||||
/** |
||||
* 小于1m的文件不启用多线程 |
||||
*/ |
||||
private static final long SUB_LEN = 1024 * 1024; |
||||
|
||||
/** |
||||
* 分块文件路径 |
||||
*/ |
||||
public static final String SUB_PATH = "%s.%s.part"; |
||||
|
||||
@Deprecated private File mConfigFile; |
||||
private TaskRecord mTaskRecord; |
||||
private AbsTaskWrapper mTaskWrapper; |
||||
private AbsNormalEntity mEntity; |
||||
|
||||
RecordHandler(AbsTaskWrapper wrapper) { |
||||
mTaskWrapper = wrapper; |
||||
mEntity = (AbsNormalEntity) mTaskWrapper.getEntity(); |
||||
} |
||||
|
||||
/** |
||||
* 获取任务记录，如果任务记录存在，检查任务记录 |
||||
* 检查记录 对于分块任务： 子分块不存在或被删除，子线程将重新下载 |
||||
* 对于普通任务： 预下载文件不存在，则任务任务呗删除 |
||||
* 如果任务记录不存在或线程记录不存在，初始化记录 |
||||
* |
||||
* @return 任务记录 |
||||
*/ |
||||
TaskRecord getRecord() { |
||||
mConfigFile = new File(CommonUtil.getFileConfigPath(false, mEntity.getFileName())); |
||||
if (mConfigFile.exists()) { |
||||
convertDb(); |
||||
} else { |
||||
mTaskRecord = DbDataHelper.getTaskRecord(getFilePath()); |
||||
if (mTaskRecord == null) { |
||||
initRecord(true); |
||||
} else { |
||||
if (mTaskRecord.threadRecords == null || mTaskRecord.threadRecords.isEmpty()) { |
||||
initRecord(false); |
||||
} |
||||
|
||||
if (mTaskWrapper.getRequestType() == ITaskWrapper.M3U8_VOD) { |
||||
handleM3U8Record(); |
||||
} else if (mTaskWrapper.getRequestType() == ITaskWrapper.M3U8_LIVE) { |
||||
ALog.i(TAG, "直播下载不处理历史记录"); |
||||
} else { |
||||
if (mTaskRecord.isBlock) { |
||||
handleBlockRecord(); |
||||
} else if (!mTaskWrapper.isSupportBP()) { |
||||
handleNoSupportBPRecord(); |
||||
} else if (!mTaskRecord.isBlock && mTaskRecord.threadNum > 1) { |
||||
handleNoBlockMultiThreadRecord(); |
||||
} else { |
||||
handleSingleThreadRecord(); |
||||
} |
||||
} |
||||
} |
||||
} |
||||
saveRecord(); |
||||
return mTaskRecord; |
||||
} |
||||
|
||||
/** |
||||
* 处理m3u8记录， |
||||
* 1、如果分片文件存在，并且分片文件的记录没有完成，则需要删除该分片文件 |
||||
* 2、如果记录显示已完成，但是分片文件不存在，则重新开始该分片 |
||||
* 3、如果记录显示已完成，并且文件存在，记录当前任务进度 |
||||
*/ |
||||
private void handleM3U8Record() { |
||||
DTaskWrapper wrapper = (DTaskWrapper) mTaskWrapper; |
||||
String cacheDir = wrapper.asM3U8().getCacheDir(); |
||||
long currentProgress = 0; |
||||
int completeNum = 0; |
||||
for (ThreadRecord record : mTaskRecord.threadRecords) { |
||||
File temp = new File(BaseM3U8Loader.getTsFilePath(cacheDir, record.threadId)); |
||||
if (!record.isComplete) { |
||||
if (temp.exists()) { |
||||
temp.delete(); |
||||
} |
||||
record.startLocation = 0; |
||||
//ALog.d(TAG, String.format("分片【%s】未完成，将重新下载该分片", record.threadId));
|
||||
} else { |
||||
if (!temp.exists()) { |
||||
record.startLocation = 0; |
||||
record.isComplete = false; |
||||
ALog.w(TAG, String.format("分片【%s】不存在，将重新下载该分片", record.threadId)); |
||||
} else { |
||||
completeNum++; |
||||
currentProgress += temp.length(); |
||||
} |
||||
} |
||||
} |
||||
wrapper.asM3U8().setCompleteNum(completeNum); |
||||
wrapper.getEntity().setCurrentProgress(currentProgress); |
||||
mTaskRecord.bandWidth = wrapper.asM3U8().getBandWidth(); |
||||
} |
||||
|
||||
/** |
||||
* 处理不支持断点的记录 |
||||
*/ |
||||
private void handleNoSupportBPRecord() { |
||||
ThreadRecord tr = mTaskRecord.threadRecords.get(0); |
||||
tr.startLocation = 0; |
||||
tr.endLocation = mEntity.getFileSize(); |
||||
tr.taskKey = mTaskRecord.filePath; |
||||
tr.blockLen = tr.endLocation; |
||||
tr.isComplete = false; |
||||
} |
||||
|
||||
/** |
||||
* 处理为不分块的多线程任务 |
||||
*/ |
||||
private void handleNoBlockMultiThreadRecord() { |
||||
File file = new File(mTaskRecord.filePath); |
||||
if (!file.exists()) { |
||||
ALog.w(TAG, String.format("文件【%s】不存在，重新分配线程区间", mTaskRecord.filePath)); |
||||
DbEntity.deleteData(ThreadRecord.class, "taskKey=?", mTaskRecord.filePath); |
||||
initRecord(false); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* 处理单线程的任务的记录 |
||||
*/ |
||||
private void handleSingleThreadRecord() { |
||||
File file = new File(mTaskRecord.filePath); |
||||
ThreadRecord tr = mTaskRecord.threadRecords.get(0); |
||||
if (!file.exists()) { |
||||
ALog.w(TAG, String.format("文件【%s】不存在，任务将重新开始", file.getPath())); |
||||
tr.startLocation = 0; |
||||
tr.isComplete = false; |
||||
tr.endLocation = mEntity.getFileSize(); |
||||
} else if (mTaskRecord.isOpenDynamicFile) { |
||||
if (file.length() > mEntity.getFileSize()) { |
||||
ALog.i(TAG, String.format("文件【%s】错误，任务重新开始", file.getPath())); |
||||
file.delete(); |
||||
tr.startLocation = 0; |
||||
tr.isComplete = false; |
||||
tr.endLocation = mEntity.getFileSize(); |
||||
} else if (file.length() == mEntity.getFileSize()) { |
||||
tr.isComplete = true; |
||||
} else { |
||||
if (file.length() != tr.startLocation) { |
||||
ALog.i(TAG, String.format("修正【%s】的进度记录为：%s", file.getPath(), file.length())); |
||||
tr.startLocation = file.length(); |
||||
tr.isComplete = false; |
||||
} |
||||
} |
||||
} |
||||
mTaskWrapper.setNewTask(false); |
||||
} |
||||
|
||||
/** |
||||
* 处理分块任务的记录，分块文件（blockFileLen）长度必须需要小于等于线程区间（threadRectLen）的长度 |
||||
*/ |
||||
private void handleBlockRecord() { |
||||
// 默认线程分块长度
|
||||
long normalRectLen = mEntity.getFileSize() / mTaskRecord.threadRecords.size(); |
||||
for (ThreadRecord tr : mTaskRecord.threadRecords) { |
||||
long threadRect = tr.blockLen; |
||||
|
||||
File temp = new File(String.format(SUB_PATH, mTaskRecord.filePath, tr.threadId)); |
||||
if (!temp.exists()) { |
||||
ALog.i(TAG, String.format("分块文件【%s】不存在，该分块将重新开始", temp.getPath())); |
||||
tr.isComplete = false; |
||||
tr.startLocation = tr.threadId * normalRectLen; |
||||
} else { |
||||
if (!tr.isComplete) { |
||||
ALog.i(TAG, String.format( |
||||
"startLocation = %s; endLocation = %s; block = %s; tempLen = %s; threadId = %s", |
||||
tr.startLocation, tr.endLocation, threadRect, temp.length(), tr.threadId)); |
||||
|
||||
long blockFileLen = temp.length(); // 磁盘中的分块文件长度
|
||||
/* |
||||
* 检查磁盘中的分块文件 |
||||
*/ |
||||
if (blockFileLen > threadRect) { |
||||
ALog.i(TAG, String.format("分块【%s】错误，分块长度【%s】 > 线程区间长度【%s】，将重新开始该分块", |
||||
tr.threadId, blockFileLen, threadRect)); |
||||
temp.delete(); |
||||
tr.startLocation = tr.threadId * threadRect; |
||||
continue; |
||||
} |
||||
|
||||
long realLocation = |
||||
tr.threadId * normalRectLen + blockFileLen; //正常情况下，该线程的startLocation的位置
|
||||
/* |
||||
* 检查记录文件 |
||||
*/ |
||||
if (blockFileLen == threadRect) { |
||||
ALog.i(TAG, String.format("分块【%s】已完成，更新记录", temp.getPath())); |
||||
tr.startLocation = blockFileLen; |
||||
tr.isComplete = true; |
||||
} else if (tr.startLocation != realLocation) { // 处理记录小于分块文件长度的情况
|
||||
ALog.i(TAG, String.format("修正分块【%s】的进度记录为：%s", temp.getPath(), realLocation)); |
||||
tr.startLocation = realLocation; |
||||
} |
||||
} else { |
||||
ALog.i(TAG, String.format("分块【%s】已完成", temp.getPath())); |
||||
} |
||||
} |
||||
} |
||||
mTaskWrapper.setNewTask(false); |
||||
} |
||||
|
||||
/** |
||||
* convertDb 是兼容性代码 从3.4.1开始，线程配置信息将存储在数据库中。 将配置文件的内容复制到数据库中，并将配置文件删除 |
||||
*/ |
||||
private void convertDb() { |
||||
List<RecordWrapper> records = |
||||
DbEntity.findRelationData(RecordWrapper.class, "TaskRecord.filePath=?", |
||||
getFilePath()); |
||||
if (records == null || records.size() == 0) { |
||||
Properties pro = FileUtil.loadConfig(mConfigFile); |
||||
if (pro.isEmpty()) { |
||||
ALog.d(TAG, "老版本的线程记录为空，任务为新任务"); |
||||
initRecord(true); |
||||
return; |
||||
} |
||||
|
||||
Set<Object> keys = pro.keySet(); |
||||
// 老版本记录是5s存一次，但是5s中内，如果线程执行完成，record记录是没有的，只有state记录...
|
||||
// 第一步应该是record 和 state去重取正确的线程数
|
||||
Set<Integer> set = new HashSet<>(); |
||||
for (Object key : keys) { |
||||
String str = String.valueOf(key); |
||||
int i = Integer.parseInt(str.substring(str.length() - 1)); |
||||
set.add(i); |
||||
} |
||||
int threadNum = set.size(); |
||||
if (threadNum == 0) { |
||||
ALog.d(TAG, "线程数为空，任务为新任务"); |
||||
initRecord(true); |
||||
return; |
||||
} |
||||
mTaskWrapper.setNewTask(false); |
||||
mTaskRecord = createTaskRecord(threadNum); |
||||
mTaskRecord.isOpenDynamicFile = false; |
||||
mTaskRecord.isBlock = false; |
||||
File tempFile = new File(getFilePath()); |
||||
for (int i = 0; i < threadNum; i++) { |
||||
ThreadRecord tRecord = new ThreadRecord(); |
||||
tRecord.taskKey = mTaskRecord.filePath; |
||||
Object state = pro.getProperty(tempFile.getName() + STATE + i); |
||||
Object record = pro.getProperty(tempFile.getName() + RECORD + i); |
||||
if (state != null && Integer.parseInt(String.valueOf(state)) == 1) { |
||||
tRecord.isComplete = true; |
||||
continue; |
||||
} |
||||
if (record != null) { |
||||
long temp = Long.parseLong(String.valueOf(record)); |
||||
tRecord.startLocation = temp > 0 ? temp : 0; |
||||
} else { |
||||
tRecord.startLocation = 0; |
||||
} |
||||
mTaskRecord.threadRecords.add(tRecord); |
||||
} |
||||
mConfigFile.delete(); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* 初始化任务记录，分配线程区间 |
||||
* |
||||
* @param newRecord {@code true} 需要创建新{@link TaskRecord} |
||||
*/ |
||||
private void initRecord(boolean newRecord) { |
||||
if (newRecord) { |
||||
mTaskRecord = createTaskRecord(getNewTaskThreadNum()); |
||||
} |
||||
mTaskWrapper.setNewTask(true); |
||||
int requestType = mTaskWrapper.getRequestType(); |
||||
if (requestType == ITaskWrapper.M3U8_LIVE) { |
||||
return; |
||||
} |
||||
long blockSize = mEntity.getFileSize() / mTaskRecord.threadNum; |
||||
// 处理线程区间记录
|
||||
for (int i = 0; i < mTaskRecord.threadNum; i++) { |
||||
long startL = i * blockSize, endL = (i + 1) * blockSize; |
||||
ThreadRecord tr; |
||||
tr = new ThreadRecord(); |
||||
tr.taskKey = mTaskRecord.filePath; |
||||
tr.threadId = i; |
||||
tr.startLocation = startL; |
||||
tr.isComplete = false; |
||||
if (requestType == ITaskWrapper.M3U8_VOD) { |
||||
tr.startLocation = 0; |
||||
tr.threadType = TaskRecord.TYPE_M3U8_VOD; |
||||
tr.tsUrl = ((DTaskWrapper) mTaskWrapper).asM3U8().getUrls().get(i); |
||||
} else { |
||||
tr.threadType = TaskRecord.TYPE_HTTP_FTP; |
||||
//最后一个线程的结束位置即为文件的总长度
|
||||
if (i == (mTaskRecord.threadNum - 1)) { |
||||
endL = mEntity.getFileSize(); |
||||
} |
||||
tr.endLocation = endL; |
||||
tr.blockLen = RecordUtil.getBlockLen(mEntity.getFileSize(), i, mTaskRecord.threadNum); |
||||
} |
||||
mTaskRecord.threadRecords.add(tr); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* 创建任务记录 |
||||
* |
||||
* @param threadNum 线程总数 |
||||
*/ |
||||
private TaskRecord createTaskRecord(int threadNum) { |
||||
TaskRecord record = new TaskRecord(); |
||||
record.fileName = mEntity.getFileName(); |
||||
record.filePath = getFilePath(); |
||||
record.threadRecords = new ArrayList<>(); |
||||
record.threadNum = threadNum; |
||||
int requestType = mTaskWrapper.getRequestType(); |
||||
if (requestType == ITaskWrapper.M3U8_VOD) { |
||||
record.taskType = TaskRecord.TYPE_M3U8_VOD; |
||||
record.isOpenDynamicFile = true; |
||||
record.bandWidth = ((DTaskWrapper)mTaskWrapper).asM3U8().getBandWidth(); |
||||
} else if (requestType == ITaskWrapper.M3U8_LIVE) { |
||||
record.taskType = TaskRecord.TYPE_M3U8_LIVE; |
||||
record.isOpenDynamicFile = true; |
||||
record.bandWidth = ((DTaskWrapper)mTaskWrapper).asM3U8().getBandWidth(); |
||||
} else { |
||||
if (getRecordType() == TYPE_DOWNLOAD) { |
||||
record.isBlock = threadNum > 1 && Configuration.getInstance().downloadCfg.isUseBlock(); |
||||
// 线程数为1，或者使用了分块，则认为是使用动态长度文件
|
||||
record.isOpenDynamicFile = threadNum == 1 || record.isBlock; |
||||
} else { |
||||
record.isBlock = false; |
||||
} |
||||
record.taskType = TaskRecord.TYPE_HTTP_FTP; |
||||
record.isGroupRecord = mEntity.isGroupChild(); |
||||
if (record.isGroupRecord) { |
||||
if (mEntity instanceof DownloadEntity) { |
||||
record.dGroupHash = ((DownloadEntity) mEntity).getGroupHash(); |
||||
} |
||||
} |
||||
} |
||||
|
||||
return record; |
||||
} |
||||
|
||||
/** |
||||
* 保存任务记录 |
||||
*/ |
||||
private void saveRecord() { |
||||
mTaskRecord.threadNum = mTaskRecord.threadRecords.size(); |
||||
mTaskRecord.save(); |
||||
if (mTaskRecord.threadRecords != null && !mTaskRecord.threadRecords.isEmpty()) { |
||||
DbEntity.saveAll(mTaskRecord.threadRecords); |
||||
} |
||||
ALog.d(TAG, String.format("保存记录，线程记录数：%s", mTaskRecord.threadRecords.size())); |
||||
} |
||||
|
||||
/** |
||||
* 获取记录类型 |
||||
* |
||||
* @return {@link #TYPE_DOWNLOAD}、{@link #TYPE_UPLOAD} |
||||
*/ |
||||
private int getRecordType() { |
||||
if (mEntity instanceof DownloadEntity) { |
||||
return TYPE_DOWNLOAD; |
||||
} else { |
||||
return TYPE_UPLOAD; |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* 获取任务路径 |
||||
* |
||||
* @return 任务文件路径 |
||||
*/ |
||||
private String getFilePath() { |
||||
if (mEntity instanceof DownloadEntity) { |
||||
return ((DownloadEntity) mTaskWrapper.getEntity()).getFilePath(); |
||||
} else { |
||||
return ((UploadEntity) mTaskWrapper.getEntity()).getFilePath(); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* 小于1m的文件或是任务组的子任务、线程数强制为1 |
||||
* 不支持断点或chunked模式的线程数都为，线程数强制为1 |
||||
*/ |
||||
private int getNewTaskThreadNum() { |
||||
if (getRecordType() == TYPE_DOWNLOAD) { |
||||
if (mTaskWrapper.getRequestType() == ITaskWrapper.M3U8_VOD) { |
||||
return ((DTaskWrapper) mTaskWrapper).asM3U8().getUrls().size(); |
||||
} |
||||
if (mTaskWrapper.getRequestType() == ITaskWrapper.M3U8_LIVE) { |
||||
return 1; |
||||
} |
||||
if (!mTaskWrapper.isSupportBP() || mTaskWrapper.asHttp().isChunked()) { |
||||
return 1; |
||||
} |
||||
int threadNum = Configuration.getInstance().downloadCfg.getThreadNum(); |
||||
return mEntity.getFileSize() <= SUB_LEN |
||||
|| mEntity.isGroupChild() |
||||
|| threadNum == 1 |
||||
? 1 |
||||
: threadNum; |
||||
} else { |
||||
return 1; |
||||
} |
||||
} |
||||
} |
||||
@ -1,129 +0,0 @@ |
||||
/* |
||||
* Copyright (C) 2016 AriaLyy(https://github.com/AriaLyy/Aria)
|
||||
* |
||||
* 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.arialyy.aria.core.download.downloader; |
||||
|
||||
import com.arialyy.aria.core.common.AbsThreadTask; |
||||
import com.arialyy.aria.core.common.NormalFileer; |
||||
import com.arialyy.aria.core.common.RecordHandler; |
||||
import com.arialyy.aria.core.common.SubThreadConfig; |
||||
import com.arialyy.aria.core.download.DTaskWrapper; |
||||
import com.arialyy.aria.core.download.DownloadEntity; |
||||
import com.arialyy.aria.core.event.Event; |
||||
import com.arialyy.aria.core.event.SpeedEvent; |
||||
import com.arialyy.aria.core.inf.IDownloadListener; |
||||
import com.arialyy.aria.core.inf.ITaskWrapper; |
||||
import com.arialyy.aria.exception.BaseException; |
||||
import com.arialyy.aria.exception.TaskException; |
||||
import com.arialyy.aria.util.ALog; |
||||
import com.arialyy.aria.util.BufferedRandomAccessFile; |
||||
import java.io.File; |
||||
import java.io.IOException; |
||||
|
||||
/** |
||||
* Created by AriaL on 2017/7/1. 文件下载器 |
||||
*/ |
||||
public class Downloader extends NormalFileer<DownloadEntity, DTaskWrapper> { |
||||
private String TAG = "Downloader"; |
||||
|
||||
public Downloader(IDownloadListener listener, DTaskWrapper taskWrapper) { |
||||
super(listener, taskWrapper); |
||||
mTempFile = new File(mEntity.getFilePath()); |
||||
setUpdateInterval(taskWrapper.getConfig().getUpdateInterval()); |
||||
} |
||||
|
||||
@Override protected boolean handleNewTask() { |
||||
if (!mRecord.isBlock) { |
||||
if (mTempFile.exists()) { |
||||
mTempFile.delete(); |
||||
} |
||||
//CommonUtil.createFile(mTempFile.getPath());
|
||||
} else { |
||||
for (int i = 0; i < mTotalThreadNum; i++) { |
||||
File blockFile = new File(String.format(RecordHandler.SUB_PATH, mTempFile.getPath(), i)); |
||||
if (blockFile.exists()) { |
||||
ALog.d(TAG, String.format("分块【%s】已经存在，将删除该分块", i)); |
||||
blockFile.delete(); |
||||
} |
||||
} |
||||
} |
||||
BufferedRandomAccessFile file = null; |
||||
try { |
||||
if (mTotalThreadNum > 1 && !mRecord.isBlock) { |
||||
file = new BufferedRandomAccessFile(new File(mTempFile.getPath()), "rwd", 8192); |
||||
//设置文件长度
|
||||
file.setLength(mEntity.getFileSize()); |
||||
} |
||||
return true; |
||||
} catch (IOException e) { |
||||
failDownload(new TaskException(TAG, |
||||
String.format("下载失败，filePath: %s, url: %s", mEntity.getDownloadPath(), |
||||
mEntity.getUrl()), e)); |
||||
} finally { |
||||
if (file != null) { |
||||
try { |
||||
file.close(); |
||||
} catch (IOException e) { |
||||
e.printStackTrace(); |
||||
} |
||||
} |
||||
} |
||||
return false; |
||||
} |
||||
|
||||
/** |
||||
* 如果使用"Content-Disposition"中的文件名，需要更新{@link #mTempFile}的路径 |
||||
*/ |
||||
void updateTempFile() { |
||||
if (!mTempFile.getPath().equals(mEntity.getFilePath())) { |
||||
if (!mTempFile.exists()) { |
||||
mTempFile = new File(mEntity.getFilePath()); |
||||
} else { |
||||
boolean b = mTempFile.renameTo(new File(mEntity.getDownloadPath())); |
||||
ALog.d(TAG, String.format("更新tempFile文件名%s", b ? "成功" : "失败")); |
||||
} |
||||
} |
||||
} |
||||
|
||||
@Event |
||||
public void setMaxSpeed(SpeedEvent event) { |
||||
setMaxSpeed(event.speed); |
||||
} |
||||
|
||||
@Override protected void onPostPre() { |
||||
super.onPostPre(); |
||||
((IDownloadListener) mListener).onPostPre(mEntity.getFileSize()); |
||||
File file = new File(mEntity.getDownloadPath()); |
||||
if (!file.getParentFile().exists()) { |
||||
file.getParentFile().mkdirs(); |
||||
} |
||||
} |
||||
|
||||
@Override protected AbsThreadTask selectThreadTask(SubThreadConfig<DTaskWrapper> config) { |
||||
switch (mTaskWrapper.getRequestType()) { |
||||
case ITaskWrapper.D_FTP: |
||||
case ITaskWrapper.D_FTP_DIR: |
||||
return new FtpThreadTask(config); |
||||
case ITaskWrapper.D_HTTP: |
||||
return new HttpThreadTask(config); |
||||
} |
||||
return null; |
||||
} |
||||
|
||||
private void failDownload(BaseException e) { |
||||
closeTimer(); |
||||
mListener.onFail(false, e); |
||||
} |
||||
} |
||||
@ -1,140 +0,0 @@ |
||||
/* |
||||
* Copyright (C) 2016 AriaLyy(https://github.com/AriaLyy/Aria)
|
||||
* |
||||
* 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.arialyy.aria.core.download.downloader; |
||||
|
||||
import com.arialyy.aria.core.common.CompleteInfo; |
||||
import com.arialyy.aria.core.common.IUtil; |
||||
import com.arialyy.aria.core.common.OnFileInfoCallback; |
||||
import com.arialyy.aria.core.download.DTaskWrapper; |
||||
import com.arialyy.aria.core.inf.AbsEntity; |
||||
import com.arialyy.aria.core.inf.IDownloadListener; |
||||
import com.arialyy.aria.core.inf.ITaskWrapper; |
||||
import com.arialyy.aria.exception.BaseException; |
||||
|
||||
/** |
||||
* Created by lyy on 2015/8/25. |
||||
* D_HTTP\FTP单任务下载工具 |
||||
*/ |
||||
public class SimpleDownloadUtil implements IUtil { |
||||
private String TAG = "SimpleDownloadUtil"; |
||||
private IDownloadListener mListener; |
||||
private Downloader mDownloader; |
||||
private DTaskWrapper mTaskWrapper; |
||||
private boolean isStop = false, isCancel = false; |
||||
|
||||
public SimpleDownloadUtil(DTaskWrapper wrapper, IDownloadListener downloadListener) { |
||||
mTaskWrapper = wrapper; |
||||
mListener = downloadListener; |
||||
mDownloader = new Downloader(downloadListener, wrapper); |
||||
} |
||||
|
||||
@Override public String getKey() { |
||||
return mTaskWrapper.getKey(); |
||||
} |
||||
|
||||
@Override public long getFileSize() { |
||||
return mDownloader.getFileSize(); |
||||
} |
||||
|
||||
/** |
||||
* 获取当前下载位置 |
||||
*/ |
||||
@Override public long getCurrentLocation() { |
||||
return mDownloader.getCurrentLocation(); |
||||
} |
||||
|
||||
@Override public boolean isRunning() { |
||||
return mDownloader.isRunning(); |
||||
} |
||||
|
||||
/** |
||||
* 取消下载 |
||||
*/ |
||||
@Override public void cancel() { |
||||
isCancel = true; |
||||
mDownloader.cancel(); |
||||
} |
||||
|
||||
/** |
||||
* 停止下载 |
||||
*/ |
||||
@Override public void stop() { |
||||
isStop = true; |
||||
mDownloader.stop(); |
||||
} |
||||
|
||||
/** |
||||
* 多线程断点续传下载文件，开始下载 |
||||
*/ |
||||
@Override public void start() { |
||||
if (isStop || isCancel) { |
||||
return; |
||||
} |
||||
mListener.onPre(); |
||||
// 如果网址没有变，而服务器端端文件改变，以下代码就没有用了
|
||||
//if (mTaskWrapper.getEntity().getFileSize() <= 1
|
||||
// || mTaskWrapper.isRefreshInfo()
|
||||
// || mTaskWrapper.getRequestType() == AbsTaskWrapper.D_FTP
|
||||
// || mTaskWrapper.getState() == IEntity.STATE_FAIL) {
|
||||
// new Thread(createInfoThread()).create();
|
||||
//} else {
|
||||
// mDownloader.create();
|
||||
//}
|
||||
new Thread(createInfoThread()).start(); |
||||
} |
||||
|
||||
private void failDownload(BaseException e, boolean needRetry) { |
||||
if (isStop || isCancel) { |
||||
return; |
||||
} |
||||
mListener.onFail(needRetry, e); |
||||
mDownloader.onDestroy(); |
||||
} |
||||
|
||||
/** |
||||
* 通过链接类型创建不同的获取文件信息的线程 |
||||
*/ |
||||
private Runnable createInfoThread() { |
||||
switch (mTaskWrapper.getRequestType()) { |
||||
case ITaskWrapper.D_FTP: |
||||
return new FtpFileInfoThread(mTaskWrapper, new OnFileInfoCallback() { |
||||
@Override public void onComplete(String url, CompleteInfo info) { |
||||
mDownloader.updateTempFile(); |
||||
mDownloader.start(); |
||||
} |
||||
|
||||
@Override public void onFail(AbsEntity entity, BaseException e, boolean needRetry) { |
||||
failDownload(e, needRetry); |
||||
mDownloader.closeTimer(); |
||||
} |
||||
}); |
||||
case ITaskWrapper.D_HTTP: |
||||
return new HttpFileInfoThread(mTaskWrapper, new OnFileInfoCallback() { |
||||
@Override public void onComplete(String url, CompleteInfo info) { |
||||
mDownloader.updateTempFile(); |
||||
mDownloader.start(); |
||||
} |
||||
|
||||
@Override public void onFail(AbsEntity entity, BaseException e, boolean needRetry) { |
||||
failDownload(e, needRetry); |
||||
mDownloader.closeTimer(); |
||||
} |
||||
}); |
||||
} |
||||
return null; |
||||
} |
||||
} |
||||
@ -1,133 +0,0 @@ |
||||
/* |
||||
* Copyright (C) 2016 AriaLyy(https://github.com/AriaLyy/Aria)
|
||||
* |
||||
* 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.arialyy.aria.core.download.group; |
||||
|
||||
import android.os.Handler; |
||||
import com.arialyy.aria.core.common.CompleteInfo; |
||||
import com.arialyy.aria.core.common.IUtil; |
||||
import com.arialyy.aria.core.common.OnFileInfoCallback; |
||||
import com.arialyy.aria.core.download.DTaskWrapper; |
||||
import com.arialyy.aria.core.download.DownloadEntity; |
||||
import com.arialyy.aria.core.download.downloader.Downloader; |
||||
import com.arialyy.aria.core.download.downloader.HttpFileInfoThread; |
||||
import com.arialyy.aria.core.inf.AbsEntity; |
||||
import com.arialyy.aria.core.inf.ITaskWrapper; |
||||
import com.arialyy.aria.core.scheduler.ISchedulers; |
||||
import com.arialyy.aria.exception.BaseException; |
||||
import com.arialyy.aria.util.ALog; |
||||
|
||||
/** |
||||
* 子任务下载器，负责创建{@link Downloader} |
||||
*/ |
||||
class SubDLoadUtil implements IUtil { |
||||
private final String TAG = "SubDownloadLoader"; |
||||
|
||||
private Downloader mDownloader; |
||||
private DTaskWrapper mWrapper; |
||||
private Handler mSchedulers; |
||||
private ChildDLoadListener mListener; |
||||
private boolean needGetInfo; |
||||
|
||||
/** |
||||
* @param schedulers 调度器 |
||||
* @param needGetInfo {@code true} 需要获取文件信息。{@code false} 不需要获取文件信息 |
||||
*/ |
||||
SubDLoadUtil(Handler schedulers, DTaskWrapper taskWrapper, boolean needGetInfo) { |
||||
mWrapper = taskWrapper; |
||||
mSchedulers = schedulers; |
||||
this.needGetInfo = needGetInfo; |
||||
mListener = new ChildDLoadListener(mSchedulers, SubDLoadUtil.this); |
||||
} |
||||
|
||||
@Override public String getKey() { |
||||
return mWrapper.getKey(); |
||||
} |
||||
|
||||
public DTaskWrapper getWrapper() { |
||||
return mWrapper; |
||||
} |
||||
|
||||
public DownloadEntity getEntity() { |
||||
return mWrapper.getEntity(); |
||||
} |
||||
|
||||
/** |
||||
* 重新开始任务 |
||||
*/ |
||||
void reStart() { |
||||
if (mDownloader != null) { |
||||
mDownloader.retryTask(); |
||||
} |
||||
} |
||||
|
||||
public Downloader getDownloader() { |
||||
return mDownloader; |
||||
} |
||||
|
||||
@Override public long getFileSize() { |
||||
return mDownloader == null ? -1 : mDownloader.getFileSize(); |
||||
} |
||||
|
||||
@Override public long getCurrentLocation() { |
||||
return mDownloader == null ? -1 : mDownloader.getCurrentLocation(); |
||||
} |
||||
|
||||
@Override public boolean isRunning() { |
||||
return mDownloader != null && mDownloader.isRunning(); |
||||
} |
||||
|
||||
@Override public void cancel() { |
||||
if (mDownloader != null && isRunning()) { |
||||
mDownloader.cancel(); |
||||
} else { |
||||
mSchedulers.obtainMessage(ISchedulers.CANCEL, this).sendToTarget(); |
||||
} |
||||
} |
||||
|
||||
@Override public void stop() { |
||||
if (mDownloader != null && isRunning()) { |
||||
mDownloader.stop(); |
||||
} else { |
||||
mSchedulers.obtainMessage(ISchedulers.STOP, this).sendToTarget(); |
||||
} |
||||
} |
||||
|
||||
@Override public void start() { |
||||
if (mWrapper.getRequestType() == ITaskWrapper.D_HTTP) { |
||||
if (needGetInfo) { |
||||
new Thread(new HttpFileInfoThread(mWrapper, new OnFileInfoCallback() { |
||||
|
||||
@Override public void onComplete(String url, CompleteInfo info) { |
||||
mDownloader = new Downloader(mListener, mWrapper); |
||||
mDownloader.start(); |
||||
} |
||||
|
||||
@Override public void onFail(AbsEntity entity, BaseException e, boolean needRetry) { |
||||
mSchedulers.obtainMessage(ISchedulers.FAIL, SubDLoadUtil.this).sendToTarget(); |
||||
} |
||||
})).start(); |
||||
} else { |
||||
mDownloader = new Downloader(mListener, mWrapper); |
||||
mDownloader.start(); |
||||
} |
||||
} else if (mWrapper.getRequestType() == ITaskWrapper.D_FTP) { |
||||
mDownloader = new Downloader(mListener, mWrapper); |
||||
mDownloader.start(); |
||||
} else { |
||||
ALog.w(TAG, String.format("不识别的类型，requestType：%s", mWrapper.getRequestType())); |
||||
} |
||||
} |
||||
} |
||||
@ -1,145 +0,0 @@ |
||||
/* |
||||
* Copyright (C) 2016 AriaLyy(https://github.com/AriaLyy/Aria)
|
||||
* |
||||
* 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.arialyy.aria.core.download.m3u8; |
||||
|
||||
import android.text.TextUtils; |
||||
import com.arialyy.aria.core.common.CompleteInfo; |
||||
import com.arialyy.aria.core.common.IUtil; |
||||
import com.arialyy.aria.core.common.OnFileInfoCallback; |
||||
import com.arialyy.aria.core.download.DTaskWrapper; |
||||
import com.arialyy.aria.core.download.M3U8Listener; |
||||
import com.arialyy.aria.core.inf.AbsEntity; |
||||
import com.arialyy.aria.exception.BaseException; |
||||
import com.arialyy.aria.exception.M3U8Exception; |
||||
import java.io.File; |
||||
import java.util.ArrayList; |
||||
import java.util.Collection; |
||||
import java.util.List; |
||||
|
||||
/** |
||||
* M3U8点播文件下载工具 |
||||
* 工作流程： |
||||
* 1、创建一个和文件同父路径并且同名隐藏文件夹 |
||||
* 2、将所有m3u8的ts文件下载到该文件夹中 |
||||
* 3、完成所有分片下载后，合并ts文件 |
||||
* 4、删除该隐藏文件夹 |
||||
*/ |
||||
public class M3U8VodUtil implements IUtil { |
||||
private final String TAG = "M3U8DownloadUtil"; |
||||
|
||||
private DTaskWrapper mWrapper; |
||||
private M3U8Listener mListener; |
||||
private boolean isStop = false, isCancel = false; |
||||
private List<String> mUrls = new ArrayList<>(); |
||||
private M3U8VodLoader mLoader; |
||||
|
||||
public M3U8VodUtil(DTaskWrapper wrapper, M3U8Listener listener) { |
||||
mWrapper = wrapper; |
||||
mListener = listener; |
||||
mLoader = new M3U8VodLoader(mListener, mWrapper); |
||||
} |
||||
|
||||
@Override public String getKey() { |
||||
return mWrapper.getKey(); |
||||
} |
||||
|
||||
@Override public long getFileSize() { |
||||
return 0; |
||||
} |
||||
|
||||
@Override public long getCurrentLocation() { |
||||
return 0; |
||||
} |
||||
|
||||
@Override public boolean isRunning() { |
||||
return mLoader.isRunning(); |
||||
} |
||||
|
||||
@Override public void cancel() { |
||||
isCancel = true; |
||||
mLoader.cancel(); |
||||
} |
||||
|
||||
@Override public void stop() { |
||||
isStop = true; |
||||
mLoader.stop(); |
||||
} |
||||
|
||||
@Override public void start() { |
||||
if (isStop || isCancel) { |
||||
return; |
||||
} |
||||
mListener.onPre(); |
||||
// peer数量小于0，
|
||||
M3U8Entity m3U8Entity = mWrapper.getEntity().getM3U8Entity(); |
||||
if (m3U8Entity.getPeerNum() <= 0 || (m3U8Entity.isGenerateIndexFile() && !new File( |
||||
String.format(M3U8InfoThread.M3U8_INDEX_FORMAT, |
||||
mWrapper.getEntity().getFilePath())).exists())) { |
||||
getVodInfo(); |
||||
} else { |
||||
mLoader.start(); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* 获取点播文件信息 |
||||
*/ |
||||
private void getVodInfo() { |
||||
M3U8InfoThread thread = new M3U8InfoThread(mWrapper, new OnFileInfoCallback() { |
||||
@Override public void onComplete(String key, CompleteInfo info) { |
||||
IVodTsUrlConverter converter = mWrapper.asM3U8().getVodUrlConverter(); |
||||
if (converter != null) { |
||||
if (TextUtils.isEmpty(mWrapper.asM3U8().getBandWidthUrl())) { |
||||
mUrls.addAll(converter.convert(mWrapper.getEntity().getUrl(), (List<String>) info.obj)); |
||||
} else { |
||||
mUrls.addAll( |
||||
converter.convert(mWrapper.asM3U8().getBandWidthUrl(), (List<String>) info.obj)); |
||||
} |
||||
} else { |
||||
mUrls.addAll((Collection<? extends String>) info.obj); |
||||
} |
||||
if (mUrls.isEmpty()) { |
||||
failDownload(new M3U8Exception(TAG, "获取地址失败"), false); |
||||
return; |
||||
} else if (!mUrls.get(0).startsWith("http")) { |
||||
failDownload(new M3U8Exception(TAG, "地址错误，请使用IM3U8UrlExtInfHandler处理你的url信息"), false); |
||||
return; |
||||
} |
||||
mWrapper.asM3U8().setUrls(mUrls); |
||||
if (isStop) { |
||||
mListener.onStop(mWrapper.getEntity().getCurrentProgress()); |
||||
} else if (isCancel) { |
||||
mListener.onCancel(); |
||||
} else { |
||||
mLoader.start(); |
||||
} |
||||
} |
||||
|
||||
@Override public void onFail(AbsEntity entity, BaseException e, boolean needRetry) { |
||||
failDownload(e, needRetry); |
||||
} |
||||
}); |
||||
new Thread(thread).start(); |
||||
} |
||||
|
||||
private void failDownload(BaseException e, boolean needRetry) { |
||||
if (isStop || isCancel) { |
||||
return; |
||||
} |
||||
mListener.onFail(needRetry, e); |
||||
mLoader.onDestroy(); |
||||
} |
||||
} |
||||
@ -1,36 +0,0 @@ |
||||
/* |
||||
* Copyright (C) 2016 AriaLyy(https://github.com/AriaLyy/Aria)
|
||||
* |
||||
* 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.arialyy.aria.core.inf; |
||||
|
||||
/** |
||||
* Created by lyy on 2017/6/3. |
||||
*/ |
||||
public abstract class AbsNormalTask<TASK_WRAPPER extends AbsTaskWrapper> |
||||
extends AbsTask<TASK_WRAPPER> { |
||||
|
||||
/** |
||||
* 最高优先级命令，最高优先级命令有以下属性 |
||||
* 1、在下载队列中，有且只有一个最高优先级任务 |
||||
* 2、最高优先级任务会一直存在，直到用户手动暂停或任务完成 |
||||
* 3、任务调度器不会暂停最高优先级任务 |
||||
* 4、用户手动暂停或任务完成后，第二次重新执行该任务，该命令将失效 |
||||
* 5、如果下载队列中已经满了，则会停止队尾的任务 |
||||
* 6、把任务设置为最高优先级任务后，将自动执行任务，不需要重新调用start()启动任务 |
||||
*/ |
||||
public void setHighestPriority(boolean isHighestPriority) { |
||||
isHeighestTask = isHighestPriority; |
||||
} |
||||
} |
||||
@ -1,116 +0,0 @@ |
||||
/* |
||||
* Copyright (C) 2016 AriaLyy(https://github.com/AriaLyy/Aria)
|
||||
* |
||||
* 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.arialyy.aria.core.upload.uploader; |
||||
|
||||
import com.arialyy.aria.core.common.CompleteInfo; |
||||
import com.arialyy.aria.core.common.IUtil; |
||||
import com.arialyy.aria.core.common.OnFileInfoCallback; |
||||
import com.arialyy.aria.core.inf.AbsEntity; |
||||
import com.arialyy.aria.core.inf.AbsTaskWrapper; |
||||
import com.arialyy.aria.core.inf.IUploadListener; |
||||
import com.arialyy.aria.core.upload.UTaskWrapper; |
||||
import com.arialyy.aria.exception.BaseException; |
||||
import com.arialyy.aria.util.CheckUtil; |
||||
|
||||
/** |
||||
* Created by lyy on 2017/2/9. |
||||
* 简单的文件上传工具 |
||||
*/ |
||||
public class SimpleUploadUtil implements IUtil, Runnable { |
||||
private static final String TAG = "SimpleUploadUtil"; |
||||
|
||||
private UTaskWrapper mTaskWrapper; |
||||
private IUploadListener mListener; |
||||
private Uploader mUploader; |
||||
private boolean isStop = false, isCancel = false; |
||||
|
||||
public SimpleUploadUtil(UTaskWrapper taskWrapper, IUploadListener listener) { |
||||
mTaskWrapper = taskWrapper; |
||||
CheckUtil.checkTaskEntity(taskWrapper); |
||||
if (listener == null) { |
||||
throw new IllegalArgumentException("上传监听不能为空"); |
||||
} |
||||
mListener = listener; |
||||
mUploader = new Uploader(mListener, taskWrapper); |
||||
} |
||||
|
||||
@Override public void run() { |
||||
mListener.onPre(); |
||||
switch (mTaskWrapper.getRequestType()) { |
||||
case AbsTaskWrapper.U_FTP: |
||||
FtpFileInfoThread infoThread = |
||||
new FtpFileInfoThread(mTaskWrapper, new OnFileInfoCallback() { |
||||
@Override public void onComplete(String url, CompleteInfo info) { |
||||
if (info.code == FtpFileInfoThread.CODE_COMPLETE) { |
||||
mListener.onComplete(); |
||||
} else { |
||||
mUploader.start(); |
||||
} |
||||
} |
||||
|
||||
@Override public void onFail(AbsEntity entity, BaseException e, boolean needRetry) { |
||||
failUpload(e, needRetry); |
||||
} |
||||
}); |
||||
new Thread(infoThread).start(); |
||||
break; |
||||
case AbsTaskWrapper.U_HTTP: |
||||
mUploader.start(); |
||||
break; |
||||
} |
||||
} |
||||
|
||||
private void failUpload(BaseException e, boolean needRetry) { |
||||
if (isStop || isCancel) { |
||||
return; |
||||
} |
||||
mListener.onFail(needRetry, e); |
||||
mUploader.onDestroy(); |
||||
} |
||||
|
||||
@Override public String getKey() { |
||||
return mTaskWrapper.getKey(); |
||||
} |
||||
|
||||
@Override public long getFileSize() { |
||||
return mUploader.getFileSize(); |
||||
} |
||||
|
||||
@Override public long getCurrentLocation() { |
||||
return mUploader.getCurrentLocation(); |
||||
} |
||||
|
||||
@Override public boolean isRunning() { |
||||
return mUploader.isRunning(); |
||||
} |
||||
|
||||
@Override public void cancel() { |
||||
isCancel = true; |
||||
mUploader.cancel(); |
||||
} |
||||
|
||||
@Override public void stop() { |
||||
isStop = true; |
||||
mUploader.stop(); |
||||
} |
||||
|
||||
@Override public void start() { |
||||
if (isStop || isCancel) { |
||||
return; |
||||
} |
||||
new Thread(this).start(); |
||||
} |
||||
} |
||||
@ -1,61 +0,0 @@ |
||||
/* |
||||
* Copyright (C) 2016 AriaLyy(https://github.com/AriaLyy/Aria)
|
||||
* |
||||
* 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.arialyy.aria.core.upload.uploader; |
||||
|
||||
import com.arialyy.aria.core.AriaManager; |
||||
import com.arialyy.aria.core.common.AbsThreadTask; |
||||
import com.arialyy.aria.core.common.NormalFileer; |
||||
import com.arialyy.aria.core.common.SubThreadConfig; |
||||
import com.arialyy.aria.core.event.Event; |
||||
import com.arialyy.aria.core.event.SpeedEvent; |
||||
import com.arialyy.aria.core.inf.AbsTaskWrapper; |
||||
import com.arialyy.aria.core.inf.IUploadListener; |
||||
import com.arialyy.aria.core.upload.UTaskWrapper; |
||||
import com.arialyy.aria.core.upload.UploadEntity; |
||||
import java.io.File; |
||||
|
||||
/** |
||||
* Created by Aria.Lao on 2017/7/27. |
||||
* 文件上传器 |
||||
*/ |
||||
class Uploader extends NormalFileer<UploadEntity, UTaskWrapper> { |
||||
|
||||
Uploader(IUploadListener listener, UTaskWrapper taskEntity) { |
||||
super(listener, taskEntity); |
||||
mTempFile = new File(mEntity.getFilePath()); |
||||
setUpdateInterval( |
||||
AriaManager.getInstance().getUploadConfig().getUpdateInterval()); |
||||
} |
||||
|
||||
@Override protected boolean handleNewTask() { |
||||
return true; |
||||
} |
||||
|
||||
@Override protected AbsThreadTask selectThreadTask(SubThreadConfig<UTaskWrapper> config) { |
||||
switch (mTaskWrapper.getRequestType()) { |
||||
case AbsTaskWrapper.U_FTP: |
||||
return new FtpThreadTask(config); |
||||
case AbsTaskWrapper.U_HTTP: |
||||
return new HttpThreadTask(config); |
||||
} |
||||
return null; |
||||
} |
||||
|
||||
@Event |
||||
public void setMaxSpeed(SpeedEvent event) { |
||||
setMaxSpeed(event.speed); |
||||
} |
||||
} |
||||
@ -1,26 +0,0 @@ |
||||
package com.example.ariaftpcomponent; |
||||
|
||||
import android.content.Context; |
||||
import androidx.test.platform.app.InstrumentationRegistry; |
||||
import androidx.test.ext.junit.runners.AndroidJUnit4; |
||||
|
||||
import org.junit.Test; |
||||
import org.junit.runner.RunWith; |
||||
|
||||
import static org.junit.Assert.*; |
||||
|
||||
/** |
||||
* Instrumented test, which will execute on an Android device. |
||||
* |
||||
* @see <a href="http://d.android.com/tools/testing">Testing documentation</a> |
||||
*/ |
||||
@RunWith(AndroidJUnit4.class) |
||||
public class ExampleInstrumentedTest { |
||||
@Test |
||||
public void useAppContext() { |
||||
// Context of the app under test.
|
||||
Context appContext = InstrumentationRegistry.getInstrumentation().getTargetContext(); |
||||
|
||||
assertEquals("com.example.ariaftpcomponent.test", appContext.getPackageName()); |
||||
} |
||||
} |
||||
@ -1,4 +0,0 @@ |
||||
package com.arialyy.aria.ftpcomponent; |
||||
|
||||
public class bb { |
||||
} |
||||
@ -1,17 +0,0 @@ |
||||
package com.example.ariaftpcomponent; |
||||
|
||||
import org.junit.Test; |
||||
|
||||
import static org.junit.Assert.*; |
||||
|
||||
/** |
||||
* Example local unit test, which will execute on the development machine (host). |
||||
* |
||||
* @see <a href="http://d.android.com/tools/testing">Testing documentation</a> |
||||
*/ |
||||
public class ExampleUnitTest { |
||||
@Test |
||||
public void addition_isCorrect() { |
||||
assertEquals(4, 2 + 2); |
||||
} |
||||
} |
||||
@ -0,0 +1,33 @@ |
||||
apply plugin: 'com.android.library' |
||||
|
||||
android { |
||||
compileSdkVersion rootProject.ext.compileSdkVersion |
||||
buildToolsVersion rootProject.ext.buildToolsVersion |
||||
|
||||
defaultConfig { |
||||
minSdkVersion rootProject.ext.minSdkVersion |
||||
targetSdkVersion rootProject.ext.targetSdkVersion |
||||
versionCode rootProject.ext.versionCode |
||||
versionName rootProject.ext.versionName |
||||
|
||||
testInstrumentationRunner "androidx.test.runner.AndroidJUnitRunner" |
||||
consumerProguardFiles 'consumer-rules.pro' |
||||
} |
||||
|
||||
buildTypes { |
||||
release { |
||||
minifyEnabled false |
||||
proguardFiles getDefaultProguardFile('proguard-android-optimize.txt'), 'proguard-rules.pro' |
||||
} |
||||
} |
||||
} |
||||
|
||||
dependencies { |
||||
implementation fileTree(dir: 'libs', include: ['*.jar']) |
||||
|
||||
implementation "androidx.appcompat:appcompat:${rootProject.ext.XAppcompatVersion}" |
||||
|
||||
testImplementation 'junit:junit:4.12' |
||||
implementation project(path: ':AriaFtpPlug') |
||||
implementation project(path: ':PublicComponent') |
||||
} |
||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in new issue