Refactor: Add write method to PgStream and BufStream

Co-authored-by: contact <[email protected]>

Author: cursoragent

sqlx-core/src/io/buf_stream.rs

@@ -35,7 +35,13 @@ where
         }
     }
 
-    // Note: simple `write` is unused in current code paths; keep `write_with`.
+    #[allow(dead_code)]
+    pub fn write<'en, T>(&mut self, value: T)
+    where
+        T: Encode<'en, ()>,
+    {
+        self.write_with(value, ())
+    }
 
     pub fn write_with<'en, T, C>(&mut self, value: T, context: C)
     where

sqlx-core/src/postgres/connection/stream.rs

@@ -60,6 +60,19 @@ impl PgStream {
         self.flush().await
     }
 
+    #[inline]
+    pub(crate) fn write<'en, T>(&mut self, message: T)
+    where
+        T: Encode<'en>,
+    {
+        self.inner.write(message)
+    }
+
+    #[inline]
+    pub(crate) fn flush(&mut self) -> impl core::future::Future<Output = Result<(), Error>> + '_ {
+        self.inner.flush()
+    }
+
     // Expect a specific type and format
     pub(crate) async fn recv_expect<'de, T: Decode<'de>>(
         &mut self,

sqlx-core/src/postgres/copy.rs

@@ -304,7 +304,7 @@ impl<C: DerefMut<Target = PgConnection>> PgCopyIn<C> {
 impl<C: DerefMut<Target = PgConnection>> Drop for PgCopyIn<C> {
     fn drop(&mut self) {
         if let Some(mut conn) = self.conn.take() {
-            conn.stream.write(CopyFail::new(
+            conn.stream.write(message::CopyFail::new(
                 "PgCopyIn dropped without calling finish() or fail()",
             ));
         }