Line data    Source code

       1              : use std::{
       2              :     pin::Pin,
       3              :     task::{Context, Poll},
       4              : };
       5              : 
       6              : use futures_util::Stream;
       7              : 
       8              : pin_project_lite::pin_project! {
       9              :     /// An `AsyncRead` adapter which carries a permit for the lifetime of the value.
      10              :     pub(crate) struct PermitCarrying<S> {
      11              :         permit: tokio::sync::OwnedSemaphorePermit,
      12              :         #[pin]
      13              :         inner: S,
      14              :     }
      15              : }
      16              : 
      17              : impl<S> PermitCarrying<S> {
      18        10252 :     pub(crate) fn new(permit: tokio::sync::OwnedSemaphorePermit, inner: S) -> Self {
      19        10252 :         Self { permit, inner }
      20        10252 :     }
      21              : }
      22              : 
      23              : impl<S: Stream> Stream for PermitCarrying<S> {
      24              :     type Item = <S as Stream>::Item;
      25              : 
      26       164202 :     fn poll_next(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
      27       164202 :         self.project().inner.poll_next(cx)
      28       164202 :     }
      29              : 
      30            0 :     fn size_hint(&self) -> (usize, Option<usize>) {
      31            0 :         self.inner.size_hint()
      32            0 :     }
      33              : }