laoyuyu
2 years ago
parent
6b8c051b30
commit
c06b1fd5fb
@ -1,221 +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.dua.group; |
||||
|
||||
import com.arialyy.aria.core.common.AbsEntity; |
||||
import com.arialyy.aria.core.common.CompleteInfo; |
||||
import com.arialyy.aria.core.download.DGTaskWrapper; |
||||
import com.arialyy.aria.core.download.DTaskWrapper; |
||||
import com.arialyy.aria.core.download.DownloadEntity; |
||||
import com.arialyy.aria.core.loader.IInfoTask; |
||||
import com.arialyy.aria.core.loader.ILoaderVisitor; |
||||
import com.arialyy.aria.exception.AriaException; |
||||
import com.arialyy.aria.exception.AriaHTTPException; |
||||
import com.arialyy.aria.http.HttpTaskOption; |
||||
import com.arialyy.aria.util.ALog; |
||||
import com.arialyy.aria.util.CommonUtil; |
||||
import java.util.concurrent.ExecutorService; |
||||
import java.util.concurrent.Executors; |
||||
import java.util.concurrent.atomic.AtomicInteger; |
||||
|
||||
/** |
||||
* 组合任务文件信息，用于获取长度未知时，组合任务的长度 |
||||
*/ |
||||
public final class HttpDGInfoTask implements IInfoTask { |
||||
private String TAG = CommonUtil.getClassName(this); |
||||
private Callback callback; |
||||
private DGTaskWrapper wrapper; |
||||
private final Object LOCK = new Object(); |
||||
private ExecutorService mPool = null; |
||||
private boolean getLenComplete = false; |
||||
private AtomicInteger count = new AtomicInteger(); |
||||
private AtomicInteger failCount = new AtomicInteger(); |
||||
private boolean isStop = false, isCancel = false; |
||||
|
||||
public interface DGInfoCallback extends Callback { |
||||
|
||||
/** |
||||
* 子任务失败 |
||||
*/ |
||||
void onSubFail(DownloadEntity subEntity, AriaHTTPException e, boolean needRetry); |
||||
|
||||
/** |
||||
* 组合任务停止 |
||||
*/ |
||||
void onStop(long len); |
||||
} |
||||
|
||||
/** |
||||
* 子任务回调 |
||||
*/ |
||||
private Callback subCallback = new Callback() { |
||||
@Override public void onSucceed(String url, CompleteInfo info) { |
||||
count.getAndIncrement(); |
||||
checkGetSizeComplete(count.get(), failCount.get()); |
||||
ALog.d(TAG, "获取子任务信息完成"); |
||||
} |
||||
|
||||
@Override public void onFail(AbsEntity entity, AriaException e, boolean needRetry) { |
||||
ALog.e(TAG, String.format("获取文件信息失败，url：%s", ((DownloadEntity) entity).getUrl())); |
||||
count.getAndIncrement(); |
||||
failCount.getAndIncrement(); |
||||
((DGInfoCallback) callback).onSubFail((DownloadEntity) entity, new AriaHTTPException( |
||||
String.format("子任务获取文件长度失败，url：%s", ((DownloadEntity) entity).getUrl())), needRetry); |
||||
checkGetSizeComplete(count.get(), failCount.get()); |
||||
} |
||||
}; |
||||
|
||||
HttpDGInfoTask(DGTaskWrapper wrapper) { |
||||
this.wrapper = wrapper; |
||||
} |
||||
|
||||
/** |
||||
* 停止 |
||||
*/ |
||||
@Override |
||||
public void stop() { |
||||
isStop = true; |
||||
if (mPool != null) { |
||||
mPool.shutdown(); |
||||
} |
||||
} |
||||
|
||||
@Override public void cancel() { |
||||
isCancel = true; |
||||
if (mPool != null) { |
||||
mPool.shutdown(); |
||||
} |
||||
} |
||||
|
||||
@Override public void run() { |
||||
// 如果是isUnknownSize()标志，并且获取大小没有完成，则直接回调onStop
|
||||
if (mPool != null && !getLenComplete) { |
||||
ALog.d(TAG, "获取长度未完成的情况下，停止组合任务"); |
||||
mPool.shutdown(); |
||||
((DGInfoCallback)callback).onStop(0); |
||||
return; |
||||
} |
||||
// 处理组合任务大小未知的情况
|
||||
if (wrapper.isUnknownSize()) { |
||||
mPool = Executors.newCachedThreadPool(); |
||||
getGroupSize(); |
||||
try { |
||||
synchronized (LOCK) { |
||||
LOCK.wait(); |
||||
} |
||||
} catch (InterruptedException e) { |
||||
e.printStackTrace(); |
||||
} |
||||
if (!mPool.isShutdown()) { |
||||
mPool.shutdown(); |
||||
} |
||||
} else { |
||||
for (DTaskWrapper wrapper : wrapper.getSubTaskWrapper()) { |
||||
cloneHeader(wrapper); |
||||
} |
||||
callback.onSucceed(wrapper.getKey(), new CompleteInfo()); |
||||
} |
||||
} |
||||
|
||||
/* |
||||
* 获取组合任务大小，使用该方式获取到的组合任务大小，子任务不需要再重新获取文件大小 |
||||
*/ |
||||
private void getGroupSize() { |
||||
new Thread(new Runnable() { |
||||
@Override public void run() { |
||||
for (DTaskWrapper dTaskWrapper : wrapper.getSubTaskWrapper()) { |
||||
DownloadEntity subEntity = dTaskWrapper.getEntity(); |
||||
if (subEntity.getFileSize() > 0) { |
||||
count.getAndIncrement(); |
||||
if (subEntity.getCurrentProgress() < subEntity.getFileSize()) { |
||||
// 如果没有完成需要拷贝一份数据
|
||||
cloneHeader(dTaskWrapper); |
||||
} |
||||
checkGetSizeComplete(count.get(), failCount.get()); |
||||
continue; |
||||
} |
||||
cloneHeader(dTaskWrapper); |
||||
HttpDFileInfoTask infoTask = new HttpDFileInfoTask(dTaskWrapper); |
||||
infoTask.setCallback(subCallback); |
||||
mPool.execute(infoTask); |
||||
} |
||||
} |
||||
}).start(); |
||||
} |
||||
|
||||
/** |
||||
* 检查组合任务大小是否获取完成，获取完成后取消阻塞，并设置组合任务大小 |
||||
*/ |
||||
private void checkGetSizeComplete(int count, int failCount) { |
||||
if (isStop || isCancel) { |
||||
ALog.w(TAG, "任务已停止或已取消，isStop = " + isStop + ", isCancel = " + isCancel); |
||||
notifyLock(); |
||||
return; |
||||
} |
||||
if (failCount == wrapper.getSubTaskWrapper().size()) { |
||||
callback.onFail(wrapper.getEntity(), new AriaHTTPException("获取子任务长度失败"), false); |
||||
notifyLock(); |
||||
return; |
||||
} |
||||
if (count == wrapper.getSubTaskWrapper().size()) { |
||||
long size = 0; |
||||
for (DTaskWrapper wrapper : wrapper.getSubTaskWrapper()) { |
||||
size += wrapper.getEntity().getFileSize(); |
||||
} |
||||
wrapper.getEntity().setConvertFileSize(CommonUtil.formatFileSize(size)); |
||||
wrapper.getEntity().setFileSize(size); |
||||
wrapper.getEntity().update(); |
||||
getLenComplete = true; |
||||
ALog.d(TAG, String.format("获取组合任务长度完成，组合任务总长度：%s，失败的子任务数：%s", size, failCount)); |
||||
callback.onSucceed(wrapper.getKey(), new CompleteInfo()); |
||||
notifyLock(); |
||||
} |
||||
} |
||||
|
||||
private void notifyLock() { |
||||
synchronized (LOCK) { |
||||
LOCK.notifyAll(); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* 子任务使用父包裹器的属性 |
||||
*/ |
||||
private void cloneHeader(DTaskWrapper taskWrapper) { |
||||
HttpTaskOption groupOption = (HttpTaskOption) wrapper.getTaskOption(); |
||||
HttpTaskOption subOption = new HttpTaskOption(); |
||||
|
||||
// 设置属性
|
||||
subOption.setFileLenAdapter(groupOption.getFileLenAdapter()); |
||||
subOption.setFileNameAdapter(groupOption.getFileNameAdapter()); |
||||
subOption.setUseServerFileName(groupOption.isUseServerFileName()); |
||||
|
||||
subOption.setFileNameAdapter(groupOption.getFileNameAdapter()); |
||||
subOption.setRequestEnum(groupOption.getRequestEnum()); |
||||
subOption.setHeaders(groupOption.getHeaders()); |
||||
subOption.setProxy(groupOption.getProxy()); |
||||
subOption.setParams(groupOption.getParams()); |
||||
taskWrapper.setTaskOption(subOption); |
||||
} |
||||
|
||||
@Override public void setCallback(Callback callback) { |
||||
this.callback = callback; |
||||
} |
||||
|
||||
@Override public void accept(ILoaderVisitor visitor) { |
||||
visitor.addComponent(this); |
||||
} |
||||
} |
||||
@ -1,59 +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.dua.group; |
||||
|
||||
import android.os.Handler; |
||||
import com.arialyy.aria.core.download.DTaskWrapper; |
||||
import com.arialyy.aria.core.group.AbsSubDLoadAdapter; |
||||
import com.arialyy.aria.core.group.SubRecordHandler; |
||||
import com.arialyy.aria.core.loader.GroupSubThreadStateManager; |
||||
import com.arialyy.aria.core.loader.LoaderStructure; |
||||
import com.arialyy.aria.core.loader.NormalTTBuilder; |
||||
import com.arialyy.aria.core.loader.SubLoader; |
||||
|
||||
/** |
||||
* @author lyy |
||||
* Date: 2019-09-28 |
||||
*/ |
||||
final class HttpSubDLoaderAdapter extends AbsSubDLoadAdapter { |
||||
/** |
||||
* @param schedulers 调度器 |
||||
* @param needGetInfo {@code true} 需要获取文件信息。{@code false} 不需要获取文件信息 |
||||
*/ |
||||
HttpSubDLoaderAdapter( Handler schedulers, boolean needGetInfo, String parentKey) { |
||||
super(schedulers, needGetInfo, parentKey); |
||||
} |
||||
|
||||
@Override protected SubLoader getLoader() { |
||||
if (mDLoader == null) { |
||||
mDLoader = new SubLoader(getWrapper(), getSchedulers()); |
||||
mDLoader.setNeedGetInfo(isNeedGetInfo()); |
||||
mDLoader.setParentKey(getParentKey()); |
||||
} |
||||
return mDLoader; |
||||
} |
||||
|
||||
@Override protected LoaderStructure buildLoaderStructure() { |
||||
LoaderStructure structure = new LoaderStructure(); |
||||
structure.addComponent(new SubRecordHandler(getWrapper())) |
||||
.addComponent(new GroupSubThreadStateManager(getSchedulers(),getKey())) |
||||
.addComponent(new NormalTTBuilder(getWrapper(), new HttpDTTBuilderAdapter())) |
||||
.addComponent(new HttpDFileInfoTask(getWrapper())); |
||||
structure.accept(getLoader()); |
||||
return structure; |
||||
} |
||||
} |
||||
|
||||
Loading…
Reference in new issue