backend: add next record lsn to interpreted batch

neondatabase · Nov 15, 2024 · e6c92ba · e6c92ba
1 parent 9fe5b8c
commit e6c92ba
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/postgres-protocol/src/message/backend.rs b/postgres-protocol/src/message/backend.rs
@@ -375,9 +375,15 @@ impl ReplicationMessage<Bytes> {
             INTERPRETED_WAL_RECORD_TAG => {
                 let streaming_lsn = buf.read_u64::<BigEndian>()?;
                 let wal_end = buf.read_u64::<BigEndian>()?;
+                let next_record_lsn = match buf.read_u64::<BigEndian>()? {
+                    0 => None,
+                    lsn => Some(lsn),
+                };
+
                 ReplicationMessage::RawInterpretedWalRecords(RawInterpretedWalRecordsBody {
                     streaming_lsn,
                     wal_end,
+                    next_record_lsn,
                     data: buf.read_all(),
                 })
             }
@@ -965,6 +971,7 @@ impl<D> XLogDataBody<D> {
 pub struct RawInterpretedWalRecordsBody<D> {
     streaming_lsn: u64,
     wal_end: u64,
+    next_record_lsn: Option<u64>,
     data: D,
 }
 
@@ -979,6 +986,11 @@ impl<D> RawInterpretedWalRecordsBody<D> {
         self.wal_end
     }
 
+    #[inline]
+    pub fn next_record_lsn(&self) -> Option<u64> {
+        self.next_record_lsn
+    }
+
     #[inline]
     pub fn data(&self) -> &D {
         &self.data