Skip to content

Commit 2313682

Browse files
committed
Fix custom FeedableBodyGenerator support, close AsyncHttpClient#1028
1 parent 6be404f commit 2313682

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

client/src/main/java/org/asynchttpclient/netty/request/body/NettyBodyBody.java

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,9 @@
2929
import org.asynchttpclient.request.body.Body;
3030
import org.asynchttpclient.request.body.RandomAccessBody;
3131
import org.asynchttpclient.request.body.generator.BodyGenerator;
32-
import org.asynchttpclient.request.body.generator.SimpleFeedableBodyGenerator;
32+
import org.asynchttpclient.request.body.generator.FeedableBodyGenerator;
3333
import org.asynchttpclient.request.body.generator.FeedableBodyGenerator.FeedListener;
34+
import org.asynchttpclient.request.body.generator.ReactiveStreamsBodyGenerator;
3435

3536
public class NettyBodyBody implements NettyBody {
3637

@@ -67,8 +68,8 @@ public void write(final Channel channel, NettyResponseFuture<?> future) throws I
6768
msg = new BodyChunkedInput(body);
6869

6970
BodyGenerator bg = future.getTargetRequest().getBodyGenerator();
70-
if (bg instanceof SimpleFeedableBodyGenerator) {
71-
SimpleFeedableBodyGenerator.class.cast(bg).setListener(new FeedListener() {
71+
if (bg instanceof FeedableBodyGenerator && !(bg instanceof ReactiveStreamsBodyGenerator)) {
72+
FeedableBodyGenerator.class.cast(bg).setListener(new FeedListener() {
7273
@Override
7374
public void onContentAdded() {
7475
channel.pipeline().get(ChunkedWriteHandler.class).resumeTransfer();

0 commit comments

Comments
 (0)