2a80118e87
The common case is to simply want a vec of rows to work with, so make that the main API, wrapping the old version returning an iterator.
38 lines
960 B
Rust
38 lines
960 B
Rust
use fallible_iterator::FallibleIterator;
|
|
use futures::stream::{self, Stream};
|
|
use std::marker::PhantomData;
|
|
use tokio_postgres::impls;
|
|
use tokio_postgres::{Error, SimpleQueryMessage};
|
|
|
|
pub struct SimpleQueryIter<'a> {
|
|
it: stream::Wait<impls::SimpleQuery>,
|
|
_p: PhantomData<&'a mut ()>,
|
|
}
|
|
|
|
// no-op impl to extend borrow until drop
|
|
impl<'a> Drop for SimpleQueryIter<'a> {
|
|
fn drop(&mut self) {}
|
|
}
|
|
|
|
impl<'a> SimpleQueryIter<'a> {
|
|
pub(crate) fn new(stream: impls::SimpleQuery) -> SimpleQueryIter<'a> {
|
|
SimpleQueryIter {
|
|
it: stream.wait(),
|
|
_p: PhantomData,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<'a> FallibleIterator for SimpleQueryIter<'a> {
|
|
type Item = SimpleQueryMessage;
|
|
type Error = Error;
|
|
|
|
fn next(&mut self) -> Result<Option<SimpleQueryMessage>, Error> {
|
|
match self.it.next() {
|
|
Some(Ok(row)) => Ok(Some(row)),
|
|
Some(Err(e)) => Err(e),
|
|
None => Ok(None),
|
|
}
|
|
}
|
|
}
|