xitca-postgres 0.4.0

an async postgres client
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

use core::{
    pin::Pin,
    task::{Context, Poll, ready},
};

use futures_core::Stream;

use crate::{error::Error, protocol::message::backend, row::RowOwned};

use super::RowStreamOwned;

impl Stream for RowStreamOwned {
    type Item = Result<RowOwned, Error>;

    fn poll_next(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
        let this = self.get_mut();
        loop {
            match ready!(this.res.poll_recv(cx))? {
                backend::Message::DataRow(body) => {
                    return Poll::Ready(Some(RowOwned::try_new(
                        this.col.clone(),
                        body,
                        Vec::with_capacity(this.col.len()),
                    )));
                }
                backend::Message::BindComplete
                | backend::Message::EmptyQueryResponse
                | backend::Message::CommandComplete(_)
                | backend::Message::PortalSuspended => {}
                backend::Message::ReadyForQuery(_) => return Poll::Ready(None),
                _ => return Poll::Ready(Some(Err(Error::unexpected()))),
            }
        }
    }
}