Method try_encode_file_format

datafusion/proto/src/logical_plan/mod.rs:271–321

(
        &self,
        buf: &mut Vec<u8>,
        node: Arc<dyn FileFormatFactory>,
    )

    }

    fn try_encode_file_format(
        &self,
        buf: &mut Vec<u8>,
        node: Arc<dyn FileFormatFactory>,
    ) -> Result<()> {
        let mut encoded_file_format = Vec::new();

        let kind = if node.downcast_ref::<CsvFormatFactory>().is_some() {
            file_formats::CsvLogicalExtensionCodec
                .try_encode_file_format(&mut encoded_file_format, Arc::clone(&node))?;
            protobuf::FileFormatKind::Csv
        } else if node.downcast_ref::<JsonFormatFactory>().is_some() {
            file_formats::JsonLogicalExtensionCodec
                .try_encode_file_format(&mut encoded_file_format, Arc::clone(&node))?;
            protobuf::FileFormatKind::Json
        } else if node.downcast_ref::<ArrowFormatFactory>().is_some() {
            file_formats::ArrowLogicalExtensionCodec
                .try_encode_file_format(&mut encoded_file_format, Arc::clone(&node))?;
            protobuf::FileFormatKind::Arrow
        } else {
            #[cfg(feature = "parquet")]
            {
                if node.downcast_ref::<ParquetFormatFactory>().is_some() {
                    file_formats::ParquetLogicalExtensionCodec.try_encode_file_format(
                        &mut encoded_file_format,
                        Arc::clone(&node),
                    )?;
                    protobuf::FileFormatKind::Parquet
                } else {
                    return not_impl_err!(
                        "Unsupported FileFormatFactory type for DefaultLogicalExtensionCodec"
                    );
                }
            }
            #[cfg(not(feature = "parquet"))]
            {
                return not_impl_err!(
                    "Unsupported FileFormatFactory type for DefaultLogicalExtensionCodec"
                );
            }
        };

        let proto = protobuf::FileFormatProto {
            kind: kind as i32,
            encoded_file_format,
        };
        proto.encode(buf).map_err(|e| {
            internal_datafusion_err!("Failed to encode FileFormatProto: {e}")
        })?;
        Ok(())
    }
}

#[macro_export]