package com.videogo.ui.common;
|
|
import java.util.ArrayDeque;
|
import java.util.concurrent.BlockingQueue;
|
import java.util.concurrent.Callable;
|
import java.util.concurrent.CancellationException;
|
import java.util.concurrent.ExecutionException;
|
import java.util.concurrent.Executor;
|
import java.util.concurrent.FutureTask;
|
import java.util.concurrent.LinkedBlockingQueue;
|
import java.util.concurrent.ThreadFactory;
|
import java.util.concurrent.ThreadPoolExecutor;
|
import java.util.concurrent.TimeUnit;
|
import java.util.concurrent.TimeoutException;
|
import java.util.concurrent.atomic.AtomicBoolean;
|
import java.util.concurrent.atomic.AtomicInteger;
|
|
import android.os.AsyncTask;
|
import android.os.Handler;
|
import android.os.Looper;
|
import android.os.Message;
|
import android.os.Process;
|
|
public abstract class HikAsyncTask<Params, Progress, Result> {
|
|
private static final String LOG_TAG = HikAsyncTask.class.getSimpleName();
|
|
private static final int CPU_COUNT = Runtime.getRuntime().availableProcessors();
|
private static final int CORE_POOL_SIZE = CPU_COUNT + 1;
|
private static final int MAXIMUM_POOL_SIZE = CPU_COUNT * 2 + 1;
|
private static final int KEEP_ALIVE = 1;
|
|
private static final ThreadFactory sThreadFactory = new ThreadFactory() {
|
private final AtomicInteger mCount = new AtomicInteger(1);
|
|
public Thread newThread(Runnable r) {
|
return new Thread(r, "AsyncTask #" + mCount.getAndIncrement());
|
}
|
};
|
|
private static final BlockingQueue<Runnable> sPoolWorkQueue = new LinkedBlockingQueue<Runnable>(128);
|
|
public static final Executor THREAD_POOL_EXECUTOR = new ThreadPoolExecutor(CORE_POOL_SIZE, MAXIMUM_POOL_SIZE,
|
KEEP_ALIVE, TimeUnit.SECONDS, sPoolWorkQueue, sThreadFactory);
|
|
public static final Executor SERIAL_EXECUTOR = new SerialExecutor();
|
|
private static final int MESSAGE_POST_RESULT = 0x1;
|
private static final int MESSAGE_POST_PROGRESS = 0x2;
|
|
private static InternalHandler sHandler;
|
|
private final WorkerRunnable<Params, Result> mWorker;
|
private final FutureTask<Result> mFuture;
|
|
private volatile Status mStatus = Status.PENDING;
|
|
private final AtomicBoolean mCancelled = new AtomicBoolean();
|
private final AtomicBoolean mTaskInvoked = new AtomicBoolean();
|
|
private static class SerialExecutor implements Executor {
|
final ArrayDeque<Runnable> mTasks = new ArrayDeque<Runnable>();
|
Runnable mActive;
|
|
public synchronized void execute(final Runnable r) {
|
mTasks.offer(new Runnable() {
|
public void run() {
|
try {
|
r.run();
|
} finally {
|
scheduleNext();
|
}
|
}
|
});
|
if (mActive == null) {
|
scheduleNext();
|
}
|
}
|
|
protected synchronized void scheduleNext() {
|
if ((mActive = mTasks.poll()) != null) {
|
THREAD_POOL_EXECUTOR.execute(mActive);
|
}
|
}
|
}
|
|
public enum Status {
|
PENDING,
|
RUNNING,
|
FINISHED,
|
}
|
|
private static Handler getHandler() {
|
synchronized (HikAsyncTask.class) {
|
if (sHandler == null) {
|
sHandler = new InternalHandler();
|
}
|
return sHandler;
|
}
|
}
|
|
public HikAsyncTask() {
|
mWorker = new WorkerRunnable<Params, Result>() {
|
public Result call() throws Exception {
|
mTaskInvoked.set(true);
|
|
Process.setThreadPriority(Process.THREAD_PRIORITY_BACKGROUND);
|
// noinspection unchecked
|
return postResult(doInBackground(mParams));
|
}
|
};
|
|
mFuture = new FutureTask<Result>(mWorker) {
|
@Override
|
protected void done() {
|
try {
|
postResultIfNotInvoked(get());
|
} catch (InterruptedException e) {
|
android.util.Log.w(LOG_TAG, e);
|
} catch (ExecutionException e) {
|
throw new RuntimeException("An error occured while executing doInBackground()", e.getCause());
|
} catch (CancellationException e) {
|
postResultIfNotInvoked(null);
|
}
|
}
|
};
|
}
|
|
private void postResultIfNotInvoked(Result result) {
|
final boolean wasTaskInvoked = mTaskInvoked.get();
|
if (!wasTaskInvoked) {
|
postResult(result);
|
}
|
}
|
|
private Result postResult(Result result) {
|
@SuppressWarnings("unchecked")
|
Message message = getHandler().obtainMessage(MESSAGE_POST_RESULT, new AsyncTaskResult<Result>(this, result));
|
message.sendToTarget();
|
return result;
|
}
|
|
public final Status getStatus() {
|
return mStatus;
|
}
|
|
protected abstract Result doInBackground(Params... params);
|
|
protected void onPreExecute() {
|
}
|
|
protected void onPostExecute(Result result) {
|
}
|
|
protected void onProgressUpdate(Progress... values) {
|
}
|
|
protected void onCancelled(Result result) {
|
onCancelled();
|
}
|
|
protected void onCancelled() {
|
}
|
|
public final boolean isCancelled() {
|
return mCancelled.get();
|
}
|
|
public final boolean cancel(boolean mayInterruptIfRunning) {
|
mCancelled.set(true);
|
return mFuture.cancel(mayInterruptIfRunning);
|
}
|
|
public final Result get() throws InterruptedException, ExecutionException {
|
return mFuture.get();
|
}
|
|
public final Result get(long timeout, TimeUnit unit) throws InterruptedException, ExecutionException,
|
TimeoutException {
|
return mFuture.get(timeout, unit);
|
}
|
|
public final HikAsyncTask<Params, Progress, Result> execute(Params... params) {
|
return executeOnExecutor(THREAD_POOL_EXECUTOR, params);
|
}
|
|
public final HikAsyncTask<Params, Progress, Result> serialExecute(Params... params) {
|
return executeOnExecutor(SERIAL_EXECUTOR, params);
|
}
|
|
public final HikAsyncTask<Params, Progress, Result> executeOnExecutor(Executor exec, Params... params) {
|
if (mStatus != Status.PENDING) {
|
switch (mStatus) {
|
case RUNNING:
|
throw new IllegalStateException("Cannot execute task:" + " the task is already running.");
|
case FINISHED:
|
throw new IllegalStateException("Cannot execute task:" + " the task has already been executed "
|
+ "(a task can be executed only once)");
|
}
|
}
|
|
mStatus = Status.RUNNING;
|
|
onPreExecute();
|
|
mWorker.mParams = params;
|
exec.execute(mFuture);
|
|
return this;
|
}
|
|
public static void execute(boolean serial, Runnable runnable) {
|
THREAD_POOL_EXECUTOR.execute(runnable);
|
}
|
|
public static void serialExecute(Runnable runnable) {
|
SERIAL_EXECUTOR.execute(runnable);
|
}
|
|
protected final void publishProgress(Progress... values) {
|
if (!isCancelled()) {
|
getHandler().obtainMessage(MESSAGE_POST_PROGRESS, new AsyncTaskResult<Progress>(this, values))
|
.sendToTarget();
|
}
|
}
|
|
private void finish(Result result) {
|
if (isCancelled()) {
|
onCancelled(result);
|
} else {
|
onPostExecute(result);
|
}
|
mStatus = Status.FINISHED;
|
}
|
|
private static class InternalHandler extends Handler {
|
public InternalHandler() {
|
super(Looper.getMainLooper());
|
}
|
|
@SuppressWarnings({
|
"unchecked", "RawUseOfParameterizedType"})
|
@Override
|
public void handleMessage(Message msg) {
|
AsyncTaskResult<?> result = (AsyncTaskResult<?>) msg.obj;
|
if (msg.what==MESSAGE_POST_RESULT) {
|
result.mTask.finish(result.mData[0]);
|
}else if (msg.what==MESSAGE_POST_PROGRESS) {
|
result.mTask.onProgressUpdate(result.mData);
|
}
|
}
|
}
|
|
private static abstract class WorkerRunnable<Params, Result> implements Callable<Result> {
|
Params[] mParams;
|
}
|
|
@SuppressWarnings({"RawUseOfParameterizedType"})
|
private static class AsyncTaskResult<Data> {
|
final HikAsyncTask mTask;
|
final Data[] mData;
|
|
AsyncTaskResult(HikAsyncTask task, Data... data) {
|
mTask = task;
|
mData = data;
|
}
|
}
|
}
|
