All Downloads are FREE. Search and download functionalities are using the official Maven repository.

hu.akarnokd.rxjava2.internal.disposables.NbpFullArbiter Maven / Gradle / Ivy

There is a newer version: 2.0.0-RC3
Show newest version
/**
 * Copyright 2015 David Karnok and Netflix, Inc.
 * 
 * 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 hu.akarnokd.rxjava2.internal.disposables;

import java.util.concurrent.atomic.AtomicIntegerFieldUpdater;

import hu.akarnokd.rxjava2.NbpObservable.NbpSubscriber;
import hu.akarnokd.rxjava2.disposables.Disposable;
import hu.akarnokd.rxjava2.internal.queue.SpscLinkedArrayQueue;
import hu.akarnokd.rxjava2.internal.util.NotificationLite;
import hu.akarnokd.rxjava2.plugins.RxJavaPlugins;

/**
 * Performs full arbitration of Subscriber events with strict drain (i.e., old emissions of another
 * subscriber are dropped).
 *
 * @param  the value type
 */
public final class NbpFullArbiter extends FullArbiterPad1 implements Disposable {
    final NbpSubscriber actual;
    final SpscLinkedArrayQueue queue;

    volatile Disposable s;
    static final Disposable INITIAL = new Disposable() {
        @Override
        public void dispose() { }
    };
    
    
    Disposable resource;

    volatile boolean cancelled;

    public NbpFullArbiter(NbpSubscriber actual, Disposable resource, int capacity) {
        this.actual = actual;
        this.resource = resource;
        this.queue = new SpscLinkedArrayQueue(capacity);
        this.s = INITIAL;
    }

    @Override
    public void dispose() {
        if (!cancelled) {
            cancelled = true;
            disposeResource();
        }
    }
    
    void disposeResource() {
        Disposable d = resource;
        resource = null;
        if (d != null) {
            d.dispose();
        }
    }

    public boolean setSubscription(Disposable s) {
        if (cancelled) {
            return false;
        }

        queue.offer(this.s, NotificationLite.disposable(s));
        drain();
        return true;
    }

    public boolean onNext(T value, Disposable s) {
        if (cancelled) {
            return false;
        }

        queue.offer(s, NotificationLite.next(value));
        drain();
        return true;
    }

    public void onError(Throwable value, Disposable s) {
        if (cancelled) {
            RxJavaPlugins.onError(value);
            return;
        }
        queue.offer(s, NotificationLite.error(value));
        drain();
    }

    public void onComplete(Disposable s) {
        queue.offer(s, NotificationLite.complete());
        drain();
    }

    void drain() {
        if (WIP.getAndIncrement(this) != 0) {
            return;
        }
        
        int missed = 1;
        
        final SpscLinkedArrayQueue q = queue;
        final NbpSubscriber a = actual;
        
        for (;;) {
            
            for (;;) {
                Object o = q.peek();
                
                if (o == null) {
                    break;
                }
                
                q.poll();
                Object v = q.poll();
                
                if (o != s) {
                    continue;
                } else
                if (NotificationLite.isDisposable(v)) {
                    Disposable next = NotificationLite.getDisposable(v);
                    if (s != null) {
                        s.dispose();
                    }
                    s = next;
                } else 
                if (NotificationLite.isError(v)) {
                    q.clear();
                    disposeResource();
                    
                    Throwable ex = NotificationLite.getError(v);
                    if (!cancelled) {
                        cancelled = true;
                        a.onError(ex);
                    } else {
                        RxJavaPlugins.onError(ex);
                    }
                } else
                if (NotificationLite.isComplete(v)) {
                    q.clear();
                    disposeResource();

                    if (!cancelled) {
                        cancelled = true;
                        a.onComplete();
                    }
                } else {
                    a.onNext(NotificationLite.getValue(v));
                }
            }
            
            missed = WIP.addAndGet(this, -missed);
            if (missed == 0) {
                break;
            }
        }
    }
}

/** Pads the object header away. */
class FullArbiterPad0 {
    volatile long p1a, p2a, p3a, p4a, p5a, p6a, p7a;
    volatile long p8a, p9a, p10a, p11a, p12a, p13a, p14a, p15a;
}

/** The work-in-progress counter. */
class FullArbiterWip extends FullArbiterPad0 {
    volatile int wip;
    static final AtomicIntegerFieldUpdater WIP =
    AtomicIntegerFieldUpdater.newUpdater(FullArbiterWip.class, "wip");
}

/** Pads the wip counter away. */
class FullArbiterPad1 extends FullArbiterWip {
    volatile long p1b, p2b, p3b, p4b, p5b, p6b, p7b;
    volatile long p8b, p9b, p10b, p11b, p12b, p13b, p14b, p15b;
}