feat(ws): HTTP/2 support for WebSocket

src/async_impl/client.rs

@@ -309,6 +309,7 @@ impl Client {
     ///     headers: typing.Optional[typing.Dict[str, bytes]]
     ///     cookies: typing.Optional[typing.Dict[str, str]]
     ///     protocols: typing.Optional[builtins.list[builtins.str]]
+    ///     use_http2: typing.Optional[builtins.bool]
     ///     auth: typing.Optional[builtins.str]
     ///     bearer_auth: typing.Optional[builtins.str]
     ///     basic_auth: typing.Optional[tuple[builtins.str, typing.Optional[builtins.str]]]

src/async_impl/request.rs

@@ -182,6 +182,15 @@ where
         accept_unmasked_frames
     );
 
+    // Use http2 options.
+    apply_option!(
+        apply_option_or_default,
+        builder,
+        params.use_http2,
+        use_http2,
+        false
+    );
+
     // Network options.
     apply_option!(
         apply_transformed_option_ref,

src/blocking/client.rs

@@ -296,6 +296,7 @@ impl BlockingClient {
     ///     headers: typing.Optional[typing.Dict[str, bytes]]
     ///     cookies: typing.Optional[typing.Dict[str, str]]
     ///     protocols: typing.Optional[builtins.list[builtins.str]]
+    ///     use_http2: typing.Optional[builtins.bool]
     ///     auth: typing.Optional[builtins.str]
     ///     bearer_auth: typing.Optional[builtins.str]
     ///     basic_auth: typing.Optional[tuple[builtins.str, typing.Optional[builtins.str]]]

src/lib.rs

@@ -279,6 +279,7 @@ fn request(
 ///     headers: typing.Optional[typing.Dict[str, bytes]]
 ///     cookies: typing.Optional[typing.Dict[str, str]]
 ///     protocols: typing.Optional[builtins.list[builtins.str]]
+///     use_http2: typing.Optional[builtins.bool]
 ///     auth: typing.Optional[builtins.str]
 ///     bearer_auth: typing.Optional[builtins.str]
 ///     basic_auth: typing.Optional[tuple[builtins.str, typing.Optional[builtins.str]]]

src/param/ws.rs

@@ -23,6 +23,9 @@ pub struct WebSocketParams {
     /// The protocols to use for the request.
     pub protocols: Option<Vec<String>>,
 
+    /// Whether to use HTTP/2 for the websocket.
+    pub use_http2: Option<bool>,
+
     /// The authentication to use for the request.
     pub auth: Option<PyBackedStr>,
 
@@ -91,6 +94,7 @@ impl<'py> FromPyObject<'py> for WebSocketParams {
         extract_option!(ob, params, local_address);
         extract_option!(ob, params, interface);
 
+        extract_option!(ob, params, use_http2);
         extract_option!(ob, params, headers);
         extract_option!(ob, params, cookies);
         extract_option!(ob, params, protocols);