From b566ddf0e5921423160abbf57779b132e086d988 Mon Sep 17 00:00:00 2001
From: Lev Kokotov <lev.kokotov@gmail.com>
Date: Sun, 26 Jan 2025 10:38:28 -0800
Subject: [PATCH] clearer debug messages

---
 pgdog/src/backend/server.rs   |  4 ++--
 pgdog/src/net/messages/mod.rs | 10 +++++-----
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/pgdog/src/backend/server.rs b/pgdog/src/backend/server.rs
index f5973ff..097f789 100644
--- a/pgdog/src/backend/server.rs
+++ b/pgdog/src/backend/server.rs
@@ -199,7 +199,7 @@ impl Server {
         self.stats.state(State::Active);
 
         #[cfg(debug_assertions)]
-        message.debug()?;
+        message.debug("→")?;
 
         match self.stream().send(message).await {
             Ok(sent) => self.stats.send(sent),
@@ -259,7 +259,7 @@ impl Server {
         }
 
         #[cfg(debug_assertions)]
-        message.debug()?;
+        message.debug("←")?;
 
         Ok(message)
     }
diff --git a/pgdog/src/net/messages/mod.rs b/pgdog/src/net/messages/mod.rs
index 3bc4a74..de2b106 100644
--- a/pgdog/src/net/messages/mod.rs
+++ b/pgdog/src/net/messages/mod.rs
@@ -62,27 +62,27 @@ pub trait Protocol: ToBytes + FromBytes {
         Ok(Message::new(self.to_bytes()?))
     }
 
-    fn debug(&self) -> Result<(), Error> {
+    fn debug(&self, direction: &str) -> Result<(), Error> {
         let message = self.message()?;
         match message.code() {
             'd' => {
                 let copy_data = CopyData::from_bytes(message.to_bytes()?)?;
                 if let Some(xlog) = copy_data.xlog_data() {
-                    debug!("{:#?}", xlog.payload());
+                    debug!("{} {:#?}", direction, xlog.payload());
                 }
                 if let Some(meta) = copy_data.replication_meta() {
-                    debug!("{:#?}", meta);
+                    debug!("{} {:#?}", direction, meta);
                 }
             }
 
             'D' => {
                 let data_row = DataRow::from_bytes(message.to_bytes()?)?;
-                debug!("{:#?}", data_row);
+                debug!("{} {:#?}", direction, data_row);
             }
 
             'T' => {
                 let rd = RowDescription::from_bytes(message.to_bytes()?)?;
-                debug!("{:#?}", rd);
+                debug!("{} {:#?}", direction, rd);
             }
 
             _ => (),