Skip to content

Commit

Permalink
feat(transport): Expose hyper's H2 adaptive window on server (#1071)
Browse files Browse the repository at this point in the history
Co-authored-by: Adam Chalmers <adamschalmers@gmail.com>
  • Loading branch information
adamchalmers and Adam Chalmers committed Sep 6, 2022
1 parent 2a83f8b commit 919d28b
Showing 1 changed file with 16 additions and 0 deletions.
16 changes: 16 additions & 0 deletions tonic/src/transport/server/mod.rs
Expand Up @@ -90,6 +90,7 @@ pub struct Server<L = Identity> {
tcp_nodelay: bool,
http2_keepalive_interval: Option<Duration>,
http2_keepalive_timeout: Option<Duration>,
http2_adaptive_window: Option<bool>,
max_frame_size: Option<u32>,
accept_http1: bool,
service_builder: ServiceBuilder<L>,
Expand All @@ -110,6 +111,7 @@ impl Default for Server<Identity> {
tcp_nodelay: false,
http2_keepalive_interval: None,
http2_keepalive_timeout: None,
http2_adaptive_window: None,
max_frame_size: None,
accept_http1: false,
service_builder: Default::default(),
Expand Down Expand Up @@ -258,6 +260,17 @@ impl<L> Server<L> {
}
}

/// Sets whether to use an adaptive flow control. Defaults to false.
/// Enabling this will override the limits set in http2_initial_stream_window_size and
/// http2_initial_connection_window_size.
#[must_use]
pub fn http2_adaptive_window(self, enabled: Option<bool>) -> Self {
Server {
http2_adaptive_window: enabled,
..self
}
}

/// Set whether TCP keepalive messages are enabled on accepted connections.
///
/// If `None` is specified, keepalive is disabled, otherwise the duration
Expand Down Expand Up @@ -439,6 +452,7 @@ impl<L> Server<L> {
tcp_nodelay: self.tcp_nodelay,
http2_keepalive_interval: self.http2_keepalive_interval,
http2_keepalive_timeout: self.http2_keepalive_timeout,
http2_adaptive_window: self.http2_adaptive_window,
max_frame_size: self.max_frame_size,
accept_http1: self.accept_http1,
}
Expand Down Expand Up @@ -476,6 +490,7 @@ impl<L> Server<L> {
let http2_keepalive_timeout = self
.http2_keepalive_timeout
.unwrap_or_else(|| Duration::new(DEFAULT_HTTP2_KEEPALIVE_TIMEOUT_SECS, 0));
let http2_adaptive_window = self.http2_adaptive_window;

let svc = self.service_builder.service(svc);

Expand All @@ -497,6 +512,7 @@ impl<L> Server<L> {
.http2_max_concurrent_streams(max_concurrent_streams)
.http2_keep_alive_interval(http2_keepalive_interval)
.http2_keep_alive_timeout(http2_keepalive_timeout)
.http2_adaptive_window(http2_adaptive_window.unwrap_or_default())
.http2_max_frame_size(max_frame_size);

if let Some(signal) = signal {
Expand Down

0 comments on commit 919d28b

Please sign in to comment.