writeHttp2StreamFrame() — netty Function Reference

Source Code

codec-http2/src/main/java/io/netty/handler/codec/http2/AbstractHttp2StreamChannel.java lines 1074–1113

        private void writeHttp2StreamFrame(Http2StreamFrame frame, final ChannelPromise promise) {
            if (!firstFrameWritten && !isStreamIdValid(stream().id()) && !(frame instanceof Http2HeadersFrame)) {
                ReferenceCountUtil.release(frame);
                promise.setFailure(
                    new IllegalArgumentException("The first frame must be a headers frame. Was: "
                        + frame.name()));
                return;
            }

            final boolean firstWrite;
            if (firstFrameWritten) {
                firstWrite = false;
            } else {
                firstWrite = firstFrameWritten = true;
            }

            // Let's keep track of what we send as the stream state itself will only be updated once the frame
            // was written which might cause problems if we try to close the channel in a write future.
            sentEndOfStream |= isEndOfStream(frame);
            ChannelFuture f = write0(parentContext(), frame);
            if (f.isDone()) {
                if (firstWrite) {
                    firstWriteComplete(f, promise);
                } else {
                    writeComplete(f, promise);
                }
            } else {
                final long bytes = FlowControlledFrameSizeEstimator.HANDLE_INSTANCE.size(frame);
                incrementPendingOutboundBytes(bytes, false);
                f.addListener(future -> {
                    if (firstWrite) {
                        firstWriteComplete(future, promise);
                    } else {
                        writeComplete(future, promise);
                    }
                    decrementPendingOutboundBytes(bytes, false);
                });
                writeDoneAndNoFlush = true;
            }
        }