laoyuyu
2 years ago
parent
d5cc67391e
commit
61ba26e58a
@ -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() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,388 @@ |
|||||||
|
package com.arialyy.aria.util |
||||||
|
|
||||||
|
import android.annotation.SuppressLint |
||||||
|
import android.content.* |
||||||
|
import android.database.Cursor |
||||||
|
import android.net.Uri |
||||||
|
import android.os.Build |
||||||
|
import android.os.Environment |
||||||
|
import android.provider.* |
||||||
|
import androidx.core.content.FileProvider |
||||||
|
import com.arialyy.aria.core.DuaContext |
||||||
|
import timber.log.Timber |
||||||
|
import java.io.* |
||||||
|
|
||||||
|
/** |
||||||
|
* # FileUri |
||||||
|
* |
||||||
|
* - Uri & Path Tool |
||||||
|
* |
||||||
|
* @author javakam |
||||||
|
* @date 2020/8/24 11:24 |
||||||
|
*/ |
||||||
|
object FileUri { |
||||||
|
|
||||||
|
//Android R |
||||||
|
//---------------------------------------------------------------- |
||||||
|
|
||||||
|
/** |
||||||
|
* `MANAGE_EXTERNAL_STORAGE` 权限检查 |
||||||
|
* |
||||||
|
* @return `true` Have permission |
||||||
|
*/ |
||||||
|
fun isExternalStorageManager(): Boolean = |
||||||
|
if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.R) Environment.isExternalStorageManager() else false |
||||||
|
|
||||||
|
/** |
||||||
|
* 跳转到 `MANAGE_EXTERNAL_STORAGE` 权限设置页面 |
||||||
|
* |
||||||
|
* @return `true` Has been set |
||||||
|
*/ |
||||||
|
fun jumpManageAppAllFilesPermissionSetting( |
||||||
|
context: Context, |
||||||
|
isNewTask: Boolean = false, |
||||||
|
): Boolean { |
||||||
|
if (isExternalStorageManager()) return true |
||||||
|
|
||||||
|
if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.R) { |
||||||
|
try { |
||||||
|
val intent = Intent(Settings.ACTION_MANAGE_APP_ALL_FILES_ACCESS_PERMISSION) |
||||||
|
intent.data = Uri.parse("package:${context.packageName}") |
||||||
|
if (isNewTask) intent.addFlags(Intent.FLAG_ACTIVITY_NEW_TASK) |
||||||
|
context.startActivity(intent) |
||||||
|
} catch (e: Exception) { |
||||||
|
Timber.e("jumpManageAppAllFilesPermissionSetting: $e") |
||||||
|
} |
||||||
|
} |
||||||
|
return false |
||||||
|
} |
||||||
|
|
||||||
|
//从 FilePath 中获取 Uri (Get Uri from FilePath) |
||||||
|
//---------------------------------------------------------------- |
||||||
|
|
||||||
|
fun getUriByPath(path: String?): Uri? = |
||||||
|
if (path.isNullOrBlank()) null else getUriByFile(File(path)) |
||||||
|
|
||||||
|
/** |
||||||
|
* Return a content URI for a given file. |
||||||
|
* |
||||||
|
* @param file The file. |
||||||
|
* @param isOriginal true content:// or file:// ; false file://xxx |
||||||
|
* @return a content URI for a given file |
||||||
|
*/ |
||||||
|
fun getUriByFile(file: File?, isOriginal: Boolean = false): Uri? { |
||||||
|
return if (isOriginal) Uri.fromFile(file) |
||||||
|
else { |
||||||
|
if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.N) { |
||||||
|
val authority = DuaContext.context.packageName + AUTHORITY |
||||||
|
FileProvider.getUriForFile(DuaContext.context, authority, file ?: return null) |
||||||
|
} else { |
||||||
|
Uri.fromFile(file) |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
fun getShareUri(path: String?): Uri? = |
||||||
|
if (path.isNullOrBlank()) null else getUriByFile(File(path), isOriginal = false) |
||||||
|
|
||||||
|
/** |
||||||
|
* @return content:// or file:// |
||||||
|
*/ |
||||||
|
fun getShareUri(file: File?): Uri? = getUriByFile(file, isOriginal = false) |
||||||
|
|
||||||
|
fun getOriginalUri(path: String?): Uri? = |
||||||
|
if (path.isNullOrBlank()) null else getUriByFile(File(path), isOriginal = true) |
||||||
|
|
||||||
|
/** |
||||||
|
* @return file://xxx |
||||||
|
*/ |
||||||
|
fun getOriginalUri(file: File?): Uri? = getUriByFile(file, isOriginal = true) |
||||||
|
|
||||||
|
//获取Uri对应的文件路径, Compatible with API 26 |
||||||
|
//---------------------------------------------------------------- |
||||||
|
|
||||||
|
/** |
||||||
|
* ### Get the file path through Uri |
||||||
|
* |
||||||
|
* - Need permission: RequiresPermission(permission.READ_EXTERNAL_STORAGE) |
||||||
|
* |
||||||
|
* - Modified from: https://github.com/coltoscosmin/FileUtils/blob/master/FileUtils.java |
||||||
|
* |
||||||
|
* @return file path |
||||||
|
*/ |
||||||
|
fun getPathByUri(uri: Uri?): String? { |
||||||
|
return uri?.use { |
||||||
|
Timber.i( |
||||||
|
"FileUri getPathByUri -> " + |
||||||
|
"Uri: " + uri + |
||||||
|
", Authority: " + uri.authority + |
||||||
|
", Fragment: " + uri.fragment + |
||||||
|
", Port: " + uri.port + |
||||||
|
", Query: " + uri.query + |
||||||
|
", Scheme: " + uri.scheme + |
||||||
|
", Host: " + uri.host + |
||||||
|
", Segments: " + uri.pathSegments.toString() |
||||||
|
) |
||||||
|
|
||||||
|
// 以 file:// 开头的使用第三方应用打开 (open with third-party applications starting with file://) |
||||||
|
if (ContentResolver.SCHEME_FILE.equals(uri.scheme, ignoreCase = true)) return getDataColumn( |
||||||
|
uri |
||||||
|
) |
||||||
|
|
||||||
|
@SuppressLint("ObsoleteSdkInt") |
||||||
|
val isKitKat = Build.VERSION.SDK_INT >= Build.VERSION_CODES.KITKAT |
||||||
|
|
||||||
|
// Before 4.4 , API 19 content:// 开头, 比如 content://media/external/images/media/123 |
||||||
|
if (!isKitKat && ContentResolver.SCHEME_CONTENT.equals(uri.scheme, true)) { |
||||||
|
if (isGooglePhotosUri(uri)) return uri.lastPathSegment |
||||||
|
return getDataColumn(uri) |
||||||
|
} |
||||||
|
|
||||||
|
val context = DuaContext.context |
||||||
|
// After 4.4 , API 19 |
||||||
|
// DocumentProvider |
||||||
|
if (isKitKat && DocumentsContract.isDocumentUri(context, uri)) { |
||||||
|
// LocalStorageProvider |
||||||
|
if (isLocalStorageDocument(uri)) { |
||||||
|
// The path is the id |
||||||
|
return DocumentsContract.getDocumentId(uri); |
||||||
|
} |
||||||
|
// ExternalStorageProvider |
||||||
|
if (isExternalStorageDocument(uri)) { |
||||||
|
val docId = DocumentsContract.getDocumentId(uri) |
||||||
|
val split = docId.split(":").toTypedArray() |
||||||
|
val type = split[0] |
||||||
|
if ("primary".equals(type, ignoreCase = true)) { |
||||||
|
if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.Q) { |
||||||
|
return context.getExternalFilesDir(Environment.DIRECTORY_DOCUMENTS) |
||||||
|
.toString() + File.separator + split[1] |
||||||
|
} else { |
||||||
|
@Suppress("DEPRECATION") |
||||||
|
return Environment.getExternalStorageDirectory() |
||||||
|
.toString() + File.separator + split[1] |
||||||
|
} |
||||||
|
} else if ("home".equals(type, ignoreCase = true)) { |
||||||
|
if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.Q) { |
||||||
|
return context.getExternalFilesDir(Environment.DIRECTORY_DOCUMENTS) |
||||||
|
.toString() + File.separator + "documents" + File.separator + split[1] |
||||||
|
} else { |
||||||
|
@Suppress("DEPRECATION") |
||||||
|
return Environment.getExternalStorageDirectory() |
||||||
|
.toString() + File.separator + "documents" + File.separator + split[1] |
||||||
|
} |
||||||
|
} else { |
||||||
|
@Suppress("DEPRECATION") |
||||||
|
val sdcardPath = |
||||||
|
Environment.getExternalStorageDirectory() |
||||||
|
.toString() + File.separator + "documents" + File.separator + split[1] |
||||||
|
return if (sdcardPath.startsWith("file://")) { |
||||||
|
sdcardPath.replace("file://", "") |
||||||
|
} else { |
||||||
|
sdcardPath |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
// DownloadsProvider |
||||||
|
else if (isDownloadsDocument(uri)) { |
||||||
|
val id = DocumentsContract.getDocumentId(uri) |
||||||
|
if (id != null && id.startsWith("raw:")) { |
||||||
|
return id.substring(4) |
||||||
|
} |
||||||
|
if (Build.VERSION.SDK_INT < Build.VERSION_CODES.O) { |
||||||
|
val contentUriPrefixesToTry = arrayOf( |
||||||
|
"content://downloads/public_downloads", |
||||||
|
"content://downloads/my_downloads", |
||||||
|
"content://downloads/all_downloads" |
||||||
|
) |
||||||
|
for (contentUriPrefix in contentUriPrefixesToTry) { |
||||||
|
val contentUri = ContentUris.withAppendedId(Uri.parse(contentUriPrefix), id.toLong()) |
||||||
|
try { |
||||||
|
val path = getDataColumn(contentUri) |
||||||
|
if (!path.isNullOrBlank()) return path |
||||||
|
} catch (e: Exception) { |
||||||
|
Timber.e(e.toString()) |
||||||
|
} |
||||||
|
} |
||||||
|
} else { |
||||||
|
//testPath(uri) |
||||||
|
return getDataColumn(uri) |
||||||
|
} |
||||||
|
} |
||||||
|
// MediaProvider |
||||||
|
else if (isMediaDocument(uri)) { |
||||||
|
val docId = DocumentsContract.getDocumentId(uri) |
||||||
|
val split = docId.split(":").toTypedArray() |
||||||
|
val contentUri: Uri? = when (split[0]) { |
||||||
|
"image" -> MediaStore.Images.Media.EXTERNAL_CONTENT_URI |
||||||
|
"video" -> MediaStore.Video.Media.EXTERNAL_CONTENT_URI |
||||||
|
"audio" -> MediaStore.Audio.Media.EXTERNAL_CONTENT_URI |
||||||
|
"download" -> if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.Q) { |
||||||
|
MediaStore.Downloads.EXTERNAL_CONTENT_URI |
||||||
|
} else null |
||||||
|
else -> null |
||||||
|
} |
||||||
|
val selectionArgs = arrayOf(split[1]) |
||||||
|
return getDataColumn(contentUri, "_id=?", selectionArgs) |
||||||
|
} |
||||||
|
|
||||||
|
//GoogleDriveProvider |
||||||
|
else if (isGoogleDriveUri(uri)) { |
||||||
|
return getGoogleDriveFilePath(uri, context) |
||||||
|
} |
||||||
|
} |
||||||
|
// MediaStore (and general) |
||||||
|
else if ("content".equals(uri.scheme, ignoreCase = true)) { |
||||||
|
// Return the remote address |
||||||
|
if (isGooglePhotosUri(uri)) { |
||||||
|
return uri.lastPathSegment |
||||||
|
} |
||||||
|
// Google drive legacy provider |
||||||
|
else if (isGoogleDriveUri(uri)) { |
||||||
|
return getGoogleDriveFilePath(uri, context) |
||||||
|
} |
||||||
|
// Huawei |
||||||
|
else if (isHuaWeiUri(uri)) { |
||||||
|
val uriPath = getDataColumn(uri) ?: uri.toString() |
||||||
|
//content://com.huawei.hidisk.fileprovider/root/storage/emulated/0/Android/data/com.xxx.xxx/ |
||||||
|
if (uriPath.startsWith("/root")) { |
||||||
|
return uriPath.replace("/root".toRegex(), "") |
||||||
|
} |
||||||
|
} |
||||||
|
return getDataColumn(uri) |
||||||
|
} |
||||||
|
return getDataColumn(uri) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* BUG : 部分机型进入"文件管理器" 执行到 cursor.getColumnIndexOrThrow(column);出现 |
||||||
|
* Caused by: java.lang.IllegalArgumentException: column '_data' does not exist. Available columns: [] |
||||||
|
* |
||||||
|
* Fixed : |
||||||
|
* https://stackoverflow.com/questions/42508383/illegalargumentexception-column-data-does-not-exist |
||||||
|
* |
||||||
|
*/ |
||||||
|
private fun getDataColumn( |
||||||
|
uri: Uri?, |
||||||
|
selection: String? = null, |
||||||
|
selectionArgs: Array<String>? = null |
||||||
|
): String? { |
||||||
|
@Suppress("DEPRECATION") |
||||||
|
val column = MediaStore.Files.FileColumns.DATA |
||||||
|
val projection = arrayOf(column) |
||||||
|
try { |
||||||
|
DuaContext.context.contentResolver.query( |
||||||
|
uri ?: return null, |
||||||
|
projection, |
||||||
|
selection, |
||||||
|
selectionArgs, |
||||||
|
null |
||||||
|
)?.use { c: Cursor -> |
||||||
|
if (c.moveToFirst()) { |
||||||
|
val columnIndex = c.getColumnIndex(column) |
||||||
|
return c.getString(columnIndex) |
||||||
|
} |
||||||
|
} |
||||||
|
} catch (e: Throwable) { |
||||||
|
Timber.e("getDataColumn -> ${e.message}") |
||||||
|
} |
||||||
|
return null |
||||||
|
} |
||||||
|
|
||||||
|
//The Uri to check |
||||||
|
//---------------------------------------------------------------- |
||||||
|
|
||||||
|
private fun getGoogleDriveFilePath(uri: Uri, context: Context): String? { |
||||||
|
context.contentResolver.query(uri, null, null, null, null)?.use { c: Cursor -> |
||||||
|
/* |
||||||
|
Get the column indexes of the data in the Cursor, |
||||||
|
move to the first row in the Cursor, get the data, and display it. |
||||||
|
*/ |
||||||
|
val nameIndex: Int = c.getColumnIndex(OpenableColumns.DISPLAY_NAME) |
||||||
|
//val sizeIndex: Int = c.getColumnIndex(OpenableColumns.SIZE) |
||||||
|
if (!c.moveToFirst()) { |
||||||
|
return uri.toString() |
||||||
|
} |
||||||
|
val name: String = c.getString(nameIndex) |
||||||
|
//val size = c.getLong(sizeIndex).toString() |
||||||
|
val file = File(context.cacheDir, name) |
||||||
|
|
||||||
|
var inputStream: InputStream? = null |
||||||
|
var outputStream: FileOutputStream? = null |
||||||
|
try { |
||||||
|
inputStream = context.contentResolver.openInputStream(uri) |
||||||
|
outputStream = FileOutputStream(file) |
||||||
|
var read = 0 |
||||||
|
val maxBufferSize = 1 * 1024 * 1024 |
||||||
|
val bytesAvailable: Int = inputStream?.available() ?: 0 |
||||||
|
val bufferSize = bytesAvailable.coerceAtMost(maxBufferSize) |
||||||
|
val buffers = ByteArray(bufferSize) |
||||||
|
while (inputStream?.read(buffers)?.also { read = it } != -1) { |
||||||
|
outputStream.write(buffers, 0, read) |
||||||
|
} |
||||||
|
} catch (e: Exception) { |
||||||
|
e.printStackTrace() |
||||||
|
} finally { |
||||||
|
inputStream?.close() |
||||||
|
outputStream?.close() |
||||||
|
} |
||||||
|
return file.path |
||||||
|
} |
||||||
|
return uri.toString() |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* @param uri The Uri to check. |
||||||
|
* @return Whether the Uri authority is Google Photos. |
||||||
|
*/ |
||||||
|
fun isGooglePhotosUri(uri: Uri?): Boolean { |
||||||
|
return "com.google.android.apps.photos.content".equals(uri?.authority, true) |
||||||
|
} |
||||||
|
|
||||||
|
fun isGoogleDriveUri(uri: Uri?): Boolean { |
||||||
|
return "com.google.android.apps.docs.storage.legacy" == uri?.authority || "com.google.android.apps.docs.storage" == uri?.authority |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* @param uri The Uri to check. |
||||||
|
* @return Whether the Uri authority is local. |
||||||
|
*/ |
||||||
|
fun isLocalStorageDocument(uri: Uri?): Boolean { |
||||||
|
return AUTHORITY.equals(uri?.authority, true) |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* @param uri The Uri to check. |
||||||
|
* @return Whether the Uri authority is ExternalStorageProvider. |
||||||
|
*/ |
||||||
|
private fun isExternalStorageDocument(uri: Uri?): Boolean { |
||||||
|
return "com.android.externalstorage.documents".equals(uri?.authority, true) |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* @param uri The Uri to check. |
||||||
|
* @return Whether the Uri authority is DownloadsProvider. |
||||||
|
*/ |
||||||
|
private fun isDownloadsDocument(uri: Uri?): Boolean { |
||||||
|
return "com.android.providers.downloads.documents".equals(uri?.authority, true) |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* @param uri The Uri to check. |
||||||
|
* @return Whether the Uri authority is MediaProvider. |
||||||
|
*/ |
||||||
|
private fun isMediaDocument(uri: Uri?): Boolean { |
||||||
|
return "com.android.providers.media.documents".equals(uri?.authority, true) |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* content://com.huawei.hidisk.fileprovider/root/storage/emulated/0/Android/data/com.xxx.xxx/ |
||||||
|
* |
||||||
|
* @param uri |
||||||
|
* @return |
||||||
|
*/ |
||||||
|
private fun isHuaWeiUri(uri: Uri?): Boolean { |
||||||
|
return "com.huawei.hidisk.fileprovider".equals(uri?.authority, true) |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
Loading…
Reference in new issue