Jefffrey commented on code in PR #10428:
URL: https://github.com/apache/arrow-rs/pull/10428#discussion_r3652446870
##########
arrow-flight/tests/encode_decode.rs:
##########
@@ -414,6 +417,110 @@ async fn test_mismatched_schema_message() {
.await;
}
+/// Encode `batch` to IPC, shift its body buffer by `alignment_offset` bytes
from a
+/// 64-byte-aligned base, then decode with alignment enforcement enabled.
+fn decode_misaligned(batch: RecordBatch, alignment_offset: usize) {
+ use arrow_buffer::{Buffer, MutableBuffer};
+ use arrow_ipc::writer::{DictionaryTracker, IpcDataGenerator};
+
+ let ipc_gen = IpcDataGenerator {};
+ let mut dict_tracker = DictionaryTracker::new(false);
+ let (_, encoded) = ipc_gen
+ .encode(
+ &batch,
+ &mut dict_tracker,
+ &Default::default(),
+ &mut Default::default(),
+ )
+ .unwrap();
+
+ // MutableBuffer guarantees 64-byte alignment, so slicing at
`alignment_offset`
+ // produces a pointer misaligned for types with stricter requirements.
+ let mut mis = MutableBuffer::with_capacity(encoded.arrow_data.len() +
alignment_offset);
+ for _ in 0..alignment_offset {
+ mis.push(0_u8);
+ }
+ mis.extend_from_slice(&encoded.arrow_data);
+ let misaligned_buf = Buffer::from(mis).slice(alignment_offset);
+
+ let message = arrow_ipc::root_as_message(&encoded.ipc_message).unwrap();
+ let ipc_batch = message.header_as_record_batch().unwrap();
+
+ reader::RecordBatchDecoder::try_new(
+ &misaligned_buf,
+ ipc_batch,
+ batch.schema(),
+ &Default::default(),
+ &message.version(),
+ )
+ .unwrap()
+ .with_require_alignment(true)
+ .read_record_batch()
+ .unwrap();
+}
+
+#[test]
+#[should_panic(
Review Comment:
nit: we can probably assert on the error instead of relying on should_panic
here (since we explicitly call unwrap in the helper)?
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
