AsyncTask源码模拟

这章将研究AsyncTask的实现原理,并自己尝试编写一个相同的异步操作类
这章你将学习到的关键词:
AsyncTask

线程相关:
ThreadPoolExector,Exector, ArrayDeque双向队列

任务操作相关:
FutureTask<?>(Callable<?> r) [?表示泛型]

其他类:
AtomicBoolean或者Atomic* [*表示基本数据类型, Atomic表示自增长]

下面我们便开始着手编写我们的异步线程并结合AsyncTask源码

需求:
1.我们要写一个异步线程,这个线程需要写在线程池里面
2.我们的线程需要传入自定义类型,回调能够有自定义类型参数回调
3.我们的线程必须要能有相应的回调,以及能够取消

1.编写异步线程
你需要一个线程池TheadPoolExector, 你需要一个Exector来实现runnable

一般你可能会这样写:

ExecutorService service = Executors.newFixedThreadPool(3);

这样就创建了Exector与TheadPoolExector, 但是阿里的编码规范里面推荐手动创建线程池,那么我们可以自定义实现Executor接口来创建我们的线程池TheadPoolExector,这里直接上代码:

/**
 * Created by Qzhu on 2018/5/18.
 */
public class ThreadExector implements Executor{
    private static final int CPU_COUNT = Runtime.getRuntime().availableProcessors();
    private static final int CORE_POOL_SIZE = Math.max(2, Math.min(CPU_COUNT - 1, 4));
    private static final int MAXIMUM_POOL_SIZE = CPU_COUNT * 2 + 1;
    private static final int KEEP_ALIVE_SECONDS = 30;

    private static final BlockingQueue<Runnable> sPoolWorkQueue =
            new LinkedBlockingQueue<Runnable>(128);

    private static final ThreadFactory sThreadFactory = new ThreadFactory() {
        private final AtomicInteger atomicInteger = new AtomicInteger(1);

        @Override
        public Thread newThread(Runnable r) {
            return new Thread("ThreadExector #" + atomicInteger.getAndIncrement());
        }
    };

    static {
        ThreadPoolExecutor threadPoolExecutor = new ThreadPoolExecutor(
                CORE_POOL_SIZE, MAXIMUM_POOL_SIZE, KEEP_ALIVE_SECONDS, TimeUnit.SECONDS,
                sPoolWorkQueue, sThreadFactory);
        threadPoolExecutor.allowCoreThreadTimeOut(true);
        THREAD_POOL_EXECUTOR = threadPoolExecutor;
    }

    /**
     * An {@link Executor} that can be used to execute tasks in parallel.
     */
    public static final Executor THREAD_POOL_EXECUTOR;

    final ArrayDeque<Runnable> queue = new ArrayDeque<>();
    Runnable point = null;

    @Override
    public void execute(final Runnable command) {
        queue.offer(new Runnable() {
            @Override
            public void run() {
                try {
                    command.run();
                }finally {
                    scheduleNext();
                }
            }
        });

        if(point == null) {
            scheduleNext();
        }
    }

    private void scheduleNext() {
        if((point = queue.poll()) != null) {
            THREAD_POOL_EXECUTOR.execute(point);
        }
    }
}

类解析:
我们可以看到这是一个自定义的Exector实现类,里面初始化了ThreadPoolExector,定制了参数(初始化请自行百度~),实现的execute里面的runnable使用了双向队列ArrayDeque包装,这样可以不停取出里面的runnable并放入ThreadPoolExector执行,只要execute执行多少次都是排入队列并等待弹出执行(等下我们将分析怎样限制其之执行一次[其实只要加个状态就行啦])

以上我们便准备好了线程执行者,以及可以先这样用了

new ThreadExector().execute(Runnable r);

下面我们需要对参数Runnaber r进行包装使用的是FutureTask<?>(Callable<?> r),先来大概了解以下FutureTask<?>(Callable<?> r)这个类对Runnable的包装吧,FutureTask实现了Runnable将执行Run方法,然后里面执行Callable的call方法,或者其他情况将执行自身的done方法,比如取消或者异常等,而?泛型代表返回参数,这就符合条件2的自定义结果参数,下面缺少自定义入参,这一点只要添加一个泛型就行了
我们先实现FutureTask<?>(Callable<?> r)里面的Callable<?>

public abstract class WorkParam<R,P> implements Callable<R>{
    P[] param;
}

其中泛型R代表结果参数,P代表传入的参数
初始化如下:

private final AtomicBoolean mTaskInvoked = new AtomicBoolean(false);
private final AtomicBoolean mCancelled = new AtomicBoolean(false);

WorkParam worker = new WorkParam<Result,Params>() {
    @Override
    public Result call() throws Exception {
        Result r = null;
        mTaskInvoked.set(true);
        android.os.Process.setThreadPriority(android.os.Process.THREAD_PRIORITY_BACKGROUND);
        try {
            r = doInBackgroud(param);
        }catch (Exception e) {
            mCancelled.set(true);
            r = null;
        }finally {
            postResult(r);
        }
        Binder.flushPendingCommands();
        return r;
    }
};

这段代码编写参考AsyncTask,实现call的方法,在里面回调doInBackgroud执行具体异步操作,同名AsyncTask接口方法,而onPostExecute也是AsyncTask接口方法,这里先调用doInBackgroud放到后台让Exector工作,然后执行postResult(见下面具体实现),最后就是回调onPostExecute接口方法

以上是Callable<?> r的包装,还有一个FutureTask<?>如下:

futureTask = new FutureTask<Result>(worker){
    @Override
    protected void done() {
        super.done();
        try {
            postIfNotInvoke(get());
        } catch (InterruptedException e) {
            e.printStackTrace();
        } catch (ExecutionException e) {
            e.printStackTrace();
        }catch (CancellationException e) {
            postIfNotInvoke(null);
        }
    }
};
private void postIfNotInvoke(Result r){
    if(!mTaskInvoked.get()) {
        postResult(r);
    }
}
private Result postResult(Result result) {
    @SuppressWarnings("unchecked")
    Message message = innerHandller.obtainMessage(MESSAGE_POST_RESULT,
            new AsyncTaskResult<Result>(this, result));
    message.sendToTarget();
    return result;
}

//定义一个InnerHandler来处理线程切换回调
private static final int MESSAGE_POST_RESULT = 0x1;
private static final int MESSAGE_POST_PROGRESS = 0x2;
private Handler innerHandller = new Handler(Looper.getMainLooper()){
    @Override
    public void handleMessage(Message msg) {
        AsyncTaskResult<?> result = (AsyncTaskResult<?>) msg.obj;
        switch (msg.what) {
            case MESSAGE_POST_RESULT:
                result.mTask.onPostExecute(result.mData[0]);
                break;
            case MESSAGE_POST_PROGRESS:
                result.mTask.onProgressUpdate(result.mData);
                break;
            default:
                break;
        }
    }
};
private static class AsyncTaskResult<Data> {
    final AsyncTask mTask;
    final Data[] mData;

    AsyncTaskResult(AsyncTask task, Data... data) {
        mTask = task;
        mData = data;
    }
}

这个方法表示如果操作被其他行为中断,比如取消,执行postIfNotInvoke方法,而这个方法将判断线程是否执行过Callable<?>的call方法,里面的mTaskInvoked是一个标志,用于区分是否被执行过,如果执行过将回调postExecute,使用handler来切换到主线程回调,这样我们就实现了doInBackgroud与onPostExecute 回调
至于如何实现onpreExecute其实只要在execute方法里面调用就行的,如下:

public AsyncTask<Params, Process, Result> execute(Params... params){
    return executeOnExecutor(new ThreadExector(), params);
}

private final AsyncTask<Params, Process, Result> executeOnExecutor(Executor exec,Params... params){
    onPreExecute();
    worker.param = params;
    exec.execute(futureTask);
    return this;
}

关于实现进度更新,其实上面自定义的InnerHandler里面有切换到主线程回调跟新,只需要定义公开方法就行,如下:

protected final void publishProgress(Progress... values) {
    if (!isCancelled()) {
        getHandler().obtainMessage(MESSAGE_POST_PROGRESS,
                new AsyncTaskResult<Progress>(this, values)).sendToTarget();
    }
}
protected void onProgressUpdate(Progress... values) {
}

以上过程便是AsyncTask的实现流程,首先在构造函数里面初始化FutureTask<?>(Callable<?> r),然后执行execute的时候将准备好的线程池Exector来执行该FutureTask,回调对应实现体内部,线程的取消可以由FutureTask控制,使用几个AtomicBoolean作为标志位来控制结果的回调,状态是否取消等等,在Callable执行call方法里面可以使用InnerHandler来切换到主线程,这便是AsyncTask源码的实现方式

总结以下:
1.[重要]自定义Exector线程池(ArrayDeque双向无限扩充队列)
2.[重要]Runnable包装类FutureTask<?>(Callable<?> r)的使用,泛型的传递
3.线程切换回调方法InnerHandler的定义与参数的定义
4.Atomic*类型控制结果

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,294评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,493评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,790评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,595评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,718评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,906评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,053评论 3 410
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,797评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,250评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,570评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,711评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,388评论 4 332
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,018评论 3 316
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,796评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,023评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,461评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,595评论 2 350

推荐阅读更多精彩内容