foster/denoland-deno
1
0
Fork 0
mirror of https://github.com/denoland/deno.git synced 2024-11-21 15:04:11 -05:00
Code Issues Packages 1 Wiki Activity

fix(ext/node): send data frame with end_stream flag on _final call (#24147)

Browse source
This commit is contained in:
Satya Rohith 2024-06-10 20:00:56 +05:30 committed by GitHub
parent 3be0a1e8b4
commit 4fd3d5a86e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 21 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
ext/node/ops/http2.rs
View file

@ -344,6 +344,7 @@ pub async fn op_http2_client_send_data(
state: Rc<RefCell<OpState>>,
#[smi] stream_rid: ResourceId,
#[buffer] data: JsBuffer,
end_of_stream: bool,
) -> Result<(), AnyError> {
let resource = state
.borrow()
@ -351,8 +352,7 @@ pub async fn op_http2_client_send_data(
.get::<Http2ClientStream>(stream_rid)?;
let mut stream = RcRef::map(&resource, |r| &r.stream).borrow_mut().await;
// TODO(bartlomieju): handle end of stream
stream.send_data(data.to_vec().into(), false)?;
stream.send_data(data.to_vec().into(), end_of_stream)?;
Ok(())
}

23
ext/node/polyfills/http2.ts
View file

@ -978,7 +978,7 @@ export class ClientHttp2Stream extends Duplex {
return;
}
shutdownWritable(this, cb);
shutdownWritable(this, cb, this.#rid);
}
// TODO(bartlomieju): needs a proper cleanup
@ -1176,15 +1176,30 @@ export class ClientHttp2Stream extends Duplex {
}
}
function shutdownWritable(stream, callback) {
function shutdownWritable(stream, callback, streamRid) {
debugHttp2(">>> shutdownWritable", callback);
const state = stream[kState];
if (state.shutdownWritableCalled) {
debugHttp2(">>> shutdownWritable() already called");
return callback();
}
state.shutdownWritableCalled = true;
onStreamTrailers(stream);
callback();
if (state.flags & STREAM_FLAGS_HAS_TRAILERS) {
onStreamTrailers(stream);
callback();
} else {
op_http2_client_send_data(streamRid, new Uint8Array(), true)
.then(() => {
callback();
stream[kMaybeDestroy]();
core.tryClose(streamRid);
})
.catch((e) => {
callback(e);
core.tryClose(streamRid);
stream._destroy(e);
});
}
// TODO(bartlomieju): might have to add "finish" event listener here,
// check it.
}