From a5d94e3de8ae27e7f0795efcd9452bbdfa2516d8 Mon Sep 17 00:00:00 2001
From: Lucas Kent <rubickent@gmail.com>
Date: Tue, 23 Aug 2022 16:12:40 +1000
Subject: [PATCH] Fix debug cassandra message dump (#751)

---
 shotover-proxy/src/codec/cassandra.rs | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/shotover-proxy/src/codec/cassandra.rs b/shotover-proxy/src/codec/cassandra.rs
index e6d889272..b6f5bb3e0 100644
--- a/shotover-proxy/src/codec/cassandra.rs
+++ b/shotover-proxy/src/codec/cassandra.rs
@@ -42,10 +42,6 @@ impl CassandraCodec {
 impl CassandraCodec {
     fn encode_raw(&mut self, item: CassandraFrame, dst: &mut BytesMut) {
         let buffer = item.encode().encode_with(self.compressor).unwrap();
-        tracing::debug!(
-            "outgoing cassandra message:\n{}",
-            pretty_hex::pretty_hex(&buffer)
-        );
         if buffer.is_empty() {
             info!("trying to send 0 length frame");
         }
@@ -124,11 +120,16 @@ impl Encoder<Messages> for CassandraCodec {
         dst: &mut BytesMut,
     ) -> std::result::Result<(), Self::Error> {
         for m in item {
+            let start = dst.len();
             // TODO: always check if cassandra message
             match m.into_encodable(MessageType::Cassandra)? {
                 Encodable::Bytes(bytes) => dst.extend_from_slice(&bytes),
                 Encodable::Frame(frame) => self.encode_raw(frame.into_cassandra().unwrap(), dst),
             }
+            tracing::debug!(
+                "outgoing cassandra message:\n{}",
+                pretty_hex::pretty_hex(&&dst[start..])
+            );
         }
         Ok(())
     }