# Conflicts: # PublicComponent/src/main/java/com/arialyy/aria/util/FileUri.ktv4
laoyuyu
2 years ago
commit
f0c63d72b3
@ -0,0 +1,33 @@ |
|||||||
|
/* |
||||||
|
* 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.http.download |
||||||
|
|
||||||
|
import com.arialyy.aria.core.listener.AbsEventListener |
||||||
|
import com.arialyy.aria.core.task.DownloadTask |
||||||
|
import com.arialyy.aria.util.FileUtils |
||||||
|
|
||||||
|
class HttpDEventListener(task: DownloadTask) : AbsEventListener(task) { |
||||||
|
|
||||||
|
override fun handleCancel() { |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
override fun handleComplete() { |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
} |
||||||
@ -1,249 +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.http.download; |
|
||||||
|
|
||||||
import com.arialyy.aria.core.common.RequestEnum; |
|
||||||
import com.arialyy.aria.core.task.AbsThreadTaskAdapter; |
|
||||||
import com.arialyy.aria.core.task.ThreadConfig; |
|
||||||
import com.arialyy.aria.exception.AriaHTTPException; |
|
||||||
import com.arialyy.aria.http.ConnectionHelp; |
|
||||||
import com.arialyy.aria.http.HttpOption; |
|
||||||
import com.arialyy.aria.http.request.IRequest; |
|
||||||
import com.arialyy.aria.util.ALog; |
|
||||||
import com.arialyy.aria.util.BufferedRandomAccessFile; |
|
||||||
import java.io.BufferedInputStream; |
|
||||||
import java.io.FileOutputStream; |
|
||||||
import java.io.IOException; |
|
||||||
import java.io.InputStream; |
|
||||||
import java.io.OutputStreamWriter; |
|
||||||
import java.net.HttpURLConnection; |
|
||||||
import java.net.MalformedURLException; |
|
||||||
import java.net.URLEncoder; |
|
||||||
import java.nio.ByteBuffer; |
|
||||||
import java.nio.channels.Channels; |
|
||||||
import java.nio.channels.FileChannel; |
|
||||||
import java.nio.channels.ReadableByteChannel; |
|
||||||
import java.util.Map; |
|
||||||
import java.util.Set; |
|
||||||
import timber.log.Timber; |
|
||||||
|
|
||||||
/** |
|
||||||
* Created by lyy on 2017/1/18. 下载线程 |
|
||||||
*/ |
|
||||||
final class HttpDThreadTaskAdapter extends AbsThreadTaskAdapter { |
|
||||||
|
|
||||||
HttpDThreadTaskAdapter(ThreadConfig threadConfig) { |
|
||||||
super(threadConfig); |
|
||||||
} |
|
||||||
|
|
||||||
private HttpDTaskOption getTaskOption() { |
|
||||||
return (HttpDTaskOption) getThreadConfig().getOption(); |
|
||||||
} |
|
||||||
|
|
||||||
@Override protected void handlerThreadTask() { |
|
||||||
HttpURLConnection conn = null; |
|
||||||
BufferedInputStream is = null; |
|
||||||
BufferedRandomAccessFile file = null; |
|
||||||
try { |
|
||||||
HttpDTaskOption taskOption = getTaskOption(); |
|
||||||
HttpOption option = taskOption.getHttpOption(); |
|
||||||
conn = IRequest.Companion.getRequest(option).getDConnection(taskOption.getSourUrl(), option); |
|
||||||
if (!taskOption.isSupportResume()) { |
|
||||||
Timber.w("this task not support resume, url: %s", taskOption.getSourUrl()); |
|
||||||
}else { |
|
||||||
conn.setRequestProperty("Range", |
|
||||||
String.format("bytes=%s-%s", getThreadConfig().getBlockRecord().getStartLocation(), |
|
||||||
(getThreadConfig().getBlockRecord().getEndLocation() - 1))); |
|
||||||
} |
|
||||||
|
|
||||||
ConnectionHelp.setConnectParam(mTaskOption, conn); |
|
||||||
conn.setConnectTimeout(getTaskConfig().getConnectTimeOut()); |
|
||||||
conn.setReadTimeout(getTaskConfig().getIOTimeOut()); //设置读取流的等待时间,必须设置该参数
|
|
||||||
if (mTaskOption.isChunked()) { |
|
||||||
conn.setDoInput(true); |
|
||||||
conn.setChunkedStreamingMode(0); |
|
||||||
} |
|
||||||
conn.connect(); |
|
||||||
|
|
||||||
is = new BufferedInputStream(ConnectionHelp.convertInputStream(conn)); |
|
||||||
if (mTaskOption.isChunked()) { |
|
||||||
readChunked(is); |
|
||||||
} else if (getThreadConfig().isBlock) { |
|
||||||
readDynamicFile(is); |
|
||||||
} else { |
|
||||||
//创建可设置位置的文件
|
|
||||||
file = |
|
||||||
new BufferedRandomAccessFile(getThreadConfig().tempFile, "rwd", |
|
||||||
getTaskConfig().getBuffSize()); |
|
||||||
//设置每条线程写入文件的位置
|
|
||||||
if (getThreadRecord().startLocation > 0) { |
|
||||||
file.seek(getThreadRecord().startLocation); |
|
||||||
} |
|
||||||
readNormal(is, file); |
|
||||||
handleComplete(); |
|
||||||
} |
|
||||||
} catch (MalformedURLException e) { |
|
||||||
fail(new AriaHTTPException(String.format("任务【%s】下载失败，filePath: %s, url: %s", getFileName(), |
|
||||||
getEntity().getFilePath(), getEntity().getUrl()), e), false); |
|
||||||
} catch (IOException e) { |
|
||||||
fail(new AriaHTTPException(String.format("任务【%s】下载失败，filePath: %s, url: %s", getFileName(), |
|
||||||
getEntity().getFilePath(), getEntity().getUrl()), e), true); |
|
||||||
} catch (ArrayIndexOutOfBoundsException e) { |
|
||||||
fail(new AriaHTTPException(String.format("任务【%s】下载失败，filePath: %s, url: %s", getFileName(), |
|
||||||
getEntity().getFilePath(), getEntity().getUrl()), e), false); |
|
||||||
} catch (Exception e) { |
|
||||||
fail(new AriaHTTPException(String.format("任务【%s】下载失败，filePath: %s, url: %s", getFileName(), |
|
||||||
getEntity().getFilePath(), getEntity().getUrl()), e), false); |
|
||||||
} finally { |
|
||||||
try { |
|
||||||
if (file != null) { |
|
||||||
file.close(); |
|
||||||
} |
|
||||||
if (is != null) { |
|
||||||
is.close(); |
|
||||||
} |
|
||||||
if (conn != null) { |
|
||||||
conn.getInputStream().close(); |
|
||||||
conn.disconnect(); |
|
||||||
} |
|
||||||
} catch (IOException e) { |
|
||||||
e.printStackTrace(); |
|
||||||
} |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
/** |
|
||||||
* 读取chunked数据 |
|
||||||
*/ |
|
||||||
private void readChunked(InputStream is) { |
|
||||||
FileOutputStream fos = null; |
|
||||||
try { |
|
||||||
fos = new FileOutputStream(getThreadConfig().tempFile, true); |
|
||||||
byte[] buffer = new byte[getTaskConfig().getBuffSize()]; |
|
||||||
int len; |
|
||||||
while (getThreadTask().isLive() && (len = is.read(buffer)) != -1) { |
|
||||||
if (getThreadTask().isBreak()) { |
|
||||||
break; |
|
||||||
} |
|
||||||
if (mSpeedBandUtil != null) { |
|
||||||
mSpeedBandUtil.limitNextBytes(len); |
|
||||||
} |
|
||||||
fos.write(buffer, 0, len); |
|
||||||
progress(len); |
|
||||||
} |
|
||||||
handleComplete(); |
|
||||||
} catch (IOException e) { |
|
||||||
fail(new AriaHTTPException( |
|
||||||
String.format("文件下载失败，savePath: %s, url: %s", getEntity().getFilePath(), |
|
||||||
getThreadConfig().url), e), true); |
|
||||||
} finally { |
|
||||||
if (fos != null) { |
|
||||||
try { |
|
||||||
fos.close(); |
|
||||||
} catch (IOException e) { |
|
||||||
e.printStackTrace(); |
|
||||||
} |
|
||||||
} |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
/** |
|
||||||
* 动态长度文件读取方式 |
|
||||||
*/ |
|
||||||
private void readDynamicFile(InputStream is) { |
|
||||||
FileOutputStream fos = null; |
|
||||||
FileChannel foc = null; |
|
||||||
ReadableByteChannel fic = null; |
|
||||||
try { |
|
||||||
int len; |
|
||||||
fos = new FileOutputStream(getThreadConfig().tempFile, true); |
|
||||||
foc = fos.getChannel(); |
|
||||||
fic = Channels.newChannel(is); |
|
||||||
|
|
||||||
ByteBuffer bf = ByteBuffer.allocate(getTaskConfig().getBuffSize()); |
|
||||||
//如果要通过 Future 的 cancel 方法取消正在运行的任务，那么该任务必定是可以 对线程中断做出响应 的任务。
|
|
||||||
|
|
||||||
while (getThreadTask().isLive() && (len = fic.read(bf)) != -1) { |
|
||||||
if (getThreadTask().isBreak()) { |
|
||||||
break; |
|
||||||
} |
|
||||||
if (mSpeedBandUtil != null) { |
|
||||||
mSpeedBandUtil.limitNextBytes(len); |
|
||||||
} |
|
||||||
if (getRangeProgress() + len >= getThreadRecord().endLocation) { |
|
||||||
len = (int) (getThreadRecord().endLocation - getRangeProgress()); |
|
||||||
bf.flip(); |
|
||||||
fos.write(bf.array(), 0, len); |
|
||||||
bf.compact(); |
|
||||||
progress(len); |
|
||||||
break; |
|
||||||
} else { |
|
||||||
bf.flip(); |
|
||||||
foc.write(bf); |
|
||||||
bf.compact(); |
|
||||||
progress(len); |
|
||||||
} |
|
||||||
} |
|
||||||
handleComplete(); |
|
||||||
} catch (IOException e) { |
|
||||||
fail(new AriaHTTPException( |
|
||||||
String.format("文件下载失败，savePath: %s, url: %s", getEntity().getFilePath(), |
|
||||||
getThreadConfig().url), e), true); |
|
||||||
} finally { |
|
||||||
try { |
|
||||||
if (fos != null) { |
|
||||||
fos.flush(); |
|
||||||
fos.close(); |
|
||||||
} |
|
||||||
if (foc != null) { |
|
||||||
foc.close(); |
|
||||||
} |
|
||||||
if (fic != null) { |
|
||||||
fic.close(); |
|
||||||
} |
|
||||||
} catch (IOException e) { |
|
||||||
e.printStackTrace(); |
|
||||||
} |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
/** |
|
||||||
* 读取普通的文件流 |
|
||||||
*/ |
|
||||||
private void readNormal(InputStream is, BufferedRandomAccessFile file) |
|
||||||
throws IOException { |
|
||||||
byte[] buffer = new byte[getTaskConfig().getBuffSize()]; |
|
||||||
int len; |
|
||||||
while (getThreadTask().isLive() && (len = is.read(buffer)) != -1) { |
|
||||||
if (getThreadTask().isBreak()) { |
|
||||||
break; |
|
||||||
} |
|
||||||
if (mSpeedBandUtil != null) { |
|
||||||
mSpeedBandUtil.limitNextBytes(len); |
|
||||||
} |
|
||||||
file.write(buffer, 0, len); |
|
||||||
progress(len); |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
@Override public void cancel() { |
|
||||||
|
|
||||||
} |
|
||||||
|
|
||||||
@Override public void stop() { |
|
||||||
|
|
||||||
} |
|
||||||
} |
|
||||||
@ -1,265 +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.manager; |
|
||||||
|
|
||||||
import android.text.TextUtils; |
|
||||||
import com.arialyy.aria.core.task.ITask; |
|
||||||
import com.arialyy.aria.core.task.IThreadTask; |
|
||||||
import com.arialyy.aria.util.ALog; |
|
||||||
import com.arialyy.aria.util.CommonUtil; |
|
||||||
import java.util.HashSet; |
|
||||||
import java.util.Map; |
|
||||||
import java.util.Set; |
|
||||||
import java.util.concurrent.ConcurrentHashMap; |
|
||||||
import java.util.concurrent.Future; |
|
||||||
import java.util.concurrent.SynchronousQueue; |
|
||||||
import java.util.concurrent.ThreadPoolExecutor; |
|
||||||
import java.util.concurrent.TimeUnit; |
|
||||||
import java.util.concurrent.locks.ReentrantLock; |
|
||||||
|
|
||||||
/** |
|
||||||
* 线程任务管理器 |
|
||||||
*/ |
|
||||||
@Deprecated |
|
||||||
public class ThreadTaskManager1 { |
|
||||||
private final String TAG = CommonUtil.getClassName(this); |
|
||||||
private static volatile ThreadTaskManager INSTANCE = null; |
|
||||||
private static final int CORE_POOL_NUM = 20; |
|
||||||
private static final ReentrantLock LOCK = new ReentrantLock(); |
|
||||||
private final ThreadPoolExecutor mExePool; |
|
||||||
private final Map<Integer, Set<FutureContainer>> mThreadTasks = new ConcurrentHashMap<>(); |
|
||||||
|
|
||||||
public static synchronized ThreadTaskManager getInstance() { |
|
||||||
if (INSTANCE == null) { |
|
||||||
INSTANCE = new ThreadTaskManager(); |
|
||||||
} |
|
||||||
return INSTANCE; |
|
||||||
} |
|
||||||
|
|
||||||
private ThreadTaskManager() { |
|
||||||
mExePool = new ThreadPoolExecutor(CORE_POOL_NUM, Integer.MAX_VALUE, |
|
||||||
60L, TimeUnit.SECONDS, |
|
||||||
new SynchronousQueue<Runnable>()); |
|
||||||
mExePool.allowsCoreThreadTimeOut(); |
|
||||||
} |
|
||||||
|
|
||||||
/** |
|
||||||
* 删除所有线程任务 |
|
||||||
*/ |
|
||||||
public void removeAllThreadTask() { |
|
||||||
if (mThreadTasks.isEmpty()) { |
|
||||||
return; |
|
||||||
} |
|
||||||
try { |
|
||||||
LOCK.tryLock(2, TimeUnit.SECONDS); |
|
||||||
for (Set<FutureContainer> threads : mThreadTasks.values()) { |
|
||||||
for (FutureContainer container : threads) { |
|
||||||
if (container.future.isDone() || container.future.isCancelled()) { |
|
||||||
continue; |
|
||||||
} |
|
||||||
container.threadTask.destroy(); |
|
||||||
} |
|
||||||
threads.clear(); |
|
||||||
} |
|
||||||
mThreadTasks.clear(); |
|
||||||
} catch (InterruptedException e) { |
|
||||||
e.printStackTrace(); |
|
||||||
} finally { |
|
||||||
LOCK.unlock(); |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
/** |
|
||||||
* 启动线程任务 |
|
||||||
* |
|
||||||
* @param taskId {@link ITask#getTaskId()} |
|
||||||
* @param threadTask 线程任务{@link IThreadTask} |
|
||||||
*/ |
|
||||||
public void startThread(Integer taskId, IThreadTask threadTask) { |
|
||||||
try { |
|
||||||
LOCK.tryLock(2, TimeUnit.SECONDS); |
|
||||||
if (mExePool.isShutdown()) { |
|
||||||
ALog.e(TAG, "线程池已经关闭"); |
|
||||||
return; |
|
||||||
} |
|
||||||
Set<FutureContainer> temp = mThreadTasks.get(taskId); |
|
||||||
if (temp == null) { |
|
||||||
temp = new HashSet<>(); |
|
||||||
mThreadTasks.put(taskId, temp); |
|
||||||
} |
|
||||||
FutureContainer container = new FutureContainer(); |
|
||||||
container.threadTask = threadTask; |
|
||||||
container.future = mExePool.submit(threadTask); |
|
||||||
temp.add(container); |
|
||||||
} catch (Exception e) { |
|
||||||
e.printStackTrace(); |
|
||||||
} finally { |
|
||||||
LOCK.unlock(); |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
/** |
|
||||||
* 任务是否在执行 |
|
||||||
* |
|
||||||
* @return {@code true} 任务正在运行 |
|
||||||
*/ |
|
||||||
public boolean taskIsRunning(Integer taskId) { |
|
||||||
return mThreadTasks.get(taskId) != null; |
|
||||||
} |
|
||||||
|
|
||||||
/** |
|
||||||
* 停止任务的所有线程 |
|
||||||
*/ |
|
||||||
public void removeTaskThread(Integer taskId) { |
|
||||||
try { |
|
||||||
LOCK.tryLock(2, TimeUnit.SECONDS); |
|
||||||
if (mExePool.isShutdown()) { |
|
||||||
ALog.e(TAG, "线程池已经关闭"); |
|
||||||
return; |
|
||||||
} |
|
||||||
Set<FutureContainer> temp = mThreadTasks.get(taskId); |
|
||||||
if (temp != null && temp.size() > 0) { |
|
||||||
for (FutureContainer container : temp) { |
|
||||||
if (container.future.isDone() || container.future.isCancelled()) { |
|
||||||
continue; |
|
||||||
} |
|
||||||
container.threadTask.destroy(); |
|
||||||
} |
|
||||||
temp.clear(); |
|
||||||
mThreadTasks.remove(taskId); |
|
||||||
} |
|
||||||
} catch (Exception e) { |
|
||||||
e.printStackTrace(); |
|
||||||
} finally { |
|
||||||
LOCK.unlock(); |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
/** |
|
||||||
* 根据线程名删除任务的中的线程 |
|
||||||
* |
|
||||||
* @param threadName 线程名 |
|
||||||
* @return true 删除线程成功；false 删除线程失败 |
|
||||||
*/ |
|
||||||
public boolean removeSingleTaskThread(Integer taskId, String threadName) { |
|
||||||
try { |
|
||||||
LOCK.tryLock(2, TimeUnit.SECONDS); |
|
||||||
if (mExePool.isShutdown()) { |
|
||||||
ALog.e(TAG, "线程池已经关闭"); |
|
||||||
return false; |
|
||||||
} |
|
||||||
if (TextUtils.isEmpty(threadName)) { |
|
||||||
ALog.e(TAG, "线程名为空"); |
|
||||||
return false; |
|
||||||
} |
|
||||||
|
|
||||||
Set<FutureContainer> temp = mThreadTasks.get(taskId); |
|
||||||
if (temp != null && temp.size() > 0) { |
|
||||||
FutureContainer tempC = null; |
|
||||||
for (FutureContainer container : temp) { |
|
||||||
if (container.threadTask.getThreadName().equals(threadName)) { |
|
||||||
tempC = container; |
|
||||||
break; |
|
||||||
} |
|
||||||
} |
|
||||||
if (tempC != null) { |
|
||||||
tempC.threadTask.destroy(); |
|
||||||
temp.remove(tempC); |
|
||||||
return true; |
|
||||||
} |
|
||||||
} |
|
||||||
} catch (Exception e) { |
|
||||||
e.printStackTrace(); |
|
||||||
} finally { |
|
||||||
LOCK.unlock(); |
|
||||||
} |
|
||||||
return false; |
|
||||||
} |
|
||||||
|
|
||||||
/** |
|
||||||
* 删除单个线程任务 |
|
||||||
* |
|
||||||
* @param task 线程任务 |
|
||||||
*/ |
|
||||||
public boolean removeSingleTaskThread(Integer taskId, IThreadTask task) { |
|
||||||
try { |
|
||||||
LOCK.tryLock(2, TimeUnit.SECONDS); |
|
||||||
if (mExePool.isShutdown()) { |
|
||||||
ALog.e(TAG, "线程池已经关闭"); |
|
||||||
return false; |
|
||||||
} |
|
||||||
if (task == null) { |
|
||||||
ALog.e(TAG, "线程任务为空"); |
|
||||||
return false; |
|
||||||
} |
|
||||||
Set<FutureContainer> temp = mThreadTasks.get(taskId); |
|
||||||
if (temp != null && temp.size() > 0) { |
|
||||||
FutureContainer tempC = null; |
|
||||||
for (FutureContainer container : temp) { |
|
||||||
if (container.threadTask == task) { |
|
||||||
tempC = container; |
|
||||||
break; |
|
||||||
} |
|
||||||
} |
|
||||||
if (tempC != null) { |
|
||||||
task.destroy(); |
|
||||||
temp.remove(tempC); |
|
||||||
return true; |
|
||||||
} |
|
||||||
} |
|
||||||
} catch (Exception e) { |
|
||||||
e.printStackTrace(); |
|
||||||
} finally { |
|
||||||
LOCK.unlock(); |
|
||||||
} |
|
||||||
return false; |
|
||||||
} |
|
||||||
|
|
||||||
/** |
|
||||||
* 重试线程任务 |
|
||||||
* |
|
||||||
* @param task 线程任务 |
|
||||||
*/ |
|
||||||
public void retryThread(IThreadTask task) { |
|
||||||
try { |
|
||||||
LOCK.tryLock(2, TimeUnit.SECONDS); |
|
||||||
if (mExePool.isShutdown()) { |
|
||||||
ALog.e(TAG, "线程池已经关闭"); |
|
||||||
return; |
|
||||||
} |
|
||||||
try { |
|
||||||
if (task == null || task.isDestroy()) { |
|
||||||
ALog.e(TAG, "线程为空或线程已经中断"); |
|
||||||
return; |
|
||||||
} |
|
||||||
} catch (Exception e) { |
|
||||||
ALog.e(TAG, "", e); |
|
||||||
return; |
|
||||||
} |
|
||||||
mExePool.submit(task); |
|
||||||
} catch (Exception e) { |
|
||||||
e.printStackTrace(); |
|
||||||
} finally { |
|
||||||
LOCK.unlock(); |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
private static class FutureContainer { |
|
||||||
Future future; |
|
||||||
IThreadTask threadTask; |
|
||||||
} |
|
||||||
} |
|
||||||
@ -0,0 +1,82 @@ |
|||||||
|
/* |
||||||
|
* 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.task |
||||||
|
|
||||||
|
import java.util.concurrent.ConcurrentHashMap |
||||||
|
import java.util.concurrent.TimeUnit.SECONDS |
||||||
|
import java.util.concurrent.locks.ReentrantLock |
||||||
|
|
||||||
|
object ThreadTaskManager2 { |
||||||
|
private val mThreadTasks: ConcurrentHashMap<Int, MutableSet<IThreadTask>> = ConcurrentHashMap() |
||||||
|
private val LOCK = ReentrantLock() |
||||||
|
|
||||||
|
/** |
||||||
|
* 任务是否在执行 |
||||||
|
* |
||||||
|
* @return `true` 任务正在运行 |
||||||
|
*/ |
||||||
|
fun taskIsRunning(taskId: Int): Boolean { |
||||||
|
return mThreadTasks[taskId] != null |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* stop thread task |
||||||
|
* @param isRemoveTask if true, remove task and block |
||||||
|
*/ |
||||||
|
fun stopThreadTask(taskId: Int, isRemoveTask: Boolean = false) { |
||||||
|
try { |
||||||
|
LOCK.tryLock(2, SECONDS) |
||||||
|
val threadTaskList: MutableSet<IThreadTask>? = mThreadTasks[taskId] |
||||||
|
threadTaskList?.forEach { |
||||||
|
if (isRemoveTask) { |
||||||
|
it.cancel() |
||||||
|
return@forEach |
||||||
|
} |
||||||
|
it.stop() |
||||||
|
} |
||||||
|
mThreadTasks.remove(taskId) |
||||||
|
|
||||||
|
} catch (e: Exception) { |
||||||
|
e.printStackTrace() |
||||||
|
} finally { |
||||||
|
LOCK.unlock() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 删除所有线程任务 |
||||||
|
*/ |
||||||
|
fun removeAllThreadTask() { |
||||||
|
if (mThreadTasks.isEmpty()) { |
||||||
|
return |
||||||
|
} |
||||||
|
try { |
||||||
|
LOCK.tryLock(2, SECONDS) |
||||||
|
for (threads in mThreadTasks.values) { |
||||||
|
for (tt in threads) { |
||||||
|
tt.stop() |
||||||
|
} |
||||||
|
threads.clear() |
||||||
|
} |
||||||
|
mThreadTasks.clear() |
||||||
|
} catch (e: InterruptedException) { |
||||||
|
e.printStackTrace() |
||||||
|
} finally { |
||||||
|
LOCK.unlock() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
Loading…
Reference in new issue