dnim/rust-postgres
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Log notices in Connection::poll impl

Browse Source
This commit is contained in:
Steven Fackler 2020-03-05 15:51:08 -08:00
parent 9f6d03d38b
commit e51028385b
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tokio-postgres/src/connection.rs
View File

@ -8,7 +8,7 @@ use fallible_iterator::FallibleIterator;
use futures::channel::mpsc; use futures::channel::mpsc;
use futures::stream::FusedStream; use futures::stream::FusedStream;
use futures::{ready, Sink, Stream, StreamExt}; use futures::{ready, Sink, Stream, StreamExt};
use log::trace; use log::{info, trace};
use postgres_protocol::message::backend::Message; use postgres_protocol::message::backend::Message;
use postgres_protocol::message::frontend; use postgres_protocol::message::frontend;
use std::collections::{HashMap, VecDeque}; use std::collections::{HashMap, VecDeque};
@ -330,7 +330,11 @@ where
type Output = Result<(), Error>; type Output = Result<(), Error>;
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Result<(), Error>> { fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Result<(), Error>> {
while let Some(_) = ready!(self.poll_message(cx)?) {} while let Some(message) = ready!(self.poll_message(cx)?) {
if let AsyncMessage::Notice(notice) = message {
info!("{}: {}", notice.severity(), notice.message());
}
}
Poll::Ready(Ok(())) Poll::Ready(Ok(()))
} }
} }