Fork/Join框架之Fork、Join操作

Fork

Fork就是一个不断分枝的过程，在当前任务的基础上长出n多个子任务。

当一个ForkJoinTask任务调用fork()方法时，当前线程会把这个任务放入到queue数组的queueTop位置，然后执行以下两句代码：

if ((s -= queueBase) <= 2)    pool.signalWork();else if (s == m)    growQueue();

其中s=queueTop,m为数组length减1。else if部分，表示数组所有元素都满了，需要扩容，不难理解。if部分表示当数组元素比较少时（1或者2），就调用signalWork()方法。signalWork（）方法做了两件事：1、唤配当前线程;2、当没有活动线程时或者线程数较少时，添加新的线程。

Join

Join是一个不断等待，获取任务执行结果的过程。

private int doJoin() {    Thread t; ForkJoinWorkerThread w; int s; boolean completed;    if ((t = Thread.currentThread()) instanceof ForkJoinWorkerThread) {        if ((s = status) < 0)            return s;        if ((w = (ForkJoinWorkerThread)t).unpushTask(this)) {            try {                completed = exec();            } catch (Throwable rex) {                return setExceptionalCompletion(rex);            }            if (completed)                return setCompletion(NORMAL);        }        return w.joinTask(this);    }    else        return externalAwaitDone();}

(1)第4行，(s=status)<0表示这个任务被执行完，直接返回执行结果状态，上层捕获到状态后，决定是要获取结果还是进行错误处理；

(2)第6行，从queue中取出这个任务来执行，如果执行完了，就设置状态为NORMAL；

(3)前面unpushTask()方法在队列中没有这个任务时会返回false，15行调用joinTask等待这个任务完成。

(4)由于ForkJoinPool中有一个数组叫submissionQueue，通过submit方法调用而且非ForkJoinTask这种任务会被放到这个队列中。这种任务有可能被非ForkJoinWorkerThread线程执行，第18行表示如果是这种任务，等待它执行完成。

下面来看joinTask方法

final int joinTask(ForkJoinTask<?> joinMe) {    ForkJoinTask<?> prevJoin = currentJoin;    currentJoin = joinMe;    for (int s, retries = MAX_HELP;;) {        if ((s = joinMe.status) < 0) {            currentJoin = prevJoin;            return s;        }        if (retries > 0) {            if (queueTop != queueBase) {                if (!localHelpJoinTask(joinMe))                    retries = 0;           // cannot help            }            else if (retries == MAX_HELP >>> 1) {                --retries;                 // check uncommon case                if (tryDeqAndExec(joinMe) >= 0)                    Thread.yield();        // for politeness            }            else                retries = helpJoinTask(joinMe) ? MAX_HELP : retries - 1;        }        else {            retries = MAX_HELP;           // restart if not done            pool.tryAwaitJoin(joinMe);        }    }}

(1)这里有个常量MAX_HELP=16,表示帮助join的次数。第11行，queueTop!=queueBase表示本地队列中有任务，如果这个任务刚好在队首，则尝试自己执行；否则返回false。这时retries被设置为0，表示不能帮助,因为自已队列不为空，自己并不空闲。在下一次循环就会进入第24行，等待这个任务执行完成。

(2)第20行helpJoinTask（）方法返回false时，retries-1，连续8次都没有帮到忙，就会进入第14行，调用yield让权等待。没办法人口太差，想做点好事都不行，只有停下来休息一下。

(3)当执行到第20行，表示自己队列为空，可以去帮助这个任务了，下面来看是怎么帮助的？

outer:for (ForkJoinWorkerThread thread = this;;) {    // Try to find v, the stealer of task, by first using hint    ForkJoinWorkerThread v = ws[thread.stealHint & m];    if (v == null || v.currentSteal != task) {        for (int j = 0; ;) {        // search array            if ((v = ws[j]) != null && v.currentSteal == task) {                thread.stealHint = j;                break;              // save hint for next time            }            if (++j > m)                break outer;        // can't find stealer        }    }    // Try to help v, using specialized form of deqTask    for (;;) {        ForkJoinTask<?>[] q; int b, i;        if (joinMe.status < 0)            break outer;        if ((b = v.queueBase) == v.queueTop ||            (q = v.queue) == null ||            (i = (q.length-1) & b) < 0)            break;                  // empty        long u = (i << ASHIFT) + ABASE;        ForkJoinTask<?> t = q[i];        if (task.status < 0)            break outer;            // stale        if (t != null && v.queueBase == b &&            UNSAFE.compareAndSwapObject(q, u, t, null)) {            v.queueBase = b + 1;            v.stealHint = poolIndex;            ForkJoinTask<?> ps = currentSteal;            currentSteal = t;            t.doExec();            currentSteal = ps;            helped = true;        }    }    // Try to descend to find v's stealer    ForkJoinTask<?> next = v.currentJoin;    if (--levels > 0 && task.status >= 0 &&        next != null && next != task) {        task = next;        thread = v;    }}

(1)通过查看stealHint这个字段的注释可以知道，它表示最近一次谁来偷过我的queue中的任务。因此通过stealHint并不能找到当前任务被谁偷了？所以第4行v.currentSteal != task完全可能。这时还有一个办法找到这个任务被谁偷了，看看currentSteal这个字段的注释表示最近偷的哪个任务。这里扫描所有偷来的任务与当前任务比较，如果相等，就是这个线程偷的。如果这两种方法都不能找到小偷，只能等待了。

(2)当找到了小偷后，以其人之身还之其人之道，从小偷那里偷任务过来，相当于你和小偷共同执行你的任务，会加速你的任务完成。

(3)小偷也是爷，如果小偷也在等待一个任务完成，权利反转（小偷等待的这个任务做为当前任务，小偷扮演当事人角色把前面的流程走一遍），这是一个递归的过程。