perf: Only compute checksums on debug builds (#2446)

Signed-off-by: jthomson04 <jwillthomson19@gmail.com>

lib/runtime/src/pipeline/network/codec/two_part.rs

@@ -84,14 +84,27 @@ impl Decoder for TwoPartCodec {
         // Advance the buffer past the lengths and checksum
         src.advance(24);
 
-        let bytes_to_hash = header_len + body_len;
+        #[cfg(debug_assertions)]
+        {
+            // If the server sent a dummy checksum, skip it.
+            if checksum != 0 {
+                let bytes_to_hash =
+                    header_len
+                        .checked_add(body_len)
+                        .ok_or(TwoPartCodecError::InvalidMessage(
+                            "Message exceeds max allowed length.".to_string(),
+                        ))?;
+
                 let data_to_hash = &src[..bytes_to_hash];
+
                 let computed_checksum = xxh3_64(data_to_hash);
 
                 // Compare checksums
                 if checksum != computed_checksum {
                     return Err(TwoPartCodecError::ChecksumMismatch);
                 }
+            }
+        }
 
         // Read header and body data
         let header = src.split_to(header_len).freeze();
@@ -117,16 +130,25 @@ impl Encoder<TwoPartMessage> for TwoPartCodec {
             }
         }
 
+        dst.put_u64(header_len as u64);
+        dst.put_u64(body_len as u64);
+
+        // Only compute the checksum in debug mode.
+        // If we're in release mode, put a dummy value.
+        #[cfg(debug_assertions)]
+        {
             // Compute checksum of the data
             let mut data_to_hash = BytesMut::with_capacity(header_len + body_len);
             data_to_hash.extend_from_slice(&item.header);
             data_to_hash.extend_from_slice(&item.data);
             let checksum = xxh3_64(&data_to_hash);
 
-        // Write header and body sizes and checksum
-        dst.put_u64(header_len as u64);
-        dst.put_u64(body_len as u64);
             dst.put_u64(checksum);
+        }
+        #[cfg(not(debug_assertions))]
+        {
+            dst.put_u64(0);
+        }
 
         // Write header and body
         dst.put_slice(&item.header);