feat(server): add service property to server::conn::Parts
This allows getting the original service back. Closes #1471 Cherry-pick of commit bf7c0bbf4f55fdf465407874b0b2d4bd748e6783 from the 0.11.x branch.
This commit is contained in:
committed by
Ran Benita
parent
f98f168f07
commit
18c5f640e2
@@ -354,7 +354,7 @@ where
|
||||
///
|
||||
/// Only works for HTTP/1 connections. HTTP/2 connections will panic.
|
||||
pub fn into_parts(self) -> Parts<T> {
|
||||
let (io, read_buf) = match self.inner {
|
||||
let (io, read_buf, _) = match self.inner {
|
||||
Either::A(h1) => h1.into_inner(),
|
||||
Either::B(_h2) => {
|
||||
panic!("http2 cannot into_inner");
|
||||
|
||||
@@ -27,7 +27,7 @@ pub(crate) trait Dispatch {
|
||||
|
||||
pub struct Server<S: Service> {
|
||||
in_flight: Option<S::Future>,
|
||||
service: S,
|
||||
pub(crate) service: S,
|
||||
}
|
||||
|
||||
pub struct Client<B> {
|
||||
@@ -58,8 +58,9 @@ where
|
||||
self.conn.disable_keep_alive()
|
||||
}
|
||||
|
||||
pub fn into_inner(self) -> (I, Bytes) {
|
||||
self.conn.into_inner()
|
||||
pub fn into_inner(self) -> (I, Bytes, D) {
|
||||
let (io, buf) = self.conn.into_inner();
|
||||
(io, buf, self.dispatch)
|
||||
}
|
||||
|
||||
/// The "Future" poll function. Runs this dispatcher until the
|
||||
|
||||
@@ -98,7 +98,7 @@ where
|
||||
/// This allows taking apart a `Connection` at a later time, in order to
|
||||
/// reclaim the IO object, and additional related pieces.
|
||||
#[derive(Debug)]
|
||||
pub struct Parts<T> {
|
||||
pub struct Parts<T, S> {
|
||||
/// The original IO object used in the handshake.
|
||||
pub io: T,
|
||||
/// A buffer of bytes that have been read but not processed as HTTP.
|
||||
@@ -110,6 +110,8 @@ pub struct Parts<T> {
|
||||
/// You will want to check for any existing bytes if you plan to continue
|
||||
/// communicating on the IO object.
|
||||
pub read_buf: Bytes,
|
||||
/// The `Service` used to serve this connection.
|
||||
pub service: S,
|
||||
_inner: (),
|
||||
}
|
||||
|
||||
@@ -335,8 +337,8 @@ where
|
||||
/// This should only be called after `poll_without_shutdown` signals
|
||||
/// that the connection is "done". Otherwise, it may not have finished
|
||||
/// flushing all necessary HTTP bytes.
|
||||
pub fn into_parts(self) -> Parts<I> {
|
||||
let (io, read_buf) = match self.conn {
|
||||
pub fn into_parts(self) -> Parts<I, S> {
|
||||
let (io, read_buf, dispatch) = match self.conn {
|
||||
Either::A(h1) => {
|
||||
h1.into_inner()
|
||||
},
|
||||
@@ -347,6 +349,7 @@ where
|
||||
Parts {
|
||||
io: io,
|
||||
read_buf: read_buf,
|
||||
service: dispatch.service,
|
||||
_inner: (),
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user