AriaLyy
8 years ago
parent
2fabf1a168
commit
ad6ea4fac9
@ -0,0 +1,190 @@ |
||||
/* |
||||
* 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.scheduler; |
||||
|
||||
import android.os.CountDownTimer; |
||||
import android.os.Message; |
||||
import android.util.Log; |
||||
import com.arialyy.aria.core.AriaManager; |
||||
import com.arialyy.aria.core.inf.AbsEntity; |
||||
import com.arialyy.aria.core.inf.AbsTaskEntity; |
||||
import com.arialyy.aria.core.inf.IEntity; |
||||
import com.arialyy.aria.core.inf.ITask; |
||||
import com.arialyy.aria.core.queue.ITaskQueue; |
||||
import java.util.Iterator; |
||||
import java.util.Map; |
||||
import java.util.Set; |
||||
import java.util.concurrent.ConcurrentHashMap; |
||||
|
||||
/** |
||||
* Created by AriaL on 2017/6/4. |
||||
*/ |
||||
public abstract class AbsSchedulers<TASK_ENTITY extends AbsTaskEntity, ENTITY extends AbsEntity, TASK extends ITask<ENTITY>, QUEUE extends ITaskQueue<TASK, TASK_ENTITY, ENTITY>> |
||||
implements ISchedulers<TASK> { |
||||
private static final String TAG = "AbsSchedulers"; |
||||
|
||||
protected QUEUE mQueue; |
||||
|
||||
private Map<String, IDownloadSchedulerListener<TASK>> mSchedulerListeners = |
||||
new ConcurrentHashMap<>(); |
||||
|
||||
@Override |
||||
public void addSchedulerListener(String targetName, ISchedulerListener<TASK> schedulerListener) { |
||||
mSchedulerListeners.put(targetName, (IDownloadSchedulerListener<TASK>) schedulerListener); |
||||
} |
||||
|
||||
@Override public void removeSchedulerListener(String targetName, |
||||
ISchedulerListener<TASK> schedulerListener) { |
||||
//该内存泄露解决方案：http://stackoverflow.com/questions/14585829/how-safe-is-to-delete-already-removed-concurrenthashmap-element
|
||||
for (Iterator<Map.Entry<String, IDownloadSchedulerListener<TASK>>> iter = |
||||
mSchedulerListeners.entrySet().iterator(); iter.hasNext(); ) { |
||||
Map.Entry<String, IDownloadSchedulerListener<TASK>> entry = iter.next(); |
||||
if (entry.getKey().equals(targetName)) iter.remove(); |
||||
} |
||||
} |
||||
|
||||
@Override public boolean handleMessage(Message msg) { |
||||
TASK task = (TASK) msg.obj; |
||||
if (task == null) { |
||||
Log.e(TAG, "请传入下载任务"); |
||||
return true; |
||||
} |
||||
callback(msg.what, task); |
||||
ENTITY entity = task.getEntity(); |
||||
switch (msg.what) { |
||||
case STOP: |
||||
case CANCEL: |
||||
mQueue.removeTask(entity); |
||||
if (mQueue.executePoolSize() < AriaManager.getInstance(AriaManager.APP) |
||||
.getUploadConfig() |
||||
.getMaxTaskNum()) { |
||||
startNextTask(); |
||||
} |
||||
break; |
||||
case COMPLETE: |
||||
mQueue.removeTask(entity); |
||||
startNextTask(); |
||||
break; |
||||
case FAIL: |
||||
handleFailTask(task); |
||||
break; |
||||
} |
||||
return true; |
||||
} |
||||
|
||||
/** |
||||
* 回调 |
||||
* |
||||
* @param state 状态 |
||||
*/ |
||||
private void callback(int state, TASK task) { |
||||
if (mSchedulerListeners.size() > 0) { |
||||
Set<String> keys = mSchedulerListeners.keySet(); |
||||
for (String key : keys) { |
||||
callback(state, task, mSchedulerListeners.get(key)); |
||||
} |
||||
} |
||||
} |
||||
|
||||
private void callback(int state, TASK task, IDownloadSchedulerListener<TASK> listener) { |
||||
if (listener != null) { |
||||
if (task == null) { |
||||
Log.e(TAG, "TASK 为null，回调失败"); |
||||
return; |
||||
} |
||||
switch (state) { |
||||
case PRE: |
||||
listener.onPre(task); |
||||
break; |
||||
case POST_PRE: |
||||
listener.onTaskPre(task); |
||||
break; |
||||
case RUNNING: |
||||
listener.onTaskRunning(task); |
||||
break; |
||||
case START: |
||||
listener.onTaskStart(task); |
||||
break; |
||||
case STOP: |
||||
listener.onTaskStop(task); |
||||
break; |
||||
case RESUME: |
||||
listener.onTaskResume(task); |
||||
break; |
||||
case CANCEL: |
||||
listener.onTaskCancel(task); |
||||
break; |
||||
case COMPLETE: |
||||
listener.onTaskComplete(task); |
||||
break; |
||||
case FAIL: |
||||
listener.onTaskFail(task); |
||||
break; |
||||
case SUPPORT_BREAK_POINT: |
||||
listener.onNoSupportBreakPoint(task); |
||||
break; |
||||
} |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* 处理下载任务下载失败的情形 |
||||
* |
||||
* @param task 下载任务 |
||||
*/ |
||||
private void handleFailTask(final TASK task) { |
||||
final long interval = |
||||
AriaManager.getInstance(AriaManager.APP).getUploadConfig().getReTryInterval(); |
||||
final int reTryNum = AriaManager.getInstance(AriaManager.APP).getUploadConfig().getReTryNum(); |
||||
|
||||
CountDownTimer timer = new CountDownTimer(interval, 1000) { |
||||
@Override public void onTick(long millisUntilFinished) { |
||||
|
||||
} |
||||
|
||||
@Override public void onFinish() { |
||||
ENTITY entity = task.getEntity(); |
||||
if (entity.getFailNum() < reTryNum) { |
||||
TASK task = mQueue.getTask(entity); |
||||
mQueue.reTryStart(task); |
||||
try { |
||||
Thread.sleep(interval); |
||||
} catch (InterruptedException e) { |
||||
e.printStackTrace(); |
||||
} |
||||
} else { |
||||
mQueue.removeTask(entity); |
||||
startNextTask(); |
||||
} |
||||
} |
||||
}; |
||||
timer.start(); |
||||
} |
||||
|
||||
/** |
||||
* 启动下一个任务，条件：任务停止，取消下载，任务完成 |
||||
*/ |
||||
private void startNextTask() { |
||||
TASK newTask = mQueue.getNextTask(); |
||||
if (newTask == null) { |
||||
Log.w(TAG, "没有下一任务"); |
||||
return; |
||||
} |
||||
if (newTask.getEntity().getState() == IEntity.STATE_WAIT) { |
||||
mQueue.startTask(newTask); |
||||
} |
||||
} |
||||
} |
||||
Loading…
Reference in new issue