Add deserialize_record_batch_message

Author: Hind-M

include/serialize.hpp

@@ -15,5 +15,6 @@ namespace sparrow_ipc
         SPARROW_IPC_API void serialize_record_batch_message(const ArrowArray& arrow_arr, const std::vector<int64_t>& buffers_sizes, std::vector<uint8_t>& final_buffer);
 
         SPARROW_IPC_API void deserialize_schema_message(const uint8_t* buf_ptr, size_t& current_offset, std::optional<std::string>& name, std::optional<std::vector<sparrow::metadata_pair>>& metadata);
+        SPARROW_IPC_API const org::apache::arrow::flatbuf::RecordBatch* deserialize_record_batch_message(const uint8_t* buf_ptr, size_t& current_offset);
     }
 }

include/serialize_null_array.hpp

@@ -58,14 +58,7 @@ namespace sparrow_ipc
         details::deserialize_schema_message(buf_ptr, current_offset, name, metadata);
 
         // II - Deserialize the RecordBatch message
-        uint32_t batch_meta_len = *(reinterpret_cast<const uint32_t*>(buf_ptr + current_offset));
-        current_offset += sizeof(uint32_t);
-        auto batch_message = org::apache::arrow::flatbuf::GetMessage(buf_ptr + current_offset);
-        if (batch_message->header_type() != org::apache::arrow::flatbuf::MessageHeader::RecordBatch)
-        {
-            throw std::runtime_error("Expected RecordBatch message, but got a different type.");
-        }
-        auto record_batch = static_cast<const org::apache::arrow::flatbuf::RecordBatch*>(batch_message->header());
+        const auto* record_batch = details::deserialize_record_batch_message(buf_ptr, current_offset);
 
         // The body is empty, so we don't need to read any further.
         // Construct the null_array from the deserialized metadata.

include/serialize_primitive_array.hpp

@@ -73,13 +73,8 @@ namespace sparrow_ipc
 
         // II - Deserialize the RecordBatch message
         uint32_t batch_meta_len = *(reinterpret_cast<const uint32_t*>(buf_ptr + current_offset));
-        current_offset += sizeof(uint32_t);
-        auto batch_message = org::apache::arrow::flatbuf::GetMessage(buf_ptr + current_offset);
-        if (batch_message->header_type() != org::apache::arrow::flatbuf::MessageHeader::RecordBatch)
-        {
-            throw std::runtime_error("Expected RecordBatch message, but got a different type.");
-        }
-        auto record_batch = static_cast<const org::apache::arrow::flatbuf::RecordBatch*>(batch_message->header());
+        const auto* record_batch = details::deserialize_record_batch_message(buf_ptr, current_offset);
+
         current_offset += utils::align_to_8(batch_meta_len);
         const uint8_t* body_ptr = buf_ptr + current_offset;
 

src/serialize.cpp

@@ -196,5 +196,17 @@ namespace sparrow_ipc
             }
             current_offset += schema_meta_len;
         }
+
+        const org::apache::arrow::flatbuf::RecordBatch* deserialize_record_batch_message(const uint8_t* buf_ptr, size_t& current_offset)
+        {
+            current_offset += sizeof(uint32_t);
+            auto batch_message = org::apache::arrow::flatbuf::GetMessage(buf_ptr + current_offset);
+            if (batch_message->header_type() != org::apache::arrow::flatbuf::MessageHeader::RecordBatch)
+            {
+                throw std::runtime_error("Expected RecordBatch message, but got a different type.");
+            }
+            return static_cast<const org::apache::arrow::flatbuf::RecordBatch*>(batch_message->header());
+        }
+
     } // namespace details
 } // namespace sparrow-ipc