Expose conection.stream

That way our proxy can take back stream for proxying.

Author: kelvich

tokio-postgres/src/connection.rs

@@ -49,7 +49,8 @@ enum State {
 /// occurred, or because its associated `Client` has dropped and all outstanding work has completed.
 #[must_use = "futures do nothing unless polled"]
 pub struct Connection<S, T> {
-    stream: Framed<MaybeTlsStream<S, T>, PostgresCodec>,
+    /// HACK: we need this in the Neon Proxy.
+    pub stream: Framed<MaybeTlsStream<S, T>, PostgresCodec>,
     /// HACK: we need this in the Neon Proxy to forward params.
     pub parameters: HashMap<String, String>,
     receiver: mpsc::UnboundedReceiver<Request>,

tokio-postgres/src/lib.rs

@@ -165,7 +165,7 @@ mod copy_out;
 pub mod error;
 mod generic_client;
 mod keepalive;
-mod maybe_tls_stream;
+pub mod maybe_tls_stream;
 mod portal;
 mod prepare;
 mod query;

tokio-postgres/src/maybe_tls_stream.rs

@@ -1,11 +1,17 @@
+//! MaybeTlsStream.
+//!
+//! Represents a stream that may or may not be encrypted with TLS.
 use crate::tls::{ChannelBinding, TlsStream};
 use std::io;
 use std::pin::Pin;
 use std::task::{Context, Poll};
 use tokio::io::{AsyncRead, AsyncWrite, ReadBuf};
 
+/// A stream that may or may not be encrypted with TLS.
 pub enum MaybeTlsStream<S, T> {
+    /// An unencrypted stream.
     Raw(S),
+    /// An encrypted stream.
     Tls(T),
 }