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