生产者消费者的例子,在生产者中有背压支持
此示例中的 TestProducer
在给定范围内生成 Integer
objects 并将它们推送到 Subscriber
。它扩展了 Flowable<Integer>
类。对于新订户,它创建一个 Subscription
对象,其 request(long)
方法用于创建和发布 Integer 值。
对于传递给 subscriber
的 Subscription
来说,重要的是可以在此 onNext()
调用中递归调用在订户上调用 onNext()
的 request()
方法。为防止堆栈溢出,所示的实现使用 outStandingRequests
计数器和 isProducing
标志。
class TestProducer extends Flowable<Integer> {
static final Logger logger = LoggerFactory.getLogger(TestProducer.class);
final int from, to;
public TestProducer(int from, int to) {
this.from = from;
this.to = to;
}
@Override
protected void subscribeActual(Subscriber<? super Integer> subscriber) {
subscriber.onSubscribe(new Subscription() {
/** the next value. */
public int next = from;
/** cancellation flag. */
private volatile boolean cancelled = false;
private volatile boolean isProducing = false;
private AtomicLong outStandingRequests = new AtomicLong(0);
@Override
public void request(long n) {
if (!cancelled) {
outStandingRequests.addAndGet(n);
// check if already fulfilling request to prevent call between request() an subscriber .onNext()
if (isProducing) {
return;
}
// start producing
isProducing = true;
while (outStandingRequests.get() > 0) {
if (next > to) {
logger.info("producer finished");
subscriber.onComplete();
break;
}
subscriber.onNext(next++);
outStandingRequests.decrementAndGet();
}
isProducing = false;
}
}
@Override
public void cancel() {
cancelled = true;
}
});
}
}
此示例中的 Consumer 扩展了 DefaultSubscriber<Integer>
,并且在开始和消费整数后请求下一个。在消耗整数值时,会有一点延迟,因此背压将为生产者建立。
class TestConsumer extends DefaultSubscriber<Integer> {
private static final Logger logger = LoggerFactory.getLogger(TestConsumer.class);
@Override
protected void onStart() {
request(1);
}
@Override
public void onNext(Integer i) {
logger.info("consuming {}", i);
if (0 == (i % 5)) {
try {
Thread.sleep(500);
} catch (InterruptedException ignored) {
// can be ignored, just used for pausing
}
}
request(1);
}
@Override
public void onError(Throwable throwable) {
logger.error("error received", throwable);
}
@Override
public void onComplete() {
logger.info("consumer finished");
}
}
在以下测试类的主要方法中,创建并连接了生产者和消费者:
public static void main(String[] args) {
try {
final TestProducer testProducer = new TestProducer(1, 1_000);
final TestConsumer testConsumer = new TestConsumer();
testProducer
.subscribeOn(Schedulers.computation())
.observeOn(Schedulers.single())
.blockingSubscribe(testConsumer);
} catch (Throwable t) {
t.printStackTrace();
}
}
运行该示例时,日志文件显示使用者连续运行,而生成器仅在需要重新填充 rxjava2 的内部 Flowable 缓冲区时才会生效。