laoyuyu
6 years ago
parent
a63222a220
commit
6d5398de40
@ -1,78 +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 java.util.ArrayList; |
|
||||||
import java.util.Iterator; |
|
||||||
import java.util.List; |
|
||||||
import java.util.Map; |
|
||||||
import java.util.concurrent.ConcurrentHashMap; |
|
||||||
|
|
||||||
/** |
|
||||||
* 线程任务管理器 |
|
||||||
*/ |
|
||||||
class ThreadTaskManager { |
|
||||||
private static volatile ThreadTaskManager INSTANCE = null; |
|
||||||
private Map<String, List<AbsThreadTask>> mThreadTasks = new ConcurrentHashMap<>(); |
|
||||||
|
|
||||||
private ThreadTaskManager() { |
|
||||||
|
|
||||||
} |
|
||||||
|
|
||||||
public static ThreadTaskManager getInstance() { |
|
||||||
if (INSTANCE == null) { |
|
||||||
synchronized (ThreadTaskManager.class) { |
|
||||||
INSTANCE = new ThreadTaskManager(); |
|
||||||
} |
|
||||||
} |
|
||||||
return INSTANCE; |
|
||||||
} |
|
||||||
|
|
||||||
/** |
|
||||||
* 添加单条线程记录 |
|
||||||
* |
|
||||||
* @param key 任务对应的key |
|
||||||
* @param threadTask 线程任务 |
|
||||||
*/ |
|
||||||
public void addTask(String key, AbsThreadTask threadTask) { |
|
||||||
if (mThreadTasks.get(key) == null) { |
|
||||||
mThreadTasks.put(key, new ArrayList<AbsThreadTask>()); |
|
||||||
} |
|
||||||
mThreadTasks.get(key).add(threadTask); |
|
||||||
} |
|
||||||
|
|
||||||
/** |
|
||||||
* 删除对应的任务的线程记录 |
|
||||||
* |
|
||||||
* @param key 任务对应的key |
|
||||||
*/ |
|
||||||
public void removeTask(String key) { |
|
||||||
for (Iterator<Map.Entry<String, List<AbsThreadTask>>> iter = mThreadTasks.entrySet().iterator(); |
|
||||||
iter.hasNext(); ) { |
|
||||||
Map.Entry<String, List<AbsThreadTask>> entry = iter.next(); |
|
||||||
if (key.equals(entry.getKey())) { |
|
||||||
List<AbsThreadTask> list = mThreadTasks.get(key); |
|
||||||
if (list != null && !list.isEmpty()) { |
|
||||||
list.clear(); |
|
||||||
} |
|
||||||
iter.remove(); |
|
||||||
} |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
} |
|
||||||
@ -1,4 +1,116 @@ |
|||||||
|
/* |
||||||
|
* 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; |
package com.arialyy.aria.core.manager; |
||||||
|
|
||||||
|
import com.arialyy.aria.core.common.AbsThreadTask; |
||||||
|
import com.arialyy.aria.core.inf.AbsTaskEntity; |
||||||
|
import com.arialyy.aria.util.ALog; |
||||||
|
import com.arialyy.aria.util.CommonUtil; |
||||||
|
import java.util.HashMap; |
||||||
|
import java.util.HashSet; |
||||||
|
import java.util.Map; |
||||||
|
import java.util.Set; |
||||||
|
import java.util.concurrent.ExecutorService; |
||||||
|
import java.util.concurrent.Executors; |
||||||
|
import java.util.concurrent.Future; |
||||||
|
|
||||||
|
/** |
||||||
|
* 线程管理器 |
||||||
|
*/ |
||||||
public class ThreadTaskManager { |
public class ThreadTaskManager { |
||||||
|
private static volatile ThreadTaskManager INSTANCE = null; |
||||||
|
private static final Object LOCK = new Object(); |
||||||
|
private final String TAG = "ThreadTaskManager"; |
||||||
|
private ExecutorService mExePool; |
||||||
|
private Map<String, Set<Future>> mThreadTasks = new HashMap<>(); |
||||||
|
|
||||||
|
public static ThreadTaskManager getInstance() { |
||||||
|
if (INSTANCE == null) { |
||||||
|
synchronized (LOCK) { |
||||||
|
INSTANCE = new ThreadTaskManager(); |
||||||
|
} |
||||||
|
} |
||||||
|
return INSTANCE; |
||||||
|
} |
||||||
|
|
||||||
|
private ThreadTaskManager() { |
||||||
|
mExePool = Executors.newFixedThreadPool(Runtime.getRuntime().availableProcessors()); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 启动线程任务 |
||||||
|
* |
||||||
|
* @param key 任务对应的key{@link AbsTaskEntity#getKey()} |
||||||
|
* @param threadTask 线程任务{@link AbsThreadTask} |
||||||
|
*/ |
||||||
|
public void startThread(String key, AbsThreadTask threadTask) { |
||||||
|
if (mExePool.isShutdown()) { |
||||||
|
ALog.e(TAG, "线程池已经关闭"); |
||||||
|
return; |
||||||
|
} |
||||||
|
key = getKey(key); |
||||||
|
Set<Future> temp = mThreadTasks.get(key); |
||||||
|
if (temp == null) { |
||||||
|
temp = new HashSet<>(); |
||||||
|
mThreadTasks.put(key, temp); |
||||||
|
} |
||||||
|
temp.add(mExePool.submit(threadTask)); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 停止任务的所有线程 |
||||||
|
* |
||||||
|
* @param key 任务对应的key{@link AbsTaskEntity#getKey()} |
||||||
|
*/ |
||||||
|
public void stopTaskThread(String key) { |
||||||
|
if (mExePool.isShutdown()) { |
||||||
|
ALog.e(TAG, "线程池已经关闭"); |
||||||
|
return; |
||||||
|
} |
||||||
|
Set<Future> temp = mThreadTasks.get(getKey(key)); |
||||||
|
for (Future future : temp) { |
||||||
|
if (future.isDone() || future.isCancelled()) { |
||||||
|
continue; |
||||||
|
} |
||||||
|
future.cancel(true); |
||||||
|
} |
||||||
|
mThreadTasks.remove(key); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 重试线程任务 |
||||||
|
* |
||||||
|
* @param task 线程任务 |
||||||
|
*/ |
||||||
|
public void retryThread(AbsThreadTask task) { |
||||||
|
if (mExePool.isShutdown()) { |
||||||
|
ALog.e(TAG, "线程池已经关闭"); |
||||||
|
return; |
||||||
|
} |
||||||
|
mExePool.submit(task); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* map中的key |
||||||
|
* |
||||||
|
* @param key 任务的key{@link AbsTaskEntity#getKey()} |
||||||
|
* @return 转换后的map中的key |
||||||
|
*/ |
||||||
|
private String getKey(String key) { |
||||||
|
return CommonUtil.getStrMd5(key); |
||||||
|
} |
||||||
} |
} |
||||||
|
|||||||
Loading…
Reference in new issue