1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101
use super::DeDataType; use crate::prelude::*; use serde::ser::SerializeStruct; use serde::{Serialize, Serializer}; use std::cell::RefCell; pub struct IterSer<I> where I: IntoIterator, <I as IntoIterator>::Item: Serialize, { iter: RefCell<Option<I>>, } impl<I> IterSer<I> where I: IntoIterator, <I as IntoIterator>::Item: Serialize, { fn new(iter: I) -> Self { IterSer { iter: RefCell::new(Some(iter)), } } } impl<I> Serialize for IterSer<I> where I: IntoIterator, <I as IntoIterator>::Item: Serialize, { fn serialize<S>( &self, serializer: S, ) -> std::result::Result<<S as Serializer>::Ok, <S as Serializer>::Error> where S: Serializer, { let iter: I = self.iter.borrow_mut().take().unwrap(); serializer.collect_seq(iter.into_iter()) } } impl<T> Serialize for ChunkedArray<T> where T: PolarsNumericType, T::Native: Serialize, { fn serialize<S>( &self, serializer: S, ) -> std::result::Result<<S as Serializer>::Ok, <S as Serializer>::Error> where S: Serializer, { let mut state = serializer.serialize_struct("series", 3)?; state.serialize_field("name", self.name())?; let dtype: DeDataType = self.dtype().into(); state.serialize_field("datatype", &dtype)?; state.serialize_field("values", &IterSer::new(self.into_iter()))?; state.end() } } macro_rules! impl_serialize { ($ca: ident) => { impl Serialize for $ca { fn serialize<S>( &self, serializer: S, ) -> std::result::Result<<S as Serializer>::Ok, <S as Serializer>::Error> where S: Serializer, { let mut state = serializer.serialize_struct("series", 3)?; state.serialize_field("name", self.name())?; let dtype: DeDataType = self.dtype().into(); state.serialize_field("datatype", &dtype)?; state.serialize_field("values", &IterSer::new(self.into_iter()))?; state.end() } } }; } impl_serialize!(Utf8Chunked); impl_serialize!(BooleanChunked); impl_serialize!(ListChunked); impl Serialize for CategoricalChunked { fn serialize<S>( &self, serializer: S, ) -> std::result::Result<<S as Serializer>::Ok, <S as Serializer>::Error> where S: Serializer, { let ca = self.cast::<Utf8Type>().unwrap(); ca.serialize(serializer) } }