-
Notifications
You must be signed in to change notification settings - Fork 7.6k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #869 from benjchristensen/subscribeOn+groupBy
subscribeOn + groupBy
- Loading branch information
Showing
8 changed files
with
581 additions
and
84 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
193 changes: 193 additions & 0 deletions
193
rxjava-core/src/main/java/rx/operators/BufferUntilSubscriber.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,193 @@ | ||
| /** | ||
| * Copyright 2014 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 rx.operators; | ||
|
|
||
| import java.util.LinkedList; | ||
| import java.util.Queue; | ||
| import rx.Subscriber; | ||
| import rx.subscriptions.CompositeSubscription; | ||
|
|
||
| /** | ||
| * Buffers the incoming events until notified, then replays the | ||
| * buffered events and continues as a simple pass-through subscriber. | ||
| * @param <T> the streamed value type | ||
| */ | ||
| public class BufferUntilSubscriber<T> extends Subscriber<T> { | ||
| /** The actual subscriber. */ | ||
| private final Subscriber<? super T> actual; | ||
| /** Indicate the pass-through mode. */ | ||
| private volatile boolean passthroughMode; | ||
| /** Protect mode transition. */ | ||
| private final Object gate = new Object(); | ||
| /** The buffered items. */ | ||
| private final Queue<Object> queue = new LinkedList<Object>(); | ||
| /** The queue capacity. */ | ||
| private final int capacity; | ||
| /** Null sentinel (in case queue type is changed). */ | ||
| private static final Object NULL_SENTINEL = new Object(); | ||
| /** Complete sentinel. */ | ||
| private static final Object COMPLETE_SENTINEL = new Object(); | ||
| /** | ||
| * Container for an onError event. | ||
| */ | ||
| private static final class ErrorSentinel { | ||
| final Throwable t; | ||
|
|
||
| public ErrorSentinel(Throwable t) { | ||
| this.t = t; | ||
| } | ||
|
|
||
| } | ||
| /** | ||
| * Constructor that wraps the actual subscriber and shares its subscription. | ||
| * @param capacity the queue capacity to accept before blocking, negative value indicates an unbounded queue | ||
| * @param actual | ||
| */ | ||
| public BufferUntilSubscriber(int capacity, Subscriber<? super T> actual) { | ||
| super(actual); | ||
| this.actual = actual; | ||
| this.capacity = capacity; | ||
| } | ||
| /** | ||
| * Constructor that wraps the actual subscriber and uses the given composite | ||
| * subscription. | ||
| * @param capacity the queue capacity to accept before blocking, negative value indicates an unbounded queue | ||
| * @param actual | ||
| * @param cs | ||
| */ | ||
| public BufferUntilSubscriber(int capacity, Subscriber<? super T> actual, CompositeSubscription cs) { | ||
| super(cs); | ||
| this.actual = actual; | ||
| this.capacity = capacity; | ||
| } | ||
|
|
||
| /** | ||
| * Call this method to replay the buffered events and continue as a pass-through subscriber. | ||
| * If already in pass-through mode, this method is a no-op. | ||
| */ | ||
| public void enterPassthroughMode() { | ||
| if (!passthroughMode) { | ||
| synchronized (gate) { | ||
| if (!passthroughMode) { | ||
| while (!queue.isEmpty()) { | ||
| Object o = queue.poll(); | ||
| if (!actual.isUnsubscribed()) { | ||
| if (o == NULL_SENTINEL) { | ||
| actual.onNext(null); | ||
| } else | ||
| if (o == COMPLETE_SENTINEL) { | ||
| actual.onCompleted(); | ||
| } else | ||
| if (o instanceof ErrorSentinel) { | ||
| actual.onError(((ErrorSentinel)o).t); | ||
| } else | ||
| if (o != null) { | ||
| @SuppressWarnings("unchecked") | ||
| T v = (T)o; | ||
| actual.onNext(v); | ||
| } else { | ||
| throw new NullPointerException(); | ||
| } | ||
| } | ||
| } | ||
| passthroughMode = true; | ||
| gate.notifyAll(); | ||
| } | ||
| } | ||
| } | ||
| } | ||
| @Override | ||
| public void onNext(T t) { | ||
| if (!passthroughMode) { | ||
| synchronized (gate) { | ||
| if (!passthroughMode) { | ||
| if (capacity < 0 || queue.size() < capacity) { | ||
| queue.offer(t != null ? t : NULL_SENTINEL); | ||
| return; | ||
| } | ||
| try { | ||
| while (!passthroughMode) { | ||
| gate.wait(); | ||
| } | ||
| if (actual.isUnsubscribed()) { | ||
| return; | ||
| } | ||
| } catch (InterruptedException ex) { | ||
| Thread.currentThread().interrupt(); | ||
| actual.onError(ex); | ||
| return; | ||
| } | ||
| } | ||
| } | ||
| } | ||
| actual.onNext(t); | ||
| } | ||
|
|
||
| @Override | ||
| public void onError(Throwable e) { | ||
| if (!passthroughMode) { | ||
| synchronized (gate) { | ||
| if (!passthroughMode) { | ||
| if (capacity < 0 || queue.size() < capacity) { | ||
| queue.offer(new ErrorSentinel(e)); | ||
| return; | ||
| } | ||
| try { | ||
| while (!passthroughMode) { | ||
| gate.wait(); | ||
| } | ||
| if (actual.isUnsubscribed()) { | ||
| return; | ||
| } | ||
| } catch (InterruptedException ex) { | ||
| Thread.currentThread().interrupt(); | ||
| actual.onError(ex); | ||
| return; | ||
| } | ||
| } | ||
| } | ||
| } | ||
| actual.onError(e); | ||
| } | ||
|
|
||
| @Override | ||
| public void onCompleted() { | ||
| if (!passthroughMode) { | ||
| synchronized (gate) { | ||
| if (!passthroughMode) { | ||
| if (capacity < 0 || queue.size() < capacity) { | ||
| queue.offer(COMPLETE_SENTINEL); | ||
| return; | ||
| } | ||
| try { | ||
| while (!passthroughMode) { | ||
| gate.wait(); | ||
| } | ||
| if (actual.isUnsubscribed()) { | ||
| return; | ||
| } | ||
| } catch (InterruptedException ex) { | ||
| Thread.currentThread().interrupt(); | ||
| actual.onError(ex); | ||
| return; | ||
| } | ||
| } | ||
| } | ||
| } | ||
| actual.onCompleted(); | ||
| } | ||
|
|
||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.