multipart_write/write/
complete.rs

1use crate::write::MultipartWriteExt;
2use crate::{FusedMultipartWrite, MultipartWrite};
3
4use futures_core::future::{FusedFuture, Future};
5use futures_core::ready;
6use std::fmt::{self, Debug, Formatter};
7use std::pin::Pin;
8use std::task::{Context, Poll};
9
10/// Future for [`complete`](super::MultipartWriteExt::complete).
11#[must_use = "futures do nothing unless polled"]
12pub struct Complete<'a, Wr: ?Sized, Part> {
13    writer: &'a mut Wr,
14    is_terminated: bool,
15    _p: std::marker::PhantomData<Part>,
16}
17
18impl<Wr: ?Sized + Unpin, Part> Unpin for Complete<'_, Wr, Part> {}
19
20impl<'a, Wr: MultipartWrite<Part> + ?Sized + Unpin, Part> Complete<'a, Wr, Part> {
21    pub(super) fn new(writer: &'a mut Wr) -> Self {
22        Self {
23            writer,
24            is_terminated: false,
25            _p: std::marker::PhantomData,
26        }
27    }
28}
29
30impl<Wr: ?Sized + FusedMultipartWrite<Part> + Unpin, Part> FusedFuture for Complete<'_, Wr, Part> {
31    fn is_terminated(&self) -> bool {
32        self.writer.is_terminated()
33    }
34}
35
36impl<Wr: ?Sized + MultipartWrite<Part> + Unpin, Part> Future for Complete<'_, Wr, Part> {
37    type Output = Result<Wr::Output, Wr::Error>;
38    fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
39        ready!(self.writer.poll_flush_unpin(cx))?;
40        let output = ready!(self.writer.poll_complete_unpin(cx));
41        self.is_terminated = true;
42        Poll::Ready(output)
43    }
44}
45
46impl<Wr: Debug, Part> Debug for Complete<'_, Wr, Part> {
47    fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result {
48        f.debug_struct("Complete")
49            .field("writer", &self.writer)
50            .field("is_terminated", &self.is_terminated)
51            .finish()
52    }
53}