Add arrow conversions for SchemaRef

rerun-io · Jan 13, 2025 · e857670 · e857670
1 parent 9ff456a
commit e857670
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/src/datatypes/schema.rs b/src/datatypes/schema.rs
@@ -70,6 +70,17 @@ impl From<arrow_schema::Schema> for Schema {
     }
 }
 
+#[cfg(feature = "arrow")]
+impl From<arrow_schema::SchemaRef> for Schema {
+    fn from(schema: arrow_schema::SchemaRef) -> Self {
+        let arrow_schema::Schema { fields, metadata } = schema.as_ref();
+        Self {
+            fields: fields.iter().map(Into::into).collect(),
+            metadata: metadata.clone().into_iter().collect(),
+        }
+    }
+}
+
 #[cfg(feature = "arrow")]
 impl From<Schema> for arrow_schema::Schema {
     fn from(schema: Schema) -> Self {
@@ -79,3 +90,12 @@ impl From<Schema> for arrow_schema::Schema {
         Self::new_with_metadata(fields, metadata.into_iter().collect())
     }
 }
+
+impl From<Schema> for arrow_schema::SchemaRef {
+    fn from(schema: Schema) -> Self {
+        let Schema { fields, metadata } = schema;
+        let fields: arrow_schema::Fields =
+            fields.into_iter().map(arrow_schema::Field::from).collect();
+        arrow_schema::Schema::new_with_metadata(fields, metadata.into_iter().collect()).into()
+    }
+}