chore: Fix Arrow IPC deprecated function

abdolence · Jan 1, 2025 · 1e9f133 · 1e9f133
1 parent e0ade0a
commit 1e9f133
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -28,6 +28,7 @@ axum-streams = { version = "0.18", features=["json", "csv", "protobuf", "text"]
 
 | axum | axum-streams |
 |------|--------------|
+| 0.8  | v0.20+       |
 | 0.7  | v0.11+       |
 | 0.6  | v0.9-v0.10   |
 | 0.5  | 0.7          |

diff --git a/src/arrow_format.rs b/src/arrow_format.rs
@@ -46,7 +46,11 @@ impl StreamingFormat<RecordBatch> for ArrowRecordBatchIpcStreamFormat {
             let mut writer = BytesMut::new().writer();
 
             if let Some(prepend_schema) = prepend_schema {
-                let encoded_message = ipc_data_gen.schema_to_bytes(&prepend_schema, write_options);
+                let encoded_message = ipc_data_gen.schema_to_bytes_with_dictionary_tracker(
+                    &prepend_schema,
+                    dictionary_tracker,
+                    write_options,
+                );
                 write_message(&mut writer, encoded_message, write_options)?;
             }