balancer: add support for http2 (#1548)

Author: dyc3

crates/ott-balancer/src/lib.rs

@@ -6,7 +6,8 @@ use balancer::{start_dispatcher, Balancer, BalancerContext};
 use clap::Parser;
 use futures_util::stream::FuturesUnordered;
 use futures_util::{FutureExt, StreamExt};
-use hyper::server::conn::http1;
+use hyper_util::rt::TokioExecutor;
+use hyper_util::server::conn::auto;
 use tokio::net::TcpListener;
 use tokio::sync::RwLock;
 use tracing::{error, info};
@@ -185,9 +186,8 @@ pub async fn run() -> anyhow::Result<()> {
         let task = tokio::task::Builder::new()
             .name("serve http")
             .spawn(async move {
-                let conn = http1::Builder::new()
-                    .serve_connection(io, service)
-                    .with_upgrades();
+                let serve = auto::Builder::new(TokioExecutor::new());
+                let conn = serve.serve_connection_with_upgrades(io, service);
                 if let Err(err) = conn.await {
                     error!("Error serving connection: {:?}", err);
                 }

crates/ott-balancer/src/service.rs

@@ -62,6 +62,7 @@ impl Service<Request<IncomingBody>> for BalancerService {
             Level::INFO,
             "http_request",
             request_id = request_id,
+            version = ?req.version(),
             method = %req.method(),
             path = %req.uri().path(),
             handler = field::Empty,

deploy/fly.prod.balancer.toml

@@ -17,6 +17,9 @@ force_https = true
 # auto_start_machines = true
 min_machines_running = 1
 
+[http_service.http_options]
+h2_backend = true
+
 [[http_service.checks]]
 grace_period = "10s"
 interval = "30s"

deploy/fly.staging.balancer.toml

@@ -17,6 +17,9 @@ force_https = true
 # auto_start_machines = true
 min_machines_running = 1
 
+[http_service.http_options]
+h2_backend = true
+
 [[http_service.checks]]
 grace_period = "10s"
 interval = "30s"