feat(upgrade): allow http upgrades with any body type

Allow using `Request<T>`/`Response<T>` for any given T with `upgrade::on` instead of just restricting it to `hyper::Body`.
hyperium · Nov 3, 2021 · ab469eb · ab469eb
1 parent b5022f3
commit ab469eb
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/upgrade.rs b/src/upgrade.rs
@@ -258,31 +258,31 @@ mod sealed {
         fn on_upgrade(self) -> OnUpgrade;
     }
 
-    impl CanUpgrade for http::Request<crate::Body> {
+    impl<B> CanUpgrade for http::Request<B> {
         fn on_upgrade(mut self) -> OnUpgrade {
             self.extensions_mut()
                 .remove::<OnUpgrade>()
                 .unwrap_or_else(OnUpgrade::none)
         }
     }
 
-    impl CanUpgrade for &'_ mut http::Request<crate::Body> {
+    impl<B> CanUpgrade for &'_ mut http::Request<B> {
         fn on_upgrade(self) -> OnUpgrade {
             self.extensions_mut()
                 .remove::<OnUpgrade>()
                 .unwrap_or_else(OnUpgrade::none)
         }
     }
 
-    impl CanUpgrade for http::Response<crate::Body> {
+    impl<B> CanUpgrade for http::Response<B> {
         fn on_upgrade(mut self) -> OnUpgrade {
             self.extensions_mut()
                 .remove::<OnUpgrade>()
                 .unwrap_or_else(OnUpgrade::none)
         }
     }
 
-    impl CanUpgrade for &'_ mut http::Response<crate::Body> {
+    impl<B> CanUpgrade for &'_ mut http::Response<B> {
         fn on_upgrade(self) -> OnUpgrade {
             self.extensions_mut()
                 .remove::<OnUpgrade>()