提交 0605c0f3 编写于 作者: V Violeta Georgieva

Make AbstractResponseBodySubscriber.onSubscribe thread-safe

When there are simultaneous invocations of onSubscribe, only the first one should succeed, the rest should cancel the provided subscriptions
上级 11ed847a
......@@ -18,6 +18,7 @@ package org.springframework.http.server.reactive;
import java.io.IOException;
import java.nio.channels.Channel;
import java.util.Objects;
import java.util.concurrent.atomic.AtomicReference;
import javax.servlet.WriteListener;
......@@ -169,11 +170,13 @@ abstract class AbstractResponseBodySubscriber implements Subscriber<DataBuffer>
@Override
void onSubscribe(AbstractResponseBodySubscriber subscriber,
Subscription subscription) {
if (BackpressureUtils.validate(subscriber.subscription, subscription)) {
Objects.requireNonNull(subscription, "Subscription cannot be null");
if (subscriber.changeState(this, REQUESTED)) {
subscriber.subscription = subscription;
if (subscriber.changeState(this, REQUESTED)) {
subscription.request(1);
}
subscription.request(1);
}
else {
super.onSubscribe(subscriber, subscription);
}
}
},
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册